added assistant_message function

phelps-sg · Nov 17, 2023 · 613c4d1 · 613c4d1
1 parent 28a4228
commit 613c4d1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/openai_pygenerator/openai_pygenerator.py b/src/openai_pygenerator/openai_pygenerator.py
@@ -177,6 +177,10 @@ def user_message(text: str) -> Completion:
     return ChatCompletionUserMessageParam({"role": "user", "content": text})
 
 
+def assistant_message(text: str) -> Completion:
+    return ChatCompletionAssistantMessageParam({"role": "assistant", "content": text})
+
+
 def content(completion: Completion) -> str:
     return str(completion["content"])