Create assistant_api.py

assistant_api.py

@@ -0,0 +1,45 @@
+class  MathTutor:
+  def __init__(self):
+    from openai import OpenAI
+    self.client = OpenAI(api_key='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')
+
+    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(len(last_message)):
+      last_message = messages.data[i]
+      text = last_message.content[0].text.value
+      my_text.append(text)
+    return my_text