futurehouse-client 0.3.15.dev71__tar.gz → 0.3.16__tar.gz

{futurehouse_client-0.3.15.dev71 → futurehouse_client-0.3.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.15.dev71
+Version: 0.3.16
 Summary: A client for interacting with endpoints of the FutureHouse service.
 Author-email: FutureHouse technical staff <hello@futurehouse.org>
 Classifier: Operating System :: OS Independent

{futurehouse_client-0.3.15.dev71 → futurehouse_client-0.3.16}/futurehouse_client/clients/rest_client.py

@@ -118,7 +118,7 @@ class TaskResponse(BaseModel):
 
     status: str
     query: str
-    user: str
+    user: str | None = None
     created_at: datetime
     job_name: str
     public: bool
@@ -388,29 +388,30 @@ class RestClient:
             url = f"/v0.1/trajectories/{task_id}"
             full_url = f"{self.base_url}{url}"
 
-            with external_trace(
-                url=full_url,
-                method="GET",
-                library="httpx",
-                custom_params={
-                    "operation": "get_job",
-                    "job_id": task_id,
-                },
+            with (
+                external_trace(
+                    url=full_url,
+                    method="GET",
+                    library="httpx",
+                    custom_params={
+                        "operation": "get_job",
+                        "job_id": task_id,
+                    },
+                ),
+                self.client.stream("GET", url, params={"history": history}) as response,
             ):
-                response = self.client.get(
-                    url,
-                    params={"history": history},
-                )
-            response.raise_for_status()
-            verbose_response = TaskResponseVerbose(**response.json())
+                json_data = "".join(response.iter_text(chunk_size=1024))
+                data = json.loads(json_data)
+                verbose_response = TaskResponseVerbose(**data)
+
             if verbose:
                 return verbose_response
             if any(
                 JobNames.from_string(job_name) in verbose_response.job_name
                 for job_name in ["crow", "falcon", "owl", "dummy"]
             ):
-                return PQATaskResponse(**response.json())
-            return TaskResponse(**response.json())
+                return PQATaskResponse(**data)
+            return TaskResponse(**data)
         except ValueError as e:
             raise ValueError("Invalid task ID format. Must be a valid UUID.") from e
         except Exception as e:

{futurehouse_client-0.3.15.dev71 → futurehouse_client-0.3.16}/futurehouse_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.15.dev71
+Version: 0.3.16
 Summary: A client for interacting with endpoints of the FutureHouse service.
 Author-email: FutureHouse technical staff <hello@futurehouse.org>
 Classifier: Operating System :: OS Independent