uipath 2.1.56__py3-none-any.whl → 2.1.57__py3-none-any.whl

uipath/agent/_utils.py

@@ -0,0 +1,55 @@
+import logging
+from pathlib import PurePath
+
+from httpx import Response
+from pydantic import TypeAdapter
+
+from uipath._cli._utils._studio_project import (
+    ProjectFile,
+    ProjectFolder,
+    StudioClient,
+    resolve_path,
+)
+from uipath.agent.models.agent import AgentDefinition
+
+logger = logging.getLogger(__name__)
+
+
+async def get_file(
+    folder: ProjectFolder, path: PurePath, studio_client: StudioClient
+) -> Response:
+    resolved = resolve_path(folder, path)
+    assert isinstance(resolved, ProjectFile), "Path file not found."
+    return await studio_client.download_file_async(resolved.id)
+
+
+async def load_agent_definition(project_id: str):
+    studio_client = StudioClient(project_id=project_id)
+    project_structure = await studio_client.get_project_structure_async()
+
+    agent = (
+        await get_file(project_structure, PurePath("agent.json"), studio_client)
+    ).json()
+
+    resolved_path = resolve_path(project_structure, PurePath("resources"))
+    if isinstance(resolved_path, ProjectFolder):
+        resource_folders = resolved_path.folders
+    else:
+        logger.warning(
+            "Unable to read resource information from project. Defaulting to empty resources."
+        )
+        resource_folders = []
+
+    resources = []
+    for resource in resource_folders:
+        resources.append(
+            (await get_file(resource, PurePath("resource.json"), studio_client)).json()
+        )
+
+    agent_definition = {
+        "id": project_id,
+        "name": project_structure.name,
+        "resources": resources,
+        **agent,
+    }
+    return TypeAdapter(AgentDefinition).validate_python(agent_definition)

uipath/agent/models/agent.py

