chore: 迁移服务器前同步本地修改
This commit is contained in:
@@ -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} 未注册")
|
||||
|
||||
Reference in New Issue
Block a user