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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.14
+Version: 0.3.15.dev15
 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.14 → futurehouse_client-0.3.15.dev15}/futurehouse_client/clients/rest_client.py

@@ -26,6 +26,7 @@ from httpx import (
     ReadTimeout,
     RemoteProtocolError,
 )
+from ldp.agent import AgentConfig
 from pydantic import BaseModel, ConfigDict, model_validator
 from requests.exceptions import RequestException, Timeout
 from tenacity import (
@@ -575,7 +576,11 @@ class RestClient:
                 )
             logger.debug(f"Sending files: {[f[1][0] for f in files]}")
             data = {
-                "agent": config.agent,
+                "agent": (
+                    config.agent.model_dump_json()
+                    if isinstance(config.agent, AgentConfig)
+                    else config.agent
+                ),
                 "job_name": config.job_name,
                 "organization": selected_org,
                 "environment": config.environment,

{futurehouse_client-0.3.14 → futurehouse_client-0.3.15.dev15}/futurehouse_client/models/app.py

@@ -1,3 +1,4 @@
+import json
 import os
 import re
 from enum import StrEnum, auto
@@ -401,10 +402,10 @@ class JobDeploymentConfig(BaseModel):
         description="The python version your docker image should build with.",
     )
 
-    agent: Agent | str = Field(
+    agent: Agent | AgentConfig | str = Field(
         default="ldp.agent.SimpleAgent",
         description="Your desired agent path, should be a module reference and a fully qualified name. "
-        "example: ldp.agent.SimpleAgent",
+        "example: ldp.agent.SimpleAgent or by instantiating AgentConfig to further customize the agent.",
     )
 
     requires_aviary_internal: bool = Field(
@@ -531,7 +532,12 @@ class JobDeploymentConfig(BaseModel):
 
     @field_validator("agent")
     @classmethod
-    def validate_agent_path(cls, value: Agent | str) -> Agent | str:
+    def validate_agent(
+        cls, value: str | Agent | AgentConfig
+    ) -> str | Agent | AgentConfig:
+        if isinstance(value, AgentConfig):
+            return value
+
         if isinstance(value, Agent):
             return value
 
@@ -584,6 +590,13 @@ class RuntimeConfig(BaseModel):
             "agent selected during Crow deployment in the JobDeploymentConfig object."
         ),
     )
+    environment_config: dict[str, Any] | None = Field(
+        default=None,
+        description=(
+            "Kwargs to be passed to the environment constructor at runtime. "
+            "Not all environments support this functionality."
+        ),
+    )
     continued_job_id: UUID | None = Field(
         default=None,
         description="Optional job identifier for a continued job",
@@ -606,6 +619,19 @@ class RuntimeConfig(BaseModel):
             raise ValueError(f"Invalid agent path format: {value}")
         return value
 
+    @field_validator("environment_config")
+    @classmethod
+    def validate_environment_config(
+        cls, value: dict[str, Any] | None
+    ) -> dict[str, Any] | None:
+        if value is None:
+            return None
+        try:
+            json.dumps(value)
+        except (TypeError, OverflowError) as err:
+            raise ValueError("Environment config must be JSON serializable") from err
+        return value
+
 
 class TaskRequest(BaseModel):
     task_id: UUID | None = Field(

{futurehouse_client-0.3.14 → futurehouse_client-0.3.15.dev15}/futurehouse_client.egg-info/PKG-INFO

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