Merge pull request #389 from UiPath/feat/bindings

infer bindings

Author: radu-mocanu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath"
-version = "2.0.59"
+version = "2.0.60"
 description = "Python SDK and CLI for UiPath Platform, enabling programmatic interaction with automation services, process management, and deployment tools."
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"

src/uipath/_cli/_utils/_constants.py

@@ -0,0 +1 @@
+BINDINGS_VERSION = "2.2"

src/uipath/_cli/_utils/_parse_ast.py

@@ -3,10 +3,16 @@
 import ast
 import os
 from dataclasses import dataclass, field
-from typing import Any, Dict, List, Optional
-
-from ..._services import AssetsService, BucketsService, ProcessesService
+from typing import Any, Dict, List, Optional, Tuple
+
+from ..._services import (
+    AssetsService,
+    BucketsService,
+    ContextGroundingService,
+    ProcessesService,
+)
 from ..._utils import get_inferred_bindings_names
+from ._constants import BINDINGS_VERSION
 
 
 @dataclass
@@ -33,6 +39,14 @@ def extract_string_arg(self, index: int = 0) -> Optional[str]:
     "processes": "process",
     "buckets": "bucket",
     "connections": "connection",
+    "context_grounding": "index",
+}
+
+supported_bindings_by_service = {
+    "assets": AssetsService,
+    "processes": ProcessesService,
+    "buckets": BucketsService,
+    "context_grounding": ContextGroundingService,
 }
 
 
@@ -67,48 +81,10 @@ class ServiceUsage:
     def get_component_info(self) -> List[Dict[str, str]]:
         """Extract component names and folders based on the service type."""
         result = []
-
-        if self.service_name == "assets":
-            for call in self.method_calls:
-                inferred_bindings = get_inferred_bindings_names(AssetsService)
-                if call.method_name in inferred_bindings:
-                    name = extract_parameter(
-                        call, inferred_bindings[call.method_name]["name"], 0
-                    )
-                    folder_path = extract_parameter(
-                        call, inferred_bindings[call.method_name]["folder_path"]
-                    )
-                    if name:
-                        result.append(
-                            {
-                                "name": name,
-                                "folder": folder_path or "",
-                                "method": call.method_name,
-                            }
-                        )
-
-        elif self.service_name == "processes":
-            for call in self.method_calls:
-                inferred_bindings = get_inferred_bindings_names(ProcessesService)
-                if call.method_name in inferred_bindings:
-                    name = extract_parameter(
-                        call, inferred_bindings[call.method_name]["name"], 0
-                    )
-                    folder_path = extract_parameter(
-                        call, inferred_bindings[call.method_name]["folder_path"]
-                    )
-                    if name:
-                        result.append(
-                            {
-                                "name": name,
-                                "folder": folder_path or "",
-                                "method": call.method_name,
-                            }
-                        )
-
-        elif self.service_name == "buckets":
+        has_support, service_cls = self._support_for_bindings_inference()
+        if has_support:
             for call in self.method_calls:
-                inferred_bindings = get_inferred_bindings_names(BucketsService)
+                inferred_bindings = get_inferred_bindings_names(service_cls)
                 if call.method_name in inferred_bindings:
                     name = extract_parameter(
                         call, inferred_bindings[call.method_name]["name"], 0
@@ -125,9 +101,9 @@ def get_component_info(self) -> List[Dict[str, str]]:
                             }
                         )
 
+        # custom logic for connections bindings
         elif self.service_name == "connections":
             for call in self.method_calls:
-                connection_id = None
                 if len(call.args) > 0:
                     connection_id = call.args[0]
                     if connection_id:
@@ -141,6 +117,12 @@ def get_component_info(self) -> List[Dict[str, str]]:
 
         return result
 
+    def _support_for_bindings_inference(self) -> Tuple[bool, Any]:
+        return (
+            self.service_name in supported_bindings_by_service,
+            supported_bindings_by_service.get(self.service_name, None),
+        )
+
 
 def extract_parameter(
     method_call: ServiceMethodCall,
@@ -462,11 +444,10 @@ def convert_to_bindings_format(sdk_usage_data):
                             "defaultValue": connection_id,
                             "isExpression": is_connection_id_expression,
                             "displayName": "Connection",
-                            "description": "The connection to be used",
                         }
                     },
                     "metadata": {
-                        "BindingsVersion": "2.1",
+                        "BindingsVersion": BINDINGS_VERSION,
                         "Connector": connector_name,
                         "UseConnectionService": "True",
                     },
@@ -500,7 +481,7 @@ def convert_to_bindings_format(sdk_usage_data):
                 },
                 "metadata": {
                     "ActivityName": method_name,
-                    "BindingsVersion": "2.1",
+                    "BindingsVersion": BINDINGS_VERSION,
                     "DisplayLabel": "FullName",
                 },
             }

src/uipath/_cli/cli_init.py

@@ -6,6 +6,7 @@
 from typing import Optional
 
 import click
+from dotenv import load_dotenv
 
 from ..telemetry import track
 from ._utils._console import ConsoleLogger
@@ -56,6 +57,9 @@ def get_user_script(directory: str, entrypoint: Optional[str] = None) -> Optiona
 @track
 def init(entrypoint: str) -> None:
     """Create uipath.json with input/output schemas and bindings."""
+    current_path = os.getcwd()
+    load_dotenv(os.path.join(current_path, ".env"), override=True)
+
     with console.spinner("Initializing UiPath project ..."):
         current_directory = os.getcwd()
         generate_env_file(current_directory)