feat(system):优化后端
1.新增后端测试 2.增加了后端的加密 3.增加了i18n(国际化)
This commit is contained in:
@@ -52,17 +52,21 @@ class BaseIndividual:
|
||||
self.agent_id = agent_config.get("agent_id")
|
||||
self.agent: Agent | None = None
|
||||
|
||||
async def _init_agent(self, agent_name: str, system_prompt: str):
|
||||
async def _init_agent(self, agent_name: str, system_prompt: str, toolsets=None):
|
||||
"""根据 agent_config 拉起一个 pydantic-ai Agent 实例。
|
||||
|
||||
从 GlobalStateMachine 取出 Provider,按 agent_config 中的 provider_title
|
||||
和 model_id 选择模型,加载工具列表,并把 system_prompt 注册为动态提示词。
|
||||
若调用方未显式提供 ``toolsets``,会自动从全局状态机拉取 MCP toolsets 注入。
|
||||
|
||||
Args:
|
||||
agent_name: Agent 的人类可读名称,用于日志与展示。
|
||||
system_prompt: 该 Agent 的基础系统提示词,会和 task_event 拼接成动态提示词。
|
||||
toolsets: 显式传入的外部工具集;为 ``None`` 时会自动拉取 MCP toolsets。
|
||||
"""
|
||||
from kilostar.utils.get_tool import load_tools_from_list
|
||||
from kilostar.utils.mcp_helper import get_all_toolsets_for_scope
|
||||
from kilostar.core.global_state_machine.gsm_snapshot import fetch_snapshot
|
||||
|
||||
global_state_machine = ray_actor_hook(
|
||||
"global_state_machine"
|
||||
@@ -73,13 +77,18 @@ class BaseIndividual:
|
||||
model_id = self.agent_config.get("model_id", "gpt-4o") # default fallback
|
||||
tools_list = self.agent_config.get("tools", None)
|
||||
|
||||
provider: Provider = await global_state_machine.get_provider.remote(
|
||||
provider_title
|
||||
)
|
||||
# 直读快照,避开 actor RPC 单线程串行
|
||||
snapshot = await fetch_snapshot(gsm_actor=global_state_machine)
|
||||
provider: Provider = snapshot.providers.get(provider_title)
|
||||
if provider is None:
|
||||
raise ValueError(f"Provider {provider_title!r} 未注册")
|
||||
agent_factory = AgentFactory()
|
||||
|
||||
callables = load_tools_from_list(tools_list)
|
||||
|
||||
if toolsets is None:
|
||||
toolsets = await get_all_toolsets_for_scope(agent_name)
|
||||
|
||||
self.agent = agent_factory.create_agent(
|
||||
provider=provider,
|
||||
model_id=model_id,
|
||||
@@ -88,6 +97,7 @@ class BaseIndividual:
|
||||
deps_type=WorkerIndividualDeps,
|
||||
agent_name=agent_name,
|
||||
tools=callables,
|
||||
toolsets=toolsets,
|
||||
)
|
||||
|
||||
@self.agent.system_prompt
|
||||
|
||||
Reference in New Issue
Block a user