Christian Krakau-Louis
125 lines
5.2 KiB
Python
125 lines
5.2 KiB
Python
"""
|
|
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)}"
|
|
}
|