change model

This commit is contained in:
xuyue committed 2026-03-31 05:50:27 -06:00
1 parent e356a2631d
commit e2449869aa
3 files changed
+36 -9

No files matched your search

+2 -1
View File
@@ -39,6 +39,7 @@ class PromptConsumer(AsyncWebsocketConsumer):
return
prompt = data.get("content", "").strip()
model = data.get("model", "")
if not prompt:
return
@@ -52,7 +53,7 @@ class PromptConsumer(AsyncWebsocketConsumer):
# Stream AI response
full_response = ""
try:
async for chunk in stream_chat(history):
async for chunk in stream_chat(history, model=model):
full_response += chunk
await self.send(text_data=json.dumps({
"type": "stream",