[AUR-389] Add base interface and embedding model (#17)

Browse Source

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")
```

...

This commit is contained in:

Nguyen Trung Duc (john)

2023-09-14 14:08:58 +07:00

committed by GitHub

parent 1061192731

commit c339912312

6 changed files with 4772 additions and 3 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

1552

tests/resources/embedding_openai.json Normal file

File diff suppressed because it is too large Load Diff

3094

tests/resources/embedding_openai_batch.json Normal file

File diff suppressed because it is too large Load Diff