diff --git a/examples/fizzbuzz/run.py b/examples/fizzbuzz/run.py new file mode 100644 index 0000000..0b3cde2 --- /dev/null +++ b/examples/fizzbuzz/run.py @@ -0,0 +1,23 @@ +from typing import List +from openai import OpenAI +import instructor + +client = instructor.patch(OpenAI()) + + +def fizzbuzz_gpt(n) -> List[int | str]: + return client.chat.completions.create( + model="gpt-3.5-turbo", + response_model=List[int | str], + messages=[ + { + "role": "user", + "content": f"Return the first {n} numbers in fizzbuzz", + }, + ], + ) # type: ignore + + +if __name__ == "__main__": + print(fizzbuzz_gpt(n=15)) + # > [1, 2, 'Fizz', 4, 'Buzz', 'Fizz', 7, 8, 'Fizz', 'Buzz', 11, 'Fizz', 13, 14, 'FizzBuzz']