Public/kotaemon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use new Langchain's dedicated Azure OpenAI embedding class (#76)

Browse Source 
* Use new Langchain's dedicated Azure OpenAI embedding class

* Update test
This commit is contained in:
Nguyen Trung Duc (john)
2023-11-15 14:46:32 +07:00
committed by GitHub
parent b159897ac6
commit b52f312d8e
2 changed files with 6 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
knowledgehub/embeddings/openai.py
View File

@@ -1,4 +1,4 @@
from langchain.embeddings import OpenAIEmbeddings as LCOpenAIEmbeddings
from langchain import embeddings as lcembeddings
from .base import LangchainEmbeddings
@@ -9,23 +9,13 @@ class OpenAIEmbeddings(LangchainEmbeddings):
This method is wrapped around the Langchain OpenAIEmbeddings class.
"""
_lc_class = LCOpenAIEmbeddings
_lc_class = lcembeddings.OpenAIEmbeddings
class AzureOpenAIEmbeddings(LangchainEmbeddings):
"""Azure OpenAI embeddings.
This method is wrapped around the Langchain OpenAIEmbeddings class.
This method is wrapped around the Langchain AzureOpenAIEmbeddings class.
"""
_lc_class = LCOpenAIEmbeddings
def __init__(self, **params):
params["openai_api_type"] = "azure"
# openai.error.InvalidRequestError: Too many inputs. The max number of
# inputs is 16. We hope to increase the number of inputs per request
# soon. Please contact us through an Azure support request at:
# https://go.microsoft.com/fwlink/?linkid=2213926 for further questions.
params["chunk_size"] = 16
super().__init__(**params)
_lc_class = lcembeddings.AzureOpenAIEmbeddings