Merge pull request #37 from Axiomatic-AI/update-tools-client

javieraspuru-ax · web-flow · commit 30e4c1be4faf · 2025-03-17T08:39:06.000-06:00
Add job_id to tools helper and magic
diff --git a/src/axiomatic/client.py b/src/axiomatic/client.py
@@ -223,6 +223,7 @@ def tool_exec(self, tool: str, code: str, poll_interval: int = 3, debug: bool =
                                 return result.output
                             else:
                                 return {
+                                    "job_id": job_id,
                                     "messages": output['messages'],
                                     "objects": self._load_objects_from_base64(output['objects'])
                                 }
diff --git a/src/axiomatic/magic.py b/src/axiomatic/magic.py
@@ -126,6 +126,7 @@ def ax_tool(self, tool, cell):
                                 get_ipython().user_ns["tool_result"] = output
                             else:
                                 get_ipython().user_ns["tool_result"] = {
+                                    "job_id": job_id,
                                     "messages": output['messages'],
                                     "objects": self._load_objects_from_base64(output['objects'])
                                 }

Original file line number	Diff line number	Diff line change
`@@ -223,6 +223,7 @@ def tool_exec(self, tool: str, code: str, poll_interval: int = 3, debug: bool =`
`223`	`223`	`return result.output`
`224`	`224`	`else:`
`225`	`225`	`return {`
	`226`	`+ "job_id": job_id,`
`226`	`227`	`"messages": output['messages'],`
`227`	`228`	`"objects": self._load_objects_from_base64(output['objects'])`
`228`	`229`	`}`
Original file line number	Diff line number	Diff line change
`@@ -126,6 +126,7 @@ def ax_tool(self, tool, cell):`
`126`	`126`	`get_ipython().user_ns["tool_result"] = output`
`127`	`127`	`else:`
`128`	`128`	`get_ipython().user_ns["tool_result"] = {`
	`129`	`+ "job_id": job_id,`
`129`	`130`	`"messages": output['messages'],`
`130`	`131`	`"objects": self._load_objects_from_base64(output['objects'])`
`131`	`132`	`}`