Skip to content

Commit 51148be

Browse files
committed
Modify _get_base_url to use self-hosted mode when token is not set.
1 parent 088ee99 commit 51148be

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

.fernignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
# Specify files that shouldn't be modified by Fern
22

33
tests/custom/test_client.py
4+
src/letta/client.py
45
.github/workflows/ci.yml

src/letta/client.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def __init__(
7070
):
7171
_defaulted_timeout = timeout if timeout is not None else 60 if httpx_client is None else None
7272
self._client_wrapper = SyncClientWrapper(
73-
base_url=_get_base_url(base_url=base_url, environment=environment),
73+
base_url=_get_base_url(token=token, base_url=base_url, environment=environment),
7474
token=token,
7575
httpx_client=httpx_client
7676
if httpx_client is not None
@@ -137,7 +137,7 @@ def __init__(
137137
):
138138
_defaulted_timeout = timeout if timeout is not None else 60 if httpx_client is None else None
139139
self._client_wrapper = AsyncClientWrapper(
140-
base_url=_get_base_url(base_url=base_url, environment=environment),
140+
base_url=_get_base_url(token=token, base_url=base_url, environment=environment),
141141
token=token,
142142
httpx_client=httpx_client
143143
if httpx_client is not None
@@ -155,9 +155,11 @@ def __init__(
155155
self.health = AsyncHealthClient(client_wrapper=self._client_wrapper)
156156

157157

158-
def _get_base_url(*, base_url: typing.Optional[str] = None, environment: LettaEnvironment) -> str:
158+
def _get_base_url(*, token: typing.Optional[str] = None, base_url: typing.Optional[str] = None, environment: LettaEnvironment) -> str:
159159
if base_url is not None:
160160
return base_url
161+
elif token is None:
162+
return LettaEnvironment.SELF_HOSTED.value
161163
elif environment is not None:
162164
return environment.value
163165
else:

0 commit comments

Comments
 (0)