feat(system):优化后端

1.新增后端测试
2.增加了后端的加密
3.增加了i18n(国际化)
This commit is contained in:
2026-05-31 15:39:34 +00:00
parent affe460180
commit 99520c69d7
118 changed files with 8174 additions and 1491 deletions
+14 -4
View File
@@ -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