diff --git a/examples/translate_text.py b/examples/translate_text.py index e7d1d3e..b9e38d2 100644 --- a/examples/translate_text.py +++ b/examples/translate_text.py @@ -1,9 +1,13 @@ from _context import sm -conversation = sm.create_conversation(llm_model="gpt-4o", llm_provider="openai") -conversation.add_message( - "user", "Translate the following text to French: 'Hello, world!'" -) +def translate_to_french(text: str) -> str: + conversation = sm.create_conversation(llm_model="gpt-4o", llm_provider="openai") -print(conversation.send().text) + conversation.add_message( + "user", f"Translate the following text to French: {text!r}" + ) + return conversation.send().text + + +print(translate_to_french("Hello, world!"))