Spaces:
Sleeping
Sleeping
| from typing import Dict, Any | |
| from agents.base_agent import BaseAgent | |
| class ADKAgentAdapter: | |
| """ | |
| Adapter wrapping existing BaseAgent specialized agents to conform to Google ADK | |
| agent invocation schemas. | |
| """ | |
| def __init__(self, agent: BaseAgent): | |
| self.agent = agent | |
| self.name = agent.__class__.__name__ | |
| async def execute_task(self, context: Dict[str, Any], query: str) -> Dict[str, Any]: | |
| profile = context.get("profile", {}) | |
| graph = context.get("graph", {}) | |
| summary = context.get("summary", {}) | |
| report = context.get("report", "") | |
| # Call the underlying agent | |
| return await self.agent.run( | |
| profile=profile, | |
| graph=graph, | |
| summary=summary, | |
| report=report, | |
| query=query | |
| ) | |