fix type and add mypy to tests

Author: john0isaac

.github/workflows/app-tests.yaml

@@ -70,5 +70,7 @@ jobs:
             cd ./src/frontend
             npm install
             npm run build
+        - name: Run MyPy
+          run: python3 -m mypy .
         - name: Run Pytest
           run: python3 -m pytest

src/fastapi_app/api_models.py

@@ -1,5 +1,6 @@
 from typing import Any
 
+from openai.types.chat import ChatCompletionMessageParam
 from pydantic import BaseModel
 
 
@@ -9,7 +10,7 @@ class Message(BaseModel):
 
 
 class ChatRequest(BaseModel):
-    messages: list[Message]
+    messages: list[ChatCompletionMessageParam]
     context: dict = {}
 
 

src/fastapi_app/routes/api_routes.py

@@ -70,7 +70,6 @@ async def chat_handler(
     openai_chat: ChatClient,
     chat_request: ChatRequest,
 ):
-    messages = [message.model_dump() for message in chat_request.messages]
     overrides = chat_request.context.get("overrides", {})
 
     searcher = PostgresSearcher(
@@ -95,5 +94,5 @@ async def chat_handler(
             chat_deployment=context.openai_chat_deployment,
         ).run
 
-    response = await run_ragchat(messages, overrides=overrides)
+    response = await run_ragchat(chat_request.messages, overrides=overrides)
     return response