bug fix

Sage-Bionetworks · Jan 14, 2025 · fd457a2 · fd457a2
1 parent 3ff650d
commit fd457a2
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/synapseclient/models/agent.py b/synapseclient/models/agent.py
@@ -101,7 +101,8 @@ def fill_from_dict(self, synapse_response: Dict[str, str]) -> "AgentPrompt":
         Returns:
             The AgentPrompt object.
         """
-        self.id = synapse_response.get("sessionId", None)
+        self.id = synapse_response.get("jobId", None)
+        self.session_id = synapse_response.get("sessionId", None)
         self.response = synapse_response.get("responseText", None)
         return self
 

diff --git a/synapseclient/models/mixins/asynchronous_job.py b/synapseclient/models/mixins/asynchronous_job.py
@@ -221,12 +221,15 @@ async def send_job_and_wait_async(
         SynapseTimeoutError: If the job does not complete within the timeout.
     """
     job_id = await send_job_async(request=request, synapse_client=synapse_client)
-    return await get_job_async(
-        job_id=job_id,
-        request_type=AGENT_CHAT_REQUEST,
-        synapse_client=synapse_client,
-        endpoint=endpoint,
-    )
+    return {
+        "jobId": job_id,
+        **await get_job_async(
+            job_id=job_id,
+            request_type=AGENT_CHAT_REQUEST,
+            synapse_client=synapse_client,
+            endpoint=endpoint,
+        ),
+    }
 
 
 async def send_job_async(