Tiny fix

xusenlinzy · Nov 7, 2023 · cf3f92e · cf3f92e
1 parent eec8154
commit cf3f92e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/api/generation/qwen.py b/api/generation/qwen.py
@@ -181,7 +181,6 @@ def process_qwen_messages(messages: List[ChatMessage], functions: Union[dict, Li
                         content = f"Thought: 我可以使用 {f_name} API。"
                     else:
                         content = f"Thought: I can use {f_name}."
-                content = f"\n{content}\nAction: {f_name}\nAction Input: {f_args}"
 
             if messages[-1].role == Role.USER:
                 messages.append(
@@ -247,7 +246,7 @@ def parse_response(response):
 
     if func_name:
         function_call = {"name": func_name, "arguments": func_args}
-        return response[:i], function_call
+        return response[:k], function_call
 
     z = response.rfind("\nFinal Answer: ")
     if z >= 0:

diff --git a/examples/qwen-7b-chat/get_weather.py b/examples/qwen-7b-chat/get_weather.py
@@ -107,7 +107,8 @@ def run_conversation(query: str, stream=False, functions=None, max_retry=5):
                     params["messages"].append(
                         {
                             "role": "assistant",
-                            "content": output
+                            "content": output,
+                            "function_call": function_call,
                         }
                     )
                     params["messages"].append(