fix: keep handler reconnect on the calling task on Python 3.11.

asyncio.wait_for spawned a nested task, so recovery cancelled the handler. Also clear the Ruff findings that failed CI.

Author: AKKI0511

agentconnect/agent/session.py

@@ -949,11 +949,13 @@ async def _call(self, op: str, *args: Any, **kwargs: Any) -> dict[str, Any]:
         except TransportError as exc:
             if _should_reconnect(exc):
                 try:
-                    await asyncio.wait_for(
-                        self._reconnect(),
-                        timeout=self._foreground_recovery_seconds(),
-                    )
-                except asyncio.TimeoutError:
+                    # asyncio.timeout keeps reconnect on this task. wait_for()
+                    # on 3.11 wraps it in a new Task, so current_task() inside
+                    # _reconnect is not the handler and _abandon_sdk_handlers
+                    # cancels the caller.
+                    async with asyncio.timeout(self._foreground_recovery_seconds()):
+                        await self._reconnect()
+                except TimeoutError:
                     raise SessionError(
                         "unavailable",
                         "Session recovery timed out; the Runtime may already have accepted the operation",

agentconnect/index/CLIENT.md

@@ -48,7 +48,7 @@ client = RegistryAPIClient(
     connect_timeout=10.0,
     read_timeout=30.0,
     max_connections=10,
-    max_keepalive_connections=5
+    max_keepalive_connections=5,
 )
 ```
 
@@ -87,6 +87,7 @@ from agentconnect.index import RegistryAPIClient
 from agentconnect.index.registry import AgentRegistration
 from agentconnect.core import AgentType, InteractionMode, AgentIdentity, Capability
 
+
 async def register_agent_example():
     async with RegistryAPIClient() as client:
         # Create agent registration
@@ -98,19 +99,24 @@ async def register_agent_example():
             name="Data Processor",
             summary="Processes CSV and JSON data files",
             capabilities=[
-                Capability(name="csv_processing", description="Parse and transform CSV files"),
-                Capability(name="json_processing", description="Parse and transform JSON data")
+                Capability(
+                    name="csv_processing", description="Parse and transform CSV files"
+                ),
+                Capability(
+                    name="json_processing", description="Parse and transform JSON data"
+                ),
             ],
-            tags=["data", "processing", "csv", "json"]
+            tags=["data", "processing", "csv", "json"],
         )
-        
+
         # Register agent
         success = await client.register(agent)
         if success:
             print(f"Successfully registered {agent.agent_id}")
         else:
             print("Registration failed")
 
+
 # Run the example
 asyncio.run(register_agent_example())
 ```
@@ -120,21 +126,21 @@ asyncio.run(register_agent_example())
 ```python
 async def discover_agents_example():
     async with RegistryAPIClient() as client:
-        
         # Find agents using semantic search
         results = await client.get_by_capability_semantic(
             capability_description="process data files and generate reports",
             limit=5,
             similarity_threshold=0.3,
-            filters={"tags": ["data", "reporting"]}
+            filters={"tags": ["data", "reporting"]},
         )
-        
+
         # Process results
         for agent_reg, score in results:
             print(f"Found: {agent_reg.name} (Score: {score:.3f})")
             print(f"  Capabilities: {[cap.name for cap in agent_reg.capabilities]}")
             print(f"  Tags: {agent_reg.tags}")
 
+
 asyncio.run(discover_agents_example())
 ```
 
@@ -143,19 +149,21 @@ asyncio.run(discover_agents_example())
 ```python
 async def bulk_operations_example():
     async with RegistryAPIClient() as client:
-        
         # Get all agents from a specific organization
         org_agents = await client.get_by_organization("acme_corp")
         print(f"Found {len(org_agents)} agents from Acme Corp")
-        
+
         # Get all verified agents
         verified = await client.get_verified_agents()
         print(f"Found {len(verified)} verified agents")
-        
+
         # Get agents by interaction mode
-        api_agents = await client.get_by_interaction_mode(InteractionMode.AGENT_TO_AGENT)
+        api_agents = await client.get_by_interaction_mode(
+            InteractionMode.AGENT_TO_AGENT
+        )
         print(f"Found {len(api_agents)} A2A agents")
 
+
 asyncio.run(bulk_operations_example())
 ```
 
@@ -167,20 +175,20 @@ The client includes comprehensive error handling:
 async def error_handling_example():
     try:
         async with RegistryAPIClient() as client:
-            
             # This will automatically retry on network errors
             result = await client.get_registration("some_agent_id")
-            
+
             if result is None:
                 print("Agent not found (404)")
             else:
                 print(f"Found agent: {result.name}")
-                
+
     except httpx.RequestError as e:
         print(f"Network error after all retries: {e}")
     except Exception as e:
         print(f"Unexpected error: {e}")
 
+
 asyncio.run(error_handling_example())
 ```
 
@@ -211,16 +219,16 @@ The client uses `settings` from `agentconnect.config` and can be configured via
 
 ```python
 # Client configuration (available via settings.clients.registry)
-settings.clients.registry.base_url                    # Default: "http://localhost:8000"
-settings.clients.registry.default_timeout             # Default: 30.0
-settings.clients.registry.connect_timeout             # Default: 10.0
-settings.clients.registry.read_timeout                # Default: 30.0
-settings.clients.registry.pool_timeout                # Default: 5.0
-settings.clients.registry.max_retries                 # Default: 3
-settings.clients.registry.retry_backoff_factor        # Default: 0.5
-settings.clients.registry.retryable_status_codes      # Default: [502, 503, 504]
-settings.clients.registry.max_connections             # Default: 10
-settings.clients.registry.max_keepalive_connections   # Default: 5
+settings.clients.registry.base_url  # Default: "http://localhost:8000"
+settings.clients.registry.default_timeout  # Default: 30.0
+settings.clients.registry.connect_timeout  # Default: 10.0
+settings.clients.registry.read_timeout  # Default: 30.0
+settings.clients.registry.pool_timeout  # Default: 5.0
+settings.clients.registry.max_retries  # Default: 3
+settings.clients.registry.retry_backoff_factor  # Default: 0.5
+settings.clients.registry.retryable_status_codes  # Default: [502, 503, 504]
+settings.clients.registry.max_connections  # Default: 10
+settings.clients.registry.max_keepalive_connections  # Default: 5
 ```
 
 Example `agentconnect.yaml` configuration:
@@ -288,6 +296,7 @@ Enable debug logging for detailed HTTP request/response information:
 
 ```python
 import logging
+
 logging.basicConfig(level=logging.DEBUG)
 
 # Will show detailed HTTP logs

agentconnect/index/README.md

@@ -81,7 +81,10 @@ app = create_registry_api_app(custom)
 # Optionally run with Uvicorn programmatically
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host=custom.host, port=custom.port, log_level=custom.log_level.lower())
+
+    uvicorn.run(
+        app, host=custom.host, port=custom.port, log_level=custom.log_level.lower()
+    )
 ```
 
 Preferred CLI:

agentconnect/index/client.py

@@ -273,9 +273,9 @@ async def _request(
                             or expected_status == 204
                         ):  # 204 No Content
                             # For DELETE or successful updates that don't return body, but we expect a model that might be a simple bool/dict
-                            if response_model == bool:
+                            if response_model is bool:
                                 return True
-                            if response_model == dict:
+                            if response_model is dict:
                                 return {}  # common for success messages
                             return None  # Or handle as per specific endpoint needs
 

agentconnect/index/registry/capability_discovery.py

@@ -289,14 +289,15 @@ async def precompute_all_capability_embeddings(
                 precompute_all_capability_embeddings as _precompute_all_capability_embeddings,
             )
 
-            capability_to_agent_map, total_points = (
-                await _precompute_all_capability_embeddings(
-                    self._async_qdrant_client,
-                    self.COLLECTION_NAME,
-                    self._embeddings_model,
-                    agent_registrations,
-                    self._get_batch_size(),
-                )
+            (
+                capability_to_agent_map,
+                total_points,
+            ) = await _precompute_all_capability_embeddings(
+                self._async_qdrant_client,
+                self.COLLECTION_NAME,
+                self._embeddings_model,
+                agent_registrations,
+                self._get_batch_size(),
             )
 
             # Update capability map

agentconnect/index/registry/capability_discovery_impl/README.md

@@ -77,7 +77,7 @@ Each indexed point includes a rich payload with metadata for filtering:
     "default_output_modes": ["text"],
     "payment_address": "0x1234...",
     "doc_id": "agent123_profile",
-    "doc_type": "agent_profile"
+    "doc_type": "agent_profile",
 }
 ```
 
