fix: call golang api failing using multipartencoder

tensorplex-labs · Jul 10, 2024 · 0a8d504 · 0a8d504
1 parent d59ec0c
commit 0a8d504
Showing 1 changed file with 17 additions and 16 deletions.
diff --git a/commons/human_feedback/dojo.py b/commons/human_feedback/dojo.py
@@ -4,7 +4,6 @@
 
 import httpx
 from loguru import logger
-from requests_toolbelt import MultipartEncoder
 
 import template
 from commons.utils import loaddotenv
@@ -92,29 +91,30 @@ async def create_task(
             else:
                 logger.error(f"Unrecognized criteria type: {type(criteria_type)}")
 
-        body = {
-            "title": "LLM Code Generation Task",
-            "body": ranking_request.prompt,
+        json_body = {
+            "title": ("", "LLM Code Generation Task"),
+            "body": ("", ranking_request.prompt),
             "expireAt": (
-                datetime.datetime.utcnow()
-                + datetime.timedelta(seconds=template.TASK_DEADLINE)
-            )
-            .replace(microsecond=0, tzinfo=datetime.timezone.utc)
-            .isoformat()
-            .replace("+00:00", "Z"),
-            "taskData": json.dumps([taskData]),
-            "maxResults": "1",
+                "",
+                (
+                    datetime.datetime.utcnow()
+                    + datetime.timedelta(seconds=template.TASK_DEADLINE)
+                )
+                .replace(microsecond=0, tzinfo=datetime.timezone.utc)
+                .isoformat()
+                .replace("+00:00", "Z"),
+            ),
+            "taskData": ("", json.dumps([taskData])),
+            "maxResults": ("", "1"),
         }
 
-        mp = MultipartEncoder(fields=body)
         DOJO_API_KEY = loaddotenv("DOJO_API_KEY")
-        print("DOJO API KEY: ", DOJO_API_KEY)
+
         response = await cls._http_client.post(
             path,
-            data=mp.to_string(),
+            files=json_body,
             headers={
                 "x-api-key": DOJO_API_KEY,
-                "content-type": mp.content_type,
             },
             timeout=15.0,
         )
@@ -132,6 +132,7 @@ async def create_task(
                 print("Tried to export create task request as curl, but failed.")
                 print(f"Exception: {e}")
 
+        task_ids = []
         if response.status_code == 200:
             task_ids = response.json()["body"]
             logger.success(f"Successfully created task with\ntask ids:{task_ids}")