bohr-agent-sdk 0.1.17__tar.gz → 0.1.18__tar.gz

{bohr_agent_sdk-0.1.17 → bohr_agent_sdk-0.1.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bohr-agent-sdk
-Version: 0.1.17
+Version: 0.1.18
 Summary: SDK for scientific agents
 Home-page: https://github.com/dptech-corp/bohr-agent-sdk/
 Author: DP Technology

{bohr_agent_sdk-0.1.17 → bohr_agent_sdk-0.1.18}/setup.py

@@ -9,7 +9,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="bohr-agent-sdk",
-    version="0.1.17",
+    version="0.1.18",
     description="SDK for science agent and mcp tools",
     long_description=long_description,
     long_description_content_type="text/markdown",

{bohr_agent_sdk-0.1.17 → bohr_agent_sdk-0.1.18}/src/bohr_agent_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bohr-agent-sdk
-Version: 0.1.17
+Version: 0.1.18
 Summary: SDK for scientific agents
 Home-page: https://github.com/dptech-corp/bohr-agent-sdk/
 Author: DP Technology

{bohr_agent_sdk-0.1.17 → bohr_agent_sdk-0.1.18}/src/dp/agent/adapter/adk/utils.py

@@ -116,6 +116,7 @@ def search_error_in_memory_handler(toolset):
 def extract_job_info(events: List[Event]) -> dict:
     jobs = {}
     artifacts = {}
+    events.sort(key=lambda event: event.timestamp)
     for event in events:
         if event.content and event.content.parts:
             for part in event.content.parts:
@@ -135,39 +136,42 @@ def extract_job_info(events: List[Event]) -> dict:
                         }
                     job = jobs[resp.id]
                     job["timestamp"] = event.timestamp
-                    if "result" in resp.response and isinstance(
-                            resp.response["result"], types.CallToolResult):
-                        res = resp.response["result"]
+                    res = resp.response.get("result")
+                    if isinstance(res, dict) and "content" in res \
+                            and "isError" in res:
+                        res = types.CallToolResult(
+                            content=res["content"], isError=res["isError"])
+                    if isinstance(res, types.CallToolResult):
                         if res.isError:
                             err_msg = res.content[0].text
                             if err_msg.startswith("Error executing tool"):
                                 err_msg = err_msg[err_msg.find(":")+2:]
                             job["err_msg"] = err_msg
-                        else:
+                        elif hasattr(res.content[0], "text"):
                             result = jsonpickle.loads(res.content[0].text)
                             if "job_id" not in result:
                                 job["result"] = result
-                    if hasattr(res.content[0], "job_info"):
-                        job_info = res.content[0].job_info
-                        job.update(job_info)
-                        for name, art in job_info.get(
-                                "input_artifacts", {}).items():
-                            if art["uri"] not in artifacts:
-                                artifacts[art["uri"]] = {
-                                    "type": "input",
-                                    "name": name,
-                                    "job_id": job_info["job_id"],
-                                    **art,
-                                }
-                        for name, art in job_info.get(
-                                "output_artifacts", {}).items():
-                            if art["uri"] not in artifacts:
-                                artifacts[art["uri"]] = {
-                                    "type": "output",
-                                    "name": name,
-                                    "job_id": job_info["job_id"],
-                                    **art,
-                                }
+                        if hasattr(res.content[0], "job_info"):
+                            job_info = res.content[0].job_info
+                            job.update(job_info)
+                            for name, art in job_info.get(
+                                    "input_artifacts", {}).items():
+                                if art["uri"] not in artifacts:
+                                    artifacts[art["uri"]] = {
+                                        "type": "input",
+                                        "name": name,
+                                        "job_id": job_info["job_id"],
+                                        **art,
+                                    }
+                            for name, art in job_info.get(
+                                    "output_artifacts", {}).items():
+                                if art["uri"] not in artifacts:
+                                    artifacts[art["uri"]] = {
+                                        "type": "output",
+                                        "name": name,
+                                        "job_id": job_info["job_id"],
+                                        **art,
+                                    }
     return {
         "jobs": list(jobs.values()),
         "artifacts": list(artifacts.values()),