uipath 2.0.10__py3-none-any.whl → 2.0.12__py3-none-any.whl

uipath/_cli/cli_init.py

@@ -3,6 +3,7 @@ import json
 import os
 import traceback
 import uuid
+from pathlib import Path
 from typing import Optional
 
 import click
@@ -75,7 +76,7 @@ def init(entrypoint: str) -> None:
     try:
         args = generate_args(script_path)
 
-        relative_path = os.path.relpath(script_path, current_directory)
+        relative_path = Path(script_path).relative_to(current_directory).as_posix()
 
         config_data = {
             "entryPoints": [

uipath/_cli/cli_pack.py

@@ -232,7 +232,7 @@ def pack_fn(projectName, description, entryPoints, version, authors, directory):
     os.makedirs(".uipath", exist_ok=True)
 
     # Define the allowlist of file extensions to include
-    file_extensions_allowlist = [".py", ".mermaid", ".json"]
+    file_extensions_allowlist = [".py", ".mermaid", ".json", ".yaml", ".yml"]
 
     with zipfile.ZipFile(
         f".uipath/{projectName}.{version}.nupkg", "w", zipfile.ZIP_DEFLATED

uipath/_services/actions_service.py

@@ -14,6 +14,7 @@ from .._utils.constants import (
     HEADER_TENANT_ID,
 )
 from ..models import Action, ActionSchema
+from ..tracing._traced import traced
 from ._base_service import BaseService
 
 
@@ -175,6 +176,7 @@ class ActionsService(FolderContext, BaseService):
         """
         super().__init__(config=config, execution_context=execution_context)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def create_async(
         self,
         title: str,
@@ -232,6 +234,7 @@ class ActionsService(FolderContext, BaseService):
             await self.request_async(spec.method, spec.endpoint, content=spec.content)
         return Action.model_validate(json_response)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def create(
         self,
         title: str,
@@ -287,6 +290,7 @@ class ActionsService(FolderContext, BaseService):
             self.request(spec.method, spec.endpoint, content=spec.content)
         return Action.model_validate(json_response)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def retrieve(
         self, action_key: str, app_folder_path: str = "", app_folder_key: str = ""
     ) -> Action:
@@ -311,6 +315,7 @@ class ActionsService(FolderContext, BaseService):
 
         return Action.model_validate(response.json())
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def retrieve_async(
         self, action_key: str, app_folder_path: str = "", app_folder_key: str = ""
     ) -> Action:

uipath/_services/assets_service.py

@@ -7,6 +7,7 @@ from .._execution_context import ExecutionContext
 from .._folder_context import FolderContext
 from .._utils import Endpoint, RequestSpec, header_folder, infer_bindings
 from ..models import UserAsset
+from ..tracing._traced import traced
 from ._base_service import BaseService
 
 
@@ -21,6 +22,7 @@ class AssetsService(FolderContext, BaseService):
         super().__init__(config=config, execution_context=execution_context)
 
     @infer_bindings()
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def retrieve(
         self,
         name: str,
@@ -56,6 +58,7 @@ class AssetsService(FolderContext, BaseService):
 
         return UserAsset.model_validate(response.json())
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def retrieve_async(
         self,
         name: str,
@@ -83,6 +86,7 @@ class AssetsService(FolderContext, BaseService):
         return UserAsset.model_validate(response.json())
 
     @infer_bindings()
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def retrieve_credential(
         self,
         name: str,
@@ -118,6 +122,7 @@ class AssetsService(FolderContext, BaseService):
         return user_asset.credential_password
 
     @infer_bindings()
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def retrieve_credential_async(
         self,
         name: str,
@@ -153,6 +158,7 @@ class AssetsService(FolderContext, BaseService):
 
         return user_asset.credential_password
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def update(
         self,
         robot_asset: UserAsset,
@@ -181,6 +187,7 @@ class AssetsService(FolderContext, BaseService):
             headers=spec.headers,
         )
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def update_async(
         self,
         robot_asset: UserAsset,

uipath/_services/buckets_service.py

@@ -6,6 +6,7 @@ from .._config import Config
 from .._execution_context import ExecutionContext
 from .._folder_context import FolderContext
 from .._utils import Endpoint, RequestSpec, infer_bindings
+from ..tracing._traced import traced
 from ._base_service import BaseService
 
 
@@ -19,6 +20,7 @@ class BucketsService(FolderContext, BaseService):
     def __init__(self, config: Config, execution_context: ExecutionContext) -> None:
         super().__init__(config=config, execution_context=execution_context)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def download(
         self,
         bucket_key: str,
@@ -57,6 +59,7 @@ class BucketsService(FolderContext, BaseService):
                 file_content = request("GET", read_uri, headers=headers).content
             file.write(file_content)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def upload(
         self,
         *,
@@ -108,6 +111,7 @@ class BucketsService(FolderContext, BaseService):
             else:
                 request("PUT", write_uri, headers=headers, files={"file": file})
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def upload_from_memory(
         self,
         *,
@@ -163,6 +167,7 @@ class BucketsService(FolderContext, BaseService):
             request("PUT", write_uri, headers=headers, content=content)
 
     @infer_bindings()
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def retrieve(self, name: str) -> Any:
         """Retrieve bucket information by its name.
 
@@ -187,6 +192,7 @@ class BucketsService(FolderContext, BaseService):
         return response.json()["value"][0]
 
     @infer_bindings()
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def retrieve_async(self, name: str) -> Any:
         """Asynchronously retrieve bucket information by its name.
 
@@ -210,6 +216,7 @@ class BucketsService(FolderContext, BaseService):
 
         return response.json()["value"][0]
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def retrieve_by_key(self, key: str) -> Any:
         """Retrieve bucket information by its key.
 
@@ -229,6 +236,7 @@ class BucketsService(FolderContext, BaseService):
 
         return response.json()
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def retrieve_by_key_async(self, key: str) -> Any:
         """Asynchronously retrieve bucket information by its key.
 

uipath/_services/connections_service.py

@@ -7,6 +7,7 @@ from .._execution_context import ExecutionContext
 from .._utils import Endpoint, RequestSpec
 from .._utils.constants import ENTRYPOINT
 from ..models import Connection, ConnectionToken
+from ..tracing._traced import traced
 from ._base_service import BaseService
 
 T_co = TypeVar("T_co", covariant=True)
@@ -72,6 +73,7 @@ class ConnectionsService(BaseService):
         except AttributeError as e:
             raise PluginNotFoundError(f"Plugin '{name}' is not installed") from e
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def retrieve(self, key: str) -> Connection:
         """Retrieve connection details by its key.
 
@@ -89,6 +91,7 @@ class ConnectionsService(BaseService):
         response = self.request(spec.method, url=spec.endpoint)
         return Connection.model_validate(response.json())
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def retrieve_async(self, key: str) -> Connection:
         """Asynchronously retrieve connection details by its key.
 
@@ -106,6 +109,7 @@ class ConnectionsService(BaseService):
         response = await self.request_async(spec.method, url=spec.endpoint)
         return Connection.model_validate(response.json())
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def retrieve_token(self, key: str) -> ConnectionToken:
         """Retrieve an authentication token for a connection.
 
@@ -124,6 +128,7 @@ class ConnectionsService(BaseService):
         response = self.request(spec.method, url=spec.endpoint, params=spec.params)
         return ConnectionToken.model_validate(response.json())
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def retrieve_token_async(self, key: str) -> ConnectionToken:
         """Asynchronously retrieve an authentication token for a connection.
 

uipath/_services/context_grounding_service.py

@@ -15,6 +15,7 @@ from .._utils.constants import (
 from ..models import IngestionInProgressException
 from ..models.context_grounding import ContextGroundingQueryResponse
 from ..models.context_grounding_index import ContextGroundingIndex
+from ..tracing._traced import traced
 from ._base_service import BaseService
 from .folder_service import FolderService
 
@@ -41,6 +42,7 @@ class ContextGroundingService(FolderContext, BaseService):
         self._folders_service = folders_service
         super().__init__(config=config, execution_context=execution_context)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def retrieve(self, name: str) -> Optional[ContextGroundingIndex]:
         """Retrieve context grounding index information by its name.
 
@@ -70,6 +72,7 @@ class ContextGroundingService(FolderContext, BaseService):
             None,
         )
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def retrieve_async(self, name: str) -> Optional[ContextGroundingIndex]:
         """Retrieve asynchronously context grounding index information by its name.
 
@@ -102,6 +105,7 @@ class ContextGroundingService(FolderContext, BaseService):
             None,
         )
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def retrieve_by_id(self, id: str) -> Any:
         """Retrieve context grounding index information by its ID.
 
@@ -122,6 +126,7 @@ class ContextGroundingService(FolderContext, BaseService):
             params=spec.params,
         ).json()
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def retrieve_by_id_async(self, id: str) -> Any:
         """Retrieve asynchronously context grounding index information by its ID.
 
@@ -145,6 +150,7 @@ class ContextGroundingService(FolderContext, BaseService):
 
         return response.json()
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def search(
         self,
         name: str,
@@ -183,6 +189,7 @@ class ContextGroundingService(FolderContext, BaseService):
             response.json()
         )
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def search_async(
         self,
         name: str,
@@ -220,6 +227,7 @@ class ContextGroundingService(FolderContext, BaseService):
             response.json()
         )
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def get_or_create_index(
         self,
         name: str,
@@ -248,6 +256,7 @@ class ContextGroundingService(FolderContext, BaseService):
         ).json()
         return ContextGroundingIndex.model_validate(response)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def get_or_create_index_async(
         self,
         name: str,
@@ -278,6 +287,7 @@ class ContextGroundingService(FolderContext, BaseService):
         ).json()
         return ContextGroundingIndex.model_validate(response)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def ingest_data(self, index: ContextGroundingIndex) -> None:
         if not index.id:
             return
@@ -288,6 +298,7 @@ class ContextGroundingService(FolderContext, BaseService):
             headers=spec.headers,
         )
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def ingest_data_async(self, index: ContextGroundingIndex) -> None:
         if not index.id:
             return
@@ -298,6 +309,7 @@ class ContextGroundingService(FolderContext, BaseService):
             headers=spec.headers,
         )
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def delete_index(self, index: ContextGroundingIndex) -> None:
         if not index.id:
             return
@@ -308,6 +320,7 @@ class ContextGroundingService(FolderContext, BaseService):
             headers=spec.headers,
         )
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def delete_index_async(self, index: ContextGroundingIndex) -> None:
         if not index.id:
             return

uipath/_services/folder_service.py

@@ -1,5 +1,7 @@
 from typing import Optional
 
+from uipath.tracing._traced import traced
+
 from .._config import Config
 from .._execution_context import ExecutionContext
 from .._utils import Endpoint, RequestSpec
@@ -31,6 +33,7 @@ class FolderService(BaseService):
     def __init__(self, config: Config, execution_context: ExecutionContext) -> None:
         super().__init__(config=config, execution_context=execution_context)
 
+    @traced(run_type="uipath")
     def retrieve_key_by_folder_path(self, folder_path: str) -> Optional[str]:
         spec = _retrieve_spec(folder_path)
         response = self.request(

uipath/_services/jobs_service.py

@@ -6,6 +6,7 @@ from .._execution_context import ExecutionContext
 from .._folder_context import FolderContext
 from .._utils import Endpoint, RequestSpec, header_folder
 from ..models.job import Job
+from ..tracing._traced import traced
 from ._base_service import BaseService
 
 
@@ -26,6 +27,7 @@ class JobsService(FolderContext, BaseService):
     @overload
     def resume(self, *, job_id: str, payload: Any) -> None: ...
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def resume(
         self,
         *,

uipath/_services/llm_gateway_service.py

@@ -12,6 +12,7 @@ from ..models.llm_gateway import (
     ToolDefinition,
     UsageInfo,
 )
+from ..tracing._traced import traced
 from ._base_service import BaseService
 
 # Common constants
@@ -53,6 +54,7 @@ class UiPathOpenAIService(BaseService):
     def __init__(self, config: Config, execution_context: ExecutionContext) -> None:
         super().__init__(config=config, execution_context=execution_context)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def embeddings_usage(
         self, input: str, embedding_model: str = EmbeddingModels.text_embedding_ada_002
     ):
@@ -79,6 +81,7 @@ class UiPathOpenAIService(BaseService):
 
         return UsageInfo.model_validate(response.json())
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def embeddings(
         self, input: str, embedding_model: str = EmbeddingModels.text_embedding_ada_002
     ):
@@ -104,6 +107,7 @@ class UiPathOpenAIService(BaseService):
 
         return TextEmbedding.model_validate(response.json())
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def chat_completions(
         self,
         messages: List[Dict[str, str]],
@@ -153,6 +157,7 @@ class UiPathOpenAIService(BaseService):
 
         return ChatCompletion.model_validate(response.json())
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def chat_completions_usage(
         self,
         messages: List[Dict[str, str]],
@@ -211,6 +216,7 @@ class UiPathLlmChatService(BaseService):
     def __init__(self, config: Config, execution_context: ExecutionContext) -> None:
         super().__init__(config=config, execution_context=execution_context)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def chat_completions(
         self,
         messages: List[Dict[str, str]],

uipath/_services/processes_service.py

@@ -8,6 +8,7 @@ from .._folder_context import FolderContext
 from .._utils import Endpoint, RequestSpec, header_folder, infer_bindings
 from .._utils.constants import ENV_JOB_ID, HEADER_JOB_KEY
 from ..models.job import Job
+from ..tracing._traced import traced
 from ._base_service import BaseService
 
 
@@ -22,6 +23,7 @@ class ProcessesService(FolderContext, BaseService):
     def __init__(self, config: Config, execution_context: ExecutionContext) -> None:
         super().__init__(config=config, execution_context=execution_context)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def invoke(
         self,
         name: str,
@@ -79,6 +81,7 @@ class ProcessesService(FolderContext, BaseService):
         return Job.model_validate(response.json()["value"][0])
 
     @infer_bindings()
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def invoke_async(
         self,
         name: str,

uipath/_services/queues_service.py

@@ -7,6 +7,7 @@ from .._execution_context import ExecutionContext
 from .._folder_context import FolderContext
 from .._utils import Endpoint, RequestSpec
 from ..models import CommitType, QueueItem, TransactionItem, TransactionItemResult
+from ..tracing._traced import traced
 from ._base_service import BaseService
 
 
@@ -20,6 +21,7 @@ class QueuesService(FolderContext, BaseService):
     def __init__(self, config: Config, execution_context: ExecutionContext) -> None:
         super().__init__(config=config, execution_context=execution_context)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def list_items(self) -> Response:
         """Retrieves a list of queue items from the Orchestrator.
 
@@ -29,6 +31,7 @@ class QueuesService(FolderContext, BaseService):
         spec = self._list_items_spec()
         return self.request(spec.method, url=spec.endpoint)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def list_items_async(self) -> Response:
         """Asynchronously retrieves a list of queue items from the Orchestrator.
 
@@ -38,6 +41,7 @@ class QueuesService(FolderContext, BaseService):
         spec = self._list_items_spec()
         return await self.request_async(spec.method, url=spec.endpoint)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def create_item(self, item: Union[Dict[str, Any], QueueItem]) -> Response:
         """Creates a new queue item in the Orchestrator.
 
@@ -52,6 +56,7 @@ class QueuesService(FolderContext, BaseService):
         spec = self._create_item_spec(item)
         return self.request(spec.method, url=spec.endpoint, json=spec.json)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def create_item_async(
         self, item: Union[Dict[str, Any], QueueItem]
     ) -> Response:
@@ -68,6 +73,7 @@ class QueuesService(FolderContext, BaseService):
         spec = self._create_item_spec(item)
         return await self.request_async(spec.method, url=spec.endpoint, json=spec.json)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def create_items(
         self,
         items: List[Union[Dict[str, Any], QueueItem]],
@@ -87,6 +93,7 @@ class QueuesService(FolderContext, BaseService):
         spec = self._create_items_spec(items, queue_name, commit_type)
         return self.request(spec.method, url=spec.endpoint, json=spec.json)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def create_items_async(
         self,
         items: List[Union[Dict[str, Any], QueueItem]],
@@ -106,6 +113,7 @@ class QueuesService(FolderContext, BaseService):
         spec = self._create_items_spec(items, queue_name, commit_type)
         return await self.request_async(spec.method, url=spec.endpoint, json=spec.json)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def create_transaction_item(
         self, item: Union[Dict[str, Any], TransactionItem], no_robot: bool = False
     ) -> Response:
@@ -121,6 +129,7 @@ class QueuesService(FolderContext, BaseService):
         spec = self._create_transaction_item_spec(item, no_robot)
         return self.request(spec.method, url=spec.endpoint, json=spec.json)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def create_transaction_item_async(
         self, item: Union[Dict[str, Any], TransactionItem], no_robot: bool = False
     ) -> Response:
@@ -136,6 +145,7 @@ class QueuesService(FolderContext, BaseService):
         spec = self._create_transaction_item_spec(item, no_robot)
         return await self.request_async(spec.method, url=spec.endpoint, json=spec.json)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def update_progress_of_transaction_item(
         self, transaction_key: str, progress: str
     ) -> Response:
@@ -153,6 +163,7 @@ class QueuesService(FolderContext, BaseService):
         spec = self._update_progress_of_transaction_item_spec(transaction_key, progress)
         return self.request(spec.method, url=spec.endpoint, json=spec.json)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def update_progress_of_transaction_item_async(
         self, transaction_key: str, progress: str
     ) -> Response:
@@ -170,6 +181,7 @@ class QueuesService(FolderContext, BaseService):
         spec = self._update_progress_of_transaction_item_spec(transaction_key, progress)
         return await self.request_async(spec.method, url=spec.endpoint, json=spec.json)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     def complete_transaction_item(
         self, transaction_key: str, result: Union[Dict[str, Any], TransactionItemResult]
     ) -> Response:
@@ -187,6 +199,7 @@ class QueuesService(FolderContext, BaseService):
         spec = self._complete_transaction_item_spec(transaction_key, result)
         return self.request(spec.method, url=spec.endpoint, json=spec.json)
 
+    @traced(run_type="uipath", hide_input=True, hide_output=True)
     async def complete_transaction_item_async(
         self, transaction_key: str, result: Union[Dict[str, Any], TransactionItemResult]
     ) -> Response:

uipath/tracing/_traced.py

@@ -2,7 +2,7 @@ import inspect
 import json
 import logging
 from functools import wraps
-from typing import Any
+from typing import Any, Callable, Optional
 
 from opentelemetry import trace
 from opentelemetry.sdk.trace import TracerProvider
@@ -23,6 +23,16 @@ def wait_for_tracers():
     trace.get_tracer_provider().shutdown()  # type: ignore
 
 
+def _default_input_processor(inputs):
+    """Default input processor that doesn't log any actual input data."""
+    return {"redacted": "Input data not logged for privacy/security"}
+
+
+def _default_output_processor(outputs):
+    """Default output processor that doesn't log any actual output data."""
+    return {"redacted": "Output data not logged for privacy/security"}
+
+
 class TracedDecoratorRegistry:
     """Registry for tracing decorators."""
 
@@ -42,23 +52,42 @@ class TracedDecoratorRegistry:
         return cls._decorators.get(cls._active_decorator)
 
 
-def _opentelemetry_traced():
+def _opentelemetry_traced(
+    run_type: Optional[str] = None,
+    span_type: Optional[str] = None,
+    input_processor: Optional[Callable[..., Any]] = None,
+    output_processor: Optional[Callable[..., Any]] = None,
+):
     def decorator(func):
         @wraps(func)
         def sync_wrapper(*args, **kwargs):
             with tracer.start_as_current_span(func.__name__) as span:
-                span.set_attribute("span_type", "function_call_sync")
+                default_span_type = "function_call_sync"
                 span.set_attribute(
-                    "inputs",
-                    _SpanUtils.format_args_for_trace_json(
-                        inspect.signature(func), *args, **kwargs
-                    ),
+                    "span_type",
+                    span_type if span_type is not None else default_span_type,
                 )
+                if run_type is not None:
+                    span.set_attribute("run_type", run_type)
+
+                # Format arguments for tracing
+                inputs = _SpanUtils.format_args_for_trace_json(
+                    inspect.signature(func), *args, **kwargs
+                )
+                # Apply input processor if provided
+                if input_processor is not None:
+                    processed_inputs = input_processor(json.loads(inputs))
+                    inputs = json.dumps(processed_inputs, default=str)
+
+                span.set_attribute("inputs", inputs)
+
                 try:
                     result = func(*args, **kwargs)
-                    span.set_attribute(
-                        "output", json.dumps(result, default=str)
-                    )  # Record output
+                    # Process output if processor is provided
+                    output = result
+                    if output_processor is not None:
+                        output = output_processor(result)
+                    span.set_attribute("output", json.dumps(output, default=str))
                     return result
                 except Exception as e:
                     span.record_exception(e)
@@ -70,18 +99,32 @@ def _opentelemetry_traced():
         @wraps(func)
         async def async_wrapper(*args, **kwargs):
             with tracer.start_as_current_span(func.__name__) as span:
-                span.set_attribute("span_type", "function_call_async")
+                default_span_type = "function_call_async"
                 span.set_attribute(
-                    "inputs",
-                    _SpanUtils.format_args_for_trace_json(
-                        inspect.signature(func), *args, **kwargs
-                    ),
+                    "span_type",
+                    span_type if span_type is not None else default_span_type,
+                )
+                if run_type is not None:
+                    span.set_attribute("run_type", run_type)
+
+                # Format arguments for tracing
+                inputs = _SpanUtils.format_args_for_trace_json(
+                    inspect.signature(func), *args, **kwargs
                 )
+                # Apply input processor if provided
+                if input_processor is not None:
+                    processed_inputs = input_processor(json.loads(inputs))
+                    inputs = json.dumps(processed_inputs, default=str)
+
+                span.set_attribute("inputs", inputs)
+
                 try:
                     result = await func(*args, **kwargs)
-                    span.set_attribute(
-                        "output", json.dumps(result, default=str)
-                    )  # Record output
+                    # Process output if processor is provided
+                    output = result
+                    if output_processor is not None:
+                        output = output_processor(result)
+                    span.set_attribute("output", json.dumps(output, default=str))
                     return result
                 except Exception as e:
                     span.record_exception(e)
@@ -93,22 +136,39 @@ def _opentelemetry_traced():
         @wraps(func)
         def generator_wrapper(*args, **kwargs):
             with tracer.start_as_current_span(func.__name__) as span:
-                span.set_attribute("span_type", "function_call_generator_sync")
+                default_span_type = "function_call_generator_sync"
                 span.set_attribute(
-                    "inputs",
-                    _SpanUtils.format_args_for_trace_json(
-                        inspect.signature(func), *args, **kwargs
-                    ),
+                    "span_type",
+                    span_type if span_type is not None else default_span_type,
                 )
+                if run_type is not None:
+                    span.set_attribute("run_type", run_type)
+
+                # Format arguments for tracing
+                inputs = _SpanUtils.format_args_for_trace_json(
+                    inspect.signature(func), *args, **kwargs
+                )
+                # Apply input processor if provided
+                if input_processor is not None:
+                    processed_inputs = input_processor(json.loads(inputs))
+                    inputs = json.dumps(processed_inputs, default=str)
+
+                span.set_attribute("inputs", inputs)
+
                 outputs = []
                 try:
                     for item in func(*args, **kwargs):
                         outputs.append(item)
                         span.add_event(f"Yielded: {item}")  # Add event for each yield
                         yield item
+
+                    # Process output if processor is provided
+                    output_to_record = outputs
+                    if output_processor is not None:
+                        output_to_record = output_processor(outputs)
                     span.set_attribute(
-                        "output", json.dumps(outputs, default=str)
-                    )  # Record aggregated outputs
+                        "output", json.dumps(output_to_record, default=str)
+                    )
                 except Exception as e:
                     span.record_exception(e)
                     span.set_status(
@@ -119,22 +179,39 @@ def _opentelemetry_traced():
         @wraps(func)
         async def async_generator_wrapper(*args, **kwargs):
             with tracer.start_as_current_span(func.__name__) as span:
-                span.set_attribute("span_type", "function_call_generator_async")
+                default_span_type = "function_call_generator_async"
                 span.set_attribute(
-                    "inputs",
-                    _SpanUtils.format_args_for_trace_json(
-                        inspect.signature(func), *args, **kwargs
-                    ),
+                    "span_type",
+                    span_type if span_type is not None else default_span_type,
+                )
+                if run_type is not None:
+                    span.set_attribute("run_type", run_type)
+
+                # Format arguments for tracing
+                inputs = _SpanUtils.format_args_for_trace_json(
+                    inspect.signature(func), *args, **kwargs
                 )
+                # Apply input processor if provided
+                if input_processor is not None:
+                    processed_inputs = input_processor(json.loads(inputs))
+                    inputs = json.dumps(processed_inputs, default=str)
+
+                span.set_attribute("inputs", inputs)
+
                 outputs = []
                 try:
                     async for item in func(*args, **kwargs):
                         outputs.append(item)
                         span.add_event(f"Yielded: {item}")  # Add event for each yield
                         yield item
+
+                    # Process output if processor is provided
+                    output_to_record = outputs
+                    if output_processor is not None:
+                        output_to_record = output_processor(outputs)
                     span.set_attribute(
-                        "output", json.dumps(outputs, default=str)
-                    )  # Record aggregated outputs
+                        "output", json.dumps(output_to_record, default=str)
+                    )
                 except Exception as e:
                     span.record_exception(e)
                     span.set_status(
@@ -154,12 +231,39 @@ def _opentelemetry_traced():
     return decorator
 
 
-def traced():
-    """Decorator that will trace function invocations."""
+def traced(
+    run_type: Optional[str] = None,
+    span_type: Optional[str] = None,
+    input_processor: Optional[Callable[..., Any]] = None,
+    output_processor: Optional[Callable[..., Any]] = None,
+    hide_input: bool = False,
+    hide_output: bool = False,
+):
+    """Decorator that will trace function invocations.
+
+    Args:
+        run_type: Optional string to categorize the run type
+        span_type: Optional string to categorize the span type
+        input_processor: Optional function to process function inputs before recording
+            Should accept a dictionary of inputs and return a processed dictionary
+        output_processor: Optional function to process function outputs before recording
+            Should accept the function output and return a processed value
+        hide_input: If True, don't log any input data
+        hide_output: If True, don't log any output data
+    """
+    # Apply default processors selectively based on hide flags
+    if hide_input:
+        input_processor = _default_input_processor
+
+    if hide_output:
+        output_processor = _default_output_processor
+
     decorator_factory = TracedDecoratorRegistry.get_decorator()
 
     if decorator_factory:
-        return decorator_factory()
+        return decorator_factory(run_type, span_type, input_processor, output_processor)
     else:
         # Fallback to original implementation if no active decorator
-        return _opentelemetry_traced()
+        return _opentelemetry_traced(
+            run_type, span_type, input_processor, output_processor
+        )

{uipath-2.0.10.dist-info → uipath-2.0.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath
-Version: 2.0.10
+Version: 2.0.12
 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
@@ -16,6 +16,7 @@ Requires-Python: >=3.10
 Requires-Dist: click>=8.1.8
 Requires-Dist: httpx>=0.28.1
 Requires-Dist: opentelemetry-sdk>=1.32.1
+Requires-Dist: pathlib>=1.0.1
 Requires-Dist: pydantic>=2.11.1
 Requires-Dist: pytest-asyncio>=0.25.3
 Requires-Dist: python-dotenv>=1.0.1

{uipath-2.0.10.dist-info → uipath-2.0.12.dist-info}/RECORD

@@ -8,9 +8,9 @@ uipath/_cli/README.md,sha256=GLtCfbeIKZKNnGTCsfSVqRQ27V1btT1i2bSAyW_xZl4,474
 uipath/_cli/__init__.py,sha256=CAyMecQhBBD6s4MzYQNa01bz0TIXXCqkQ5DR2kCb00w,1574
 uipath/_cli/cli_auth.py,sha256=ANaYUc2q1t0hDbGBRT3ags6K6Lef_3tyC8Mmc611jow,3141
 uipath/_cli/cli_deploy.py,sha256=h8qwJkXnW6JURsg4YcocJInGA4dwkl4CZkpT1Cn9A3c,268
-uipath/_cli/cli_init.py,sha256=dPzwbfB0hJL5jRLxJJJ67DHI_i3b34_oepDHwqQN7Yw,3701
+uipath/_cli/cli_init.py,sha256=idoqlGhhzXZKmLAg-3JgZ2fYMrK7qFXYV0EhnNaI3bg,3738
 uipath/_cli/cli_new.py,sha256=SP7eWOa5valmCpc8UsOCIezL25euhglB3yJkx-N92W8,1903
-uipath/_cli/cli_pack.py,sha256=pszhSGxb6DZtYeHymS6Pys7kewx1hFvKKYilfw7aUzE,12859
+uipath/_cli/cli_pack.py,sha256=x7Je61NYzR7CR8TUoT-xeZMrXXFAsXO9PGFDUVfBU8s,12876
 uipath/_cli/cli_publish.py,sha256=_b9rehjsbxwkpH5_DtgFUaWWJqcZTg5nate-M5BnE_c,3586
 uipath/_cli/cli_run.py,sha256=dV0a-sx78T0HJHArfZP2M9YhT8d8aOuf-9OdkBqj3fE,4577
 uipath/_cli/middlewares.py,sha256=IiJgjsqrJVKSXx4RcIKHWoH-SqWqpHPbhzkQEybmAos,3937
@@ -36,18 +36,18 @@ uipath/_cli/_utils/_input_args.py,sha256=pyQhEcQXHdFHYTVNzvfWp439aii5StojoptnmCv
 uipath/_cli/_utils/_parse_ast.py,sha256=3XVjnhJNnSfjXlitct91VOtqSl0l-sqDpoWww28mMc0,20663
 uipath/_services/__init__.py,sha256=VPbwLDsvN26nWZgvR-8_-tc3i0rk5doqjTJbSrK0nN4,818
 uipath/_services/_base_service.py,sha256=3YClCoZBkVQGNJZGy-4NTk-HGsGA61XtwVQFYv9mwWk,7955
-uipath/_services/actions_service.py,sha256=muTw0jJ71pyLyrnzMZ-QIKjcmHEsLSzpd_UCLTAIrFM,14685
+uipath/_services/actions_service.py,sha256=Dl0ppz4lLiTMAvaS44Qp20FBjRIZbL-58FlifYIoXI8,14986
 uipath/_services/api_client.py,sha256=1hYLc_90dQzCGnqqirEHpPqvL3Gkv2sSKoeOV_iTmlk,2903
-uipath/_services/assets_service.py,sha256=OdnhlHfEmwA0SQqkXO4XHl3bV26QLgIxj6sFycseZ0g,8647
-uipath/_services/buckets_service.py,sha256=JeSFoEOBeGi-i_aevaMAyu5gpauq1KC_JkANRTmyxEs,8655
-uipath/_services/connections_service.py,sha256=J_eKhMuBmiDKIhr5bA4_9g8xND185lh_MuBh2hmWMs4,6853
+uipath/_services/assets_service.py,sha256=UUWzQiYruNAWk3P8qPrccDDWRUD_ycfqf3eRM-E9N44,9080
+uipath/_services/buckets_service.py,sha256=h1Rx9H4XV2cxIZ1xIcYjNYFa1YZEHgIhQpQ10jZIinU,9154
+uipath/_services/connections_service.py,sha256=5impJ5O0Wj2n0-RYnwI1TfOPAp5JStFwTjn_yCGySE8,7154
 uipath/_services/connections_service.pyi,sha256=6OOnh0aCfxhETL8n_JZ6Xoe2BE3ST_7Vz-FgLZc53lM,2465
-uipath/_services/context_grounding_service.py,sha256=eYxmkfBkLIxbMypBEzvyQ6_Jh7PF3OCrwzIe0PQXIs8,13372
-uipath/_services/folder_service.py,sha256=1BRTnfA-iMzAGZTJqTUtOXzNZLzbGCoWpH3g45YBEuQ,1556
-uipath/_services/jobs_service.py,sha256=iNs4hvr8Qyy_zbJjEkPK-0ygCM1l0OqpD1D_0PRgmf4,8126
-uipath/_services/llm_gateway_service.py,sha256=1YCgW2c0Hh3uXctUKPLcytAMJe4Qk_L8LLCd8Plgvvs,11695
-uipath/_services/processes_service.py,sha256=3A7EQrwY2fzKrnLk-MBowKnba4rmUAXIhtw1LlnU1yw,5510
-uipath/_services/queues_service.py,sha256=R9QMTIGeOa76uZJznbNXZ0PPAztah8JnSMjKsZkdNwU,12225
+uipath/_services/context_grounding_service.py,sha256=reEMo9vMJLjo-gAsJH7wL7PMD5O1lK_0PYoLnKRRnug,14201
+uipath/_services/folder_service.py,sha256=CBrEPjHFg6zAaNBbpcdrqaoPZWbgpTzWtAVz04J39mY,1630
+uipath/_services/jobs_service.py,sha256=IpsqXo0QAjUK8AdA_pSMu2g9ncPccNxHh-yMF-Gy58I,8229
+uipath/_services/llm_gateway_service.py,sha256=I1WcpIVWAMkkkY7uT_EU8Mqc_53LbRlo40GvSxNfiXg,12062
+uipath/_services/processes_service.py,sha256=nzNlxJnTiM5NVtFxFX2WuWOxE0YrVb0WG6QE7f4Csg0,5679
+uipath/_services/queues_service.py,sha256=yIDLxFEmVTt39ISSL2AaBJYq2hQH4CXxUP4Zc2MVTRI,13054
 uipath/_utils/__init__.py,sha256=y8asYKjU5j3v72TbgShEpUafAAJXJ6bngqdzXIl-Lhk,481
 uipath/_utils/_endpoint.py,sha256=yYHwqbQuJIevpaTkdfYJS9CrtlFeEyfb5JQK5osTCog,2489
 uipath/_utils/_infer_bindings.py,sha256=ysAftopcCBj4ojYyeVwbSl20qYhCDmqyldCinj6sICM,905
@@ -71,10 +71,10 @@ uipath/models/processes.py,sha256=Atvfrt6X4TYST3iA62jpS_Uxc3hg6uah11p-RaKZ6dk,20
 uipath/models/queues.py,sha256=N_s0GKucbyjh0RnO8SxPk6wlRgvq8KIIYsfaoIY46tM,6446
 uipath/tracing/__init__.py,sha256=mQEKHs47ufpy3MxphTJRh6AQsH90y6pWW80Nqbs0Q9c,157
 uipath/tracing/_otel_exporters.py,sha256=Fo10ofCWGd6PN5ndEsHkw6FZgQT5zOlob2XGnkiWvZE,2056
-uipath/tracing/_traced.py,sha256=HareJwZXD_z1EtaMrSDdK14AVH_sBmelIkG3hag6WaY,6271
+uipath/tracing/_traced.py,sha256=9EUy7-OlBOApcKx4L8pwXCUJQQSl0Qh5IPvLGd27cS0,10913
 uipath/tracing/_utils.py,sha256=5SwsTGpHkIouXBndw-u8eCLnN4p7LM8DsTCCuf2jJgs,10165
-uipath-2.0.10.dist-info/METADATA,sha256=vUcBFQGPWH6AawCi2VEudyEzy2bsucbWu3ZuIJllwqM,6048
-uipath-2.0.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-uipath-2.0.10.dist-info/entry_points.txt,sha256=9C2_29U6Oq1ExFu7usihR-dnfIVNSKc-0EFbh0rskB4,43
-uipath-2.0.10.dist-info/licenses/LICENSE,sha256=-KBavWXepyDjimmzH5fVAsi-6jNVpIKFc2kZs0Ri4ng,1058
-uipath-2.0.10.dist-info/RECORD,,
+uipath-2.0.12.dist-info/METADATA,sha256=hzh4ODa7bKR-z6ee1CqMLOkzWGIXCAXm6TkEgI9r0JE,6078
+uipath-2.0.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+uipath-2.0.12.dist-info/entry_points.txt,sha256=9C2_29U6Oq1ExFu7usihR-dnfIVNSKc-0EFbh0rskB4,43
+uipath-2.0.12.dist-info/licenses/LICENSE,sha256=-KBavWXepyDjimmzH5fVAsi-6jNVpIKFc2kZs0Ri4ng,1058
+uipath-2.0.12.dist-info/RECORD,,