@@ -0,0 +1,344 @@
+"""Agent Models."""
+
+from enum import Enum
+from typing import Annotated, Any, Dict, List, Literal, Optional, Union
+
+from pydantic import BaseModel, ConfigDict, Discriminator, Field, Tag
+
+from uipath.models import Connection
+
+
+class AgentResourceType(str, Enum):
+    """Enum for resource types."""
+
+    TOOL = "tool"
+    CONTEXT = "context"
+    ESCALATION = "escalation"
+
+
+class BaseAgentResourceConfig(BaseModel):
+    """Base resource model with common properties."""
+
+    name: str
+    description: str
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentUnknownResourceConfig(BaseAgentResourceConfig):
+    """Fallback for unknown or future resource types."""
+
+    resource_type: str = Field(alias="$resourceType")
+
+    model_config = ConfigDict(extra="allow")
+
+
+class BaseAgentToolResourceConfig(BaseAgentResourceConfig):
+    """Tool resource with tool-specific properties."""
+
+    resource_type: Literal[AgentResourceType.TOOL] = Field(alias="$resourceType")
+    input_schema: Dict[str, Any] = Field(
+        ..., alias="inputSchema", description="Input schema for the tool"
+    )
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentToolType(str, Enum):
+    """Agent tool type."""
+
+    AGENT = "agent"
+    INTEGRATION = "integration"
+
+
+class AgentToolSettings(BaseModel):
+    """Settings for tool configuration."""
+
+    max_attempts: Optional[int] = Field(None, alias="maxAttempts")
+    retry_delay: Optional[int] = Field(None, alias="retryDelay")
+    timeout: Optional[int] = Field(None)
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentProcessToolProperties(BaseModel):
+    """Properties specific to tool configuration."""
+
+    folder_path: Optional[str] = Field(None, alias="folderPath")
+    process_name: Optional[str] = Field(None, alias="processName")
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentProcessToolResourceConfig(BaseAgentToolResourceConfig):
+    """Tool resource with tool-specific properties."""
+
+    type: Literal[AgentToolType.AGENT] = AgentToolType.AGENT
+    output_schema: Dict[str, Any] = Field(
+        ..., alias="outputSchema", description="Output schema for the tool"
+    )
+    properties: AgentProcessToolProperties = Field(
+        ..., description="Tool-specific properties"
+    )
+    settings: AgentToolSettings = Field(
+        default_factory=AgentToolSettings, description="Tool settings"
+    )
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentIntegrationToolProperties(BaseModel):
+    """Properties specific to tool configuration."""
+
+    tool_path: str = Field(..., alias="toolPath")
+    object_name: str = Field(..., alias="objectName")
+    tool_display_name: str = Field(..., alias="toolDisplayName")
+    tool_description: str = Field(..., alias="toolDescription")
+    method: str = Field(..., alias="method")
+    connection: Connection = Field(..., alias="connection")
+    body_structure: dict[str, Any] = Field(..., alias="bodyStructure")
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentIntegrationToolResourceConfig(BaseAgentToolResourceConfig):
+    """Tool resource with tool-specific properties."""
+
+    type: Literal[AgentToolType.INTEGRATION] = AgentToolType.INTEGRATION
+    properties: AgentIntegrationToolProperties
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentUnknownToolResourceConfig(BaseAgentToolResourceConfig):
+    """Fallback for unknown or future tool types."""
+
+    resource_type: Literal[AgentResourceType.TOOL] = AgentResourceType.TOOL
+    type: str = Field(alias="$resourceType")
+
+    model_config = ConfigDict(extra="allow")
+
+
+class AgentContextSettings(BaseModel):
+    """Settings for context configuration."""
+
+    result_count: int = Field(alias="resultCount")
+    retrieval_mode: Literal["Semantic", "Structured"] = Field(alias="retrievalMode")
+    threshold: float = Field(default=0)
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentContextResourceConfig(BaseAgentResourceConfig):
+    """Context resource with context-specific properties."""
+
+    resource_type: Literal[AgentResourceType.CONTEXT] = Field(alias="$resourceType")
+    folder_path: str = Field(alias="folderPath")
+    index_name: str = Field(alias="indexName")
+    settings: AgentContextSettings = Field(..., description="Context settings")
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentEscalationChannelProperties(BaseModel):
+    """Agent escalation channel properties."""
+
+    app_name: str = Field(..., alias="appName")
+    app_version: int = Field(..., alias="appVersion")
+    folder_name: Optional[str] = Field(..., alias="folderName")
+    resource_key: str = Field(..., alias="resourceKey")
+    is_actionable_message_enabled: Optional[bool] = Field(
+        None, alias="isActionableMessageEnabled"
+    )
+    actionable_message_meta_data: Optional[Any] = Field(
+        None, alias="actionableMessageMetaData"
+    )
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentEscalationChannel(BaseModel):
+    """Agent escalation channel."""
+
+    id: str = Field(..., alias="id")
+    name: str = Field(..., alias="name")
+    type: str = Field(alias="type")
+    description: str = Field(..., alias="description")
+    input_schema: Dict[str, Any] = Field(
+        ..., alias="inputSchema", description="Input schema for the escalation channel"
+    )
+    output_schema: Dict[str, Any] = Field(
+        ...,
+        alias="outputSchema",
+        description="Output schema for the escalation channel",
+    )
+    properties: AgentEscalationChannelProperties = Field(..., alias="properties")
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentEscalationResourceConfig(BaseAgentResourceConfig):
+    """Escalation resource with escalation-specific properties."""
+
+    resource_type: Literal[AgentResourceType.ESCALATION] = Field(alias="$resourceType")
+    channels: List[AgentEscalationChannel] = Field(alias="channels")
+
+    # escalation_type: int = Field(..., alias="escalationType")
+    is_agent_memory_enabled: bool = Field(alias="isAgentMemoryEnabled")
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+def custom_discriminator(data: Any) -> str:
+    """Discriminator for resource types. This is required due to multi-key discrimination requirements for resources."""
+    if isinstance(data, dict):
+        resource_type = data.get("$resourceType")
+        if resource_type == AgentResourceType.CONTEXT:
+            return "AgentContextResourceConfig"
+        elif resource_type == AgentResourceType.ESCALATION:
+            return "AgentEscalationResourceConfig"
+        elif resource_type == AgentResourceType.TOOL:
+            tool_type = data.get("type")
+            if tool_type == AgentToolType.AGENT:
+                return "AgentProcessToolResourceConfig"
+            elif tool_type == AgentToolType.INTEGRATION:
+                return "AgentIntegrationToolResourceConfig"
+            else:
+                return "AgentUnknownToolResourceConfig"
+        else:
+            return "AgentUnknownResourceConfig"
+    raise ValueError("Invalid discriminator values")
+
+
+AgentResourceConfig = Annotated[
+    Union[
+        Annotated[
+            AgentProcessToolResourceConfig, Tag("AgentProcessToolResourceConfig")
+        ],
+        Annotated[
+            AgentIntegrationToolResourceConfig,
+            Tag("AgentIntegrationToolResourceConfig"),
+        ],
+        Annotated[
+            AgentUnknownToolResourceConfig, Tag("AgentUnknownToolResourceConfig")
+        ],
+        Annotated[AgentContextResourceConfig, Tag("AgentContextResourceConfig")],
+        Annotated[AgentEscalationResourceConfig, Tag("AgentEscalationResourceConfig")],
+        Annotated[AgentUnknownResourceConfig, Tag("AgentUnknownResourceConfig")],
+    ],
+    Field(discriminator=Discriminator(custom_discriminator)),
+]
+
+
+class BaseAgentDefinition(BaseModel):
+    """Main agent model."""
+
+    id: str = Field(..., description="Agent id or project name")
+    name: str = Field(..., description="Agent name or project name")
+    input_schema: Dict[str, Any] = Field(
+        ..., alias="inputSchema", description="JSON schema for input arguments"
+    )
+    output_schema: Dict[str, Any] = Field(
+        ..., alias="outputSchema", description="JSON schema for output arguments"
+    )
+    version: str = Field("1.0.0", description="Agent version")
+    resources: List[AgentResourceConfig] = Field(
+        ..., description="List of tools, context, and escalation resources"
+    )
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentType(str, Enum):
+    """Agent type."""
+
+    LOW_CODE = "lowCode"
+
+
+class AgentMessageRole(str, Enum):
+    """Enum for message roles."""
+
+    SYSTEM = "system"
+    USER = "user"
+
+
+class AgentMessage(BaseModel):
+    """Message model for agent conversations."""
+
+    role: AgentMessageRole
+    content: str
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class AgentSettings(BaseModel):
+    """Settings for agent configuration."""
+
+    engine: str = Field(..., description="Engine type, e.g., 'basic-v1'")
+    model: str = Field(..., description="LLM model identifier")
+    max_tokens: int = Field(
+        ..., alias="maxTokens", description="Maximum number of tokens"
+    )
+    temperature: float = Field(..., description="Temperature for response generation")
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class LowCodeAgentDefinition(BaseAgentDefinition):
+    """Low code agent definition."""
+
+    type: Literal[AgentType.LOW_CODE] = AgentType.LOW_CODE
+    messages: List[AgentMessage] = Field(
+        ..., description="List of system and user messages"
+    )
+    features: List[Any] = Field(
+        default_factory=list, description="Currently empty feature list"
+    )
+    settings: AgentSettings = Field(..., description="Agent settings configuration")
+
+
+KnownAgentDefinition = Annotated[
+    Union[LowCodeAgentDefinition,],
+    Field(discriminator="type"),
+]
+
+
+class UnknownAgentDefinition(BaseAgentDefinition):
+    """Fallback for unknown agent definitions."""
+
+    type: str
+
+    model_config = ConfigDict(extra="allow")
+
+
+AgentDefinition = Union[KnownAgentDefinition, UnknownAgentDefinition]

