fix print log (#84)

cubenlp · Jun 23, 2024 · 950937d · 950937d
1 parent 3b25b79
commit 950937d
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/chattool/chattype.py b/chattool/chattype.py
@@ -196,6 +196,7 @@ def display_role_content(dic:dict, sep:Union[str, None]=None):
                 return f"{sep}{role}{sep}calling tool:\n{pformat(tools)}"
             if 'function_call' in dic:
                 return f"{sep}{role}{sep}calling function:\n{pformat(dic['function_call'])}"
+            return f"{sep}{role}{sep}{content}"
         else:
             raise Exception(f"Unknown role {role}")
 

diff --git a/tests/test_chattool.py b/tests/test_chattool.py
@@ -196,7 +196,6 @@ def test_error_message():
     assert resp.error_code == "invalid_api_key"
     assert resp.is_valid() == False
 
-
 def test_usage():
     resp = Resp(response=response)
     assert resp.total_tokens == 18