Merge pull request #10 from import-ai/feature/internal_api

LucienShui · web-flow · commit 55cc0dbb041c · 2025-02-12T22:40:06.000+08:00
Update callback API endpoint
diff --git a/wizard/wand/worker.py b/wizard/wand/worker.py
@@ -145,7 +145,7 @@ async def process_task(self, task: Task, trace_info: TraceInfo) -> Task:
     async def callback(self, task: Task, trace_info: TraceInfo):
         async with httpx.AsyncClient(base_url=self.config.backend.base_url) as client:
             http_response: httpx.Response = await client.post(
-                f"/api/v1/tasks/callback",
+                f"/internal/api/tasks/callback",
                 json=task.model_dump(exclude_none=True, mode="json"),
                 headers={"X-Trace-ID": task.task_id}
             )

Original file line number	Diff line number	Diff line change
`@@ -145,7 +145,7 @@ async def process_task(self, task: Task, trace_info: TraceInfo) -> Task:`
`145`	`145`	`async def callback(self, task: Task, trace_info: TraceInfo):`
`146`	`146`	`async with httpx.AsyncClient(base_url=self.config.backend.base_url) as client:`
`147`	`147`	`http_response: httpx.Response = await client.post(`
`148`		`- f"/api/v1/tasks/callback",`
	`148`	`+ f"/internal/api/tasks/callback",`
`149`	`149`	`json=task.model_dump(exclude_none=True, mode="json"),`
`150`	`150`	`headers={"X-Trace-ID": task.task_id}`
`151`	`151`	`)`