Provide openai-key support from config (#1052)

This commit is contained in:
Sidharth Mohanty
2023-12-23 14:42:18 +05:30
committed by GitHub
parent e90673ae5b
commit 11f0d719f5
9 changed files with 31 additions and 43 deletions

View File

@@ -5,7 +5,9 @@ from embedchain.helpers.json_serializable import register_deserializable
@register_deserializable
class BaseEmbedderConfig:
def __init__(self, model: Optional[str] = None, deployment_name: Optional[str] = None):
def __init__(
self, model: Optional[str] = None, deployment_name: Optional[str] = None, api_key: Optional[str] = None
):
"""
Initialize a new instance of an embedder config class.
@@ -16,3 +18,4 @@ class BaseEmbedderConfig:
"""
self.model = model
self.deployment_name = deployment_name
self.api_key = api_key

View File

@@ -69,6 +69,7 @@ class BaseLlmConfig(BaseConfig):
where: Dict[str, Any] = None,
query_type: Optional[str] = None,
callbacks: Optional[List] = None,
api_key: Optional[str] = None,
):
"""
Initializes a configuration class instance for the LLM.
@@ -117,6 +118,7 @@ class BaseLlmConfig(BaseConfig):
self.system_prompt = system_prompt
self.query_type = query_type
self.callbacks = callbacks
self.api_key = api_key
if type(template) is str:
template = Template(template)