From 4d6c3bc8048ae0998f6a2bc9d227dd08fe45f30b Mon Sep 17 00:00:00 2001
From: mgiannakopoulos <mgiannakopoulos@singlestore.com>
Date: Tue, 7 Oct 2025 05:37:31 -0400
Subject: [PATCH] Delete deprecated, old, invalid versions for chat and
 embeddings clients.

---
 singlestoredb/ai/__init__.py   |  3 ---
 singlestoredb/ai/chat.py       | 38 ----------------------------------
 singlestoredb/ai/embeddings.py | 15 --------------
 3 files changed, 56 deletions(-)

diff --git a/singlestoredb/ai/__init__.py b/singlestoredb/ai/__init__.py
index 31a4db2e..3cc526d6 100644
--- a/singlestoredb/ai/__init__.py
+++ b/singlestoredb/ai/__init__.py
@@ -1,5 +1,2 @@
-from .chat import SingleStoreChat  # noqa: F401
 from .chat import SingleStoreChatFactory  # noqa: F401
-from .chat import SingleStoreChatOpenAI  # noqa: F401
-from .embeddings import SingleStoreEmbeddings  # noqa: F401
 from .embeddings import SingleStoreEmbeddingsFactory  # noqa: F401
diff --git a/singlestoredb/ai/chat.py b/singlestoredb/ai/chat.py
index 347833b7..f3419c28 100644
--- a/singlestoredb/ai/chat.py
+++ b/singlestoredb/ai/chat.py
@@ -29,44 +29,6 @@
 from botocore.config import Config
 
 
-class SingleStoreChatOpenAI(ChatOpenAI):
-    def __init__(self, model_name: str, api_key: Optional[str] = None, **kwargs: Any):
-        inference_api_manger = (
-            manage_workspaces().organizations.current.inference_apis
-        )
-        info = inference_api_manger.get(model_name=model_name)
-        token = (
-            api_key
-            if api_key is not None
-            else os.environ.get('SINGLESTOREDB_USER_TOKEN')
-        )
-        super().__init__(
-            base_url=info.connection_url,
-            api_key=token,
-            model=model_name,
-            **kwargs,
-        )
-
-
-class SingleStoreChat(ChatOpenAI):
-    def __init__(self, model_name: str, api_key: Optional[str] = None, **kwargs: Any):
-        inference_api_manger = (
-            manage_workspaces().organizations.current.inference_apis
-        )
-        info = inference_api_manger.get(model_name=model_name)
-        token = (
-            api_key
-            if api_key is not None
-            else os.environ.get('SINGLESTOREDB_USER_TOKEN')
-        )
-        super().__init__(
-            base_url=info.connection_url,
-            api_key=token,
-            model=model_name,
-            **kwargs,
-        )
-
-
 def SingleStoreChatFactory(
     model_name: str,
     api_key: Optional[str] = None,
diff --git a/singlestoredb/ai/embeddings.py b/singlestoredb/ai/embeddings.py
index 531d25ea..2449d94f 100644
--- a/singlestoredb/ai/embeddings.py
+++ b/singlestoredb/ai/embeddings.py
@@ -29,21 +29,6 @@
 from botocore.config import Config
 
 
-class SingleStoreEmbeddings(OpenAIEmbeddings):
-
-    def __init__(self, model_name: str, **kwargs: Any):
-        inference_api_manger = (
-            manage_workspaces().organizations.current.inference_apis
-        )
-        info = inference_api_manger.get(model_name=model_name)
-        super().__init__(
-            base_url=info.connection_url,
-            api_key=os.environ.get('SINGLESTOREDB_USER_TOKEN'),
-            model=model_name,
-            **kwargs,
-        )
-
-
 def SingleStoreEmbeddingsFactory(
     model_name: str,
     api_key: Optional[str] = None,