feat: Add OpenAI and Azure AI API endpoints with connection testing functionality

app/api/azure.py

@@ -0,0 +1,124 @@
+"""
+Azure AI API endpoints
+"""
+from fastapi import APIRouter, Request, HTTPException, status
+import logging
+import os
+
+from app.auth import require_login
+from app.config import settings
+
+# Import the Azure modules including the administration client
+from azure.core.credentials import AzureKeyCredential
+from azure.ai.documentintelligence import DocumentIntelligenceClient
+from azure.ai.documentintelligence import DocumentIntelligenceAdministrationClient
+import azure.core.exceptions
+
+# Set up logging
+logger = logging.getLogger(__name__)
+
+router = APIRouter()
+
+@router.get("/azure/test")
+@require_login
+async def test_azure_connection(request: Request):
+    """
+    Test if the configured Azure Document Intelligence connection is valid.
+    Uses the DocumentIntelligenceAdministrationClient for testing the connection.
+    """
+    try:
+        logger.info("Testing Azure Document Intelligence connection")
+        
+        # Check if Azure configuration is present
+        if not settings.azure_endpoint or not settings.azure_ai_key:
+            logger.warning("Azure Document Intelligence configuration is incomplete")
+            missing = []
+            if not settings.azure_endpoint:
+                missing.append("endpoint")
+            if not settings.azure_ai_key:
+                missing.append("API key")
+                
+            return {
+                "status": "error",
+                "message": f"Azure Document Intelligence configuration is incomplete. Missing: {', '.join(missing)}"
+            }
+        
+        # Try to initialize the admin client and make a request to list operations
+        try:
+            # Initialize the admin client with credentials
+            admin_client = DocumentIntelligenceAdministrationClient(
+                endpoint=settings.azure_endpoint,
+                credential=AzureKeyCredential(settings.azure_ai_key)
+            )
+            
+            # Test the connection by listing operations - this is a documented method in the admin client
+            operations = list(admin_client.list_operations())
+            
+            # Successfully initialized client and made a request
+            logger.info("Azure Document Intelligence Admin connection successfully tested")
+            
+            # Return success with available operations info
+            operations_info = []
+            try:
+                for op in operations:
+                    if hasattr(op, 'operation_id') and op.operation_id:
+                        op_info = {
+                            "id": op.operation_id,
+                            "status": op.status if hasattr(op, 'status') else "Unknown",
+                            "created": str(op.created_on) if hasattr(op, 'created_on') else "Unknown",
+                            "kind": op.kind if hasattr(op, 'kind') else "Unknown"
+                        }
+                        operations_info.append(op_info)
+                
+                operation_count = len(operations_info)
+                return {
+                    "status": "success",
+                    "message": f"Azure Document Intelligence connection is valid. Found {operation_count} operations.",
+                    "endpoint": settings.azure_endpoint,
+                    "operations_count": operation_count,
+                    "recent_operations": operations_info[:3] if operations_info else []
+                }
+            except Exception as e:
+                # If error occurs while processing operations info, still return success
+                logger.warning(f"Connected to Azure but couldn't parse operations: {e}")
+                return {
+                    "status": "success",
+                    "message": "Azure Document Intelligence connection is valid, but couldn't retrieve operations details.",
+                    "endpoint": settings.azure_endpoint
+                }
+                
+        except azure.core.exceptions.ClientAuthenticationError as e:
+            logger.error(f"Azure authentication error: {e}")
+            return {
+                "status": "error",
+                "message": f"Authentication error: Invalid API key or credentials",
+                "detail": str(e)
+            }
+        except azure.core.exceptions.ServiceRequestError as e:
+            logger.error(f"Azure service request error: {e}")
+            return {
+                "status": "error",
+                "message": f"Service request error: Could not reach the Azure endpoint",
+                "detail": str(e)
+            }
+        except ValueError as e:
+            logger.error(f"Azure configuration value error: {e}")
+            return {
+                "status": "error",
+                "message": f"Configuration error: {str(e)}",
+                "detail": str(e)
+            }
+        except Exception as e:
+            logger.error(f"Azure connection test failed with unexpected error: {e}")
+            return {
+                "status": "error",
+                "message": f"Connection test failed with unexpected error",
+                "detail": str(e)
+            }
+    
+    except Exception as e:
+        logger.exception("Unexpected error testing Azure Document Intelligence connection")
+        return {
+            "status": "error",
+            "message": f"Unexpected error: {str(e)}"
+        }