allowing extra fields in the response

Author: Akhilesh Negi

backend/app/api/routes/responses.py

@@ -8,7 +8,6 @@
 from sqlmodel import Session
 
 from app.api.deps import get_current_user_org, get_db
-from app.api.routes.threads import send_callback
 from app.crud.credentials import get_provider_credential
 from app.crud.assistants import get_assistant_by_id
 from app.models import UserOrganization
@@ -69,6 +68,11 @@ class _APIResponse(BaseModel):
     chunks: list[FileResultChunk]
     diagnostics: Optional[Diagnostics] = None
 
+    class Config:
+        extra = (
+            Extra.allow
+        )  # This allows additional fields to be included in the response
+
 
 class ResponsesAPIResponse(APIResponse[_APIResponse]):
     pass
@@ -101,7 +105,7 @@ def process_response(
 ):
     """Process a response and send callback with results."""
     logger.info(
-        f"[responses.process_response] Starting generating response for assistant_id={request.assistant_id}, project_id={request.project_id}, organization_id={organization_id}"
+        f"Starting generating response for assistant_id={request.assistant_id}, project_id={request.project_id}, organization_id={organization_id}"
     )
     try:
         response = client.responses.create(
@@ -121,7 +125,7 @@ def process_response(
         )
         response_chunks = get_file_search_results(response)
         logger.info(
-            f"[responses.process_response] Successfully generated response: response_id={response.id}, assistant={request.assistant_id}, project_id={request.project_id}, organization_id={organization_id}"
+            f"Successfully generated response: response_id={response.id}, assistant={request.assistant_id}, project_id={request.project_id}, organization_id={organization_id}"
         )
 
         # Convert request to dict and include all fields
@@ -155,18 +159,19 @@ def process_response(
     except openai.OpenAIError as e:
         error_message = handle_openai_error(e)
         logger.error(
-            f"[responses.process_response] OpenAI API error during response processing: {error_message}, project_id={request.project_id}, organization_id={organization_id}"
+            f"OpenAI API error during response processing: {error_message}, project_id={request.project_id}, organization_id={organization_id}"
         )
         callback_response = ResponsesAPIResponse.failure_response(error=error_message)
 
     if request.callback_url:
         logger.info(
-            f"[responses.process_response] Sending callback to URL: {request.callback_url}, assistant={request.assistant_id}, project_id={request.project_id}, organization_id={organization_id}"
+            f"Sending callback to URL: {request.callback_url}, assistant={request.assistant_id}, project_id={request.project_id}, organization_id={organization_id}"
         )
+        from app.api.routes.threads import send_callback
 
         send_callback(request.callback_url, callback_response.model_dump())
         logger.info(
-            f"[responses.process_response] Callback sent successfully, assistant={request.assistant_id}, project_id={request.project_id}, organization_id={organization_id}"
+            f"Callback sent successfully, assistant={request.assistant_id}, project_id={request.project_id}, organization_id={organization_id}"
         )
 
 
@@ -179,7 +184,7 @@ async def responses(
 ):
     """Asynchronous endpoint that processes requests in background."""
     logger.info(
-        f"[responses.responses] Processing response request for assistant_id={request.assistant_id}, project_id={request.project_id}, organization_id={_current_user.organization_id}"
+        f"Processing response request for assistant_id={request.assistant_id}, project_id={request.project_id}, organization_id={_current_user.organization_id}"
     )
 
     # Get assistant details
@@ -188,7 +193,7 @@ async def responses(
     )
     if not assistant:
         logger.error(
-            f"[responses.responses] Assistant not found: assistant_id={request.assistant_id}, project_id={request.project_id}, organization_id={_current_user.organization_id}"
+            f"Assistant not found: assistant_id={request.assistant_id}, project_id={request.project_id}, organization_id={_current_user.organization_id}"
         )
         raise HTTPException(
             status_code=404,
@@ -203,7 +208,7 @@ async def responses(
     )
     if not credentials or "api_key" not in credentials:
         logger.error(
-            f"[responses.responses] OpenAI API key not configured for org_id={_current_user.organization_id}, project_id={request.project_id}, organization_id={_current_user.organization_id}"
+            f"OpenAI API key not configured for org_id={_current_user.organization_id}, project_id={request.project_id}, organization_id={_current_user.organization_id}"
         )
         return {
             "success": False,
@@ -231,7 +236,7 @@ async def responses(
         process_response, request, client, assistant, _current_user.organization_id
     )
     logger.info(
-        f"[responses.responses] Background task scheduled for response processing: assistant_id={request.assistant_id}, project_id={request.project_id}, organization_id={_current_user.organization_id}"
+        f"Background task scheduled for response processing: assistant_id={request.assistant_id}, project_id={request.project_id}, organization_id={_current_user.organization_id}"
     )
 
     return initial_response