sapiopycommons 2025.8.12a697__py3-none-any.whl → 2025.8.12a698__py3-none-any.whl

sapiopycommons/ai/test_client.py

@@ -1,327 +0,0 @@
-import base64
-import json
-from enum import Enum
-from typing import Any
-
-import grpc
-from sapiopylib.rest.User import SapioUser
-
-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
-from sapiopycommons.ai.api.plan.item.proto.item_container_pb2 import ContentTypePbo
-from sapiopycommons.ai.api.plan.tool.proto.entry_pb2 import StepBinaryContainerPbo, StepCsvRowPbo, \
-    StepCsvHeaderRowPbo, StepCsvContainerPbo, StepJsonContainerPbo, StepTextContainerPbo, \
-    StepItemContainerPbo, StepInputBatchPbo
-from sapiopycommons.ai.api.plan.tool.proto.tool_pb2 import ProcessStepResponsePbo, ProcessStepRequestPbo, \
-    ToolDetailsRequestPbo, ToolDetailsResponsePbo, ProcessStepResponseStatusPbo
-from sapiopycommons.ai.api.plan.tool.proto.tool_pb2_grpc import ToolServiceStub
-from sapiopycommons.ai.api.session.proto.sapio_conn_info_pb2 import SapioConnectionInfoPbo, SapioUserSecretTypePbo
-from sapiopycommons.ai.protobuf_utils import ProtobufUtils
-from sapiopycommons.general.aliases import FieldValue
-
-
-class ContainerType(Enum):
-    """
-    An enum of the different container contents of a StepItemContainerPbo.
-    """
-    BINARY = "binary"
-    CSV = "csv"
-    JSON = "json"
-    TEXT = "text"
-
-
-# FR-47422: Created class.
-class ToolOutput:
-    """
-    A class for holding the output of a TestClient that calls a ToolService. ToolOutput objects an be
-    printed to show the output of the tool in a human-readable format.
-    """
-    tool_name: str
-
-    status: str
-    message: str
-
-    binary_output: list[bytes]
-    csv_output: list[dict[str, Any]]
-    json_output: list[Any]
-    text_output: list[str]
-
-    new_records: list[dict[str, FieldValue]]
-
-    logs: list[str]
-
-    def __init__(self, tool_name: str):
-        self.tool_name = tool_name
-        self.binary_output = []
-        self.csv_output = []
-        self.json_output = []
-        self.text_output = []
-        self.new_records = []
-        self.logs = []
-
-    def __str__(self):
-        ret_val: str = f"{self.tool_name} Output:\n"
-        ret_val += f"\tStatus: {self.status}\n"
-        ret_val += f"\tMessage: {self.message}\n"
-        ret_val += "-" * 25 + "\n"
-
-        ret_val += f"Binary Output: {len(self.binary_output)} item(s)\n"
-        for binary in self.binary_output:
-            ret_val += f"\t{len(binary)} byte(s)\n"
-            ret_val += f"\t{binary[:50]}...\n"
-
-        ret_val += f"CSV Output: {len(self.csv_output)} item(s)\n"
-        if self.csv_output:
-            ret_val += f"\tHeaders: {', '.join(self.csv_output[0].keys())}\n"
-            for i, csv_row in enumerate(self.csv_output):
-                ret_val += f"\t{i}: {', '.join(f'{v}' for k, v in csv_row.items())}\n"
-
-        ret_val += f"JSON Output: {len(self.json_output)} item(s)\n"
-        if self.json_output:
-            ret_val += f"{json.dumps(self.json_output, indent=2)}\n"
-
-        ret_val += f"Text Output: {len(self.text_output)} item(s)\n"
-        for text in self.text_output:
-            ret_val += f"\t{text}\n"
-
-        ret_val += f"New Records: {len(self.new_records)} item(s)\n"
-        for record in self.new_records:
-            ret_val += f"{json.dumps(record, indent=2)}\n"
-
-        ret_val += f"Logs: {len(self.logs)} item(s)\n"
-        for log in self.logs:
-            ret_val += f"\t{log}\n"
-        return ret_val
-
-
-class TestClient:
-    """
-    A client for testing a ToolService.
-    """
-    grpc_server_url: str
-    connection: SapioConnectionInfoPbo
-    _request_inputs: list[StepItemContainerPbo]
-    _config_fields: dict[str, Any]
-
-    def __init__(self, grpc_server_url: str, user: SapioUser | None = None):
-        """
-        :param grpc_server_url: The URL of the gRPC server to connect to.
-        :param user: Optional SapioUser object to use for the connection. If not provided, a default connection
-            will be created with test credentials.
-        """
-        self.grpc_server_url = grpc_server_url
-        self._create_connection(user)
-        self._request_inputs = []
-        self._config_fields = {}
-
-    def _create_connection(self, user: SapioUser | None = None):
-        """
-        Create a SapioConnectionInfoPbo object with test credentials. This method can be overridden to
-        create a user with specific credentials for testing.
-        """
-        self.connection = SapioConnectionInfoPbo()
-        self.connection.username = user.username if user else "Testing"
-        self.connection.webservice_url = user.url if user else "https://localhost:8080/webservice/api"
-        self.connection.app_guid = user.guid if user else "1234567890"
-        self.connection.rmi_host.append("Testing")
-        self.connection.rmi_port = 9001
-        if user and user.password:
-            self.connection.secret_type = SapioUserSecretTypePbo.PASSWORD
-            self.connection.secret = "Basic " + base64.b64encode(f'{user.username}:{user.password}'.encode()).decode()
-        else:
-            self.connection.secret_type = SapioUserSecretTypePbo.SESSION_TOKEN
-            self.connection.secret = user.api_token if user and user.api_token else "test_api_token"
-
-    def add_binary_input(self, input_data: list[bytes]) -> None:
-        """
-        Add a binary input to the the next request.
-        """
-        self._add_input(ContainerType.BINARY, StepBinaryContainerPbo(items=input_data))
-
-    def add_csv_input(self, input_data: list[dict[str, Any]]) -> None:
-        """
-        Add a CSV input to the next request.
-        """
-        csv_items = []
-        for row in input_data:
-            csv_items.append(StepCsvRowPbo(cells=[str(value) for value in row.values()]))
-        header = StepCsvHeaderRowPbo(cells=list(input_data[0].keys()))
-        self._add_input(ContainerType.CSV, StepCsvContainerPbo(header=header, items=csv_items))
-
-    def add_json_input(self, input_data: list[dict[str, Any]]) -> None:
-        """
-        Add a JSON input to the next request.
-        """
-        self._add_input(ContainerType.JSON, StepJsonContainerPbo(items=[json.dumps(x) for x in input_data]))
-
-    def add_text_input(self, input_data: list[str]) -> None:
-        """
-        Add a text input to the next request.
-        """
-        self._add_input(ContainerType.TEXT, StepTextContainerPbo(items=input_data))
-
-    def clear_inputs(self) -> None:
-        """
-        Clear all inputs that have been added to the next request.
-        This is useful if you want to start a new request without the previous inputs.
-        """
-        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, container_type: ContainerType, items: Any) -> None:
-        """
-        Helper method for adding inputs to the next request.
-        """
-        match container_type:
-            # The content type doesn't matter when we're just testing.
-            case ContainerType.BINARY:
-                container = StepItemContainerPbo(content_type=ContentTypePbo(), binary_container=items)
-            case ContainerType.CSV:
-                container = StepItemContainerPbo(content_type=ContentTypePbo(), csv_container=items)
-            case ContainerType.JSON:
-                container = StepItemContainerPbo(content_type=ContentTypePbo(), json_container=items)
-            case ContainerType.TEXT:
-                container = StepItemContainerPbo(content_type=ContentTypePbo(), text_container=items)
-            case _:
-                raise ValueError(f"Unsupported data type: {container_type}")
-        self._request_inputs.append(container)
-
-    def get_service_details(self) -> ToolDetailsResponsePbo:
-        """
-        Get the details of the tools from the server.
-
-        :return: A ToolDetailsResponsePbo object containing the details of the tool service.
-        """
-        with grpc.insecure_channel(self.grpc_server_url) as channel:
-            stub = ToolServiceStub(channel)
-            return stub.GetToolDetails(ToolDetailsRequestPbo(sapio_conn_info=self.connection))
-
-    def call_tool(self, tool_name: str, is_dry_run: bool = False) -> ToolOutput:
-        """
-        Send the request to the tool service for a particular tool name. This will send all the inputs that have been
-        added using the add_X_input functions.
-
-        :param tool_name: The name of the tool to call on the server.
-        :param is_dry_run: If True, the tool will not be executed, but the request will be validated.
-        :return: A ToolOutput object containing the results of the tool service call.
-        """
-        with grpc.insecure_channel(self.grpc_server_url) as channel:
-            stub = ToolServiceStub(channel)
-
-            response: ProcessStepResponsePbo = stub.ProcessData(
-                ProcessStepRequestPbo(
-                    sapio_user=self.connection,
-                    tool_name=tool_name,
-                    config_field_values=self._config_fields,
-                    dry_run=is_dry_run,
-                    verbose_logging=True,
-                    input=[
-                        StepInputBatchPbo(is_partial=False, item_container=item)
-                        for item in self._request_inputs
-                    ]
-                )
-            )
-
-            results = ToolOutput(tool_name)
-
-            match response.status:
-                case ProcessStepResponseStatusPbo.SUCCESS:
-                    results.status = "Success"
-                case ProcessStepResponseStatusPbo.FAILURE:
-                    results.status = "Failure"
-                case _:
-                    results.status = "Unknown"
-            results.message = response.status_message
-
-            for item in response.output:
-                container = item.item_container
-
-                results.binary_output.extend(container.binary_container.items)
-                for header in container.csv_container.header.cells:
-                    output_row: dict[str, Any] = {}
-                    for i, row in enumerate(container.csv_container.items):
-                        output_row[header] = row.cells[i]
-                    results.csv_output.append(output_row)
-                results.json_output.extend([json.loads(x) for x in container.json_container.items])
-                results.text_output.extend(container.text_container.items)
-
-            for record in response.new_records:
-                field_map: dict[str, Any] = {x: ProtobufUtils.field_pbo_to_value(y) for x, y in record.fields.items()}
-                results.new_records.append(field_map)
-
-            results.logs.extend(response.log)
-
-            return results
-
-
-class TestConverterClient:
-    """
-    A client for testing a ConverterService.
-    """
-    grpc_server_url: str
-
-    def __init__(self, grpc_server_url: str):
-        """
-        :param grpc_server_url: The URL of the gRPC server to connect to.
-        """
-        self.grpc_server_url = grpc_server_url
-
-    def get_converter_details(self) -> ConverterDetailsResponsePbo:
-        """
-        Get the details of the converters from the server.
-
-        :return: A ToolDetailsResponsePbo object containing the details of the converter service.
-        """
-        with grpc.insecure_channel(self.grpc_server_url) as channel:
-            stub = ConverterServiceStub(channel)
-            return stub.GetConverterDetails(ConverterDetailsRequestPbo())
-
-    def convert_content(self, input_container: StepItemContainerPbo, target_type: ContentTypePbo) \
-            -> StepItemContainerPbo:
-        """
-        Convert the content of the input container to the target content type.
-
-        :param input_container: The input container to convert. This container must have a ContentTypePbo set that
-            matches one of the input types that the converter service supports.
-        :param target_type: The target content type to convert to. This must match one of the target types that the
-            converter service supports.
-        :return: A StepItemContainerPbo object containing the converted content.
-        """
-        with grpc.insecure_channel(self.grpc_server_url) as channel:
-            stub = ConverterServiceStub(channel)
-            response: ConvertResponsePbo = stub.ConvertContent(
-                ConvertRequestPbo(item_container=input_container, target_content_type=target_type)
-            )
-            return response.item_container