Add API response endpoint and tests

backend_service.py

@@ -3,8 +3,6 @@ FastAPI Backend AI Service converted from Gradio app
 Provides OpenAI-compatible chat completion endpoints
 """
 
-import os
-import sys
 import asyncio
 import logging
 import time
@@ -573,6 +571,22 @@ async def create_completion(
         logger.error(f"Error in completion: {e}")
         raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
 
+@app.post("/api/response")
+async def api_response(request: Request):
+    """Endpoint to receive and send responses via API."""
+    try:
+        data = await request.json()
+        message = data.get("message", "No message provided")
+        response: dict[str, str] = {
+            "status": "success",
+            "received_message": message,
+            "response_message": f"You sent: {message}"
+        }
+        return JSONResponse(content=response)
+    except Exception as e:
+        logger.error(f"Error processing API response: {e}")
+        raise HTTPException(status_code=500, detail="Internal server error")
+
 @app.exception_handler(Exception)
 async def global_exception_handler(request: Any, exc: Exception) -> JSONResponse:
     """Global exception handler"""

test_health_endpoint.py

@@ -0,0 +1,44 @@
+import requests
+
+def test_health_endpoint():
+    """Test the health endpoint of the API."""
+    base_url = "http://localhost:8000"
+    health_url = f"{base_url}/health"
+
+    try:
+        response = requests.get(health_url, timeout=10)
+        response.raise_for_status()
+        data = response.json()
+
+        assert response.status_code == 200, "Health endpoint did not return status 200"
+        assert data["status"] == "healthy", "Service is not healthy"
+        assert "model" in data, "Model information missing in health response"
+        assert "version" in data, "Version information missing in health response"
+
+        print("✅ Health endpoint test passed.")
+    except Exception as e:
+        print(f"❌ Health endpoint test failed: {e}")
+
+def test_api_response():
+    """Test the new API response endpoint."""
+    base_url = "http://localhost:8000"
+    response_url = f"{base_url}/api/response"
+
+    try:
+        payload = {"message": "Hello, API!"}
+        response = requests.post(response_url, json=payload, timeout=10)
+        response.raise_for_status()
+        data = response.json()
+
+        assert response.status_code == 200, "API response endpoint did not return status 200"
+        assert data["status"] == "success", "API response status is not success"
+        assert data["received_message"] == "Hello, API!", "Received message mismatch"
+        assert "response_message" in data, "Response message missing in API response"
+
+        print("✅ API response endpoint test passed.")
+    except Exception as e:
+        print(f"❌ API response endpoint test failed: {e}")
+
+if __name__ == "__main__":
+    test_health_endpoint()
+    test_api_response()