Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 20:07:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: integration

Browse Source
This commit is contained in:
Timothy J. Baek
2024-04-14 18:47:45 -04:00
parent 9cdb5bf9fe
commit 36ce157907
3 changed files with 28 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
backend/apps/rag/utils.py
View File

@@ -32,6 +32,7 @@ def query_doc(collection_name: str, query: str, k: int, embedding_function):
def query_embeddings_doc(collection_name: str, query_embeddings, k: int):
try:
# if you use docker use the model from the environment variable
log.info("query_embeddings_doc", query_embeddings)
collection = CHROMA_CLIENT.get_collection(
name=collection_name,
)
@@ -117,6 +118,8 @@ def query_collection(
def query_embeddings_collection(collection_names: List[str], query_embeddings, k: int):
results = []
log.info("query_embeddings_collection", query_embeddings)
for collection_name in collection_names:
try:
collection = CHROMA_CLIENT.get_collection(name=collection_name)