uipath 2.1.127__py3-none-any.whl → 2.1.128__py3-none-any.whl

uipath/utils/dynamic_schema.py

@@ -1,17 +1,20 @@
 """Json schema to dynamic pydantic model."""
 
-from typing import Any, Dict, List, Optional, Type, Union
+from enum import Enum
+from typing import Any, Dict, List, Type, Union
 
 from pydantic import BaseModel, Field, create_model
 
 
 def jsonschema_to_pydantic(
     schema: dict[str, Any],
-    definitions: Optional[dict[str, Any]] = None,
 ) -> Type[BaseModel]:
     """Convert a schema dict to a pydantic model.
 
-    Modified version of https://github.com/kreneskyp/jsonschema-pydantic to account for two unresolved issues.
+    Modified version of https://github.com/kreneskyp/jsonschema-pydantic to account for three unresolved issues.
+      1. Support for title
+      2. Better representation of optionals.
+      3. Support for optional
 
     Args:
         schema: JSON schema.
@@ -19,25 +22,14 @@ def jsonschema_to_pydantic(
 
     Returns: Pydantic model.
     """
-    title = schema.get("title", "DynamicModel")
-    assert isinstance(title, str), "Title of a model must be a string."
-
-    description = schema.get("description", None)
-
-    # top level schema provides definitions
-    if definitions is None:
-        if "$defs" in schema:
-            definitions = schema["$defs"]
-        elif "definitions" in schema:
-            definitions = schema["definitions"]
-        else:
-            definitions = {}
+    dynamic_type_counter = 0
+    combined_model_counter = 0
 
     def convert_type(prop: dict[str, Any]) -> Any:
+        nonlocal dynamic_type_counter, combined_model_counter
         if "$ref" in prop:
-            ref_path = prop["$ref"].split("/")
-            ref = definitions[ref_path[-1]]
-            return jsonschema_to_pydantic(ref, definitions)
+            # This is the full path. It will be updated in update_forward_refs.
+            return prop["$ref"].split("/")[-1].capitalize()
 
         if "type" in prop:
             type_mapping = {
@@ -52,13 +44,55 @@ def jsonschema_to_pydantic(
 
             type_ = prop["type"]
 
-            if type_ == "array":
+            if "enum" in prop:
+                dynamic_members = {
+                    f"KEY_{i}": value for i, value in enumerate(prop["enum"])
+                }
+
+                base_type: Any = type_mapping.get(type_, Any)
+
+                class DynamicEnum(base_type, Enum):
+                    pass
+
+                type_ = DynamicEnum(prop.get("title", "DynamicEnum"), dynamic_members)  # type: ignore[call-arg]
+                return type_
+            elif type_ == "array":
                 item_type: Any = convert_type(prop.get("items", {}))
-                assert isinstance(item_type, type)
                 return List[item_type]  # noqa F821
             elif type_ == "object":
                 if "properties" in prop:
-                    return jsonschema_to_pydantic(prop, definitions)
+                    if "title" in prop and prop["title"]:
+                        title = prop["title"]
+                    else:
+                        title = f"DynamicType_{dynamic_type_counter}"
+                        dynamic_type_counter += 1
+
+                    fields: dict[str, Any] = {}
+                    required_fields = prop.get("required", [])
+
+                    for name, property in prop.get("properties", {}).items():
+                        pydantic_type = convert_type(property)
+                        field_kwargs = {}
+                        if "default" in property:
+                            field_kwargs["default"] = property["default"]
+                        if name not in required_fields:
+                            # Note that we do not make this optional. This is due to a limitation in Pydantic/Python.
+                            # If we convert the Optional type back to json schema, it is represented as type | None.
+                            # pydantic_type = Optional[pydantic_type]
+
+                            if "default" not in field_kwargs:
+                                field_kwargs["default"] = None
+                        if "description" in property:
+                            field_kwargs["description"] = property["description"]
+                        if "title" in property:
+                            field_kwargs["title"] = property["title"]
+
+                        fields[name] = (pydantic_type, Field(**field_kwargs))
+
+                    object_model = create_model(title, **fields)
+                    if "description" in prop:
+                        object_model.__doc__ = prop["description"]
+                    return object_model
                 else:
                     return Dict[str, Any]
             else:
@@ -67,9 +101,13 @@ def jsonschema_to_pydantic(
         elif "allOf" in prop:
             combined_fields = {}
             for sub_schema in prop["allOf"]:
-                model = jsonschema_to_pydantic(sub_schema, definitions)
+                model = convert_type(sub_schema)
                 combined_fields.update(model.__annotations__)
-            return create_model("CombinedModel", **combined_fields)
+            combined_model = create_model(
+                f"CombinedModel_{combined_model_counter}", **combined_fields
+            )
+            combined_model_counter += 1
+            return combined_model
 
         elif "anyOf" in prop:
             unioned_types = tuple(
@@ -81,31 +119,10 @@ def jsonschema_to_pydantic(
         else:
             raise ValueError(f"Unsupported schema: {prop}")
 
-    fields: dict[str, Any] = {}
-    required_fields = schema.get("required", [])
-
-    for name, prop in schema.get("properties", {}).items():
-        pydantic_type = convert_type(prop)
-        field_kwargs = {}
-        if "default" in prop:
-            field_kwargs["default"] = prop["default"]
-        if name not in required_fields:
-            # Note that we do not make this optional. This is due to a limitation in Pydantic/Python.
-            # If we convert the Optional type back to json schema, it is represented as type | None.
-            # pydantic_type = Optional[pydantic_type]
-
-            if "default" not in field_kwargs:
-                field_kwargs["default"] = None
-        if "description" in prop:
-            field_kwargs["description"] = prop["description"]
-        if "title" in prop:
-            field_kwargs["title"] = prop["title"]
-
-        fields[name] = (pydantic_type, Field(**field_kwargs))
-
-    convert_type(schema.get("properties", {}).get("choices", {}))
-
-    model = create_model(title, **fields)
-    if description:
-        model.__doc__ = description
+    namespace: dict[str, Any] = {}
+    for name, definition in schema.get("$defs", schema.get("definitions", {})).items():
+        model = convert_type(definition)
+        namespace[name.capitalize()] = model
+    model = convert_type(schema)
+    model.model_rebuild(force=True, _types_namespace=namespace)
     return model

{uipath-2.1.127.dist-info → uipath-2.1.128.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath
-Version: 2.1.127
+Version: 2.1.128
 Summary: Python SDK and CLI for UiPath Platform, enabling programmatic interaction with automation services, process management, and deployment tools.
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-python

{uipath-2.1.127.dist-info → uipath-2.1.128.dist-info}/RECORD

@@ -223,9 +223,9 @@ uipath/tracing/_traced.py,sha256=VAwEIfDHLx-AZ792SeGxCtOttEJXrLgI0YCkUMbxHsQ,203
 uipath/tracing/_utils.py,sha256=emsQRgYu-P1gj1q7XUPJD94mOa12JvhheRkuZJpLd9Y,15051
 uipath/utils/__init__.py,sha256=VD-KXFpF_oWexFg6zyiWMkxl2HM4hYJMIUDZ1UEtGx0,105
 uipath/utils/_endpoints_manager.py,sha256=tnF_FiCx8qI2XaJDQgYkMN_gl9V0VqNR1uX7iawuLp8,8230
-uipath/utils/dynamic_schema.py,sha256=w0u_54MoeIAB-mf3GmwX1A_X8_HDrRy6p998PvX9evY,3839
-uipath-2.1.127.dist-info/METADATA,sha256=2oA6gGyXHLX8-Cr7G2onsb1-XXfeY6G4_iXQlZgcK-E,6626
-uipath-2.1.127.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-uipath-2.1.127.dist-info/entry_points.txt,sha256=9C2_29U6Oq1ExFu7usihR-dnfIVNSKc-0EFbh0rskB4,43
-uipath-2.1.127.dist-info/licenses/LICENSE,sha256=-KBavWXepyDjimmzH5fVAsi-6jNVpIKFc2kZs0Ri4ng,1058
-uipath-2.1.127.dist-info/RECORD,,
+uipath/utils/dynamic_schema.py,sha256=ahgRLBWzuU0SQilaQVBJfBAVjimq3N3QJ1ztx0U_h2c,4943
+uipath-2.1.128.dist-info/METADATA,sha256=ue_yfY8MlN-zghxuxIlCR7pul3lCbAqL5xmXUMA8oUg,6626
+uipath-2.1.128.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+uipath-2.1.128.dist-info/entry_points.txt,sha256=9C2_29U6Oq1ExFu7usihR-dnfIVNSKc-0EFbh0rskB4,43
+uipath-2.1.128.dist-info/licenses/LICENSE,sha256=-KBavWXepyDjimmzH5fVAsi-6jNVpIKFc2kZs0Ri4ng,1058
+uipath-2.1.128.dist-info/RECORD,,