Quality: Code lint

clemlesne · Dec 5, 2023 · c180f1f · c180f1f
1 parent 10b0ef5
commit c180f1f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/conversation-api/ai/openai.py b/src/conversation-api/ai/openai.py
@@ -399,9 +399,13 @@ def messages(self) -> List[BaseMessage]:
         res = []
         for message in self.store.message_list(self.conversation_id) or []:
             if message.role == MessageRole.ASSISTANT:
-                obj = AIMessage(content=message.content, **message.extra)
+                obj = AIMessage(
+                    content=message.content, additional_kwargs=message.extra
+                )
             elif message.role == MessageRole.USER:
-                obj = HumanMessage(content=message.content, **message.extra)
+                obj = HumanMessage(
+                    content=message.content, additional_kwargs=message.extra
+                )
             else:
                 raise ValueError(f"Unsupported message role: {message.role}")
             res.append(obj)