admin-valentin/arcade-mcp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Engine API Keys (#24)

Browse source
This commit is contained in:
Sam Partee 2024-08-29 09:22:09 -07:00 committed by GitHub
parent 1c1403d1dd
commit e9a8d6b7f2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
arcade/arcade/client/base.py
View file

@ -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

4
arcade/arcade/client/client.py
View file

@ -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: