UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-01-30 14:29:55 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: files

Browse Source
This commit is contained in:
Timothy J. Baek 2024-06-18 14:38:23 -07:00
parent d93160799f
commit eb21750466
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
backend/main.py
View File

@ -354,10 +354,10 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
print(f"tool_context: {context}") print(f"tool_context: {context}")
# If docs field is present, generate RAG completions # If docs field is present, generate RAG completions
if "docs" in data: if "files" in data:
data = {**data} data = {**data}
rag_context, citations = get_rag_context( rag_context, citations = get_rag_context(
docs=data["docs"], docs=data["files"],
messages=data["messages"], messages=data["messages"],
embedding_function=rag_app.state.EMBEDDING_FUNCTION, embedding_function=rag_app.state.EMBEDDING_FUNCTION,
k=rag_app.state.config.TOP_K, k=rag_app.state.config.TOP_K,
@ -369,8 +369,7 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
if rag_context: if rag_context:
context += ("\n" if context != "" else "") + rag_context context += ("\n" if context != "" else "") + rag_context
del data["docs"] del data["files"]
log.debug(f"rag_context: {rag_context}, citations: {citations}") log.debug(f"rag_context: {rag_context}, citations: {citations}")
if context != "": if context != "":
@ -1380,7 +1379,6 @@ async def update_pipeline_valves(
) )
@app.get("/api/config") @app.get("/api/config")
async def get_app_config(): async def get_app_config():
# Checking and Handling the Absence of 'ui' in CONFIG_DATA # Checking and Handling the Absence of 'ui' in CONFIG_DATA