fix: chat completed

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

View File

@ -271,6 +271,11 @@ class PipelineMiddleware(BaseHTTPMiddleware):
except:
pass
model = app.state.MODELS[model_id]
if "pipeline" in model:
sorted_filters.append(model)
for filter in sorted_filters:
r = None
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"])
model = app.state.MODELS[model_id]
if "pipeline" in model:
sorted_filters = [model] + sorted_filters
for filter in sorted_filters:
r = None
try:

View File

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