change model
This commit is contained in:
1 parent
e356a2631d
commit
e2449869aa
3 files changed
+36
-9
No files matched your search
+2
-1
@@ -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",
|
||||
|
||||
Reference in new issue
Block a user