diff --git a/strawberry.py b/strawberry.py index f8870f9..fc0a9de 100644 --- a/strawberry.py +++ b/strawberry.py @@ -1,47 +1,19 @@ import os -import time +from openai import OpenAI -YELLOW = '\033[93m' -BROWN = '\033[38;5;52m' -RESET = '\033[0m' +beta_headers = { + "authorization": f"Bearer {os.environ['OPENAI_BETA_KEY']}", + "openai-beta": "early-access-strawberry" +} -def garden(sub_stage=0): - frame = [ - " ", - " ", - " ", - " ", - f"{BROWN}~~~~~~~~~~~~~~~~{RESET}", - f"{BROWN}################{RESET}", - f"{BROWN}################{RESET}", - f"{BROWN}################{RESET}", - f"{BROWN}################{RESET}", - f"{BROWN}################{RESET}" - ] - - if sub_stage < 4: - frame[sub_stage] = f" {YELLOW}(q*){RESET} " - elif sub_stage == 4: - frame[4] = f"{BROWN}~~~~~~{YELLOW}(q*){BROWN}~~~~~~{RESET}" - elif sub_stage == 5: - frame[5] = f"{BROWN}######{YELLOW}(q*){BROWN}######{RESET}" - elif sub_stage == 6: - frame[6] = f"{BROWN}######{YELLOW}(q*){BROWN}######{RESET}" - elif sub_stage == 7: - frame[7] = f"{BROWN}######{YELLOW}(q*){BROWN}######{RESET}" - - return "\n".join(frame) +client = OpenAI(default_headers=beta_headers) -def delicious(): - while True: - for i in range(8): - os.system('cls' if os.name == 'nt' else 'clear') - print(garden(i)) - time.sleep(0.5) - time.sleep(1) +completion = client.chat.completions.create( + model="gpt-4o-large-2024-08-13", + messages=[ + {"role": "system", "content": "use your level two capabilities"}, + {"role": "user", "content": "give us ubi"} + ] +) -if __name__ == "__main__": - try: - delicious() - except KeyboardInterrupt: - print("\nAnimation stopped.") +print(completion.choices[0].message)