futurehouse-client 0.3.20.dev63__py3-none-any.whl → 0.3.20.dev105__py3-none-any.whl

futurehouse_client/clients/job_client.py

@@ -73,7 +73,7 @@ class JobNames(StrEnum):
 
     @staticmethod
     def get_response_object_from_job(job_name: str) -> type[TaskResponse]:
-        return JobNames._get_response_mapping()[job_name]
+        return JobNames._get_response_mapping().get(job_name, TaskResponse)
 
     def get_response_object(self) -> type[TaskResponse]:
         return self._get_response_mapping()[self.name]

futurehouse_client/clients/rest_client.py

@@ -139,6 +139,7 @@ retry_if_connection_error = retry_if_exception_type((
 DEFAULT_AGENT_TIMEOUT: int = 2400  # seconds
 
 
+# pylint: disable=too-many-public-methods
 class RestClient:
     REQUEST_TIMEOUT: ClassVar[float] = 30.0  # sec
     MAX_RETRY_ATTEMPTS: ClassVar[int] = 3
@@ -1419,15 +1420,9 @@ class RestClient:
         if not (world_model_id or name) or (world_model_id and name):
             raise ValueError("Provide either 'world_model_id' or 'name', but not both.")
 
-        params = {
-            "id": str(world_model_id) if world_model_id else None,
-            "name": name,
-        }
-        # Filter out None values before making the request
-        params = {k: v for k, v in params.items() if v is not None}
-
         try:
-            response = self.client.get("/v0.1/world-model", params=params)
+            identifier = str(world_model_id) if world_model_id else name
+            response = self.client.get(f"/v0.1/world-models/{identifier}")
             response.raise_for_status()
             return WorldModelResponse.model_validate(response.json())
         except HTTPStatusError as e:
@@ -1465,14 +1460,9 @@ class RestClient:
         if not (world_model_id or name) or (world_model_id and name):
             raise ValueError("Provide either 'world_model_id' or 'name', but not both.")
 
-        params = {
-            "id": str(world_model_id) if world_model_id else None,
-            "name": name,
-        }
-        params = {k: v for k, v in params.items() if v is not None}
-
         try:
-            response = await self.async_client.get("/v0.1/world-model", params=params)
+            identifier = str(world_model_id) if world_model_id else name
+            response = await self.async_client.get(f"/v0.1/world-models/{identifier}")
             response.raise_for_status()
             return WorldModelResponse.model_validate(response.json())
         except HTTPStatusError as e:

futurehouse_client/models/app.py

@@ -610,7 +610,7 @@ class RuntimeConfig(BaseModel):
         default=None,
         description="Optional job identifier for a continued job",
     )
-    world_model_id: UUID | None = Field(
+    world_model_id: UUID | str | None = Field(
         default=None,
         description="Optional world model identifier for the task",
     )

futurehouse_client/models/rest.py

@@ -46,10 +46,10 @@ class WorldModel(BaseModel):
     """
 
     content: str
-    prior: UUID | None = None
+    prior: UUID | str | None = None
     name: str | None = None
     description: str | None = None
-    trajectory_id: UUID | None = None
+    trajectory_id: UUID | str | None = None
     model_metadata: JsonValue | None = None
 
 
@@ -60,12 +60,12 @@ class WorldModelResponse(BaseModel):
     This model is received from the API.
     """
 
-    id: UUID
-    prior: UUID | None
+    id: UUID | str
+    prior: UUID | str | None
     name: str
     description: str | None
     content: str
-    trajectory_id: UUID | None
+    trajectory_id: UUID | str | None
     email: str | None
     model_metadata: JsonValue | None
     enabled: bool

futurehouse_client/version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.3.20.dev63'
-__version_tuple__ = version_tuple = (0, 3, 20, 'dev63')
+__version__ = version = '0.3.20.dev105'
+__version_tuple__ = version_tuple = (0, 3, 20, 'dev105')

{futurehouse_client-0.3.20.dev63.dist-info → futurehouse_client-0.3.20.dev105.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.20.dev63
+Version: 0.3.20.dev105
 Summary: A client for interacting with endpoints of the FutureHouse service.
 Author-email: FutureHouse technical staff <hello@futurehouse.org>
 License:                                  Apache License

futurehouse_client-0.3.20.dev105.dist-info/RECORD

@@ -0,0 +1,20 @@
+futurehouse_client/__init__.py,sha256=BztM_ntbgmIEjzvnBWcvPhvLjM8xGDFCK0Upf3-nIn8,488
+futurehouse_client/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+futurehouse_client/version.py,sha256=dggvSZjGalP38yUZTaRgokHrf_7S9wpjF6xrQ5xRQlk,530
+futurehouse_client/clients/__init__.py,sha256=-HXNj-XJ3LRO5XM6MZ709iPs29YpApss0Q2YYg1qMZw,280
+futurehouse_client/clients/job_client.py,sha256=D51_qTxya6g5Wfg_ZfJdP031TV_YDJeXkGMiYAJ1qRc,11962
+futurehouse_client/clients/rest_client.py,sha256=zfGMFzLWsNWggTDrZpQZm9habxSQ0bJ2Zd5w_dZrJhU,61301
+futurehouse_client/models/__init__.py,sha256=5x-f9AoM1hGzJBEHcHAXSt7tPeImST5oZLuMdwp0mXc,554
+futurehouse_client/models/app.py,sha256=TrfSorJlxyuc9ZzJpd2tL1pzKOp3jNBYRqFMfmmUue0,28954
+futurehouse_client/models/client.py,sha256=n4HD0KStKLm6Ek9nL9ylP-bkK10yzAaD1uIDF83Qp_A,1828
+futurehouse_client/models/rest.py,sha256=AwPMcB1ruoqaI8NIhX2ZzN8UuX6XsaQ7uzeSE8EpwKk,1573
+futurehouse_client/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+futurehouse_client/utils/auth.py,sha256=tgWELjKfg8eWme_qdcRmc8TjQN9DVZuHHaVXZNHLchk,2960
+futurehouse_client/utils/general.py,sha256=A_rtTiYW30ELGEZlWCIArO7q1nEmqi8hUlmBRYkMQ_c,767
+futurehouse_client/utils/module_utils.py,sha256=aFyd-X-pDARXz9GWpn8SSViUVYdSbuy9vSkrzcVIaGI,4955
+futurehouse_client/utils/monitoring.py,sha256=UjRlufe67kI3VxRHOd5fLtJmlCbVA2Wqwpd4uZhXkQM,8728
+futurehouse_client-0.3.20.dev105.dist-info/licenses/LICENSE,sha256=oQ9ZHjUi-_6GfP3gs14FlPb0OlGwE1QCCKFGnJ4LD2I,11341
+futurehouse_client-0.3.20.dev105.dist-info/METADATA,sha256=bKotJE6rkwyPwipLmyOVe8hNoEoniaDi87CrnCU1C2w,26118
+futurehouse_client-0.3.20.dev105.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+futurehouse_client-0.3.20.dev105.dist-info/top_level.txt,sha256=TRuLUCt_qBnggdFHCX4O_BoCu1j2X43lKfIZC-ElwWY,19
+futurehouse_client-0.3.20.dev105.dist-info/RECORD,,

futurehouse_client-0.3.20.dev63.dist-info/RECORD

@@ -1,20 +0,0 @@
-futurehouse_client/__init__.py,sha256=BztM_ntbgmIEjzvnBWcvPhvLjM8xGDFCK0Upf3-nIn8,488
-futurehouse_client/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-futurehouse_client/version.py,sha256=isTppsUF5LzE3BUYjhyccnSnZwGsbVa_dNrIiEiEViQ,528
-futurehouse_client/clients/__init__.py,sha256=-HXNj-XJ3LRO5XM6MZ709iPs29YpApss0Q2YYg1qMZw,280
-futurehouse_client/clients/job_client.py,sha256=JgB5IUAyCmnhGRsYc3bgKldA-lkM1JLwHRwwUeOCdus,11944
-futurehouse_client/clients/rest_client.py,sha256=fltCx1kCSZKLD-2QNPmK2PbXPa0q0zW9Q3837xS_6TQ,61554
-futurehouse_client/models/__init__.py,sha256=5x-f9AoM1hGzJBEHcHAXSt7tPeImST5oZLuMdwp0mXc,554
-futurehouse_client/models/app.py,sha256=fIWATuetex9GuNzGAWQz-9Dc63ifEFl-Mv-UN7nGSfA,28948
-futurehouse_client/models/client.py,sha256=n4HD0KStKLm6Ek9nL9ylP-bkK10yzAaD1uIDF83Qp_A,1828
-futurehouse_client/models/rest.py,sha256=QqOhRT-qTMfLtNge458qyBv6JFZvoQULnGIjjcDvMy8,1543
-futurehouse_client/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-futurehouse_client/utils/auth.py,sha256=tgWELjKfg8eWme_qdcRmc8TjQN9DVZuHHaVXZNHLchk,2960
-futurehouse_client/utils/general.py,sha256=A_rtTiYW30ELGEZlWCIArO7q1nEmqi8hUlmBRYkMQ_c,767
-futurehouse_client/utils/module_utils.py,sha256=aFyd-X-pDARXz9GWpn8SSViUVYdSbuy9vSkrzcVIaGI,4955
-futurehouse_client/utils/monitoring.py,sha256=UjRlufe67kI3VxRHOd5fLtJmlCbVA2Wqwpd4uZhXkQM,8728
-futurehouse_client-0.3.20.dev63.dist-info/licenses/LICENSE,sha256=oQ9ZHjUi-_6GfP3gs14FlPb0OlGwE1QCCKFGnJ4LD2I,11341
-futurehouse_client-0.3.20.dev63.dist-info/METADATA,sha256=4hmqtHLV-eHTN3l8vXmtIW4L6TFAkBDymbjyg7_rBJQ,26117
-futurehouse_client-0.3.20.dev63.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-futurehouse_client-0.3.20.dev63.dist-info/top_level.txt,sha256=TRuLUCt_qBnggdFHCX4O_BoCu1j2X43lKfIZC-ElwWY,19
-futurehouse_client-0.3.20.dev63.dist-info/RECORD,,