Skip to content

Commit dca9cb1

Browse files
Fallback to connectionURL if internalConnectionURL is empty (#114)
* fallback to connectionURL if internalConnectionURL is empty * review changes
1 parent cd08bfa commit dca9cb1

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

singlestoredb/ai/chat.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,8 @@ def SingleStoreChatFactory(
5252
manage_workspaces().organizations.current.inference_apis
5353
)
5454
info = inference_api_manager.get(model_name=model_name)
55+
if not info.internal_connection_url:
56+
info.internal_connection_url = info.connection_url
5557
else:
5658
info = InferenceAPIInfo(
5759
service_id='',

singlestoredb/ai/embeddings.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,8 @@ def SingleStoreEmbeddingsFactory(
5151
manage_workspaces().organizations.current.inference_apis
5252
)
5353
info = inference_api_manager.get(model_name=model_name)
54+
if not info.internal_connection_url:
55+
info.internal_connection_url = info.connection_url
5456
else:
5557
info = InferenceAPIInfo(
5658
service_id='',

0 commit comments

Comments
 (0)