From e9a8d6b7f23affcc4fa770fbb0d0385bf6afeaad Mon Sep 17 00:00:00 2001 From: Sam Partee Date: Thu, 29 Aug 2024 09:22:09 -0700 Subject: [PATCH] Engine API Keys (#24) --- arcade/arcade/client/base.py | 3 +-- arcade/arcade/client/client.py | 4 +++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/arcade/arcade/client/base.py b/arcade/arcade/client/base.py index 64d15d64..33d89e9d 100644 --- a/arcade/arcade/client/base.py +++ b/arcade/arcade/client/base.py @@ -44,7 +44,7 @@ class BaseArcadeClient: self._base_url = base_url self._api_key = api_key or os.environ.get("ARCADE_API_KEY") or config.api.key self._headers = headers or {} - self._headers.setdefault("X-API-Key", self._api_key) + self._headers.setdefault("Authorization", f"Bearer {self._api_key}") self._headers.setdefault("Content-Type", "application/json") self._proxies = proxies self._timeout = timeout @@ -82,7 +82,6 @@ class SyncArcadeClient(BaseArcadeClient): url = self._build_url(path) for attempt in range(self._retries): try: - print(method, url, kwargs) response = self._client.request(method, url, **kwargs) response.raise_for_status() return response # noqa: TRY300 diff --git a/arcade/arcade/client/client.py b/arcade/arcade/client/client.py index 9b6e13da..f831f387 100644 --- a/arcade/arcade/client/client.py +++ b/arcade/arcade/client/client.py @@ -154,7 +154,9 @@ class Arcade(ArcadeClientMixin[SyncArcadeClient], SyncArcadeClient): def __init__(self, *args: Any, **kwargs: Any): super().__init__(*args, **kwargs) - self._openai_client = OpenAI(base_url=self._base_url + "/v1") + + # Assume we are using the LLM API of the Engine for now + self._openai_client = OpenAI(base_url=self._base_url + "/v1", api_key=self._api_key) self.chat = self._openai_client.chat def _execute_request(self, method: str, url: str, **kwargs: Any) -> Any: