Merge pull request #21 from SubstrateLabs/ben-req-id

add request_id to response

examples/basic.py

@@ -18,11 +18,10 @@
 summary = GenerateText(prompt=sb.concat("Summarize this story: ", story.future.text))
 
 response = substrate.run(story, summary)
-print(response.api_response.json)
-print(response.api_response.status_code)
+print(response)
 
-# summary_out = response.get(summary)
-# print(summary_out.text)
+summary_out = response.get(summary)
+print(summary_out.text)
 
 # viz = Substrate.visualize(ry)
 # print(viz)

substrate/substrate_response.py

@@ -14,6 +14,18 @@ class SubstrateResponse:
     def __init__(self, api_response: APIResponse):
         self._api_response = api_response
 
+    def __str__(self):
+        return f"SubstrateResponse(request_id={self.request_id}, api_response={self._api_response.status_code})"
+
+    @property
+    def request_id(self) -> Optional[str]:
+        """
+        The request id.
+        """
+        if self.json:
+            return self.json.get("substrate_request_id")
+        return None
+
     def get(self, node: CoreNode[OT]) -> OT:
         """
         Get the output of a specific node.