{uipath-2.1.56.dist-info → uipath-2.1.57.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath
-Version: 2.1.56
+Version: 2.1.57
 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.56.dist-info → uipath-2.1.57.dist-info}/RECORD

@@ -101,6 +101,7 @@ uipath/_utils/_ssl_context.py,sha256=xSYitos0eJc9cPHzNtHISX9PBvL6D2vas5G_GiBdLp8
 uipath/_utils/_url.py,sha256=-4eluSrIZCUlnQ3qU17WPJkgaC2KwF9W5NeqGnTNGGo,2512
 uipath/_utils/_user_agent.py,sha256=pVJkFYacGwaQBomfwWVAvBQgdBUo62e4n3-fLIajWUU,563
 uipath/_utils/constants.py,sha256=defPi1_4sLojgKVsbdIjkfME8O8n-iLmJnj_-n4Ox8s,1108
+uipath/agent/_utils.py,sha256=Z5x7TprkEudrd_aC9YI5ObyT8O05_WkmBJTpqkQKOGQ,1641
 uipath/agent/conversation/__init__.py,sha256=5hK-Iz131mnd9m6ANnpZZffxXZLVFDQ9GTg5z9ik1oQ,5265
 uipath/agent/conversation/async_stream.py,sha256=BA_8uU1DgE3VpU2KkJj0rkI3bAHLk_ZJKsajR0ipMpo,2055
 uipath/agent/conversation/citation.py,sha256=42dGv-wiYx3Lt7MPuPCFTkjAlSADFSzjyNXuZHdxqvo,2253
