G-Madhuri's picture
deploy to hf space
7e2f74d
Raw
History Blame Contribute Delete
827 Bytes
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
)