klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Bug fix]: Fix issue of OPENAI_API_BASE env variable being mandatory (#1305)

Browse Source
This commit is contained in:
Deshraj Yadav
2024-03-05 14:07:44 -08:00
committed by GitHub
parent 4428768eaa
commit 4dc5c7348f
3 changed files with 8 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
embedchain/embedder/openai.py
View File

@@ -17,7 +17,7 @@ class OpenAIEmbedder(BaseEmbedder):
self.config.model = "text-embedding-ada-002"
api_key = self.config.api_key or os.environ["OPENAI_API_KEY"]
api_base = self.config.api_base or os.environ["OPENAI_API_BASE"]
api_base = self.config.api_base or os.environ.get("OPENAI_API_BASE")
if self.config.deployment_name:
embeddings = AzureOpenAIEmbeddings(deployment=self.config.deployment_name)