fix an index error

2024-11-23 12:16:33 +00:00 · 2024-07-11 23:27:15 +08:00 · 2024-07-11 23:27:15 +08:00 · feaa9d6b23
commit feaa9d6b23
parent 0963521071
1 changed files with 1 additions and 1 deletions
--- a/local_mode/models/codegeex.py
+++ b/local_mode/models/codegeex.py
@ -59,7 +59,7 @@ class CodegeexChatModel:
        try:
            response, _ = self.model.chat(
                self.tokenizer,
-                query=request.messages[0].content,
+                query=request.messages[-1].content,
                history=[msg.model_dump() for msg in request.messages[:-1]],
                max_new_tokens=request.max_tokens,
                temperature=request.temperature,