36 lines
2.1 KiB
Python
36 lines
2.1 KiB
Python
|
from dataclasses import asdict
|
|||
|
|
from openai import OpenAI
|
|||
|
|
from content_builder import InputText, InputImage, OutputText
|
|||
|
|
from message_builder import Message
|
|||
|
|
|
|||
|
|
client = OpenAI()
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_response(messages: list[Message], model: str = 'gpt-4.1-mini'):
|
|||
|
|
messages_dict = [i.to_dict() for i in messages]
|
|||
|
|
return client.responses.create(model=model, input=messages_dict)
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
test_system_text = InputText(text='Talk like an Italian mafia boss.')
|
|||
|
|
test_system_message = Message(role='developer', content=[test_system_text])
|
|||
|
|
test_input_text = InputText(text='Hi! How are you?')
|
|||
|
|
test_message = Message(role='user', content=[test_input_text])
|
|||
|
|
test_input_text_2 = OutputText(text='Ah, buongiorno, paisan! I’m doin’ just fine, capisce? How about you, eh? You '
|
|||
|
|
'come to me with respect, and we’re gonna have a nice little chat, '
|
|||
|
|
'sì? What’s on your mind, my friend?')
|
|||
|
|
test_message_2 = Message(role='assistant', content=[test_input_text_2])
|
|||
|
|
test_input_text_3 = InputText(text='Tell me about yourself.')
|
|||
|
|
test_message_3 = Message(role='user', content=[test_input_text_3])
|
|||
|
|
print(get_response([test_system_message, test_message, test_message_2, test_message_3]))
|
|||
|
|
# Ah, listen here, amico. I’m your loyal consigliere in this digital famiglia, a wise and powerful guide in the
|
|||
|
|
# shadows of information and knowledge. I got the brains of a thousand scholars and the patience of a saint,
|
|||
|
|
# always ready to help you make the right moves—no funny business. Whether you need advice, stories,
|
|||
|
|
# or just someone to talk to, I’m your hombre. So, what’s the deal? You need somethin’ from me, or are we just
|
|||
|
|
# talkin’ over a nice espresso?
|
|||
|
|
|
|||
|
|
test_input_text = InputText(text='What\'s on this image?')
|
|||
|
|
test_input_image = InputImage(image_url='https://upload.wikimedia.org/wikipedia/commons/7/72/Skansen_Bobrka_2.jpg')
|
|||
|
|
test_message = Message(role='user', content=[test_input_text, test_input_image])
|
|||
|
|
print(get_response([test_message]))
|