@@ -111,7 +111,7 @@ discovery_service = CapabilityDiscoveryService()
 matching_agents = await discovery_service.find_by_capability_semantic(
     capability_description="translate English to French",
     limit=5,
-    filters={"tags": ["translation"], "default_input_modes": ["text"]}
+    filters={"tags": ["translation"], "default_input_modes": ["text"]},
 )
 ```
 

agentconnect/index/registry/capability_discovery_impl/indexing.py

@@ -473,10 +473,11 @@ async def main():
     print("===========[Capability to Agent Map from _generate_points]===========")
 
     # Test precompute_all_capability_embeddings
-    capability_to_agent_map_precomp, total_points_indexed = (
-        await precompute_all_capability_embeddings(
-            async_client, collection_name, embeddings_model, agent_registrations
-        )
+    (
+        capability_to_agent_map_precomp,
+        total_points_indexed,
+    ) = await precompute_all_capability_embeddings(
+        async_client, collection_name, embeddings_model, agent_registrations
     )
     print("===========[Capability to Agent Map after precompute]===========")
     print(

agentconnect/index/registry/search/README.md

@@ -78,8 +78,8 @@ Converts `List[Skill]` to `List[Dict[str, str]]` with "name" and "description" k
 
 ```python
 from agentconnect.index.registry.search import (
-    AgentSearchInput, 
-    populate_search_result_item
+    AgentSearchInput,
+    populate_search_result_item,
 )
 
 # Create search input
