feat: /chat/completions/{completion_id}/messages implementation completed

app/api/chat_api.py

@@ -3,16 +3,8 @@
 from typing import List
 from fastapi import APIRouter, HTTPException, Depends, Request
 from pydantic import BaseModel
-from app.schema.chat_schema import (
-    ChatCompletionRequest,
-    ChatCompletionResponse,
-    MessageResponse,
-    PlotResponse,
-)
-from app.schema.conversation_schema import (
-    ConversationResponse,
-    ConversationItemResponse,
-)
+from app.schema.chat_schema import ChatCompletionRequest, ChatCompletionResponse, MessageResponse, PlotResponse
+from app.schema.conversation_schema import ConversationResponse, ConversationItemResponse
 from app.service.chat_service import ChatService
 from app.security.auth_service import AuthService
 from app.core.api_response import api_response
@@ -40,9 +32,7 @@ async def get_version():
 @router.post("/chat/completions", response_model=ChatCompletionResponse)
 @api_response()
 async def create_chat_completion(
-    chat_completion: ChatCompletionRequest,
-    request: Request,
-    username: str = Depends(auth_service.verify_credentials),
+    chat_completion: ChatCompletionRequest, request: Request, username: str = Depends(auth_service.verify_credentials)
 ):
     """
     Chat completion API - Given a list of messages comprising a conversation, the model will return a response.
@@ -58,10 +48,7 @@ async def create_chat_completion(
 
 # get all chat completions
 @router.get("/chat/completions", response_model=List[ChatCompletionResponse], deprecated=True)
-async def list_chat_completions(
-    request: Request,
-    username: str = Depends(auth_service.verify_credentials),
-):
+async def list_chat_completions(request: Request, username: str = Depends(auth_service.verify_credentials)):
     """
     Get all chat completions
     Summary: First load the chat interface(UI) for list of chat completions on the left side.
@@ -81,12 +68,7 @@ async def list_chat_completions(
 
 # get a chat completion by id
 @router.get("/chat/completions/{completion_id}", response_model=ChatCompletionResponse)
-@api_response()
-async def retrieve_chat_completion(
-    completion_id: str,
-    request: Request,
-    username: str = Depends(auth_service.verify_credentials),
-):
+async def retrieve_chat_completion(completion_id: str, request: Request, username: str = Depends(auth_service.verify_credentials)):
     """
     Get a chat completion by id
     Summary: Click on a chat completion on the left side to load the chat completion on the right side.
@@ -99,12 +81,7 @@ async def retrieve_chat_completion(
 
 # get all messages for a chat completion
 @router.get("/chat/completions/{completion_id}/messages", response_model=List[MessageResponse], deprecated=True)
-@api_response()
-async def list_messages(
-    completion_id: str,
-    request: Request,
-    username: str = Depends(auth_service.verify_credentials),
-):
+async def list_messages(completion_id: str, request: Request, username: str = Depends(auth_service.verify_credentials)):
     """
     Get all messages for a chat completion
     Summary: Click on a chat completion on the left side to load the chat completion on the right side.
@@ -120,13 +97,7 @@ async def list_messages(
 ################
 # get a plot for a message
 @router.get("/chat/completions/{completion_id}/messages/{message_id}/plot", response_model=PlotResponse)
-@api_response()
-async def retrieve_plot(
-    completion_id: str,
-    message_id: str,
-    request: Request,
-    username: str = Depends(auth_service.verify_credentials),
-):
+async def retrieve_plot(completion_id: str, message_id: str, request: Request, username: str = Depends(auth_service.verify_credentials)):
     """
     Get a plot figure for a message to visualize the data
     Summary: Click on a message on the right side to load the plot on the right side.

app/repository/chat_repository.py

@@ -179,7 +179,7 @@ class ChatRepository:
         logger.debug(f"BEGIN REPO: find messages for chat completion id. input parameters: completion_id: {completion_id}")
         projection = {"messages": 1, "_id": 0}
         chat_doc = await self.db.chat_completion.find_one({"completion_id": completion_id}, projection)
-
+        logger.trace(f"REPO find_messages. chat_doc: {chat_doc}")
         if chat_doc and "messages" in chat_doc and chat_doc["messages"]:
             try:
                 messages_list = [ChatMessage(**item) for item in chat_doc["messages"]]

app/schema/chat_schema.py

@@ -1,5 +1,6 @@
 from typing import List, Optional
 from pydantic import BaseModel, Field
+from datetime import datetime
 
 
 class MessageRequest(BaseModel):
@@ -34,6 +35,7 @@ class MessageResponse(BaseModel):
     role: Optional[str] = Field(None, description="The role of the message", examples=["user", "assistant", "system"])
     content: Optional[str] = Field(None, description="The content of the message")
     figure: Optional[dict] = Field(None, description="The figure data to be visualized")
+    created_date: Optional[datetime] = Field(None, description="The date and time the message was created")
 
 
 class ChoiceResponse(BaseModel):

app/service/chat_service.py

@@ -1,8 +1,7 @@
 import datetime
 from typing import List
 from app.repository.chat_repository import ChatRepository
-from app.schema.chat_schema import ChatCompletionRequest, ChatCompletionResponse
-from app.model.chat_model import ChatMessage
+from app.schema.chat_schema import ChatCompletionRequest, ChatCompletionResponse, MessageResponse 
 from app.mapper.chat_mapper import ChatMapper
 from app.mapper.conversation_mapper import ConversationMapper
 import uuid
@@ -47,8 +46,12 @@ class ChatService:
         entity = await self.chat_repository.find_by_id(completion_id, project)
         return self.chat_mapper.to_schema(entity) if entity else None
 
-    async def find_messages(self, completion_id: str) -> List[ChatMessage]:
-        return await self.chat_repository.find_messages(completion_id)
+    async def find_messages(self, completion_id: str) -> List[MessageResponse]:
+        logger.debug(f"BEGIN SERVICE: find_messages for completion_id: {completion_id}")
+        messages = await self.chat_repository.find_messages(completion_id)
+        logger.debug(f"END SERVICE: find_messages for completion_id: {completion_id}, messages: {messages}")
+        messages_response = [MessageResponse(message_id=message.message_id, role=message.role, content=message.content, created_date=message.created_date) for message in messages]
+        return messages_response
 
     # conversation service
     async def find_all_conversations(self, username: str) -> List[ConversationResponse]: