Skip to content

Commit c29e5ad

Browse files
committed
fix AgentSource
1 parent 964e1bc commit c29e5ad

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

examples/pydantic/my_agent.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -128,9 +128,6 @@ def _build_router_model() -> FunctionModel:
128128
return FunctionModel(_route_demo_prompt, model_name="acpkit-structured-demo-router")
129129

130130

131-
_ROUTER_MODEL: Final = _build_router_model()
132-
133-
134131
def _available_models() -> list[AdapterModel]:
135132
models = [
136133
AdapterModel(
@@ -327,6 +324,9 @@ def _route_demo_prompt(messages: list[ModelMessage], info: AgentInfo) -> ModelRe
327324
)
328325

329326

327+
_ROUTER_MODEL: Final = _build_router_model()
328+
329+
330330
def _trim_history(messages: list[ModelMessage]) -> list[ModelMessage]:
331331
return list(messages[-4:])
332332

packages/adapters/pydantic-acp/src/pydantic_acp/agent_source.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,8 @@ async def get_deps(
4040

4141
@dataclass(slots=True)
4242
class StaticAgentSource(Generic[AgentDepsT, OutputDataT]):
43-
deps: AgentDepsT | None
4443
agent: PydanticAgent[AgentDepsT, OutputDataT]
44+
deps: AgentDepsT | None
4545

4646
async def get_agent(self, session: AcpSessionContext) -> PydanticAgent[AgentDepsT, OutputDataT]:
4747
del session

0 commit comments

Comments
 (0)