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,