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

[Bug fix] Fix Azure OpenAI related issue (#973)

Browse Source
This commit is contained in:
Deshraj Yadav
2023-11-23 01:40:54 -08:00
committed by GitHub
parent f6b80e01a1
commit 33409140b4
4 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
configs/chroma.yaml
View File

@@ -23,4 +23,3 @@ embedder:
provider: openai provider: openai
config: config:
model: 'text-embedding-ada-002' model: 'text-embedding-ada-002'
deployment_name: 'test-deployment'

4
embedchain/embedder/openai.py
View File

@@ -2,7 +2,7 @@ import os
from typing import Optional from typing import Optional
from chromadb.utils.embedding_functions import OpenAIEmbeddingFunction from chromadb.utils.embedding_functions import OpenAIEmbeddingFunction
from langchain.embeddings import OpenAIEmbeddings from langchain.embeddings import AzureOpenAIEmbeddings
from embedchain.config import BaseEmbedderConfig from embedchain.config import BaseEmbedderConfig
from embedchain.embedder.base import BaseEmbedder from embedchain.embedder.base import BaseEmbedder
@@ -17,7 +17,7 @@ class OpenAIEmbedder(BaseEmbedder):
self.config.model = "text-embedding-ada-002" self.config.model = "text-embedding-ada-002"
if self.config.deployment_name: if self.config.deployment_name:
embeddings = OpenAIEmbeddings(deployment=self.config.deployment_name) embeddings = AzureOpenAIEmbeddings(deployment=self.config.deployment_name)
embedding_fn = BaseEmbedder._langchain_default_concept(embeddings) embedding_fn = BaseEmbedder._langchain_default_concept(embeddings)
else: else:
if os.getenv("OPENAI_API_KEY") is None and os.getenv("OPENAI_ORGANIZATION") is None: if os.getenv("OPENAI_API_KEY") is None and os.getenv("OPENAI_ORGANIZATION") is None:

2
pyproject.toml
View File

@@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "embedchain" name = "embedchain"
version = "0.1.19" version = "0.1.20"
description = "Data platform for LLMs - Load, index, retrieve and sync any unstructured data" description = "Data platform for LLMs - Load, index, retrieve and sync any unstructured data"
authors = [ authors = [
"Taranjeet Singh <taranjeet@embedchain.ai>", "Taranjeet Singh <taranjeet@embedchain.ai>",

2
tests/apps/test_apps.py
View File

@@ -119,7 +119,7 @@ class TestAppFromConfig:
# Validate the Embedder config values # Validate the Embedder config values
embedder_config = config_data["embedder"]["config"] embedder_config = config_data["embedder"]["config"]
assert app.embedder.config.model == embedder_config["model"] assert app.embedder.config.model == embedder_config["model"]
assert app.embedder.config.deployment_name == embedder_config["deployment_name"] assert app.embedder.config.deployment_name == embedder_config.get("deployment_name")
def test_from_opensource_config(self, mocker): def test_from_opensource_config(self, mocker):
mocker.patch("embedchain.vectordb.chroma.chromadb.Client") mocker.patch("embedchain.vectordb.chroma.chromadb.Client")