feat: align the chat & completion response interface (#43)

Signed-off-by: Keming <kemingyang@tensorchord.ai>
tensorchord · Jun 21, 2023 · f78afce · f78afce
1 parent 9e19db6
commit f78afce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmspec/llmspec.py b/llmspec/llmspec.py
@@ -144,11 +144,11 @@ class ChatResponse(LMResponse):
     def from_message(
         cls,
         message: str,
-        role: Role,
         model: str,
         finish_reason: str,
         prompt_token: int,
         completion_token: int,
+        role: Role = Role.ASSISTANT,
     ):
         return cls(
             id=str(uuid.uuid4()),