put endpoint to update topic summary

Signed-off-by: Stephanie <[email protected]>

Author: yangcao77

src/app/endpoints/conversations_v2.py

@@ -10,9 +10,11 @@
 from configuration import configuration
 from models.cache_entry import CacheEntry
 from models.config import Action
+from models.requests import ConversationUpdateRequest
 from models.responses import (
     ConversationDeleteResponse,
     ConversationResponse,
+    ConversationUpdateResponse,
     ConversationsListResponseV2,
     UnauthorizedResponse,
 )
@@ -89,6 +91,29 @@
     }
 }
 
+conversation_update_responses: dict[int | str, dict[str, Any]] = {
+    200: {
+        "conversation_id": "123e4567-e89b-12d3-a456-426614174000",
+        "success": True,
+        "message": "Topic summary updated successfully",
+        "topic_summary": "Updated topic summary",
+    },
+    400: {
+        "description": "Missing or invalid credentials provided by client",
+        "model": UnauthorizedResponse,
+    },
+    401: {
+        "description": "Unauthorized: Invalid or missing Bearer token",
+        "model": UnauthorizedResponse,
+    },
+    404: {
+        "detail": {
+            "response": "Conversation not found",
+            "cause": "The specified conversation ID does not exist.",
+        }
+    },
+}
+
 
 @router.get("/conversations", responses=conversations_list_responses)
 @authorize(Action.LIST_CONVERSATIONS)
@@ -207,6 +232,58 @@ async def delete_conversation_endpoint_handler(
     )
 
 
+@router.put(
+    "/conversations/{conversation_id}", responses=conversation_update_responses
+)
+@authorize(Action.UPDATE_CONVERSATION)
+async def update_conversation_endpoint_handler(
+    conversation_id: str,
+    update_request: ConversationUpdateRequest,
+    auth: Any = Depends(get_auth_dependency()),
+) -> ConversationUpdateResponse:
+    """Handle request to update a conversation topic summary by ID."""
+    check_configuration_loaded(configuration)
+    check_valid_conversation_id(conversation_id)
+
+    user_id = auth[0]
+    logger.info(
+        "Updating topic summary for conversation %s for user %s",
+        conversation_id,
+        user_id,
+    )
+
+    skip_userid_check = auth[2]
+
+    if configuration.conversation_cache is None:
+        logger.warning("Conversation cache is not configured")
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail={
+                "response": "Conversation cache is not configured",
+                "cause": "Conversation cache is not configured",
+            },
+        )
+
+    check_conversation_existence(user_id, conversation_id)
+
+    # Update the topic summary in the cache
+    configuration.conversation_cache.set_topic_summary(
+        user_id, conversation_id, update_request.topic_summary, skip_userid_check
+    )
+
+    logger.info(
+        "Successfully updated topic summary for conversation %s for user %s",
+        conversation_id,
+        user_id,
+    )
+
+    return ConversationUpdateResponse(
+        conversation_id=conversation_id,
+        success=True,
+        message="Topic summary updated successfully",
+    )
+
+
 def check_valid_conversation_id(conversation_id: str) -> None:
     """Check validity of conversation ID format."""
     if not check_suid(conversation_id):

src/models/config.py

@@ -356,6 +356,9 @@ class Action(str, Enum):
 
     # Access the conversation delete endpoint
     DELETE_CONVERSATION = "delete_conversation"
+
+    # Access the conversation update endpoint
+    UPDATE_CONVERSATION = "update_conversation"
     FEEDBACK = "feedback"
     GET_MODELS = "get_models"
     GET_TOOLS = "get_tools"

src/models/requests.py

@@ -410,3 +410,29 @@ class FeedbackStatusUpdateRequest(BaseModel):
     def get_value(self) -> bool:
         """Return the value of the status attribute."""
         return self.status
+
+
+class ConversationUpdateRequest(BaseModel):
+    """Model representing a request to update a conversation topic summary.
+
+    Attributes:
+        topic_summary: The new topic summary for the conversation.
+
+    Example:
+        ```python
+        update_request = ConversationUpdateRequest(
+            topic_summary="Discussion about machine learning algorithms"
+        )
+        ```
+    """
+
+    topic_summary: str = Field(
+        ...,
+        description="The new topic summary for the conversation",
+        examples=["Discussion about machine learning algorithms"],
+        min_length=1,
+        max_length=1000,
+    )
+
+    # Reject unknown fields
+    model_config = {"extra": "forbid"}

src/models/responses.py

@@ -865,3 +865,38 @@ class FeedbackStatusUpdateResponse(BaseModel):
             ]
         }
     }
+
+
+class ConversationUpdateResponse(BaseModel):
+    """Model representing a response for updating a conversation topic summary.
+
+    Attributes:
+        conversation_id: The conversation ID (UUID) that was updated.
+        success: Whether the update was successful.
+        message: A message about the update result.
+
+    Example:
+        ```python
+        update_response = ConversationUpdateResponse(
+            conversation_id="123e4567-e89b-12d3-a456-426614174000",
+            success=True,
+            message="Topic summary updated successfully",
+        )
+        ```
+    """
+
+    conversation_id: str = Field(
+        ...,
+        description="The conversation ID (UUID) that was updated",
+        examples=["123e4567-e89b-12d3-a456-426614174000"],
+    )
+    success: bool = Field(
+        ...,
+        description="Whether the update was successful",
+        examples=[True],
+    )
+    message: str = Field(
+        ...,
+        description="A message about the update result",
+        examples=["Topic summary updated successfully"],
+    )

tests/integration/test_openapi_json.py

@@ -77,6 +77,11 @@ def test_servers_section_present(spec: dict):
             "delete",
             {"200", "400", "401", "404", "503", "422"},
         ),
+        (
+            "/v1/conversations/{conversation_id}",
+            "put",
+            {"200", "400", "401", "404", "503", "422"},
+        ),
         ("/readiness", "get", {"200", "503"}),
         ("/liveness", "get", {"200"}),
         ("/authorized", "post", {"200", "400", "401", "403"}),