mirror of
https://github.com/kennethreitz/instructor.git
synced 2026-06-05 22:50:18 +00:00
Jason Liu
24 lines
601 B
Python
24 lines
601 B
Python
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']
|