agent-factory: generate agent auto-parse-email-content-and-extrac

app/agent.py

@@ -0,0 +1,148 @@
+"""
+AutoParseEmailContentAndExtrac Agent — auto-generated by Agent Factory.
+"""
+import asyncio
+import logging
+import os
+
+import click
+import httpx
+import uvicorn
+from contextlib import asynccontextmanager
+from dotenv import load_dotenv
+from langchain_core.rate_limiters import InMemoryRateLimiter
+from langchain_openai import AzureChatOpenAI
+
+from starlette.applications import Starlette
+from starlette.requests import Request
+from starlette.responses import JSONResponse
+from starlette.routing import Route
+
+from app.config import (
+    AGENT_SELF_URL,
+    AZURE_OPENAI_API_KEY,
+    AZURE_OPENAI_API_VERSION,
+    AZURE_OPENAI_DEPLOYMENT,
+    AZURE_OPENAI_ENDPOINT,
+    LOG_LEVEL,
+    REGISTRY_URL,
+)
+from app.skills import AGENT_CONFIG, AUTO_PARSE_EMAIL_CONTENT_AND_EXTRAC_SKILLS
+from app.workflows.auto_parse_email_content_and_extrac_workflow import create_auto_parse_email_content_and_extrac_workflow
+
+load_dotenv()
+
+# ── Logging ──────────────────────────────────────────────────────────────
+logging.basicConfig(
+    level=getattr(logging, LOG_LEVEL.upper(), logging.INFO),
+    format="%(asctime)s %(name)s %(levelname)s %(message)s",
+)
+logger = logging.getLogger(__name__)
+
+# ── LLM ──────────────────────────────────────────────────────────────────
+rate_limiter = InMemoryRateLimiter(
+    requests_per_second=10 / 60,
+    check_every_n_seconds=0.1,
+    max_bucket_size=10,
+)
+
+llm = AzureChatOpenAI(
+    temperature=0,
+    azure_deployment=AZURE_OPENAI_DEPLOYMENT,
+    api_version=AZURE_OPENAI_API_VERSION,
+    azure_endpoint=AZURE_OPENAI_ENDPOINT or "",
+    api_key=AZURE_OPENAI_API_KEY or "",
+    max_retries=5,
+    timeout=120,
+    rate_limiter=rate_limiter,
+)
+
+workflow = create_auto_parse_email_content_and_extrac_workflow(llm)
+
+
+# ── Endpoints ────────────────────────────────────────────────────────────
+
+async def health_check(request: Request) -> JSONResponse:
+    return JSONResponse({"status": "healthy", "agent": 'AutoParseEmailContentAndExtrac'})
+
+
+async def agent_manifest(request: Request) -> JSONResponse:
+    """GET /.well-known/agent.json"""
+    return JSONResponse({
+        "name": AGENT_CONFIG["name"],
+        "version": AGENT_CONFIG["version"],
+        "description": AGENT_CONFIG["description"],
+        "url": "/",
+        "skills": [
+            {
+                "id": s.id,
+                "name": s.name,
+                "description": s.description,
+                "tags": s.tags,
+                "inputSchema": AGENT_CONFIG.get("input_schema"),
+                "outputSchema": AGENT_CONFIG.get("output_schema"),
+            } for s in AUTO_PARSE_EMAIL_CONTENT_AND_EXTRAC_SKILLS
+        ],
+        "capabilities": AGENT_CONFIG["capabilities"],
+    })
+
+
+async def process_endpoint(request: Request) -> JSONResponse:
+    """POST /process — run the agent workflow."""
+    try:
+        body = await request.json()
+        result = await workflow.ainvoke(body)
+        return JSONResponse(result)
+    except Exception as exc:
+        logger.error("Processing failed: %s", exc, exc_info=True)
+        return JSONResponse({"error": str(exc)}, status_code=500)
+
+
+# ── Self-registration ────────────────────────────────────────────────────
+
+async def _register_with_registry():
+    if not REGISTRY_URL:
+        logger.info("REGISTRY_URL not set — skipping self-registration")
+        return
+    await asyncio.sleep(2)
+    url = f"{REGISTRY_URL.rstrip('/')}/agents/register-url"
+    for attempt in range(3):
+        try:
+            async with httpx.AsyncClient(timeout=10.0) as client:
+                resp = await client.post(url, json={"endpoint": AGENT_SELF_URL})
+                if resp.status_code in (200, 201):
+                    logger.info("Self-registered with agent-registry at %s", REGISTRY_URL)
+                    return
+                logger.warning("Registration attempt %d: HTTP %d", attempt + 1, resp.status_code)
+        except Exception as exc:
+            logger.warning("Registration attempt %d failed: %s", attempt + 1, exc)
+        await asyncio.sleep(5)
+    logger.error("Failed to self-register after 3 attempts")
+
+
+@asynccontextmanager
+async def lifespan(app):
+    task = asyncio.create_task(_register_with_registry())
+    yield
+    task.cancel()
+
+
+app = Starlette(
+    routes=[
+        Route("/health", methods=["GET"], endpoint=health_check),
+        Route("/.well-known/agent.json", methods=["GET"], endpoint=agent_manifest),
+        Route("/process", methods=["POST"], endpoint=process_endpoint),
+    ],
+    lifespan=lifespan,
+)
+
+
+@click.command()
+@click.option("--host", default="0.0.0.0")
+@click.option("--port", default=8080, type=int)
+def main(host: str, port: int):
+    uvicorn.run(app, host=host, port=port, log_level=LOG_LEVEL.lower())
+
+
+if __name__ == "__main__":
+    main()