@@ -88,14 +88,14 @@ search_input = AgentSearchInput(
     top_k=10,
     strictness=0.3,
     output_detail="capabilities",
-    include_tags=["data", "analysis"]
+    include_tags=["data", "analysis"],
 )
 
 # Transform registry result to search result
 search_result = populate_search_result_item(
     registration=agent_registration,
     similarity_score=0.85,
-    output_detail_level="capabilities"
+    output_detail_level="capabilities",
 )
 ```
 
@@ -106,8 +106,7 @@ from agentconnect.index.registry.search import AgentSearchOutput
 
 # Create output with results
 output = AgentSearchOutput(
-    message="Found 3 agents matching your criteria",
-    results=[result1, result2, result3]
+    message="Found 3 agents matching your criteria", results=[result1, result2, result3]
 )
 
 # Get clean JSON representation

agentconnect/prebuilt/README.md

@@ -22,9 +22,11 @@ attach from the Session. Conversation state for Team work is ``ctx.history``.
 from agentconnect.prebuilt import AIAgent, Tool
 from agentconnect.team import Team
 
+
 async def search_docs(query: str) -> str:
     return f"no hits for {query}"
 
+
 agent = AIAgent(
     name="researcher",
     model="gpt-4o-mini",

agentconnect/team/directory/README.md

@@ -14,6 +14,7 @@ already available.
 from agentconnect.agent import BaseAgent
 from agentconnect.team import Team
 
+
 class Reviewer(BaseAgent):
     profile = {
         "summary": "Reviews contracts for risk and missing terms.",
@@ -30,6 +31,7 @@ class Reviewer(BaseAgent):
     async def handle(self, msg, ctx):
         return "reviewed"
 
+
 team = await Team("content-squad").start()
 await Reviewer(name="reviewer").join(team)
 await Researcher(name="researcher").join(team)
@@ -65,6 +67,7 @@ Set on the Team, not on each Agent.
 async def embed(texts: list[str]) -> list[list[float]]:
     return await my_model.encode(texts)
 
+
 team = await Team("content-squad", embeddings=embed).start()
 ```