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

kilostar/adapter/model_adapter/agent_factory.py

@@ -24,8 +24,9 @@ from pydantic_ai.providers.deepseek import DeepSeekProvider
 from pydantic_ai.providers.google import GoogleProvider
 from pydantic_ai.toolsets import AbstractToolset
 
+from pydantic import BaseModel
+
 from kilostar.core.global_state_machine.model_provider import Provider
-from kilostar.utils.agent_model import ResponseModel, DepsModel
 from kilostar.utils.error import ModelNotExistError
 
 
@@ -65,9 +66,9 @@ class AgentFactory:
         self,
         provider: Provider,
         model_id: str,
-        output_type: ResponseModel,
+        output_type: type[BaseModel],
         system_prompt: str,
-        deps_type: DepsModel,
+        deps_type: type[BaseModel],
         agent_name: str,
         tools: list = None,
         toolsets: Sequence[AbstractToolset[Any]] = None,