Formatting

autoblocksai · Feb 25, 2024 · cb2889e · cb2889e
1 parent 2c3bd8b
commit cb2889e
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 32 deletions.
diff --git a/JavaScript/openai-assistants-api/README.md b/JavaScript/openai-assistants-api/README.md
@@ -44,11 +44,13 @@ npm install
 ## Run the script
 
 Without Prompt SDK:
+
 ```
 npm run start
 ```
 
 With Prompt SDK:
+
 ```
 npm run start-with-prompt-sdk
 ```

diff --git a/Python/openai-assistants-api/src/main.py b/Python/openai-assistants-api/src/main.py
@@ -17,62 +17,64 @@
 
 
 def main():
-
-    assistant_create_params = dict(name="Math Tutor",
+    assistant_create_params = dict(
+        name="Math Tutor",
         instructions="You are a personal math tutor. Write and run code to answer math questions.",
         tools=[{"type": "code_interpreter"}],
-        model="gpt-4-turbo-preview")
-    tracer.send_event('ai.assistant.create', properties=assistant_create_params)
+        model="gpt-4-turbo-preview",
+    )
+    tracer.send_event("ai.assistant.create", properties=assistant_create_params)
     assistant = client.beta.assistants.create(**assistant_create_params)
     tracer.update_properties(dict(assistant_id=assistant.id, model=assistant.model))
-    tracer.send_event('ai.assistant.created')
+    tracer.send_event("ai.assistant.created")
 
-    tracer.send_event('ai.assistant.thread.create')
+    tracer.send_event("ai.assistant.thread.create")
     thread = client.beta.threads.create()
     tracer.update_properties(dict(thread_id=thread.id))
-    tracer.send_event('ai.assistant.thread.created')
+    tracer.send_event("ai.assistant.thread.created")
 
-    message_create_params = dict(thread_id=thread.id,
+    message_create_params = dict(
+        thread_id=thread.id,
         role="user",
-        content="I need to solve the equation `3x + 11 = 14`. Can you help me?")
-    tracer.send_event('ai.assistant.thread.message.create', properties=message_create_params)
-    message = client.beta.threads.messages.create(
-        **message_create_params
+        content="I need to solve the equation `3x + 11 = 14`. Can you help me?",
+    )
+    tracer.send_event(
+        "ai.assistant.thread.message.create", properties=message_create_params
+    )
+    message = client.beta.threads.messages.create(**message_create_params)
+    tracer.send_event(
+        "ai.assistant.thread.message.created", properties=dict(message_id=message.id)
     )
-    tracer.send_event('ai.assistant.thread.message.created', properties=dict(message_id=message.id))
 
-    run_params = dict(thread_id=thread.id,
+    run_params = dict(
+        thread_id=thread.id,
         assistant_id=assistant.id,
-        instructions="You are a personal math tutor. Write and run code to answer math questions.")
-    tracer.send_event('ai.assistant.thread.run.create', properties=run_params)
-    run = client.beta.threads.runs.create(
-        **run_params
+        instructions="You are a personal math tutor. Write and run code to answer math questions.",
     )
+    tracer.send_event("ai.assistant.thread.run.create", properties=run_params)
+    run = client.beta.threads.runs.create(**run_params)
     tracer.update_properties(dict(run_id=run.id))
-    tracer.send_event('ai.assistant.thread.run.created')
+    tracer.send_event("ai.assistant.thread.run.created")
 
     run_completed = False
 
     while not run_completed:
-        run = client.beta.threads.runs.retrieve(
-            thread_id=thread.id,
-            run_id=run.id
-        )
+        run = client.beta.threads.runs.retrieve(thread_id=thread.id, run_id=run.id)
         if run.status == "completed":
             run_completed = True
 
-    run_steps = client.beta.threads.runs.steps.list(
-        thread_id=thread.id,
-        run_id=run.id
-    )
+    run_steps = client.beta.threads.runs.steps.list(thread_id=thread.id, run_id=run.id)
 
     for step in run_steps:
-        tracer.send_event(f'ai.assistant.thread.run.step.{step.type}', properties=dict(run_id=step.id, parent_run_id=step.run_id))
+        tracer.send_event(
+            f"ai.assistant.thread.run.step.{step.type}",
+            properties=dict(run_id=step.id, parent_run_id=step.run_id),
+        )
 
-    messages = client.beta.threads.messages.list(
-        thread_id=thread.id
+    messages = client.beta.threads.messages.list(thread_id=thread.id)
+    tracer.send_event(
+        "ai.assistant.thread.run.completed",
+        properties=dict(response=messages.model_dump_json()),
     )
-    tracer.send_event('ai.assistant.thread.run.completed', properties=dict(response=messages.model_dump_json()))
 
     print(f"View your trace: https://app.autoblocks.ai/explore/trace/{tracer.trace_id}")
-