futurehouse-client 0.3.20.dev112__py3-none-any.whl → 0.3.20.dev196__py3-none-any.whl

futurehouse_client/__init__.py

@@ -8,11 +8,6 @@ from .models.app import (
     TaskResponse,
     TaskResponseVerbose,
 )
-from .utils.world_model_tools import (
-    create_world_model_tool,
-    search_world_model_tool,
-    make_world_model_tools,
-)
 
 __all__ = [
     "FinchTaskResponse",
@@ -24,7 +19,4 @@ __all__ = [
     "TaskRequest",
     "TaskResponse",
     "TaskResponseVerbose",
-    "create_world_model_tool",
-    "search_world_model_tool",
-    "make_world_model_tools",
 ]

futurehouse_client/clients/rest_client.py

@@ -111,8 +111,10 @@ class WorldModelFetchError(RestClientError):
 class WorldModelCreationError(RestClientError):
     """Raised when there's an error creating a world model."""
 
-class WorldModelDeletionError(RestClientError):
-    """Raised when there's an error deleting a world model."""
+
+class TrajectoryGroupError(RestClientError):
+    """Raised when there's an error with trajectory group operations."""
+
 
 class InvalidTaskDescriptionError(Exception):
     """Raised when the task description is invalid or empty."""
@@ -535,8 +537,15 @@ class RestClient:
         wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
         retry=retry_if_connection_error,
     )
-    def create_task(self, task_data: TaskRequest | dict[str, Any]):
-        """Create a new futurehouse task."""
+    def create_task(
+        self,
+        task_data: TaskRequest | dict[str, Any],
+    ):
+        """Create a new futurehouse task.
+
+        Args:
+            task_data: The task request data
+        """
         if isinstance(task_data, dict):
             task_data = TaskRequest.model_validate(task_data)
 
@@ -563,8 +572,15 @@ class RestClient:
         wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
         retry=retry_if_connection_error,
     )
-    async def acreate_task(self, task_data: TaskRequest | dict[str, Any]):
-        """Create a new futurehouse task."""
+    async def acreate_task(
+        self,
+        task_data: TaskRequest | dict[str, Any],
+    ):
+        """Create a new futurehouse task.
+
+        Args:
+            task_data: The task request data
+        """
         if isinstance(task_data, dict):
             task_data = TaskRequest.model_validate(task_data)
 
@@ -573,6 +589,7 @@ class RestClient:
                 task_data.name.name,
                 self.stage,
             )
+
         response = await self.async_client.post(
             "/v0.1/crows", json=task_data.model_dump(mode="json")
         )
@@ -1478,25 +1495,6 @@ class RestClient:
         except Exception as e:
             raise WorldModelFetchError(f"An unexpected error occurred: {e}") from e
 
-    @retry(
-        stop=stop_after_attempt(MAX_RETRY_ATTEMPTS),
-        wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
-        retry=retry_if_connection_error,
-    )
-    def search_world_models(self, query: str, size: int = 10) -> list[str]:
-        """Search for world models.
-
-        Args:
-            query: The search query.
-            size: The number of results to return.
-
-        Returns:
-            A list of world model names.
-        """
-        response = self.client.get(f"/v0.1/world-models/search/{query}", params={"size": size})
-        response.raise_for_status()
-        return response.json()
-
     @retry(
         stop=stop_after_attempt(MAX_RETRY_ATTEMPTS),
         wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
@@ -1575,25 +1573,134 @@ class RestClient:
         wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
         retry=retry_if_connection_error,
     )
