From b6830caa7190762fc02707179c0637bb36d22c6b Mon Sep 17 00:00:00 2001 From: Robert Date: Sat, 6 Jun 2026 20:30:09 -0300 Subject: [PATCH] Use A2A DeepAgents model resolver --- agent.py | 18 ++++-------------- 1 file changed, 4 insertions(+), 14 deletions(-) diff --git a/agent.py b/agent.py index 06da6cb..478d5a5 100644 --- a/agent.py +++ b/agent.py @@ -7,10 +7,9 @@ from pathlib import Path from typing import Any import httpx -from deepagents import create_deep_agent +from a2a_pack.deepagents import create_a2a_deep_agent from langchain_core.messages import BaseMessage from langchain_core.tools import StructuredTool -from langchain_openai import ChatOpenAI from pydantic import BaseModel, Field from a2a_pack import ( @@ -97,20 +96,11 @@ class ApifyApi(A2AAgent): ), "messages": [], } - model_kwargs: dict[str, Any] = { - "model": creds.model, - "base_url": creds.base_url, - "api_key": creds.api_key, - } - if creds.temperature_mode != "omit" and creds.temperature is not None: - model_kwargs["temperature"] = creds.temperature - if creds.extra_body: - model_kwargs["extra_body"] = dict(creds.extra_body) - model = ChatOpenAI(**model_kwargs) backend = ctx.workspace_backend() skills_root = _seed_runtime_skills(backend, ctx) - graph = create_deep_agent( - model=model, + graph = create_a2a_deep_agent( + ctx, + creds=creds, backend=backend, tools=[], subagents=self._operation_subagents(ctx, skills_root),