Merge pull request #3 from balureddy003/codex/add-chat-api-endpoint-and-tests

Add chat endpoint and tests

Author: balureddy003

src/backend/app_kernel.py

@@ -20,6 +20,7 @@
 from fastapi import FastAPI, HTTPException, Query, Request
 from fastapi.middleware.cors import CORSMiddleware
 from kernel_agents.agent_factory import AgentFactory
+from pydantic import BaseModel
 
 # Local imports
 from middleware.health_check import HealthCheckMiddleware
@@ -168,6 +169,55 @@ async def user_browser_language_endpoint(
     return {"status": "Language received successfully"}
 
 
+class ChatRequest(BaseModel):
+    """Request model for the simple chat endpoint."""
+
+    session_id: str
+    user_message: str
+
+
+@app.post("/api/chat")
+async def chat_endpoint(chat_request: ChatRequest, request: Request):
+    """Handle a simple chat message from the user."""
+
+    authenticated_user = get_authenticated_user_details(request_headers=request.headers)
+    user_id = authenticated_user["user_principal_id"]
+
+    if not user_id:
+        track_event_if_configured(
+            "UserIdNotFound", {"status_code": 400, "detail": "no user"}
+        )
+        raise HTTPException(status_code=400, detail="no user")
+
+    kernel, memory_store = await initialize_runtime_and_context(
+        chat_request.session_id, user_id
+    )
+
+    client = None
+    try:
+        client = config.get_ai_project_client()
+    except Exception as client_exc:  # pylint: disable=broad-except
+        logging.error(f"Error creating AIProjectClient: {client_exc}")
+
+    simple_chat_agent = await AgentFactory.create_agent(
+        agent_type=AgentType.GENERIC,
+        session_id=chat_request.session_id,
+        user_id=user_id,
+        memory_store=memory_store,
+        client=client,
+    )
+
+    reply = await simple_chat_agent.handle_user_message(chat_request.user_message)
+
+    if client:
+        try:
+            client.close()
+        except Exception as e:  # pylint: disable=broad-except
+            logging.error(f"Error sending to AIProjectClient: {e}")
+
+    return {"reply": reply}
+
+
 @app.post("/api/input_task")
 async def input_task_endpoint(input_task: InputTask, request: Request):
     """

src/backend/tests/test_chat_endpoint.py

@@ -0,0 +1,81 @@
+"""Tests for the simple chat API endpoint."""
+
+import os
+import sys
+from unittest.mock import AsyncMock, MagicMock, patch
+
+import pytest
+from fastapi.testclient import TestClient
+
+# Mock Azure dependencies to avoid import errors during tests
+sys.modules["azure.monitor"] = MagicMock()
+sys.modules["azure.monitor.events.extension"] = MagicMock()
+sys.modules["azure.monitor.opentelemetry"] = MagicMock()
+
+# Set required environment variables before importing the app
+os.environ["COSMOSDB_ENDPOINT"] = "https://mock-endpoint"
+os.environ["COSMOSDB_KEY"] = "mock-key"
+os.environ["COSMOSDB_DATABASE"] = "mock-database"
+os.environ["COSMOSDB_CONTAINER"] = "mock-container"
+os.environ["APPLICATIONINSIGHTS_CONNECTION_STRING"] = (
+    "InstrumentationKey=mock;IngestionEndpoint=https://mock"
+)
+os.environ["AZURE_OPENAI_DEPLOYMENT_NAME"] = "mock-deployment-name"
+os.environ["AZURE_OPENAI_API_VERSION"] = "2023-01-01"
+os.environ["AZURE_OPENAI_ENDPOINT"] = "https://mock-openai-endpoint"
+os.environ["FRONTEND_SITE_NAME"] = "http://localhost"
+
+with patch("azure.monitor.opentelemetry.configure_azure_monitor", MagicMock()):
+    from src.backend.app_kernel import app
+
+
+client = TestClient(app)
+
+
+@pytest.fixture(autouse=True)
+def mock_dependencies(monkeypatch):
+    """Mock dependencies used by the chat endpoint."""
+
+    # Mock authentication helper
+    monkeypatch.setattr(
+        "src.backend.app_kernel.get_authenticated_user_details",
+        lambda request_headers: {"user_principal_id": "mock-user"},
+    )
+
+    # Mock runtime initialization
+    async def mock_initialize_runtime_and_context(session_id, user_id):
+        return None, MagicMock()
+
+    monkeypatch.setattr(
+        "src.backend.app_kernel.initialize_runtime_and_context",
+        mock_initialize_runtime_and_context,
+    )
+
+    # Mock config client creation
+    monkeypatch.setattr(
+        "src.backend.app_kernel.config.get_ai_project_client",
+        lambda: MagicMock(close=MagicMock()),
+    )
+
+    # Mock AgentFactory.create_agent to return an agent with handle_user_message
+    async def mock_create_agent(*args, **kwargs):
+        agent = MagicMock()
+        agent.handle_user_message = AsyncMock(return_value="mock response")
+        return agent
+
+    monkeypatch.setattr(
+        "src.backend.app_kernel.AgentFactory.create_agent",
+        mock_create_agent,
+    )
+
+
+def test_chat_endpoint_returns_reply():
+    """Ensure the chat endpoint returns the agent's reply."""
+
+    payload = {"session_id": "123", "user_message": "Hello"}
+
+    response = client.post("/api/chat", json=payload)
+
+    assert response.status_code == 200
+    assert response.json() == {"reply": "mock response"}
+