tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Projects 1 Releases 2 Packages 1 Activity Actions

fix

Browse source
This commit is contained in:
Timothy J. Baek 2024-04-14 19:56:33 -04:00
parent b1b72441bb
commit 741ed5dc4c
2 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
backend/apps/rag/utils.py
View file

@ -43,6 +43,8 @@ def query_embeddings_doc(collection_name: str, query_embeddings, k: int):
query_embeddings=[query_embeddings],
n_results=k,
)
log.info(f"query_embeddings_doc:result {result}")
return result
except Exception as e:
raise e
@ -155,7 +157,9 @@ def rag_messages(
openai_key,
openai_url,
):
log.debug(f"docs: {docs}")
log.debug(
f"docs: {docs} {messages} {embedding_engine} {embedding_model} {embedding_function} {openai_key} {openai_url}"
)
last_user_message_idx = None
for i in range(len(messages) - 1, -1, -1):