assistant_api.py

class  MathTutor:
  def __init__(self):
    from openai import OpenAI
    import os
    self.client = OpenAI(api_key=os.environ['openai_api_key'])

    self.my_assistant = self.client.beta.assistants.create(
        instructions="You are a personal math tutor. When asked a question, write and run Python code to answer the question.",
        name="Math Tutor",
        tools=[{"type": "code_interpreter"}],
        model="gpt-4-1106-preview",
    )

  def ask_user(self, query):

    self.thread = self.client.beta.threads.create()
    message = self.client.beta.threads.messages.create(
    thread_id=self.thread.id,
    role="user",
    content =query
    )
    self.run = self.client.beta.threads.runs.create(
      thread_id=self.thread.id,
      assistant_id=self.my_assistant.id,
      instructions="Please address the user as Aqdas. The user has a premium account."
    )

    import time 
    while True:
      run = self.client.beta.threads.runs.retrieve(thread_id=self.thread.id, run_id=self.run.id)
      if run.completed_at:
        elapsed = run.completed_at - run.created_at
        elapsed = time.strftime('%H:%M:%S', time.gmtime(elapsed))
        print(f'Run completed in {elapsed}')
        break
      # print('wait 1 sec')
      time.sleep(1)

    messages = self.client.beta.threads.messages.list(thread_id=self.thread.id)
    last_message = messages.data
    my_text = []
    for i in range(0,len(last_message)-1):
      last_message = messages.data[i]
      text = last_message.content[0].text.value
      my_text.append(text)
    return my_text