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