shen/Bubbles
1
0
Fork 0
mirror of https://github.com/Zippland/Bubbles.git synced 2026-01-19 09:41:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

debug

Browse Source
This commit is contained in:
zihanjian
2025-09-25 12:59:27 +08:00
parent 30a7e8be66
commit 83e67961d2
1 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
function_calls/llm.py
View File

@@ -74,12 +74,22 @@ class FunctionCallLLM:
openai_functions = self._build_functions_for_openai(functions)
messages: List[Dict[str, Any]] = []
system_prompt = (
custom_prompt = None
system_msg = getattr(chat_model, "system_content_msg", None)
if isinstance(system_msg, dict):
custom_prompt = system_msg.get("content")
elif isinstance(system_msg, str):
custom_prompt = system_msg
if custom_prompt:
messages.append({"role": "system", "content": custom_prompt})
tool_prompt = (
"You are an assistant that can call tools. "
"When you invoke a function, wait for the tool response before replying to the user. "
"Only deliver a final answer once you have enough information."
)
messages.append({"role": "system", "content": system_prompt})
messages.append({"role": "system", "content": tool_prompt})
messages.append({"role": "user", "content": ctx.text})
for round_index in range(self.max_function_rounds):