Forráskód Böngészése

fix an index error

shaobo 1 éve
szülő
commit
feaa9d6b23
1 módosított fájl, 1 hozzáadás és 1 törlés
  1. 1 1
      local_mode/models/codegeex.py

+ 1 - 1
local_mode/models/codegeex.py

@@ -59,7 +59,7 @@ class CodegeexChatModel:
         try:
         try:
             response, _ = self.model.chat(
             response, _ = self.model.chat(
                 self.tokenizer,
                 self.tokenizer,
-                query=request.messages[0].content,
+                query=request.messages[-1].content,
                 history=[msg.model_dump() for msg in request.messages[:-1]],
                 history=[msg.model_dump() for msg in request.messages[:-1]],
                 max_new_tokens=request.max_tokens,
                 max_new_tokens=request.max_tokens,
                 temperature=request.temperature,
                 temperature=request.temperature,