sapiopycommons 2025.7.1a576__tar.gz → 2025.7.2a578__tar.gz

{sapiopycommons-2025.7.1a576 → sapiopycommons-2025.7.2a578}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sapiopycommons
-Version: 2025.7.1a576
+Version: 2025.7.2a578
 Summary: Official Sapio Python API Utilities Package
 Project-URL: Homepage, https://github.com/sapiosciences
 Author-email: Jonathan Steck <jsteck@sapiosciences.com>, Yechen Qiao <yqiao@sapiosciences.com>

{sapiopycommons-2025.7.1a576 → sapiopycommons-2025.7.2a578}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "sapiopycommons"
-version='2025.07.01a576'
+version='2025.07.02a578'
 authors = [
     { name="Jonathan Steck", email="jsteck@sapiosciences.com" },
     { name="Yechen Qiao", email="yqiao@sapiosciences.com" },

{sapiopycommons-2025.7.1a576 → sapiopycommons-2025.7.2a578}/src/sapiopycommons/ai/test_client.py

@@ -79,6 +79,7 @@ class TestClient:
     server_url: str
     connection: SapioConnectionInfoPbo
     _request_inputs: list[Any]
+    _config_fields: dict[str, Any]
 
     def __init__(self, server_url: str):
         """
@@ -87,6 +88,7 @@ class TestClient:
         self.create_user()
         self.server_url = server_url
         self._request_inputs = []
+        self._config_fields = {}
 
     def create_user(self):
         """
@@ -143,6 +145,38 @@ class TestClient:
         """
         self._request_inputs.clear()
 
+    def add_config_field(self, field_name: str, value: Any) -> None:
+        """
+        Add a configuration field value to the next request.
+
+        :param field_name: The name of the configuration field.
+        :param value: The value to set for the configuration field.
+        """
+        self._config_fields[field_name] = value
+
+    def add_config_fields(self, config_fields: dict[str, Any]) -> None:
+        """
+        Add multiple configuration field values to the next request.
+
+        :param config_fields: A dictionary of configuration field names and their corresponding values.
+        """
+        self._config_fields.update(config_fields)
+
+    def clear_configs(self) -> None:
+        """
+        Clear all configuration field values that have been added to the next request.
+        This is useful if you want to start a new request without the previous configurations.
+        """
+        self._config_fields.clear()
+
+    def clear_request(self) -> None:
+        """
+        Clear all inputs and configuration fields that have been added to the next request.
+        This is useful if you want to start a new request without the previous inputs and configurations.
+        """
+        self.clear_inputs()
+        self.clear_configs()
+
     def _add_input(self, data_type: DataTypePbo, items: Any) -> None:
         """
         Helper method for adding inputs to the next request.
@@ -187,6 +221,7 @@ class TestClient:
                 ProcessStepRequestPbo(
                     sapio_user=self.connection,
                     tool_name=tool_name,
+                    config_field_values=self._config_fields,
                     input=[
                         StepInputBatchPbo(is_partial=False, item_container=item)
                         for item in self._request_inputs