Delete deprecated, old, invalid versions for chat and embeddings clients. (#99)

Author: mgiannakopoulos

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

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,

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,