chore: ignore zip files
This commit is contained in:
@@ -1,67 +0,0 @@
|
||||
import json
|
||||
from dataclasses import dataclass
|
||||
from typing import Any, Optional
|
||||
@dataclass
|
||||
class StepOutcome:
|
||||
data: Any
|
||||
next_prompt: Optional[str] = None
|
||||
should_exit: bool = False
|
||||
|
||||
|
||||
def try_call_generator(func, *args, **kwargs):
|
||||
ret = func(*args, **kwargs)
|
||||
if hasattr(ret, '__iter__') and not isinstance(ret, (str, bytes, dict, list)):
|
||||
ret = yield from ret
|
||||
return ret
|
||||
|
||||
class BaseHandler:
|
||||
def tool_before_callback(self, tool_name, args, content): pass
|
||||
def tool_after_callback(self, tool_name, args, content): pass
|
||||
def dispatch(self, tool_name, args, response):
|
||||
method_name = f"do_{tool_name}"
|
||||
if hasattr(self, method_name):
|
||||
_ = yield from try_call_generator(self.tool_before_callback, tool_name, args, response)
|
||||
ret = yield from try_call_generator(getattr(self, method_name), args, response)
|
||||
_ = yield from try_call_generator(self.tool_after_callback, tool_name, args, response)
|
||||
return ret
|
||||
else:
|
||||
yield f"❌ 未知工具: {tool_name}\n"
|
||||
return StepOutcome(None, "未知工具", "ERROR")
|
||||
|
||||
def json_default(o):
|
||||
if isinstance(o, set): return list(o)
|
||||
return str(o)
|
||||
|
||||
def agent_runner_loop(client, system_prompt, user_input, handler, tools_schema, max_turns=15):
|
||||
messages = [
|
||||
{"role": "system", "content": system_prompt},
|
||||
{"role": "user", "content": user_input}
|
||||
]
|
||||
for turn in range(max_turns):
|
||||
yield f"\n[🤖 LLM Thinking (Turn {turn+1})] ..."
|
||||
response = client.chat(messages=messages, tools=tools_schema)
|
||||
|
||||
if response.thinking: yield '<thinking>' + response.thinking + '</thinking>\n'
|
||||
yield response.content
|
||||
|
||||
if not response.tool_calls:
|
||||
tool_name, args = 'no_tool', {}
|
||||
else:
|
||||
tool_call = response.tool_calls[0]
|
||||
tool_name = tool_call.function.name
|
||||
args = json.loads(tool_call.function.arguments)
|
||||
|
||||
if tool_name == 'no_tool': pass
|
||||
else: yield f"\n\n正在调用工具: {tool_name},参数: {args}\n"
|
||||
outcome = yield from handler.dispatch(tool_name, args, response)
|
||||
|
||||
if outcome.next_prompt is None: return {'result': 'CURRENT_TASK_DONE', 'data': outcome.data}
|
||||
if outcome.should_exit: return {'result': 'EXITED', 'data': outcome.data}
|
||||
|
||||
next_prompt = ""
|
||||
if outcome.data is not None:
|
||||
datastr = json.dumps(outcome.data, ensure_ascii=False, default=json_default) if type(outcome.data) in [dict, list] else str(outcome.data)
|
||||
next_prompt += f"<tool_result>\n{datastr}\n</tool_result>\n\n"
|
||||
next_prompt += outcome.next_prompt
|
||||
messages = [{"role": "user", "content": next_prompt}]
|
||||
return {'result': 'MAX_TURNS_EXCEEDED'}
|
||||
Reference in New Issue
Block a user