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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.15.dev15
+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
@@ -15,6 +15,7 @@ Requires-Dist: dm-tree<0.1.9
 Requires-Dist: fhaviary
 Requires-Dist: httpx
 Requires-Dist: ldp>=0.22.0
+Requires-Dist: litellm==1.67.4.post1
 Requires-Dist: pydantic
 Requires-Dist: python-dotenv
 Requires-Dist: tenacity

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

@@ -101,6 +101,12 @@ class InvalidTaskDescriptionError(Exception):
     """Raised when the task description is invalid or empty."""
 
 
+class SimpleOrganization(BaseModel):
+    id: int
+    name: str
+    display_name: str
+
+
 # 5 minute default for JWTs
 JWT_TOKEN_CACHE_EXPIRY: int = 300  # seconds
 
@@ -112,11 +118,11 @@ class TaskResponse(BaseModel):
 
     status: str
     query: str
-    user: str
+    user: str | None = None
     created_at: datetime
     job_name: str
     public: bool
-    shared_with: list[str]
+    shared_with: list[SimpleOrganization] | None = None
     build_owner: str | None = None
     environment_name: str | None = None
     agent_name: str | None = None
@@ -191,7 +197,7 @@ class TaskResponseVerbose(TaskResponse):
     agent_state: list[dict[str, Any]] | None = None
     environment_frame: dict[str, Any] | None = None
     metadata: dict[str, Any] | None = None
-    shared_with: list[str]
+    shared_with: list[SimpleOrganization] | None = None
 
 
 class RestClient:
@@ -382,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.dev15 → 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.dev15
+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
@@ -15,6 +15,7 @@ Requires-Dist: dm-tree<0.1.9
 Requires-Dist: fhaviary
 Requires-Dist: httpx
 Requires-Dist: ldp>=0.22.0
+Requires-Dist: litellm==1.67.4.post1
 Requires-Dist: pydantic
 Requires-Dist: python-dotenv
 Requires-Dist: tenacity

{futurehouse_client-0.3.15.dev15 → futurehouse_client-0.3.16}/futurehouse_client.egg-info/requires.txt

@@ -3,6 +3,7 @@ dm-tree<0.1.9
 fhaviary
 httpx
 ldp>=0.22.0
+litellm==1.67.4.post1
 pydantic
 python-dotenv
 tenacity

{futurehouse_client-0.3.15.dev15 → futurehouse_client-0.3.16}/pyproject.toml

@@ -19,6 +19,7 @@ dependencies = [
     "fhaviary",
     "httpx",
     "ldp>=0.22.0",
+    "litellm==1.67.4.post1",  # TODO: remove after: https://github.com/BerriAI/litellm/issues/10349
     "pydantic",
     "python-dotenv",
     "tenacity",
@@ -63,3 +64,9 @@ include = ["futurehouse_client*"]
 local_scheme = "no-local-version"
 root = "../.."
 version_file = "version.py"
+
+[[tool.uv.index]]
+explicit = true
+name = "testpypi"
+publish-url = "https://test.pypi.org/legacy/"
+url = "https://test.pypi.org/"