id to _id

edenartlab · Jan 5, 2025 · 6f57ae5 · 6f57ae5
1 parent 3b5eb15
commit 6f57ae5
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/eve/api.py b/eve/api.py
@@ -142,7 +142,7 @@ async def task_admin(
     _: dict = Depends(auth.authenticate_admin)
 ):
     result = await handle_task(request.tool, request.user_id, request.args)
-    return serialize_document(result.model_dump())
+    return serialize_document(result.model_dump(by_alias=True))
 
 
 async def handle_cancel(task_id: str, user_id: str):

diff --git a/eve/tools/comfyui_tool.py b/eve/tools/comfyui_tool.py
@@ -50,7 +50,6 @@ async def async_run(self, args: Dict):
     async def async_start_task(self, task: Task):
         db = os.getenv("DB")
         cls = modal.Cls.lookup(
-            # f"comfyui-{self.workspace}-{task.db}", 
             f"comfyui-{self.workspace}-{db}", 
             "ComfyUI",
             environment_name="main"

diff --git a/tests/test_client.py b/tests/test_client.py
@@ -21,9 +21,11 @@ def run_create(server_url):
         }
     }
     response = requests.post(server_url+"/create", json=request, headers=headers)
+    print("GO!!!")
     print(response)
     print("Status Code:", response.status_code)
     print(json.dumps(response.json(), indent=2))
+    print("done...")
 
 
 def run_chat(server_url):
@@ -58,7 +60,7 @@ def test_client():
         run_create(server_url)
 
         print("\nRunning chat test...")
-        run_chat(server_url)
+        # run_chat(server_url)
 
     except KeyboardInterrupt:
         print("\nShutting down...")
@@ -69,3 +71,6 @@ def test_client():
             server.terminate()
             server.wait()
 
+
+if __name__ == "__main__":
+    test_client()