diff --git a/bot.py b/bot.py index ef87b2b..32ef337 100644 --- a/bot.py +++ b/bot.py @@ -215,6 +215,13 @@ REMOTE_HOST_NAME = "openai" authTokens = netrc_mod.authenticators(REMOTE_HOST_NAME) openai.api_key = authTokens[2] openAIClient = openai.AsyncOpenAI(api_key=openai.api_key) +saint_harlot_friend = openAIClient.beta.assistants.create( + name="Bane", + instructions="You are a personal horoscope assistant to Saint Harlot", + ools=[{"type":"file_search"}]) + + +saint_harlot_thread = openAIClient.beta.threads.create() REMOTE_HOST_NAME = "spotipy" authTokens = netrc_mod.authenticators(REMOTE_HOST_NAME) @@ -349,10 +356,25 @@ async def on_message(message): return if isinstance(message.channel, discord.DMChannel): if message.author.id == 346956223645614080: - await message.channel.send("Dzien dobry panie i władco (hue hue hue)") - channel = client.get_channel(1064888712565100614) + message = openAIClient.beta.threads.message.create( + saint_harlot_thread.id, + role = "user", + content=message.content + ) + run = openAIClient.beta.threads.runs.create_and_poll( + thread_id = saint_harlot_thread.id, + assistant_id = saint_harlot_friend.id, + instructions = "Odpowiadaj po polsku chyba że zostaniesz przez użytkownika poinstruowany inaczej" + ) + if run.status == 'completed': + messages = openAiClient.beta.threads.messages.list(thread_id = saint_harlot_thread.id) + print(messages) + else: + print(run.status) - #await channel.send("Słyszałem ja żem że: " + message.content) + + channel = client.get_channel(1064888712565100614) + await channel.send("Słyszałem ja żem że: " + message.content) return channel = message.channel await client.process_commands(message)