fix: linting

Author: csgulati09

tests/manual_test_async_thread_query.py

@@ -10,6 +10,7 @@
 
 file_to_upload_path = "./tests/configs/threads/sample.pdf"
 
+
 async def main():
     print("Step 1: Create Assistant")
     assistant = await AsyncAssistants(portkey).create(
@@ -35,7 +36,9 @@ async def main():
     print(part)
 
     print("Step 4: Complete the upload")
-    complete_upload = await AsyncUploads(portkey).complete(upload_id=upload.id, part_ids=[part.id])
+    complete_upload = await AsyncUploads(portkey).complete(
+        upload_id=upload.id, part_ids=[part.id]
+    )
     print(complete_upload)
 
     print("Step 5: Create a run and poll")
@@ -61,7 +64,7 @@ async def main():
 
     print("Step 6: Get the list of the messages")
     messages = await AsyncMessages(portkey).list(thread_id=run.thread_id)
-    print(messages) 
+    print(messages)
 
 
-asyncio.run(main())
+asyncio.run(main())

tests/manual_test_batches.py

@@ -2,27 +2,23 @@
 from portkey_ai import Portkey
 
 portkey = Portkey(
-    api_key="Xgv7++HhSeF70UzdVSsmtJuGYe0=",
-    virtual_key="portkey-welcome-e248d3"
+    api_key="Xgv7++HhSeF70UzdVSsmtJuGYe0=", virtual_key="portkey-welcome-e248d3"
 )
 
 
 file_name = "./tests/configs/batches/seed_tasks.jsonl"
 
 
 print("Step 1: Create a batch file")
-batch_file = portkey.files.create(
-  file=open(file_name, "rb"),
-  purpose="batch"
-)
+batch_file = portkey.files.create(file=open(file_name, "rb"), purpose="batch")
 print(batch_file)
 
 
 print("Step 2: Create a batch job")
 batch_job = portkey.batches.create(
-  input_file_id=batch_file.id,
-  endpoint="/v1/chat/completions",
-  completion_window="24h"
+    input_file_id=batch_file.id,
+    endpoint="/v1/chat/completions",
+    completion_window="24h",
 )
 print(batch_job)
 
@@ -34,4 +30,4 @@
 print("Step 4: Retrieve the result")
 result_file_id = batch_job.output_file_id
 result = portkey.files.content(result_file_id).content
-print(result)
+print(result)