|
|
@@ -18,8 +18,10 @@ from openai.types.chat.chat_completion_message_tool_call import Function
|
|
|
def new_message(role: str, content: str, reasoning_content=None):
|
|
|
if role != "user" and role != "system" and role != "assistant":
|
|
|
raise ValueError(f"Invalid role {role}")
|
|
|
-
|
|
|
- return {"role": role, "content": content}
|
|
|
+ if reasoning_content is None:
|
|
|
+ return {"role": role, "content": content}
|
|
|
+ else:
|
|
|
+ return {"role": role, "content": content, "reasoning_content": reasoning_content}
|
|
|
|
|
|
|
|
|
def system_message(content: str):
|
|
|
@@ -35,7 +37,7 @@ def assistant_message(content: str):
|
|
|
|
|
|
|
|
|
def tool_calls(tool_calls, reasoning_content=None):
|
|
|
- if reasoning_content is None or reasoning_content == "":
|
|
|
+ if reasoning_content is None:
|
|
|
return {"role": "assistant", "tool_calls": tool_calls}
|
|
|
else:
|
|
|
return {"role": "assistant", "tool_calls": tool_calls, "reasoning_content": reasoning_content}
|