Merge pull request #399 from julep-ai/x/traefik-related-hotfixes

wip: Python client fix + example keys whitelist
julep-ai · Apr 1, 2024 · 87e187a · 87e187a
2 parents 6b507fa + e725718
commit 87e187a
Show file tree

Hide file tree

Showing 5 changed files with 24 additions and 9 deletions.
diff --git a/agents-api/docker-compose.yml b/agents-api/docker-compose.yml
@@ -70,11 +70,9 @@ services:
 
   text-embeddings-inference:
     container_name: text-embeddings-inference
-    env_file:
-      - path: .env
-        required: false
-      - path: ../.env
-        required: true
+    environment:
+      - DTYPE=float16
+      - MODEL_ID=BAAI/llm-embedder
 
     image: ghcr.io/huggingface/text-embeddings-inference:1.0
     ports:

diff --git a/sdks/python/julep/api/types/chat_input_data_tool_choice.py b/sdks/python/julep/api/types/chat_input_data_tool_choice.py
@@ -3,7 +3,6 @@
 import typing
 
 from .named_tool_choice import NamedToolChoice
-from .named_tool_choice_function import NamedToolChoiceFunction
 from .tool_choice_option import ToolChoiceOption
 
 ChatInputDataToolChoice = typing.Union[ToolChoiceOption, NamedToolChoice]
diff --git a/sdks/python/julep/client.py b/sdks/python/julep/client.py
@@ -119,7 +119,7 @@ def __init__(
         )
 
         self._api_client = JulepApi(
-            api_key=api_key,
+            api_key=f"Bearer {api_key}",
             base_url=base_url,
             httpx_client=httpx_client,
             *args,
@@ -240,7 +240,7 @@ def __init__(
         ), "base_url must be provided or set as env var JULEP_API_URL"
 
         self._api_client = AsyncJulepApi(
-            api_key=api_key, base_url=base_url, *args, **kwargs
+            api_key=f"Bearer {api_key}", base_url=base_url, *args, **kwargs
         )
 
         self.agents = AsyncAgentsManager(api_client=self._api_client)

diff --git a/sdks/python/poetry.lock b/sdks/python/poetry.lock
diff --git a/sdks/python/pyproject.toml b/sdks/python/pyproject.toml
@@ -26,6 +26,7 @@ ipython = "<8.13"
 pyright = "^1.1.348"
 handsdown = "^2.1.0"
 jupyterlab = "^4.1.0"
+pyjwt = "^2.8.0"
 
 [build-system]
 requires = ["poetry-core"]