sapiopycommons 2025.8.15a704__tar.gz → 2025.8.15a705__tar.gz

{sapiopycommons-2025.8.15a704 → sapiopycommons-2025.8.15a705}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sapiopycommons
-Version: 2025.8.15a704
+Version: 2025.8.15a705
 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.8.15a704 → sapiopycommons-2025.8.15a705}/pyproject.toml

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

{sapiopycommons-2025.8.15a704 → sapiopycommons-2025.8.15a705}/src/sapiopycommons/ai/test_client.py

@@ -6,6 +6,7 @@ from typing import Any
 import grpc
 from sapiopylib.rest.User import SapioUser
 
+from sapiopycommons.ai.api.fielddefinitions.proto.fields_pb2 import FieldValuePbo
 from sapiopycommons.ai.api.plan.converter.proto.converter_pb2 import ConverterDetailsRequestPbo, \
     ConverterDetailsResponsePbo, ConvertResponsePbo, ConvertRequestPbo
 from sapiopycommons.ai.api.plan.converter.proto.converter_pb2_grpc import ConverterServiceStub
@@ -103,7 +104,7 @@ class TestClient:
     options: list[tuple[str, Any]] | None
     connection: SapioConnectionInfoPbo
     _request_inputs: list[StepItemContainerPbo]
-    _config_fields: dict[str, Any]
+    _config_fields: dict[str, FieldValuePbo]
 
     def __init__(self, grpc_server_url: str, user: SapioUser | None = None,
                  options: list[tuple[str, Any]] | None = None):
@@ -172,22 +173,25 @@ class TestClient:
         """
         self._request_inputs.clear()
 
-    def add_config_field(self, field_name: str, value: Any) -> None:
+    def add_config_field(self, field_name: str, value: FieldValue) -> 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.
         """
+        if not isinstance(value, FieldValuePbo):
+            value = ProtobufUtils.value_to_field_pbo(value)
         self._config_fields[field_name] = value
 
-    def add_config_fields(self, config_fields: dict[str, Any]) -> None:
+    def add_config_fields(self, config_fields: dict[str, FieldValue]) -> 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)
+        for x, y in config_fields.items():
+            self.add_config_field(x, y)
 
     def clear_configs(self) -> None:
         """
@@ -248,7 +252,7 @@ class TestClient:
                 ProcessStepRequestPbo(
                     sapio_user=self.connection,
                     tool_name=tool_name,
-                    config_field_values={x: ProtobufUtils.value_to_field_pbo(y) for x, y in self._config_fields.items()},
+                    config_field_values=self._config_fields,
                     dry_run=is_dry_run,
                     verbose_logging=True,
                     input=[