add /agents/chat_with_tools_agent API

Author: ks6088ts

.env.template

@@ -36,8 +36,8 @@ DIFY_API_KEY="xxx"
 ## Cosmos DB Settings
 COSMOSDB_HOST="https://xxx.documents.azure.com:443/"
 COSMOSDB_KEY="xxx"
-COSMOSDB_DATABASE_NAME="langgraph"
-COSMOSDB_CONTAINER_NAME="docs_kabuto"
+COSMOSDB_DATABASE_NAME="template_langgraph"
+COSMOSDB_CONTAINER_NAME="kabuto"
 COSMOSDB_PARTITION_KEY="/id"
 
 # ---------

Makefile

@@ -118,4 +118,4 @@ langgraph-studio: ## run LangGraph Studio
 
 .PHONY: fastapi-dev
 fastapi-dev: ## run FastAPI
-	uv run fastapi dev ./template_langgraph/services/fastapis.py
+	uv run fastapi dev ./template_langgraph/services/fastapis/main.py

template_langgraph/loggers.py

@@ -1,16 +1,21 @@
 import logging
 
 
-def get_logger(name: str = "default") -> logging.Logger:
+def get_logger(
+    name: str = "default",
+    verbosity: int = logging.INFO,
+) -> logging.Logger:
     """
     Get a logger with the specified name.
 
     Args:
         name (str): The name of the logger.
+        verbosity (int): The logging level (default: logging.INFO).
     Returns:
         logging.Logger: Configured logger instance.
     """
     logger = logging.getLogger(name)
+    logger.setLevel(verbosity)
     formatter = logging.Formatter("%(asctime)s [%(levelname)8s] %(message)s (%(filename)s:%(lineno)s)")
     handler = logging.StreamHandler()
     handler.setFormatter(formatter)

template_langgraph/services/fastapis.py

@@ -0,0 +1,12 @@
+from fastapi import FastAPI
+
+from template_langgraph.services.fastapis.routers import agents as agents_router
+
+app = FastAPI()
+
+
+app.include_router(
+    agents_router.router,
+    prefix="/agents",
+    tags=["agents"],
+)

template_langgraph/services/fastapis/__init__.py

@@ -0,0 +1,56 @@
+import logging
+
+from fastapi import APIRouter
+from pydantic import BaseModel, ConfigDict
+
+from template_langgraph.agents.chat_with_tools_agent.agent import AgentState
+from template_langgraph.agents.chat_with_tools_agent.agent import graph as chat_with_tools_agent
+from template_langgraph.loggers import get_logger
+
+router = APIRouter()
+logger = get_logger(
+    name=__name__,
+    verbosity=logging.DEBUG,
+)
+
+
+class RunChatWithToolsAgentRequest(BaseModel):
+    model_config = ConfigDict(extra="ignore")
+    question: str
+
+
+class RunChatWithToolsAgentResponse(BaseModel):
+    model_config = ConfigDict(extra="ignore")
+    response: str
+
+
+@router.post(
+    "/chat_with_tools_agent/",
+    response_model=RunChatWithToolsAgentResponse,
+    operation_id="create_transcription_job",
+)
+async def run_chat_with_tools_agent(
+    request: RunChatWithToolsAgentRequest,
+) -> RunChatWithToolsAgentResponse:
+    async for event in chat_with_tools_agent.astream(
+        input=AgentState(
+            messages=[
+                {
+                    "role": "user",
+                    "content": request.question,
+                },
+            ],
+        ),
+        config={
+            "recursion_limit": 30,
+        },
+    ):
+        logger.debug(f"Event received: {event}")
+    try:
+        response = event["chat_with_tools"]["messages"][0].content
+        return RunChatWithToolsAgentResponse(response=response)
+    except Exception as e:
+        logger.error(f"Error processing event: {e}")
+        return RunChatWithToolsAgentResponse(
+            response=f"An error occurred while processing your request with {e}",
+        )