copilot-swe-agent[bot]
d08040ac4a
- Run Black formatter and isort on all app/ files - Remove unused imports (F401) across multiple files - Add # noqa: F401 for intentional re-exports in celery_worker.py, tasks/__init__.py, utils.py, frontend.py, views/base.py - Fix f-strings without placeholders (F541) in azure.py, notification.py, check_credentials.py, upload_to_onedrive.py, settings.py - Fix bare except (E722) in upload_to_sftp.py - Fix block comment format (E265) in models.py - Move imports to top of file to fix E402 in celery_app.py, celery_worker.py - Fix line-too-long (E501) by wrapping strings in multiple files - Remove unused variable (F841) in upload_to_nextcloud.py Co-authored-by: christianlouis <361235+christianlouis@users.noreply.github.com>
115 lines
4.8 KiB
Python
115 lines
4.8 KiB
Python
"""
|
|
Azure AI API endpoints
|
|
"""
|
|
|
|
import logging
|
|
|
|
import azure.core.exceptions
|
|
from azure.ai.documentintelligence import DocumentIntelligenceAdministrationClient
|
|
|
|
# Import the Azure modules including the administration client
|
|
from azure.core.credentials import AzureKeyCredential
|
|
from fastapi import APIRouter, Request
|
|
|
|
from app.auth import require_login
|
|
from app.config import settings
|
|
|
|
# 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": "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": "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": "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)}"}
|