UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 13:40:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: chat completed

Browse Source
This commit is contained in:
Timothy J. Baek 2024-05-30 21:43:10 -07:00
parent e1d65065f5
commit 995f7bc51b
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
backend/main.py
View File

@ -271,6 +271,11 @@ class PipelineMiddleware(BaseHTTPMiddleware):
except: except:
pass pass
model = app.state.MODELS[model_id]
if "pipeline" in model:
sorted_filters.append(model)
for filter in sorted_filters: for filter in sorted_filters:
r = None r = None
try: try:
@ -490,6 +495,11 @@ async def chat_completed(form_data: dict, user=Depends(get_verified_user)):
] ]
sorted_filters = sorted(filters, key=lambda x: x["pipeline"]["priority"]) sorted_filters = sorted(filters, key=lambda x: x["pipeline"]["priority"])
model = app.state.MODELS[model_id]
if "pipeline" in model:
sorted_filters = [model] + sorted_filters
for filter in sorted_filters: for filter in sorted_filters:
r = None r = None
try: try:

2
src/lib/components/chat/Chat.svelte
View File

@ -899,7 +899,7 @@
const messages = createMessagesList(responseMessageId); const messages = createMessagesList(responseMessageId);
const res = await chatCompleted(localStorage.token, { const res = await chatCompleted(localStorage.token, {
model: model, model: model.id,
messages: messages.map((m) => ({ messages: messages.map((m) => ({
id: m.id, id: m.id,
role: m.role, role: m.role,