agent-api-server 2.1.7__py3-none-any.whl

agent_api_server/api/v1/api.py

@@ -0,0 +1,25 @@
+import os
+from fastapi import APIRouter
+from agent_api_server.api.v1.schema import api_router as schema_router
+from agent_api_server.api.v1.thread import api_router as thread_router
+from agent_api_server.api.v1.config import api_router as config_router
+from agent_api_server.api.v1.graph import api_router as graph_router
+
+
+
+api_router = APIRouter()
+api_router.include_router(thread_router, prefix="/thread", tags=["thread"])
+api_router.include_router(schema_router, prefix="/schema", tags=["schema"])
+api_router.include_router(config_router, prefix="/config", tags=["config"])
+api_router.include_router(graph_router, prefix="/graph", tags=["graph"])
+
+
+
+@api_router.get("/health")
+async def health_check(
+):
+    """Health check endpoint with connection pool verification"""
+    return {
+        "message": "Welcome to the LangGraph FastAPI Server!",
+        "worker_pid": os.getpid(),
+    }

agent_api_server/api/v1/config.py

@@ -0,0 +1,57 @@
+import logging
+from fastapi import APIRouter
+from fastapi.responses import JSONResponse
+from agent_api_server.shared.base_model import error_response
+from agent_api_server.shared.util_func import load_graph_config, load_graph
+from langgraph.graph.state import CompiledStateGraph
+
+logger = logging.getLogger(__name__)
+api_router = APIRouter()
+
+@api_router.get(
+    "/",
+    responses={
+        200: {
+            "description": "Successfully returned config JSON Schema",
+            "content": {"application/schema+json": {}}
+        },
+        400: {"model": error_response, "description": "Invalid graph instance"},
+        422: {"model": error_response, "description": "Schema validation error"},
+        500: {"model": error_response, "description": "Internal server error"}
+    },
+    summary="Get JSON Schema for graph_name"
+)
+async def get_config_schema(graph_name: str):
+    try:
+        graph_cfg = await load_graph_config()
+        validated_name, graph_instance, _ = await load_graph(graph_name, graph_cfg, False)
+
+        if not isinstance(graph_instance, CompiledStateGraph):
+            raise error_response(
+                status_code=400,
+                error_type="invalid_graph",
+                message="Graph instance type mismatch",
+                graph_name=graph_name,
+                actual_type=type(graph_instance).__name__
+            )
+
+        return JSONResponse(
+            content=graph_instance.get_context_jsonschema(),
+            media_type="application/schema+json",
+        )
+    except ValueError as ve:
+        logger.error(f"Schema validation failed for {graph_name}: {str(ve)}", exc_info=True)
+        raise error_response(
+            status_code=422,
+            error_type="schema_validation",
+            message=str(ve),
+            graph_name=graph_name
+        )
+    except Exception as e:
+        logger.critical(f"Unexpected error in {graph_name}: {str(e)}", exc_info=True)
+        raise error_response(
+            status_code=500,
+            error_type="internal_error",
+            message=str(e),
+            graph_name=graph_name
+        )

agent_api_server/api/v1/graph.py

@@ -0,0 +1,59 @@
+import logging
+from fastapi import APIRouter
+from fastapi.responses import JSONResponse
+from fastapi import HTTPException
+from agent_api_server.shared.base_model import error_response
+from agent_api_server.shared.util_func import get_all_graph_names
+
+logger = logging.getLogger(__name__)
+api_router = APIRouter()
+
+
+@api_router.get(
+    "/",
+    responses={
+        200: {
+            "description": "Successfully returned list of available graphs",
+            "content": {
+                "application/json": {
+                    "example": {
+                        "graphs": ["Test_App", "iot_data_analyse"],
+                        "count": 2
+                    }
+                }
+            }
+        },
+        500: {"model": error_response, "description": "Internal server error"}
+    },
+    summary="Get all available graph names",
+    response_description="List of available graph names"
+)
+async def get_available_graphs():
+    """
+    Get all available graph names defined in the configuration file.
+
+    Returns:
+        JSON response containing:
+        - graphs: List of available graph names
+        - count: Number of available graphs
+    """
+    try:
+        graph_names = await get_all_graph_names()
+        return JSONResponse(
+            content={
+                "graphs": graph_names,
+                "count": len(graph_names)
+            },
+            status_code=200
+        )
+    except HTTPException as e:
+        raise e
+    except Exception as e:
+        logger.error(f"Failed to get graph names: {str(e)}")
+        raise HTTPException(
+            status_code=500,
+            detail={
+                "error": "internal_server_error",
+                "message": "Failed to retrieve graph names"
+            }
+        )

agent_api_server/api/v1/schema.py

@@ -0,0 +1,57 @@
+import logging
+from fastapi import APIRouter
+from fastapi.responses import JSONResponse
+from agent_api_server.shared.base_model import error_response
+from agent_api_server.shared.util_func import load_graph_config, load_graph
+from langgraph.graph.state import CompiledStateGraph
+
+logger = logging.getLogger(__name__)
+api_router = APIRouter()
+
+@api_router.get(
+    "/",
+    responses={
+        200: {
+            "description": "Successfully returned JSON Schema",
+            "content": {"application/schema+json": {}}
+        },
+        400: {"model": error_response, "description": "Invalid graph instance"},
+        422: {"model": error_response, "description": "Schema validation error"},
+        500: {"model": error_response, "description": "Internal server error"}
+    },
+    summary="Get JSON Schema for graph_name"
+)
+async def get_graph_schema(graph_name: str):
+    try:
+        graph_cfg = await load_graph_config()
+        validated_name, graph_instance, _ = await load_graph(graph_name, graph_cfg, False)
+
+        if not isinstance(graph_instance, CompiledStateGraph):
+            raise error_response(
+                status_code=400,
+                error_type="invalid_graph",
+                message="Graph instance type mismatch",
+                graph_name=graph_name,
+                actual_type=type(graph_instance).__name__
+            )
+
+        return JSONResponse(
+            content=graph_instance.get_input_jsonschema(),
+            media_type="application/schema+json",
+        )
+    except ValueError as ve:
+        logger.error(f"Schema validation failed for {graph_name}: {str(ve)}", exc_info=True)
+        raise error_response(
+            status_code=422,
+            error_type="schema_validation",
+            message=str(ve),
+            graph_name=graph_name
+        )
+    except Exception as e:
+        logger.critical(f"Unexpected error in {graph_name}: {str(e)}", exc_info=True)
+        raise error_response(
+            status_code=500,
+            error_type="internal_error",
+            message=str(e),
+            graph_name=graph_name
+        )