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

Combine docstores and vectorstores within a storages component (#72)

Browse Source
This commit is contained in:
Nguyen Trung Duc (john)
2023-11-14 17:50:57 +07:00
committed by GitHub
parent 640962e916
commit b159897ac6
18 changed files with 36 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_tools.py
View File

@@ -5,12 +5,11 @@ import pytest
from openai.resources.embeddings import Embeddings
from kotaemon.base import Document
from kotaemon.docstores import InMemoryDocumentStore
from kotaemon.embeddings.openai import AzureOpenAIEmbeddings
from kotaemon.pipelines.indexing import IndexVectorStoreFromDocumentPipeline
from kotaemon.pipelines.retrieving import RetrieveDocumentFromVectorStorePipeline
from kotaemon.pipelines.tools import ComponentTool, GoogleSearchTool, WikipediaTool
from kotaemon.vectorstores import ChromaVectorStore
from kotaemon.storages import ChromaVectorStore, InMemoryDocumentStore
with open(Path(__file__).parent / "resources" / "embedding_openai.json") as f:
openai_embedding = json.load(f)