Merge pull request #59 from eddiedunn/codex/document-functions-in-router/main.py-and-local_agent/main.py

Add missing docstrings

Author: eddiedunn

local_agent/main.py

@@ -47,6 +47,8 @@ async def _post_with_retry(path: str, payload: dict) -> None:
 
 
 async def register_with_router() -> None:
+    """Register this agent with the router service."""
+
     payload = {
         "name": AGENT_NAME,
         "endpoint": AGENT_ENDPOINT,
@@ -56,19 +58,25 @@ async def register_with_router() -> None:
 
 
 async def heartbeat_loop() -> None:
+    """Send regular heartbeat messages to the router."""
+
     while True:
         await _post_with_retry("/heartbeat", {"name": AGENT_NAME})
         await asyncio.sleep(HEARTBEAT_INTERVAL)
 
 
 @app.on_event("startup")
 async def _startup() -> None:
+    """Start background registration and heartbeat tasks."""
+
     asyncio.create_task(register_with_router())
     asyncio.create_task(heartbeat_loop())
 
 
 @app.post("/infer")
 async def infer(payload: ChatCompletionRequest):
+    """Return a trivial echo response for the given request."""
+
     user_msg = payload.messages[-1].content if payload.messages else ""
     content = f"Echo: {user_msg}"
     response = {

router/main.py

@@ -108,6 +108,8 @@
 
 
 async def cache_get(key: str) -> str | None:
+    """Return cached value if present and not expired."""
+
     entry = CACHE_STORE.get(key)
     if entry is None:
         return None
@@ -119,6 +121,8 @@ async def cache_get(key: str) -> str | None:
 
 
 async def cache_set(key: str, value: str, ttl: int = CACHE_TTL) -> None:
+    """Store a value in the cache for ``ttl`` seconds."""
+
     CACHE_STORE[key] = (time.time() + ttl, value)