Skip to content

Commit 6833cc2

Browse files
committed
feat: removed files; modified handler and tests for a2a
1 parent 67bf4ab commit 6833cc2

File tree

10 files changed

+283
-1984
lines changed

10 files changed

+283
-1984
lines changed

agentic_rag/a2a_handler.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,8 +41,8 @@ def __init__(self, rag_agent, vector_store=None):
4141
"health.check": self.handle_health_check,
4242
}
4343

44-
# Register this agent in the registry
45-
self._register_self()
44+
# Initialize registration flag
45+
self._self_registered = False
4646

4747
def _register_self(self):
4848
"""Register this agent in the agent registry"""
@@ -179,6 +179,12 @@ async def handle_document_upload(self, params: Dict[str, Any]) -> Dict[str, Any]
179179
async def handle_agent_discover(self, params: Dict[str, Any]) -> Dict[str, Any]:
180180
"""Handle agent discovery requests"""
181181
try:
182+
# Ensure self is registered
183+
if not self._self_registered:
184+
logger.info("Self not registered yet, registering now...")
185+
self._register_self()
186+
self._self_registered = True
187+
182188
discover_params = AgentDiscoverParams(**params)
183189
logger.info(f"Agent discovery request: capability={discover_params.capability}, agent_id={discover_params.agent_id}")
184190

0 commit comments

Comments
 (0)