style: formmatted code with ruff

.env.sample

@@ -2,11 +2,14 @@ LOG_LEVEL=DEBUG
 AUTH_USERNAME=admin
 AUTH_PASSWORD=admin
 
-USE_MOCK="true"
 SECRET_KEY="1234"
-API_KEY="sk-lokumai=="
+API_KEY="sk-admin=="
 BASE_URL="http://0.0.0.0:7860"
 
+
+#deprecated
+USE_MOCK="true"
+
 # mongodb, embedded
 DB_DATABASE_TYPE=embedded
 

app/api/chat_api.py

@@ -55,7 +55,6 @@ async def create_chat_completion(
 
 # get all chat completions
 @router.get("/chat/completions", response_model=List[ChatCompletionResponse])
-@api_response()
 async def list_chat_completions(
     request: Request,
     username: str = Depends(auth_service.verify_credentials),
@@ -64,7 +63,7 @@ async def list_chat_completions(
     Get all chat completions
     Summary: First load the chat interface(UI) for list of chat completions on the left side.
     """
-
+    logger.debug(f"BEGIN API: list_chat_completions for username: {username}")
     page: int = 0
     limit: int = 10
     sort: dict = {"created_date": -1}

app/core/initial_setup/data/initial_chat_completions.json

@@ -7,7 +7,7 @@
             "object": "chat.completion",
             "is_archived": false,
             "is_starred": false,
-            "created_by": "system",
+            "created_by": "admin",
             "created_date": "2024-03-20T10:00:00Z",
             "last_updated_by": "system",
             "last_updated_date": "2024-03-20T11:00:00Z",
@@ -46,7 +46,7 @@
             "object": "chat.completion",
             "is_archived": false,
             "is_starred": false,
-            "created_by": "system",
+            "created_by": "admin",
             "created_date": "2024-03-21T10:00:00Z",
             "last_updated_by": "system",
             "last_updated_date": "2024-03-21T11:00:00Z",
@@ -85,7 +85,7 @@
             "object": "chat.completion",
             "is_archived": false,
             "is_starred": false,
-            "created_by": "system",
+            "created_by": "admin",
             "created_date": "2024-03-22T10:00:00Z",
             "last_updated_by": "system",
             "last_updated_date": "2024-03-22T11:00:00Z",

app/core/initial_setup/setup.py

@@ -7,7 +7,7 @@ from app.repository.chat_repository import ChatRepository
 from app.config.db import db_config
 
 class InitialSetup:
-    """Initial setup manager for the application"""
+    """Initial setup manager for the application when database type is embedded"""
     
     def __init__(self):
         self._chat_repository: Optional[ChatRepository] = None

app/schema/chat_schema.py

@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Optional
 from pydantic import BaseModel, Field
 from enum import Enum
 
@@ -24,25 +24,18 @@ class MessageRequest(BaseModel):
     content: str = Field(..., description="The content of the message")
 
 
-class ChatCompletionBase(BaseModel):
+
+class ChatCompletionRequest(BaseModel):
     """
-    Represents a chat completion.
+    Represents a chat completion request. Starting a new chat or continuing a previous chat.
     """
-
-    completion_id: str = Field(
+    completion_id: Optional[str] = Field(
         None,
         description="The unique identifier for the chat completion. When starting a new chat, this will be a new UUID. When continuing a previous chat, this will be the same as the previous chat completion id.",
     )
-
-
-class ChatCompletionRequest(ChatCompletionBase):
-    """
-    Represents a chat completion request. Starting a new chat or continuing a previous chat.
-    """
-
-    model: str = Field(None, description="The model to use for the chat completion")
-    messages: List[MessageRequest] = Field(..., description="The messages to use for the chat completion")
-    stream: bool = Field(..., description="Whether to stream the chat completion")
+    model: Optional[str] = Field(None, description="The model to use for the chat completion")
+    messages: Optional[List[MessageRequest]] = Field(None, description="The messages to use for the chat completion")
+    stream: Optional[bool] = Field(None, description="Whether to stream the chat completion")
 
 
 class MessageResponse(BaseModel):
@@ -50,34 +43,31 @@ class MessageResponse(BaseModel):
     A chat completion message generated by the model.
     """
 
-    message_id: str = Field(..., description="The unique identifier for the message")
-    role: MessageSchemaRoleType = Field(..., description="The role of the message")
-    content: str = Field(..., description="The content of the message")
-    figure: dict = Field(..., description="The figure data to be visualized")
+    message_id: Optional[str] = Field(None, description="The unique identifier for the message")
+    role: Optional[MessageSchemaRoleType] = Field(None, description="The role of the message")
+    content: Optional[str] = Field(None, description="The content of the message")
+    figure: Optional[dict] = Field(None, description="The figure data to be visualized")
 
 
 class ChoiceResponse(BaseModel):
-    finish_reason: ChoiceSchemaFinishReasonType = Field(
-        ...,
+    finish_reason: Optional[ChoiceSchemaFinishReasonType] = Field(
+        None,
         description="The reason the model stopped generating tokens. This will be `stop` if the model hit a natural stop point or a provided stop sequence, `length` if the maximum number of tokens specified in the request was reached, `content_filter` if content was omitted due to a flag from our content filters",
     )
-    index: int = Field(..., description="The index of the choice in the list of choices.")
-    message: MessageResponse = Field(..., description="The message to use for the chat completion")
+    index: Optional[int] = Field(None, description="The index of the choice in the list of choices.")
+    message: Optional[MessageResponse] = Field(None, description="The message to use for the chat completion")
     # logprobs: str = None # not implemented yet
 
 
-class ChatCompletionResponse(ChatCompletionBase):
+class ChatCompletionResponse(BaseModel):
     """
     Represents a chat completion response returned by model, based on the provided input.
     """
 
-    completion_id: str = Field(..., description="The unique identifier for the chat completion")
-    choices: List[ChoiceResponse] = Field(..., description="A list of chat completion choices.")
-    created: int = Field(
-        ...,
-        description="The Unix timestamp (in seconds) of when the chat completion was created.",
-    )
-    model: str = Field(..., description="The model used for the chat completion")
+    completion_id: Optional[str] = Field(None, description="The unique identifier for the chat completion")
+    choices: Optional[List[ChoiceResponse]] = Field(None, description="A list of chat completion choices.")
+    created: Optional[int] = Field(None, description="The Unix timestamp (in seconds) of when the chat completion was created.")
+    model: Optional[str] = Field(None, description="The model used for the chat completion")
     # not implemented yet
     # service_tier: Optional[ServiceTier] = ServiceTier.AUTO
     # system_fingerprint: Optional[StrictStr] = Field(default=None, description="This fingerprint represents the backend configuration that the model runs with.  Can be used in conjunction with the `seed` request parameter to understand when backend changes have been made that might impact determinism. ")

app/service/chat_service.py

@@ -9,7 +9,7 @@ from app.schema.chat_schema import (
 )
 from app.model.chat_model import ChatCompletion, ChatMessage
 import uuid
-
+from loguru import logger
 from app.schema.conversation import ConversationResponse
 
 
@@ -31,7 +31,7 @@ class ChatService:
         entity.last_updated_by = username
         entity.last_updated_date = datetime.datetime.now()
 
-        entity = self.chat_repository.save(entity)
+        entity = await self.chat_repository.save(entity)
 
         result = ChatCompletionResponse(**entity.model_dump())
         messages = [MessageResponse(**{"role": "assistant", "content": response_content})]  # TODO: implement ai-agent response
@@ -46,21 +46,16 @@ class ChatService:
         ]
         return result
 
-    async def find(
-        self,
-        query: dict,
-        page: int,
-        limit: int,
-        sort: dict,
-        project: dict = None,
-    ) -> List[ChatCompletion]:
-        return self.chat_repository.find(query, page, limit, sort, project)
+    async def find(self, query: dict, page: int, limit: int, sort: dict, project: dict = None) -> List[ChatCompletionResponse]:
+        logger.debug(f"BEGIN SERVICE: find for query: {query}, page: {page}, limit: {limit}, sort: {sort}, project: {project}")
+        entities = await self.chat_repository.find(query, page, limit, sort, project)
+        return [ChatCompletionResponse(**entity.model_dump()) for entity in entities]
 
     async def find_by_id(self, completion_id: str, project: dict = None) -> ChatCompletion:
-        return self.chat_repository.find_by_id(completion_id, project)
+        return await self.chat_repository.find_by_id(completion_id, project)
 
     async def find_messages(self, completion_id: str) -> List[ChatMessage]:
-        return self.chat_repository.find_messages(completion_id)
+        return await self.chat_repository.find_messages(completion_id)
 
     # conversation service
     async def find_all_conversations(self, username: str) -> List[ConversationResponse]: