Skip to content

Commit 9861278

Browse files
gustavocidornelaswhoseoyster
authored andcommitted
Add arg include_metadata to API HTTP request to avoid sending client version for data stream endpoint
1 parent 7b05b8a commit 9861278

File tree

2 files changed

+21
-5
lines changed

2 files changed

+21
-5
lines changed

openlayer/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1109,6 +1109,7 @@ def stream_data(
11091109
self.api.post_request(
11101110
endpoint=f"inference-pipelines/{inference_pipeline_id}/data-stream",
11111111
body=body,
1112+
include_metadata=False,
11121113
)
11131114
print("Stream published!")
11141115

openlayer/api.py

Lines changed: 20 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,7 @@ def _http_request(
8787
body=None,
8888
files=None,
8989
data=None,
90+
include_metadata=True,
9091
) -> Response:
9192
with requests.Session() as https:
9293
retry_strategy = Retry(
@@ -102,7 +103,8 @@ def _http_request(
102103

103104
try:
104105
params = params or {}
105-
params.update(CLIENT_METADATA)
106+
if include_metadata:
107+
params.update(CLIENT_METADATA)
106108
res = https.request(
107109
method=method,
108110
url=url,
@@ -136,12 +138,22 @@ def _api_request(
136138
body=None,
137139
files=None,
138140
data=None,
141+
include_metadata=True,
139142
):
140143
"""Make any HTTP request + error handling."""
141144

142145
url = f"{OPENLAYER_ENDPOINT}/{endpoint}"
143146

144-
res = self._http_request(method, url, headers, params, body, files, data)
147+
res = self._http_request(
148+
method=method,
149+
url=url,
150+
headers=headers,
151+
params=params,
152+
body=body,
153+
files=files,
154+
data=data,
155+
include_metadata=include_metadata,
156+
)
145157

146158
json = None
147159
if res.ok:
@@ -155,17 +167,20 @@ def get_request(self, endpoint: str, params=None):
155167
"""Generic GET Request Wrapper."""
156168
return self._api_request("GET", endpoint, headers=self._headers, params=params)
157169

158-
def post_request(self, endpoint: str, body=None, files=None, data=None):
170+
def post_request(
171+
self, endpoint: str, body=None, files=None, data=None, include_metadata=True
172+
):
159173
"""Generic POST Request Wrapper."""
160174
return self._api_request(
161-
"POST",
162-
endpoint,
175+
method="POST",
176+
endpoint=endpoint,
163177
headers=self._headers
164178
if files is None
165179
else self._headers_multipart_form_data,
166180
body=body,
167181
files=files,
168182
data=data,
183+
include_metadata=include_metadata,
169184
)
170185

171186
def put_request(self, endpoint: str, body=None, files=None, data=None):

0 commit comments

Comments
 (0)