c339912312
This change provides the base interface of an embedding, and wrap the Langchain's OpenAI embedding. Usage as follow:
```python
from kotaemon.embeddings import AzureOpenAIEmbeddings
model = AzureOpenAIEmbeddings(
model="text-embedding-ada-002",
deployment="embedding-deployment",
openai_api_base="https://test.openai.azure.com/",
openai_api_key="some-key",
)
output = model("Hello world")
```
16 lines
394 B
Python
16 lines
394 B
Python
from langchain.embeddings import OpenAIEmbeddings as LCOpenAIEmbeddings
|
|
|
|
from .base import LangchainEmbeddings
|
|
|
|
|
|
class OpenAIEmbeddings(LangchainEmbeddings):
|
|
_lc_class = LCOpenAIEmbeddings
|
|
|
|
|
|
class AzureOpenAIEmbeddings(LangchainEmbeddings):
|
|
_lc_class = LCOpenAIEmbeddings
|
|
|
|
def __init__(self, **params):
|
|
params["openai_api_type"] = "azure"
|
|
super().__init__(**params)
|