Skip to content

Commit ebfaa13

Browse files
committed
Add httpx to setup
1 parent ef4cc4d commit ebfaa13

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

databricks/sdk/mixins/open_ai_client.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,3 @@
1-
import httpx
2-
31
from databricks.sdk.service.serving import ServingEndpointsAPI
42

53

@@ -8,6 +6,7 @@ class ServingEndpointsExt(ServingEndpointsAPI):
86
# Using the HTTP Client to pass in the databricks authorization
97
# This method will be called on every invocation, so when using with model serving will always get the refreshed token
108
def _get_authorized_http_client(self):
9+
import httpx
1110

1211
class BearerAuth(httpx.Auth):
1312

setup.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,9 +18,9 @@
1818
"yapf", "pycodestyle", "autoflake", "isort", "wheel",
1919
"ipython", "ipywidgets", "requests-mock", "pyfakefs",
2020
"databricks-connect", "pytest-rerunfailures", "openai",
21-
'langchain-openai; python_version > "3.7"'],
21+
'langchain-openai; python_version > "3.7"', "httpx"],
2222
"notebook": ["ipython>=8,<9", "ipywidgets>=8,<9"],
23-
"openai": ["openai", 'langchain-openai; python_version > "3.7"']},
23+
"openai": ["openai", 'langchain-openai; python_version > "3.7"', "httpx"]},
2424
author="Serge Smertin",
2525
author_email="[email protected]",
2626
description="Databricks SDK for Python (Beta)",

0 commit comments

Comments
 (0)