uipath 2.1.86__py3-none-any.whl → 2.1.88__py3-none-any.whl

uipath/_cli/__init__.py

@@ -3,12 +3,11 @@ import sys
 
 import click
 
-from ._utils._common import load_environment_variables
+from ._utils._common import add_cwd_to_path, load_environment_variables
 from .cli_auth import auth as auth
 from .cli_deploy import deploy as deploy  # type: ignore
 from .cli_dev import dev as dev
 from .cli_eval import eval as eval  # type: ignore
-from .cli_init import generate_agents_md as generate_agents_md  # type: ignore
 from .cli_init import init as init  # type: ignore
 from .cli_invoke import invoke as invoke  # type: ignore
 from .cli_new import new as new  # type: ignore
@@ -46,6 +45,7 @@ def _get_safe_version() -> str:
 )
 def cli(lv: bool, v: bool) -> None:
     load_environment_variables()
+    add_cwd_to_path()
     if lv:
         try:
             version = importlib.metadata.version("uipath-langchain")

uipath/_cli/_utils/_common.py

@@ -9,6 +9,14 @@ from ..._utils.constants import DOTENV_FILE
 from ..spinner import Spinner
 
 
+def add_cwd_to_path():
+    import sys
+
+    cwd = os.getcwd()
+    if cwd not in sys.path:
+        sys.path.insert(0, cwd)
+
+
 def environment_options(function):
     function = click.option(
         "--alpha",

uipath/_cli/_utils/_input_args.py

@@ -2,6 +2,7 @@ import importlib.util
 import inspect
 import sys
 from dataclasses import fields, is_dataclass
+from enum import Enum
 from types import ModuleType
 from typing import (
     Any,
@@ -16,11 +17,11 @@ from typing import (
 
 from pydantic import BaseModel
 
-SchemaType = Literal["object", "integer", "double", "string", "boolean", "array"]
+SchemaType = Literal["object", "integer", "number", "string", "boolean", "array"]
 
 TYPE_MAP: Dict[str, SchemaType] = {
     "int": "integer",
-    "float": "double",
+    "float": "number",
     "str": "string",
     "bool": "boolean",
     "list": "array",
@@ -52,7 +53,25 @@ def get_type_schema(type_hint: Any) -> Dict[str, Any]:
         return {"type": "object"}
 
     if inspect.isclass(type_hint):
-        # Handle Pydantic models
+        if issubclass(type_hint, Enum):
+            enum_values = [member.value for member in type_hint]
+            if not enum_values:
+                return {"type": "string", "enum": []}
+
+            first_value = enum_values[0]
+            if isinstance(first_value, str):
+                enum_type = "string"
+            elif isinstance(first_value, int):
+                enum_type = "integer"
+            elif isinstance(first_value, float):
+                enum_type = "number"
+            elif isinstance(first_value, bool):
+                enum_type = "boolean"
+            else:
+                enum_type = "string"
+
+            return {"type": enum_type, "enum": enum_values}
+
         if issubclass(type_hint, BaseModel):
             properties = {}
             required = []

uipath/_cli/cli_init.py

@@ -1,6 +1,7 @@
 # type: ignore
 import importlib.resources
 import json
+import logging
 import os
 import shutil
 import uuid
@@ -18,6 +19,7 @@ from ._utils._parse_ast import generate_bindings_json
 from .middlewares import Middlewares
 
 console = ConsoleLogger()
+logger = logging.getLogger(__name__)
 
 CONFIG_PATH = "uipath.json"
 
@@ -55,35 +57,34 @@ def generate_env_file(target_directory):
         relative_path = os.path.relpath(env_path, target_directory)
         with open(env_path, "w"):
             pass
-        console.success(f" Created '{relative_path}' file.")
+        console.success(f"Created '{relative_path}' file.")
 
 
-def generate_agents_md(target_directory: str) -> None:
-    """Generate AGENTS.md file from the packaged resource.
+def generate_agent_specific_file_md(target_directory: str, file_name: str) -> None:
+    """Generate an agent-specific file from the packaged resource.
 
     Args:
-        target_directory: The directory where AGENTS.md should be created.
+        target_directory: The directory where the file should be created.
+        file_name: The name of the file should be created.
     """
-    target_path = os.path.join(target_directory, "AGENTS.md")
+    agent_dir = os.path.join(target_directory, ".agent")
+    os.makedirs(agent_dir, exist_ok=True)
+
+    target_path = os.path.join(agent_dir, file_name)
 
-    # Skip if file already exists
     if os.path.exists(target_path):
-        console.info("Skipping 'AGENTS.md' creation as it already exists.")
+        logger.debug(f"File '.agent/{target_path}' already exists.")
         return
 
     try:
-        # Get the resource path using importlib.resources
-        source_path = importlib.resources.files("uipath._resources").joinpath(
-            "AGENTS.md"
-        )
+        source_path = importlib.resources.files("uipath._resources").joinpath(file_name)
 
-        # Copy the file to the target directory
         with importlib.resources.as_file(source_path) as s_path:
             shutil.copy(s_path, target_path)
 
-        console.success(" Created 'AGENTS.md' file.")
+        console.success(f"Created '{f'.agent/{file_name}'}' file.")
     except Exception as e:
-        console.warning(f"Could not create AGENTS.md: {e}")
+        console.warning(f"Could not create {file_name}: {e}")
 
 
 def get_existing_settings(config_path: str) -> Optional[Dict[str, Any]]:
@@ -160,7 +161,11 @@ def init(entrypoint: str, infer_bindings: bool) -> None:
         current_directory = os.getcwd()
         generate_env_file(current_directory)
         create_telemetry_config_file(current_directory)
-        generate_agents_md(current_directory)
+        generate_agent_specific_file_md(current_directory, "AGENTS.md")
+        generate_agent_specific_file_md(current_directory, "CLI_REFERENCE.md")
+        generate_agent_specific_file_md(current_directory, "REQUIRED_STRUCTURE.md")
+        generate_agent_specific_file_md(current_directory, "SDK_REFERENCE.md")
+        generate_agent_specific_file_md(current_directory, "CLAUDE.md")
 
         result = Middlewares.next(
             "init",