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

refac: rag prompt template

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2024-11-24 18:49:56 -08:00
parent 50c3be2136
commit bd28e1ed7d
4 changed files with 63 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
backend/open_webui/main.py
View File

@@ -49,7 +49,9 @@ from open_webui.apps.openai.main import (
get_all_models_responses as get_openai_models_responses,
)
from open_webui.apps.retrieval.main import app as retrieval_app
from open_webui.apps.retrieval.utils import get_sources_from_files, rag_template
from open_webui.apps.retrieval.utils import get_sources_from_files
from open_webui.apps.socket.main import (
app as socket_app,
periodic_usage_pool_cleanup,
@@ -122,11 +124,12 @@ from open_webui.utils.response import (
)
from open_webui.utils.security_headers import SecurityHeadersMiddleware
from open_webui.utils.task import (
moa_response_generation_template,
tags_generation_template,
query_generation_template,
emoji_generation_template,
rag_template,
title_generation_template,
query_generation_template,
tags_generation_template,
emoji_generation_template,
moa_response_generation_template,
tools_function_calling_generation_template,
)
from open_webui.utils.tools import get_tools