feat(opentrons-ai-server): update endpoint payload type

update endpoint payload type

opentrons-ai-server/api/domain/openai_predict.py

@@ -94,15 +94,15 @@ class atomic_descr(BaseModel):
                 descriptions.append(x)
         return descriptions
 
-    def refine_response(self, assitant_message: str) -> str:
-        if assitant_message is None:
+    def refine_response(self, assistant_message: str) -> str:
+        if assistant_message is None:
             return ""
         system_message: ChatCompletionMessageParam = {
             "role": "system",
             "content": f"{general_rules_1}\n Please leave useful comments for each command.",
         }
 
-        user_message: ChatCompletionMessageParam = {"role": "user", "content": assitant_message}
+        user_message: ChatCompletionMessageParam = {"role": "user", "content": assistant_message}
 
         response = self.client.chat.completions.create(
             model=self.settings.OPENAI_MODEL_NAME,

opentrons-ai-server/api/handler/function.py

@@ -29,7 +29,7 @@ def create_chat_completion(event: Dict[str, Any]) -> Dict[str, Any]:
 
         settings: Settings = Settings.build()
         openai: OpenAIPredict = OpenAIPredict(settings=settings)
-        response: Union[str, None] = openai.predict(prompt=body.message)
+        response: Union[str, None] = openai.predict(prompt=body.message, chat_completion_message_params=body.chat_history)
 
         if response is None or response == "":
             return create_response(HTTPStatus.NO_CONTENT, ChatResponse(reply="No response was generated", fake=body.fake).model_dump())

opentrons-ai-server/api/models/chat_request.py

@@ -1,6 +1,12 @@
 from pydantic import BaseModel
+from typing import List
 
 
+class Chat(BaseModel):
+    role: str
+    content: str
+
 class ChatRequest(BaseModel):
     message: str
+    chat_history: List[Chat]
     fake: bool