-    async def delete_world_model(self, world_model_id: UUID) -> None:
-        """Delete a world model snapshot by its ID.
+    def create_trajectory_group(self, name: str) -> str:
+        """Create a new trajectory group.
+
+        Args:
+            name: The name for the trajectory group
+
+        Returns:
+            UUID of the created trajectory group
+
+        Raises:
+            TrajectoryGroupError: If there's an error creating the trajectory group
+        """
+        try:
+            data = {"name": name}
+            response = self.client.post("/v0.1/trajectory-groups", json=data)
+            response.raise_for_status()
+            return response.json()
+        except HTTPStatusError as e:
+            raise TrajectoryGroupError(
+                f"Error creating trajectory group: {e.response.status_code} - {e.response.text}"
+            ) from e
+        except Exception as e:
+            raise TrajectoryGroupError(f"Error creating trajectory group: {e}") from e
+
+    @retry(
+        stop=stop_after_attempt(MAX_RETRY_ATTEMPTS),
+        wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
+        retry=retry_if_connection_error,
+    )
+    def add_trajectory_to_group(self, group_id: str, trajectory_id: str) -> None:
+        """Add a trajectory to a group.
+
+        Args:
+            group_id: The UUID of the trajectory group
+            trajectory_id: The UUID of the trajectory to add
+
+        Raises:
+            TrajectoryGroupError: If there's an error adding the trajectory to the group
+        """
+        try:
+            data = {"trajectory_id": trajectory_id}
+            response = self.client.post(
+                f"/v0.1/trajectory-groups/{group_id}/trajectories", json=data
+            )
+            response.raise_for_status()
+        except HTTPStatusError as e:
+            if e.response.status_code == codes.NOT_FOUND:
+                raise TrajectoryGroupError(
+                    f"Trajectory group {group_id} or trajectory {trajectory_id} not found"
+                ) from e
+            if e.response.status_code == codes.FORBIDDEN:
+                raise TrajectoryGroupError(
+                    f"Permission denied to add trajectory to group {group_id}"
+                ) from e
+            raise TrajectoryGroupError(
+                f"Error adding trajectory to group: {e.response.status_code} - {e.response.text}"
+            ) from e
+        except Exception as e:
+            raise TrajectoryGroupError(f"Error adding trajectory to group: {e}") from e
+
+    @retry(
+        stop=stop_after_attempt(MAX_RETRY_ATTEMPTS),
+        wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
+        retry=retry_if_connection_error,
+    )
+    async def acreate_trajectory_group(self, name: str) -> str:
+        """Asynchronously create a new trajectory group.
 
         Args:
-            world_model_id: The unique ID of the world model snapshot to delete.
+            name: The name for the trajectory group
+
+        Returns:
+            UUID of the created trajectory group
 
         Raises:
