feat(agent): Gradio-powered LangGraph agent chat with streaming, tool calls, file upload, conversation persistence
- Gradio 5.50.0 ChatInterface with type='messages' streaming - LangGraph create_react_agent with InMemorySaver checkpointer - 4 @tool functions: search_dashboards, get_health_summary, list_environments, get_task_status - Structured ChatMessage metadata (7 discriminator types: stream_token, tool_start/end/error, confirm_required, confirm_resolved, error) - HITL resume via second submit() with interrupt_before/Command - Dual-identity RBAC: service JWT + user JWT for tool calls - File upload (10 MB limit, pdfplumber/xlsx/JSON parser) - Conversation persistence via POST /api/agent/conversations/save - REST API: list, history, archive conversations; multi-tab gate; LLM config - LLM provider selection via Admin -> LLM Settings (assistant_planner_provider) - Svelte 5 AgentChatModel with stream event queue, dedup, stream_status watcher - MarkdownRenderer using svelte-markdown with semantic Tailwind tokens - ToolCallCard (3 states: executing/completed/failed) - ConversationList with search, date grouping, infinite scroll - ConnectionIndicator with Gradio health status - /agent route with two-column layout - Vite proxy /api/agent/gradio -> Gradio SSE - Fixed: not_() SQLAlchemy operator, route collision with _admin_routes - Fixed: conversation_id -> id normalization, .pyc cache staleness - Fixed: event.data array parsing (Gradio returns [jsonStr, null]) - Requirements pinned: gradio==5.50.0, pydantic>=2.7,<=2.12.3
This commit is contained in:
59
backend/src/agent/middleware.py
Normal file
59
backend/src/agent/middleware.py
Normal file
@@ -0,0 +1,59 @@
|
||||
# backend/src/agent/middleware.py
|
||||
# #region AgentChat.Middleware [C:3] [TYPE Module] [SEMANTICS agent-chat,middleware,logging,audit]
|
||||
# @defgroup AgentChat Audit logging and confirmation risk middleware for LangGraph agent.
|
||||
# @BRIEF LoggingMiddleware writes tool-call events to assistant_audit table.
|
||||
# @RELATION DEPENDS_ON -> [Models.AssistantAuditRecord]
|
||||
# @RATIONALE FR-024: All agent interactions must be logged for auditability.
|
||||
# @REJECTED ConfirmationRiskMiddleware rejected — LangGraph interrupt_before=DANGEROUS_TOOLS handles HITL natively.
|
||||
|
||||
from datetime import UTC, datetime
|
||||
import logging
|
||||
|
||||
from src.agent.context import get_user_jwt
|
||||
|
||||
logger = logging.getLogger("cot")
|
||||
|
||||
|
||||
# #region AgentChat.Middleware.LoggingMiddleware [C:3] [TYPE Function] [SEMANTICS audit,tool,logging]
|
||||
# @ingroup AgentChat
|
||||
# @BRIEF Log every tool-call event to assistant_audit table with user context.
|
||||
# @PRE agent event has 'event' key with type on_tool_start/on_tool_end/on_tool_error.
|
||||
# @POST Audit record written to assistant_audit table (async, non-blocking).
|
||||
# @SIDE_EFFECT Writes to assistant_audit table via FastAPI REST call.
|
||||
# @RELATION DISPATCHES -> [Api.Assistant.Audit]
|
||||
|
||||
async def log_tool_event(event: dict, conversation_id: str) -> None:
|
||||
"""Log a tool-call event to the audit trail.
|
||||
|
||||
Args:
|
||||
event: LangGraph event dict with 'event', 'name', and 'data' keys.
|
||||
conversation_id: Current conversation thread ID.
|
||||
"""
|
||||
kind = event.get("event", "")
|
||||
tool_name = event.get("name", "unknown")
|
||||
user_jwt = get_user_jwt()
|
||||
|
||||
audit_payload = {
|
||||
"event_type": kind,
|
||||
"tool": tool_name,
|
||||
"conversation_id": conversation_id,
|
||||
"user_jwt_present": bool(user_jwt),
|
||||
"timestamp": datetime.now(UTC).isoformat(),
|
||||
}
|
||||
|
||||
if "data" in event:
|
||||
data = event["data"]
|
||||
if kind == "on_tool_start":
|
||||
audit_payload["input"] = str(data.get("input", ""))[:500]
|
||||
elif kind == "on_tool_error":
|
||||
audit_payload["error"] = str(data.get("error", ""))[:500]
|
||||
|
||||
logger.info(
|
||||
"Tool audit: %(event_type)s — %(tool)s — conv=%(conversation_id)s",
|
||||
audit_payload,
|
||||
)
|
||||
|
||||
# TODO: Async write to assistant_audit table via REST call to FastAPI
|
||||
# This is intentionally fire-and-forget — audit failures must not block tool execution
|
||||
# #endregion AgentChat.Middleware.LoggingMiddleware
|
||||
# #endregion AgentChat.Middleware
|
||||
Reference in New Issue
Block a user