DEVOCEAN

import os
import time
import json
from openai import AzureOpenAI

client = AzureOpenAI(
    api_key=os.getenv("AZURE_OPENAI_API_KEY"),
    api_version="2024-02-15-preview",                 #Azure OpenAI Assistants를 위한 API 버전
    azure_endpoint = os.getenv("AZURE_OPENAI_ENDPOINT")
    )

# assistant 생성
assistant = client.beta.assistants.create(
    name="AI 수학 도우미2",
    instructions="당신은 수학 문제에 답하는 데 도움이 되는 코드를 작성할 수 있는 AI 도우미입니다..",
    tools=[{"type": "code_interpreter"}],  
                 # tools를 dict형태의 Assistant type을 배열 내에 다수 등록 가능.
    model="gpt-4-1106-preview"
)

# thread 생성
thread = client.beta.threads.create()

content = "어떤 정사각형의 가로를 3cm, 세로를 2cm만큼 늘였더니 넓이가 처음의 2배가 되었다. 처음 정사각형의 한 변의 길이는?"

# thread에 사용자 message를 추가
message = client.beta.threads.messages.create(
    thread_id=thread.id,
    role="user",
    content=content
)

# thread를 실행한 Assistant 활성화 
run = client.beta.threads.runs.create(
  thread_id=thread.id,
  assistant_id=assistant.id,
)

run = client.beta.threads.runs.retrieve(
  thread_id=thread.id,
  run_id=run.id
)

status = run.status

# Wait till the assistant has responded
while status not in ["completed", "cancelled", "expired", "failed"]:
    time.sleep(5)
    run = client.beta.threads.runs.retrieve(thread_id=thread.id,run_id=run.id)
    status = run.status

messages = client.beta.threads.messages.list(
  thread_id=thread.id
)

result = messages.model_dump_json()

import json

result_dict = json.loads(result)
print(result_dict["data"][0]["content"][0]["text"]["value"])

message = client.beta.threads.messages.create(
    thread_id=thread.id,
    role="user",
    content="이 문제를 풀기 위해서 사용한 코드를 알려줘"
)

run = client.beta.threads.runs.create(
  thread_id=thread.id,
  assistant_id=assistant.id
)

run = client.beta.threads.runs.retrieve(
  thread_id=thread.id,
  run_id=run.id
)

status = run.status

# Wait till the assistant has responded
while status not in ["completed", "cancelled", "expired", "failed"]:
    time.sleep(5)
    run = client.beta.threads.runs.retrieve(thread_id=thread.id,run_id=run.id)
    status = run.status

messages = client.beta.threads.messages.list(
  thread_id=thread.id
)
data = json.loads(messages.model_dump_json(indent=2))
code = data['data'][0]['content'][0]['text']['value']
print(code)