-            WorldModelDeletionError: If the API call fails.
+            TrajectoryGroupError: If there's an error creating the trajectory group
         """
         try:
-            response = await self.async_client.delete(f"/v0.1/world-models/{world_model_id}")
+            data = {"name": name}
+            response = await self.async_client.post(
+                "/v0.1/trajectory-groups", json=data
+            )
             response.raise_for_status()
+            return response.json()
         except HTTPStatusError as e:
-            raise WorldModelDeletionError(
-                f"Error deleting world model: {e.response.status_code} - {e.response.text}"
+            raise TrajectoryGroupError(
+                f"Error creating trajectory group: {e.response.status_code} - {e.response.text}"
             ) from e
         except Exception as e:
-            raise WorldModelDeletionError(f"An unexpected error occurred: {e}") from e
-    
+            raise TrajectoryGroupError(f"Error creating trajectory group: {e}") from e
+
+    @retry(
+        stop=stop_after_attempt(MAX_RETRY_ATTEMPTS),
+        wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
+        retry=retry_if_connection_error,
+    )
+    async def aadd_trajectory_to_group(self, group_id: str, trajectory_id: str) -> None:
+        """Asynchronously add a trajectory to a group.
+
+        Args:
+            group_id: The UUID of the trajectory group
+            trajectory_id: The UUID of the trajectory to add
+
+        Raises:
+            TrajectoryGroupError: If there's an error adding the trajectory to the group
+        """
+        try:
+            data = {"trajectory_id": trajectory_id}
+            response = await self.async_client.post(
+                f"/v0.1/trajectory-groups/{group_id}/trajectories", json=data
+            )
+            response.raise_for_status()
+        except HTTPStatusError as e:
+            if e.response.status_code == codes.NOT_FOUND:
+                raise TrajectoryGroupError(
+                    f"Trajectory group {group_id} or trajectory {trajectory_id} not found"
+                ) from e
+            if e.response.status_code == codes.FORBIDDEN:
+                raise TrajectoryGroupError(
+                    f"Permission denied to add trajectory to group {group_id}"
+                ) from e
+            raise TrajectoryGroupError(
+                f"Error adding trajectory to group: {e.response.status_code} - {e.response.text}"
+            ) from e
+        except Exception as e:
+            raise TrajectoryGroupError(f"Error adding trajectory to group: {e}") from e
+
+
 def get_installed_packages() -> dict[str, str]:
     """Returns a dictionary of installed packages and their versions."""
     return {

futurehouse_client/models/__init__.py

@@ -13,6 +13,7 @@ from .app import (
     TaskResponse,
     TaskResponseVerbose,
 )
+from .rest import WorldModel, WorldModelResponse
 
 __all__ = [
     "AuthType",
@@ -28,4 +29,6 @@ __all__ = [
     "TaskRequest",
     "TaskResponse",
     "TaskResponseVerbose",
+    "WorldModel",
+    "WorldModelResponse",
 ]

futurehouse_client/models/app.py

@@ -652,6 +652,11 @@ class TaskRequest(BaseModel):
         description="Optional task identifier",
         alias="id",
     )
+    group_id: UUID | None = Field(
+        default=None,
+        description="Optional group identifier for the task",
+        alias="group_id",
+    )
     name: "str | JobNames" = Field(
         description="Name of the crow to execute eg. paperqa-crow"
     )

futurehouse_client/version.py

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

{futurehouse_client-0.3.20.dev112.dist-info → futurehouse_client-0.3.20.dev196.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.20.dev112
+Version: 0.3.20.dev196
 Summary: A client for interacting with endpoints of the FutureHouse service.
 Author-email: FutureHouse technical staff <hello@futurehouse.org>
 License:                                  Apache License
@@ -217,7 +217,7 @@ Requires-Dist: cloudpickle
 Requires-Dist: fhaviary
 Requires-Dist: httpx
 Requires-Dist: ldp>=0.22.0
-Requires-Dist: litellm==1.67.4.post1
+Requires-Dist: litellm
 Requires-Dist: pydantic
 Requires-Dist: python-dotenv
 Requires-Dist: tenacity
@@ -250,7 +250,8 @@ Dynamic: license-file
 
 # FutureHouse Platform API Documentation
 
-Documentation and tutorials for futurehouse-client, a client for interacting with endpoints of the FutureHouse platform.
+Documentation and tutorials for futurehouse-client,
+a client for interacting with endpoints of the FutureHouse platform.
 
 <!--TOC-->
 
@@ -289,16 +290,25 @@ task_data = {
 (task_response,) = client.run_tasks_until_done(task_data)
 ```
 
-A quickstart example can be found in the [client_notebook.ipynb](https://futurehouse.gitbook.io/futurehouse-cookbook/futurehouse-client/docs/client_notebook) file, where we show how to submit and retrieve a job task, pass runtime configuration to the agent, and ask follow-up questions to the previous job.
+A quickstart example can be found in the [client_notebook.ipynb][1] file,
+where we show how to submit and retrieve a job task,
+pass runtime configuration to the agent,
+and ask follow-up questions to the previous job.
+
+[1]: https://futurehouse.gitbook.io/futurehouse-cookbook/futurehouse-client/docs/client_notebook
 
 ## Functionalities
 
 FutureHouse client implements a RestClient (called `FutureHouseClient`) with the following functionalities:
 
-- [Simple task running](#simple-task-running): `run_tasks_until_done(TaskRequest)` or `await arun_tasks_until_done(TaskRequest)`
-- [Asynchronous tasks](#asynchronous-tasks): `get_task(task_id)` or `aget_task(task_id)` and `create_task(TaskRequest)` or `acreate_task(TaskRequest)`
+- [Simple task running](#simple-task-running):
+  `run_tasks_until_done(TaskRequest)` or `await arun_tasks_until_done(TaskRequest)`
+- [Asynchronous tasks](#asynchronous-tasks):
+  `get_task(task_id)` or `aget_task(task_id)`
+  and `create_task(TaskRequest)` or `acreate_task(TaskRequest)`
 
-To create a `FutureHouseClient`, you need to pass an FutureHouse platform api key (see [Authentication](#authentication)):
+To create a `FutureHouseClient`,
+you need to pass an FutureHouse platform api key (see [Authentication](#authentication)):
 
 ```python
 from futurehouse_client import FutureHouseClient
@@ -311,22 +321,33 @@ client = FutureHouseClient(api_key="your_api_key")
 
 ## Authentication
 
-In order to use the `FutureHouseClient`, you need to authenticate yourself. Authentication is done by providing an API key, which can be obtained directly from your [profile page in the FutureHouse platform](https://platform.futurehouse.org/profile).
+In order to use the `FutureHouseClient`, you need to authenticate yourself.
+Authentication is done by providing an API key,
+which can be obtained directly from your [profile page in the FutureHouse platform][2].
+
+[2]: https://platform.futurehouse.org/profile
 
 ## Simple task running
 
-In the FutureHouse platform, we define the deployed combination of an agent and an environment as a `job`. To invoke a job, we need to submit a `task` (also called a `query`) to it.
-`FutureHouseClient` can be used to submit tasks/queries to available jobs in the FutureHouse platform. Using a `FutureHouseClient` instance, you can submit tasks to the platform by calling the `create_task` method, which receives a `TaskRequest` (or a dictionary with `kwargs`) and returns the task id.
-Aiming to make the submission of tasks as simple as possible, we have created a `JobNames` `enum` that contains the available task types.
+In the FutureHouse platform,
+we define the deployed combination of an agent and an environment as a `job`.
+To invoke a job, we need to submit a `task` (also called a `query`) to it.
+`FutureHouseClient` can be used to submit tasks/queries to available jobs in the FutureHouse platform.
+Using a `FutureHouseClient` instance,
+you can submit tasks to the platform by calling the `create_task` method,
+which receives a `TaskRequest` (or a dictionary with `kwargs`) and returns the task id.
+Aiming to make the submission of tasks as simple as possible,
+we have created a `JobNames` `enum` that contains the available task types.
 
 The available supported jobs are:
-| Alias | Job Name | Task type | Description |
-| --- | --- | --- | --- |
-| `JobNames.CROW` | `job-futurehouse-paperqa2` | Fast Search | Ask a question of scientific data sources, and receive a high-accuracy, cited response. Built with [PaperQA2](https://github.com/Future-House/paper-qa). |
-| `JobNames.FALCON` | `job-futurehouse-paperqa2-deep` | Deep Search | Use a plethora of sources to deeply research. Receive a detailed, structured report as a response. |
-| `JobNames.OWL` | `job-futurehouse-hasanyone` | Precedent Search | Formerly known as HasAnyone, query if anyone has ever done something in science. |
-| `JobNames.PHOENIX` | `job-futurehouse-phoenix` | Chemistry Tasks | A new iteration of ChemCrow, Phoenix uses cheminformatics tools to do chemistry. Good for planning synthesis and designing new molecules. |
-| `JobNames.DUMMY` | `job-futurehouse-dummy` | Dummy Task | This is a dummy task. Mainly for testing purposes. |
+
+| Alias              | Job Name                        | Task type        | Description                                                                                                                                              |
+| ------------------ | ------------------------------- | ---------------- | -------------------------------------------------------------------------------------------------------------------------------------------------------- |
+| `JobNames.CROW`    | `job-futurehouse-paperqa2`      | Fast Search      | Ask a question of scientific data sources, and receive a high-accuracy, cited response. Built with [PaperQA2](https://github.com/Future-House/paper-qa). |
+| `JobNames.FALCON`  | `job-futurehouse-paperqa2-deep` | Deep Search      | Use a plethora of sources to deeply research. Receive a detailed, structured report as a response.                                                       |
+| `JobNames.OWL`     | `job-futurehouse-hasanyone`     | Precedent Search | Formerly known as HasAnyone, query if anyone has ever done something in science.                                                                         |
+| `JobNames.PHOENIX` | `job-futurehouse-phoenix`       | Chemistry Tasks  | A new iteration of ChemCrow, Phoenix uses cheminformatics tools to do chemistry. Good for planning synthesis and designing new molecules.                |
+| `JobNames.DUMMY`   | `job-futurehouse-dummy`         | Dummy Task       | This is a dummy task. Mainly for testing purposes.                                                                                                       |
 
 Using `JobNames`, the task submission looks like this:
 
@@ -369,7 +390,8 @@ if __name__ == "__main__":
     task_id = asyncio.run(main())
 ```
 
-Note that in either the sync or the async code, collections of tasks can be given to the client to run them in a batch:
+Note that in either the sync or the async code,
+collections of tasks can be given to the client to run them in a batch:
 
 ```python
 import asyncio
@@ -412,7 +434,12 @@ if __name__ == "__main__":
 | query          | str           | Query or task to be executed by the job                                                                             |
 | runtime_config | RuntimeConfig | Optional runtime parameters for the job                                                                             |
 
-`runtime_config` can receive a `AgentConfig` object with the desired kwargs. Check the available `AgentConfig` fields in the [LDP documentation](https://github.com/Future-House/ldp/blob/main/src/ldp/agent/agent.py#L87). Besides the `AgentConfig` object, we can also pass `timeout` and `max_steps` to limit the execution time and the number of steps the agent can take.
+`runtime_config` can receive a `AgentConfig` object with the desired kwargs.
+Check the available `AgentConfig` fields in the [LDP documentation][3].
+Besides the `AgentConfig` object,
+we can also pass `timeout` and `max_steps` to limit the execution time and the number of steps the agent can take.
+
+[3]: https://github.com/Future-House/ldp/blob/main/src/ldp/agent/agent.py#L87
 
 ```python
 from futurehouse_client import FutureHouseClient, JobNames
@@ -429,7 +456,8 @@ client = FutureHouseClient()
 print(task_response.answer)
 ```
 
-A `TaskResponse` will be returned from using our agents. For Owl, Crow, and Falcon, we default to a subclass, `PQATaskResponse` which has some key attributes:
+A `TaskResponse` will be returned from using our agents.
+For Owl, Crow, and Falcon, we default to a subclass, `PQATaskResponse` which has some key attributes:
 
 | Field                 | Type | Description                                                                     |
 | --------------------- | ---- | ------------------------------------------------------------------------------- |
@@ -437,7 +465,8 @@ A `TaskResponse` will be returned from using our agents. For Owl, Crow, and Falc
 | formatted_answer      | str  | Specially formatted answer with references.                                     |
 | has_successful_answer | bool | Flag for whether the agent was able to find a good answer to your query or not. |
 
-If using the `verbose` setting, much more data can be pulled down from your `TaskResponse`, which will exist across all agents (not just Owl, Crow, and Falcon).
+If using the `verbose` setting, much more data can be pulled down from your `TaskResponse`,
+which will exist across all agents (not just Owl, Crow, and Falcon).
 
 ```python
 from futurehouse_client import FutureHouseClient, JobNames
@@ -460,16 +489,18 @@ print(task_response.environment_frame)
 In that case, a `TaskResponseVerbose` will have the following fields:
 
 | Field             | Type | Description                                                                                                            |
-| ----------------- | ---- | ---------------------------------------------------------------------------------------------------------------------- | --- |
+| ----------------- | ---- | ---------------------------------------------------------------------------------------------------------------------- |
 | agent_state       | dict | Large object with all agent states during the progress of your task.                                                   |
 | environment_frame | dict | Large nested object with all environment data, for PQA environments it includes contexts, paper metadata, and answers. |
-| metadata          | dict | Extra metadata about your query.                                                                                       |     |
+| metadata          | dict | Extra metadata about your query.                                                                                       |
 
 ## Task Continuation
 
-Once a task is submitted and the answer is returned, FutureHouse platform allow you to ask follow-up questions to the previous task.
+Once a task is submitted and the answer is returned,
+FutureHouse platform allow you to ask follow-up questions to the previous task.
 It is also possible through the platform API.
-To accomplish that, we can use the `runtime_config` we discussed in the [Simple task running](#simple-task-running) section.
+To accomplish that,
+we can use the `runtime_config` we discussed in the [Simple task running](#simple-task-running) section.
 
 ```python
 from futurehouse_client import FutureHouseClient, JobNames
@@ -497,7 +528,9 @@ continued_task_data = {
 
 ## Asynchronous tasks
 
-Sometimes you may want to submit many jobs, while querying results at a later time. In this way you can do other things while waiting for a response. The platform API supports this as well rather than waiting for a result.
+Sometimes you may want to submit many jobs, while querying results at a later time.
+In this way you can do other things while waiting for a response.
+The platform API supports this as well rather than waiting for a result.
 
 ```python
 from futurehouse_client import FutureHouseClient, JobNames
@@ -515,4 +548,6 @@ task_id = client.create_task(task_data)
 task_status = client.get_task(task_id)
 ```
 
-`task_status` contains information about the task. For instance, its `status`, `task`, `environment_name` and `agent_name`, and other fields specific to the job. You can continually query the status until it's `success` before moving on.
+`task_status` contains information about the task.
+For instance, its `status`, `task`, `environment_name` and `agent_name`, and other fields specific to the job.
+You can continually query the status until it's `success` before moving on.

{futurehouse_client-0.3.20.dev112.dist-info → futurehouse_client-0.3.20.dev196.dist-info}/RECORD

@@ -1,11 +1,11 @@
-futurehouse_client/__init__.py,sha256=rRAJFxDI0nDHiMkyB-d7Q8Y2vVKFVnIFwOuRdjClTPw,707
+futurehouse_client/__init__.py,sha256=BztM_ntbgmIEjzvnBWcvPhvLjM8xGDFCK0Upf3-nIn8,488
 futurehouse_client/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-futurehouse_client/version.py,sha256=nAw6MsWnD67V5v8vArdCA5jF1Rjl2uTamVcuYk2lPQo,530
+futurehouse_client/version.py,sha256=ajDpFNAiJ6I3fYkoEBeu8VxH5XgyeyKdOFsM__tugkA,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=hDV1Pr9gemChBKtklnlZlrwSYVUfHEGNGGmVDhoByIw,63015
-futurehouse_client/models/__init__.py,sha256=5x-f9AoM1hGzJBEHcHAXSt7tPeImST5oZLuMdwp0mXc,554
-futurehouse_client/models/app.py,sha256=TrfSorJlxyuc9ZzJpd2tL1pzKOp3jNBYRqFMfmmUue0,28954
+futurehouse_client/clients/rest_client.py,sha256=aN5v9iyGplEfFaospYIGGstrAtiOgv0uFIKd51y4WrE,66890
+futurehouse_client/models/__init__.py,sha256=kQ4R7VEuRxO0IQEW_sk9CndBL7zzl8rUKI24ddyYLM0,647
+futurehouse_client/models/app.py,sha256=LSaKx83Yo0yBLpqhKrDw4DWxj4apNaWFyjM_dZnTxuI,29105
 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
@@ -13,9 +13,8 @@ futurehouse_client/utils/auth.py,sha256=tgWELjKfg8eWme_qdcRmc8TjQN9DVZuHHaVXZNHL
 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/utils/world_model_tools.py,sha256=f-Q1X55HuisYyfTEHhhLMY_VamSMCkdXuHHp_V509bc,1899
-futurehouse_client-0.3.20.dev112.dist-info/licenses/LICENSE,sha256=oQ9ZHjUi-_6GfP3gs14FlPb0OlGwE1QCCKFGnJ4LD2I,11341
-futurehouse_client-0.3.20.dev112.dist-info/METADATA,sha256=a4IX4zQbKP51L71qkqjS8--fvdyIGVQZqneJn7DZuLg,26118
-futurehouse_client-0.3.20.dev112.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-futurehouse_client-0.3.20.dev112.dist-info/top_level.txt,sha256=TRuLUCt_qBnggdFHCX4O_BoCu1j2X43lKfIZC-ElwWY,19
-futurehouse_client-0.3.20.dev112.dist-info/RECORD,,
+futurehouse_client-0.3.20.dev196.dist-info/licenses/LICENSE,sha256=oQ9ZHjUi-_6GfP3gs14FlPb0OlGwE1QCCKFGnJ4LD2I,11341
+futurehouse_client-0.3.20.dev196.dist-info/METADATA,sha256=hwUUzi7oZJjmRPk8PuwIwBpE5TeXXossw2IpotAwHIA,26805
+futurehouse_client-0.3.20.dev196.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+futurehouse_client-0.3.20.dev196.dist-info/top_level.txt,sha256=TRuLUCt_qBnggdFHCX4O_BoCu1j2X43lKfIZC-ElwWY,19
+futurehouse_client-0.3.20.dev196.dist-info/RECORD,,

futurehouse_client/utils/world_model_tools.py

@@ -1,52 +0,0 @@
-from futurehouse_client.clients.rest_client import RestClient
-from futurehouse_client.models.app import Stage
-from futurehouse_client.models.rest import WorldModel
-from uuid import UUID
-from aviary.core import Tool
-
-
-class WorldModelTools:
-    CLIENT = RestClient(
-            stage = Stage.DEV,
-            api_key="Dk7WJRLpqTFNxp5dYRoj6A.platformv01.eyJqdGkiOiI4ODAyZmZiNy1hNjM2LTRkMWYtYWE4NC1lZTQzYTMzMzRjZGMiLCJzdWIiOiJuN1dJbGU5VDljZ1BkTjd2OUJlM0pEUlpZVTgyIiwiaWF0IjoxNzQ5MDc2NzYzfQ.vThmFNLChP54DZBwB+qeMTB6CvAQ1IVXkTcpB0+efZ0",
-        )
-    
-    @staticmethod
-    def create_world_model(name: str, description: str, content: str) -> UUID:
-        """Create a new world model.
-        
-        Args:
-            name: The name of the world model.
-            description: A description of the world model.
-            content: The content/data of the world model.
-            
-        Returns:
-            UUID: The ID of the newly created world model.
-        """
-        world_model = WorldModel(
-            name=name,
-            description=description,
-            content=content,
-        )
-        return WorldModelTools.CLIENT.create_world_model(world_model)
-    
-    @staticmethod
-    def search_world_models(query: str) -> list[str]:
-        """Search for world models using a text query.
-        
-        Args:
-            query: The search query string to match against world model content.
-            
-        Returns:
-            list[str]: A list of world model IDs that match the search query.
-        """
-        return WorldModelTools.CLIENT.search_world_models(query, size=1)
-    
-create_world_model_tool = Tool.from_function(WorldModelTools.create_world_model)
-search_world_model_tool = Tool.from_function(WorldModelTools.search_world_models)
-
-def make_world_model_tools() -> list[Tool]:
-    return [
-        search_world_model_tool,
-        create_world_model_tool,
-    ]