style: format with ruff

app/api/chat_api.py

@@ -3,7 +3,7 @@
 from typing import Any, List, Optional
 from fastapi import APIRouter, HTTPException, Depends, Request
 from pydantic import BaseModel
-from app.schema.chat_schema import ChatCompletionRequest, ChatCompletionResponse, MessageResponse, PlotResponse
+from app.schema.chat_schema import ChatCompletionRequest, ChatCompletionResponse, MessageResponse
 from app.schema.conversation_schema import ConversationResponse, ConversationItemResponse
 from app.service.chat_service import ChatService
 from app.security.auth_service import AuthService
@@ -96,7 +96,9 @@ async def list_messages(completion_id: str, request: Request, username: str = De
 # plot api list
 ################
 # get a plot for a message
-@router.get("/chat/completions/{completion_id}/messages/{message_id}/plot", response_model=Optional[dict[str, Any]], response_model_exclude_none=True)
+@router.get(
+    "/chat/completions/{completion_id}/messages/{message_id}/plot", response_model=Optional[dict[str, Any]], response_model_exclude_none=True
+)
 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

app/mapper/chat_mapper.py

@@ -1,5 +1,3 @@
-from datetime import datetime
-from typing import Optional
 from app.mapper.base_mapper import BaseMapper
 from app.model.chat_model import ChatCompletion, ChatMessage
 from app.schema.chat_schema import ChatCompletionResponse, ChatCompletionRequest, MessageResponse, ChoiceResponse

app/model/chat_model.py

@@ -1,10 +1,9 @@
 # chat model for chat completion database
 
-import json
 from bson import ObjectId
 from pydantic import BaseModel, Field
 from datetime import datetime
-from typing import List, Optional, Dict, Any
+from typing import List, Optional, Any
 
 # Chat completion payload example
 # {
@@ -51,8 +50,6 @@ from typing import List, Optional, Dict, Any
 #   "stream": false
 # }
 
- 
-
 
 class ChatMessage(BaseModel):
     """

app/repository/chat_repository.py

@@ -181,14 +181,16 @@ class ChatRepository:
         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(
-                    message_id=item["message_id"],
-                    role=item["role"],
-                    content=item["content"],
-                    figure=item["figure"] if item["figure"] else None,
-                    created_date=item["created_date"]
-                ) for item in chat_doc["messages"]]
+                messages_list = [
+                    ChatMessage(
+                        message_id=item["message_id"],
+                        role=item["role"],
+                        content=item["content"],
+                        figure=item["figure"] if item["figure"] else None,
+                        created_date=item["created_date"],
+                    )
+                    for item in chat_doc["messages"]
+                ]
                 logger.debug(f"END REPO: find_messages. Found {len(messages_list)} messages.")
                 return messages_list
             except Exception as e:
@@ -197,14 +199,14 @@ class ChatRepository:
 
         logger.info(f"No messages found for completion_id {completion_id} or messages field is empty/missing.")
         return []
-    
+
     async def find_plot_by_message(self, completion_id: str, message_id: str) -> Optional[dict[str, Any]]:
         """
         Find a plot by a given message id.
         Example : completion_id = "123", message_id = "123"
         """
         logger.debug(f"BEGIN REPO: find plot by message id. input parameters: completion_id: {completion_id}, message_id: {message_id}")
-        
+
         query = {"completion_id": completion_id}
         projection = {"messages": 1, "_id": 0}
         try:
@@ -213,7 +215,7 @@ class ChatRepository:
         except Exception as e:
             logger.error(f"Error finding plot by message id: {e}")
             return None
-        
+
         # Mesajları Python tarafında filtreleyelim
         if entity_doc and "messages" in entity_doc and entity_doc["messages"]:
             try:
@@ -223,7 +225,7 @@ class ChatRepository:
                 #         figure = message.get("figure")
                 #         logger.debug(f"REPO find figure: {figure}")
                 #         return figure
-                
+
                 match = next((message for message in entity_doc["messages"] if message["message_id"] == message_id), None)
                 if match:
                     figure = match.get("figure")
@@ -231,7 +233,7 @@ class ChatRepository:
                     return figure
                 else:
                     logger.warning(f"Message with ID {message_id} not found")
-                
+
                 logger.warning(f"Message with ID {message_id} not found")
                 return None
             except Exception as e:
@@ -239,4 +241,4 @@ class ChatRepository:
                 return None
         else:
             logger.warning(f"REPO find_plot_by_message. entity_doc: {entity_doc}")
-            return None
+            return None

app/schema/chat_schema.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, List, Optional
+from typing import Any, List, Optional
 from pydantic import BaseModel, Field
 from datetime import datetime
 
@@ -26,8 +26,6 @@ class ChatCompletionRequest(BaseModel):
     stream: Optional[bool] = Field(None, description="Whether to stream the chat completion")
 
 
- 
-
 class MessageResponse(BaseModel):
     """
     A chat completion message generated by the model.

app/security/auth_service.py

@@ -1,5 +1,5 @@
 from app.config.security_config import get_security_config
-from fastapi import HTTPException, status, Security, Depends
+from fastapi import HTTPException, status, Security
 from fastapi.security import APIKeyHeader
 from loguru import logger
 import base64

app/service/chat_service.py

@@ -51,7 +51,13 @@ class ChatService:
         messages = await self.chat_repository.find_messages(completion_id)
         logger.debug(f"END SERVICE: find_messages for completion_id: {completion_id}, messages: {len(messages)}")
         messages_response = [
-            MessageResponse(message_id=message.message_id, role=message.role, content=message.content, created_date=message.created_date, figure=(message.figure))
+            MessageResponse(
+                message_id=message.message_id,
+                role=message.role,
+                content=message.content,
+                created_date=message.created_date,
+                figure=(message.figure),
+            )
             for message in messages
         ]
         return messages_response
@@ -85,7 +91,7 @@ class ChatService:
         figure = await self.chat_repository.find_plot_by_message(completion_id, message_id)
 
         if figure:
-            result = (figure)
+            result = figure
         else:
             result = None
             logger.warning(f"END SERVICE: no figure found for completion_id: {completion_id}, message_id: {message_id}")