chore: 迁移服务器前同步本地修改

This commit is contained in:
2026-08-05 21:26:13 +08:00
parent aa47a19e98
commit 1a914a2517
73 changed files with 908 additions and 1961 deletions
+6 -11
View File
@@ -13,30 +13,28 @@
# limitations under the License.
from pydantic_ai import Agent, RunContext
from pydantic import Field
from pydantic import BaseModel, Field
from kilostar.adapter.model_adapter.agent_factory import AgentFactory
from kilostar.core.global_state_machine.model_provider.base_provider import Provider
from kilostar.utils.agent_model import ResponseModel, RequestModel, DepsModel
from kilostar.utils.ray_hook import ray_actor_hook
from kilostar.utils.logger import get_logger
logger = get_logger("worker_individual")
class WorkerIndividualResponse(ResponseModel):
class WorkerIndividualResponse(BaseModel):
"""Worker Individual 的输出模型,承载一次任务执行后的结果文本。"""
output: str = Field(..., description="Worker执行任务的输出结果")
class WorkerIndividualDeps(DepsModel):
class WorkerIndividualDeps(BaseModel):
"""Worker Individual 的运行期依赖,注入到 pydantic-ai Agent 的 RunContext。"""
task_event: dict
class WorkerIndividualInput(RequestModel):
class WorkerIndividualInput(BaseModel):
"""Worker Individual 的输入模型,承载一次任务事件的入参。"""
task_event: dict
@@ -67,17 +65,14 @@ class BaseIndividual:
from kilostar.utils.mcp_helper import get_all_tools_and_toolsets_for_scope
from kilostar.core.global_state_machine.gsm_snapshot import fetch_snapshot
global_state_machine = ray_actor_hook(
"global_state_machine"
).global_state_machine
provider_title = self.agent_config.get(
"provider_title", "openai"
) # default fallback
model_id = self.agent_config.get("model_id", "gpt-4o") # default fallback
toolset_ids = self.agent_config.get("tools", None)
# 直读快照,避开 actor RPC 单线程串行
snapshot = await fetch_snapshot(gsm_actor=global_state_machine)
# 直读进程内缓存快照
snapshot = await fetch_snapshot()
provider: Provider = snapshot.providers.get(provider_title)
if provider is None:
raise ValueError(f"Provider {provider_title!r} 未注册")