@@ -111,6 +112,7 @@ uipath/agent/conversation/exchange.py,sha256=nuk1tEMBHc_skrraT17d8U6AtyJ3h07ExGQ
 uipath/agent/conversation/message.py,sha256=1ZkEs146s79TrOAWCQwzBAEJvjAu4lQBpJ64tKXDgGE,2142
 uipath/agent/conversation/meta.py,sha256=3t0eS9UHoAPHre97QTUeVbjDhnMX4zj4-qG6ju0B8wY,315
 uipath/agent/conversation/tool.py,sha256=ol8XI8AVd-QNn5auXNBPcCzOkh9PPFtL7hTK3kqInkU,2191
+uipath/agent/models/agent.py,sha256=jhBxZwNRsB-1F-aD6ZELWaX3cFzgGIJ25cCHIb-RzhE,10965
 uipath/eval/_helpers/__init__.py,sha256=GSmZMryjuO3Wo_zdxZdrHCRRsgOxsVFYkYgJ15YNC3E,86
 uipath/eval/_helpers/helpers.py,sha256=iE2HHdMiAdAMLqxHkPKHpfecEtAuN5BTBqvKFTI8ciE,1315
 uipath/eval/evaluators/__init__.py,sha256=DJAAhgv0I5UfBod4sGnSiKerfrz1iMmk7GNFb71V8eI,494
@@ -148,8 +150,8 @@ uipath/tracing/_traced.py,sha256=CsuNMy67R5HCunuu84LnRNwcadNLhtHLMEawGaNsTYs,184
 uipath/tracing/_utils.py,sha256=wJRELaPu69iY0AhV432Dk5QYf_N_ViRU4kAUG1BI1ew,10384
 uipath/utils/__init__.py,sha256=VD-KXFpF_oWexFg6zyiWMkxl2HM4hYJMIUDZ1UEtGx0,105
 uipath/utils/_endpoints_manager.py,sha256=iRTl5Q0XAm_YgcnMcJOXtj-8052sr6jpWuPNz6CgT0Q,8408
-uipath-2.1.56.dist-info/METADATA,sha256=FwOSIJ0R0gXGh9pIvx6xTl2GjTVElDiovjDpJ35X9ik,6482
-uipath-2.1.56.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-uipath-2.1.56.dist-info/entry_points.txt,sha256=9C2_29U6Oq1ExFu7usihR-dnfIVNSKc-0EFbh0rskB4,43
-uipath-2.1.56.dist-info/licenses/LICENSE,sha256=-KBavWXepyDjimmzH5fVAsi-6jNVpIKFc2kZs0Ri4ng,1058
-uipath-2.1.56.dist-info/RECORD,,
+uipath-2.1.57.dist-info/METADATA,sha256=LZV3q6VHuyZ0YY919szKSEItcd9Ir0p4wocixl1r7T0,6482
+uipath-2.1.57.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+uipath-2.1.57.dist-info/entry_points.txt,sha256=9C2_29U6Oq1ExFu7usihR-dnfIVNSKc-0EFbh0rskB4,43
+uipath-2.1.57.dist-info/licenses/LICENSE,sha256=-KBavWXepyDjimmzH5fVAsi-6jNVpIKFc2kZs0Ri4ng,1058
+uipath-2.1.57.dist-info/RECORD,,