fix: fix the system message in simple-chatbot example (#1394)
This commit is contained in:
parent
8dd15ac71c
commit
c81ff64e94
1 changed files with 8 additions and 1 deletions
|
|
@ -291,8 +291,15 @@ class ChatSession:
|
|||
"""
|
||||
import json
|
||||
|
||||
def _clean_json_string(json_string: str) -> str:
|
||||
"""Remove ```json ... ``` or ``` ... ``` wrappers if the LLM response is fenced."""
|
||||
import re
|
||||
|
||||
pattern = r"^```(?:\s*json)?\s*(.*?)\s*```$"
|
||||
return re.sub(pattern, r"\1", json_string, flags=re.DOTALL | re.IGNORECASE).strip()
|
||||
|
||||
try:
|
||||
tool_call = json.loads(llm_response)
|
||||
tool_call = json.loads(_clean_json_string(llm_response))
|
||||
if "tool" in tool_call and "arguments" in tool_call:
|
||||
logging.info(f"Executing tool: {tool_call['tool']}")
|
||||
logging.info(f"With arguments: {tool_call['arguments']}")
|
||||
|
|
|
|||
Loading…
Reference in a new issue