feat: switch tool entities and format

This commit is contained in:
Junyan Qin
2025-06-15 12:51:51 +08:00
parent c5eeab2fd0
commit 0c2560cafb
55 changed files with 455 additions and 774 deletions
+8 -6
View File
@@ -18,13 +18,15 @@ class LocalAgentRunner(runner.RequestRunner):
req_messages = query.prompt.messages.copy() + query.messages.copy() + [query.user_message]
use_llm_model = await self.ap.model_mgr.get_model_by_uuid(query.use_llm_model_uuid)
# 首次请求
msg = await query.use_llm_model.requester.invoke_llm(
msg = await use_llm_model.requester.invoke_llm(
query,
query.use_llm_model,
use_llm_model,
req_messages,
query.use_funcs,
extra_args=query.use_llm_model.model_entity.extra_args,
extra_args=use_llm_model.model_entity.extra_args,
)
yield msg
@@ -61,12 +63,12 @@ class LocalAgentRunner(runner.RequestRunner):
req_messages.append(err_msg)
# 处理完所有调用,再次请求
msg = await query.use_llm_model.requester.invoke_llm(
msg = await use_llm_model.requester.invoke_llm(
query,
query.use_llm_model,
use_llm_model,
req_messages,
query.use_funcs,
extra_args=query.use_llm_model.model_entity.extra_args,
extra_args=use_llm_model.model_entity.extra_args,
)
yield msg