#1155: Add support for OpenAI-compatible endpoint in LLM and Embed (#1197)

embedchain/config/embedder/base.py

@@ -11,6 +11,7 @@ class BaseEmbedderConfig:
         deployment_name: Optional[str] = None,
         vector_dimension: Optional[int] = None,
         api_key: Optional[str] = None,
+        api_base: Optional[str] = None,
     ):
         """
         Initialize a new instance of an embedder config class.
@@ -24,3 +25,4 @@ class BaseEmbedderConfig:
         self.deployment_name = deployment_name
         self.vector_dimension = vector_dimension
         self.api_key = api_key
+        self.api_base = api_base

embedchain/config/llm/base.py

@@ -93,6 +93,7 @@ class BaseLlmConfig(BaseConfig):
         query_type: Optional[str] = None,
         callbacks: Optional[list] = None,
         api_key: Optional[str] = None,
+        base_url: Optional[str] = None,
         endpoint: Optional[str] = None,
         model_kwargs: Optional[dict[str, Any]] = None,
         local: Optional[bool] = False,
@@ -167,6 +168,7 @@ class BaseLlmConfig(BaseConfig):
         self.query_type = query_type
         self.callbacks = callbacks
         self.api_key = api_key
+        self.base_url = base_url
         self.endpoint = endpoint
         self.model_kwargs = model_kwargs
         self.local = local