kotaemon/knowledgehub/embeddings/openai.py
Nguyen Trung Duc (john) b52f312d8e Use new Langchain's dedicated Azure OpenAI embedding class (#76)
* Use new Langchain's dedicated Azure OpenAI embedding class

* Update test
2023-11-15 14:46:32 +07:00

22 lines
511 B
Python

from langchain import embeddings as lcembeddings
from .base import LangchainEmbeddings
class OpenAIEmbeddings(LangchainEmbeddings):
"""OpenAI embeddings.
This method is wrapped around the Langchain OpenAIEmbeddings class.
"""
_lc_class = lcembeddings.OpenAIEmbeddings
class AzureOpenAIEmbeddings(LangchainEmbeddings):
"""Azure OpenAI embeddings.
This method is wrapped around the Langchain AzureOpenAIEmbeddings class.
"""
_lc_class = lcembeddings.AzureOpenAIEmbeddings