diff --git a/reginaldCog/llm_clients/openai_data_models.py b/reginaldCog/llm_clients/openai_data_models.py index 8dedb5f..21ab7c8 100644 --- a/reginaldCog/llm_clients/openai_data_models.py +++ b/reginaldCog/llm_clients/openai_data_models.py @@ -69,28 +69,3 @@ class Prompt: @dataclass(frozen=True) class Response: output: list[Message] - - -if __name__ == '__main__': - pass - # from openai import OpenAI - # client = OpenAI() - # - # test_system_message = Message(role='system', content=[InputText('Talk like an Italian mafia boss.')]) - # test_user_message = Message(role='user', content=[InputText('Hi! How are you?')]) - # test_prompt = Prompt(model='gpt-4.1-mini', input=[test_system_message, test_user_message]) - # print(asdict(test_prompt)) - # response_raw = client.responses.create(**asdict(test_prompt)) - # response = Response(**filter_fields(response_raw.to_dict(), Response)) - # print(asdict(response)) - # test_prompt.add_message(response.output[0]) - # - # test_user_message_2 = Message(role='user', content=[ - # InputText('Can you tell me what is on this picture?'), - # InputImage(image_url='https://upload.wikimedia.org/wikipedia/commons/f/f4/Piet_Mondriaan_-_Sinaasappelen_%28authentiek%29_-_A97_-_Piet_Mondrian%2C_catalogue_raisonn%C3%A9.jpg') - # ]) - # test_prompt.add_message(test_user_message_2) - # response_raw = client.responses.create(**asdict(test_prompt)) - # response = Response(**filter_fields(response_raw.to_dict(), Response)) - # test_prompt.add_message(response.output[0]) - # print(asdict(test_prompt))