sapiopycommons 2025.8.15a704__py3-none-any.whl → 2025.8.15a705__py3-none-any.whl

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=[

{sapiopycommons-2025.8.15a704.dist-info → sapiopycommons-2025.8.15a705.dist-info}/METADATA

@@ -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.dist-info → sapiopycommons-2025.8.15a705.dist-info}/RECORD

@@ -2,7 +2,7 @@ sapiopycommons/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sapiopycommons/ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sapiopycommons/ai/converter_service_base.py,sha256=w4_PXjlk7_F5AUMlPUhKJ9gZlxY9RNG6yH9cDDFROuo,4620
 sapiopycommons/ai/protobuf_utils.py,sha256=QdYcelIvOIKBV5u_KR_Fh06kZ0eJDIb5ZVnfqV01FeE,24601
-sapiopycommons/ai/test_client.py,sha256=fo8j_1SjkR6eWAdO3_NU6CsCclOqQGBW9woSUjuOSnM,14210
+sapiopycommons/ai/test_client.py,sha256=CBacWKFVLdrStxuyRnJP5ifDen5dpvkgdF9mz-_vlu4,14397
 sapiopycommons/ai/tool_service_base.py,sha256=kfMWU0OndYl0XnKwztHGTT-L20Mpq7VUI-vhwyfJkkI,43375
 sapiopycommons/ai/api/fielddefinitions/proto/fields_pb2.py,sha256=YcZjb_YM-XeLErM8hEC_S7vGMVGvcXAMGs2b-u5zvOE,2377
 sapiopycommons/ai/api/fielddefinitions/proto/fields_pb2.pyi,sha256=FwtXmNAf7iYGEFm4kbqb04v77jNHbZg18ZmEDhle_bU,1444
@@ -97,7 +97,7 @@ sapiopycommons/webhook/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3
 sapiopycommons/webhook/webhook_context.py,sha256=D793uLsb1691SalaPnBUk3rOSxn_hYLhdvkaIxjNXss,1909
 sapiopycommons/webhook/webhook_handlers.py,sha256=7o_wXOruhT9auNh8OfhJAh4WhhiPKij67FMBSpGPICc,39939
 sapiopycommons/webhook/webservice_handlers.py,sha256=cvW6Mk_110BzYqkbk63Kg7jWrltBCDALOlkJRu8h4VQ,14300
-sapiopycommons-2025.8.15a704.dist-info/METADATA,sha256=gRXt65SiBgfQjkbYDqEgSaoVxLfxe7G7tFlc2dzoero,3143
-sapiopycommons-2025.8.15a704.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-sapiopycommons-2025.8.15a704.dist-info/licenses/LICENSE,sha256=HyVuytGSiAUQ6ErWBHTqt1iSGHhLmlC8fO7jTCuR8dU,16725
-sapiopycommons-2025.8.15a704.dist-info/RECORD,,
+sapiopycommons-2025.8.15a705.dist-info/METADATA,sha256=jzsSGPS3vU0kRu46cYHefEm0kFnErZgXJ_jQn8iq2PI,3143
+sapiopycommons-2025.8.15a705.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+sapiopycommons-2025.8.15a705.dist-info/licenses/LICENSE,sha256=HyVuytGSiAUQ6ErWBHTqt1iSGHhLmlC8fO7jTCuR8dU,16725
+sapiopycommons-2025.8.15a705.dist-info/RECORD,,