feat(ns): move stuff under chat

ionmincu · ionmincu · commit 04f3c479ea25 · 2025-12-03T19:33:31.000+02:00
diff --git a/samples/llm_chat_agent/README.md b/samples/llm_chat_agent/README.md
@@ -120,9 +120,8 @@ Tools used: ['calculate']
 ### LLM Service Initialization
 
 ```python
-from uipath._config import Config
-from uipath._execution_context import ExecutionContext
-from uipath.platform.llm_gateway._llm_gateway_service import (
+from uipath.platform import Config, ExecutionContext
+from uipath.platform.chat._llm_gateway_service import (
     ChatModels,
     UiPathLlmChatService,
 )
@@ -148,7 +147,7 @@ result = await llm_service.chat_completions(
 ### Tool Definition
 
 ```python
-from uipath.platform.llm_gateway import (
+from uipath.platform.chat import (
     ToolDefinition,
     ToolFunctionDefinition,
     ToolParametersDefinition,
diff --git a/samples/llm_chat_agent/agent.py b/samples/llm_chat_agent/agent.py
@@ -20,19 +20,18 @@
 import httpx
 from dotenv import load_dotenv
 from pydantic import BaseModel, Field
-from uipath._config import Config
-from uipath._execution_context import ExecutionContext
-from uipath.platform.llm_gateway._llm_gateway_service import (
-    ChatModels,
-    UiPathLlmChatService,
-)
-from uipath.platform.llm_gateway import (
+from uipath.platform import Config, ExecutionContext
+from uipath.platform.chat import (
     AutoToolChoice,
     ToolDefinition,
     ToolFunctionDefinition,
     ToolParametersDefinition,
     ToolPropertyDefinition,
 )
+from uipath.platform.chat._llm_gateway_service import (
+    ChatModels,
+    UiPathLlmChatService,
+)
 from uipath.tracing import traced
 
 # Load environment variables from .env file if it exists
diff --git a/src/uipath/_cli/_evals/mocks/llm_mocker.py b/src/uipath/_cli/_evals/mocks/llm_mocker.py
@@ -91,7 +91,7 @@ async def response(
             x.name for x in self.evaluation_item.mocking_strategy.tools_to_simulate
         ]:
             from uipath.platform import UiPath
-            from uipath.platform.llm_gateway._llm_gateway_service import _cleanup_schema
+            from uipath.platform.chat._llm_gateway_service import _cleanup_schema
 
             from .mocks import (
                 cache_manager_context,
diff --git a/src/uipath/eval/evaluators/legacy_llm_as_judge_evaluator.py b/src/uipath/eval/evaluators/legacy_llm_as_judge_evaluator.py
@@ -8,7 +8,7 @@
 from uipath.eval.models import NumericEvaluationResult
 
 from ..._utils.constants import COMMUNITY_agents_SUFFIX
-from ...platform.llm_gateway import UiPathLlmChatService
+from ...platform.chat import UiPathLlmChatService
 from ..models.models import AgentExecution, EvaluationResult, LLMResponse
 from .legacy_base_evaluator import (
     LegacyBaseEvaluator,
diff --git a/src/uipath/eval/evaluators/legacy_trajectory_evaluator.py b/src/uipath/eval/evaluators/legacy_trajectory_evaluator.py
@@ -9,7 +9,7 @@
 from uipath.eval.models import EvaluationResult
 
 from ..._utils.constants import COMMUNITY_agents_SUFFIX
-from ...platform.llm_gateway import UiPathLlmChatService
+from ...platform.chat import UiPathLlmChatService
 from ..models.models import (
     AgentExecution,
     LLMResponse,
diff --git a/src/uipath/platform/__init__.py b/src/uipath/platform/__init__.py
@@ -35,5 +35,6 @@
 """
 
 from ._uipath import UiPath
+from .common import Config, ExecutionContext
 
-__all__ = ["UiPath"]
+__all__ = ["Config", "ExecutionContext", "UiPath"]
diff --git a/src/uipath/platform/_uipath.py b/src/uipath/platform/_uipath.py
@@ -3,20 +3,17 @@
 
 from pydantic import ValidationError
 
-from .._config import Config
-from .._execution_context import ExecutionContext
 from .._utils._auth import resolve_config
 from .action_center import TasksService
-from .agenthub import McpService
 from .autopilot import ConversationsService
-from .common import ApiClient
+from .chat import McpService, UiPathLlmChatService, UiPathOpenAIService
+from .common import ApiClient, Config, ExecutionContext
 from .connections import ConnectionsService
 from .context_grounding import ContextGroundingService
 from .documents import DocumentsService
 from .entities import EntitiesService
 from .errors import BaseUrlMissingError, SecretMissingError
 from .guardrails import GuardrailsService
-from .llm_gateway import UiPathLlmChatService, UiPathOpenAIService
 from .orchestrator import (
     AssetsService,
     AttachmentsService,
diff --git a/src/uipath/platform/action_center/_tasks_service.py b/src/uipath/platform/action_center/_tasks_service.py
@@ -2,9 +2,6 @@
 import uuid
 from typing import Any, Dict, List, Optional, Tuple
 
-from ..._config import Config
-from ..._execution_context import ExecutionContext
-from ..._folder_context import FolderContext
 from ..._utils import Endpoint, RequestSpec, resource_override
 from ..._utils.constants import (
     ENV_TENANT_ID,
@@ -13,6 +10,7 @@
     HEADER_TENANT_ID,
 )
 from ...tracing import traced
+from ..common import Config, ExecutionContext, FolderContext
 from ..common._base_service import BaseService
 from .task_schema import TaskSchema
 from .tasks import Task
diff --git a/src/uipath/platform/agenthub/__init__.py b/src/uipath/platform/agenthub/__init__.py
diff --git a/src/uipath/platform/autopilot/_conversations_service.py b/src/uipath/platform/autopilot/_conversations_service.py
@@ -1,10 +1,8 @@
 from uipath.core.chat import UiPathConversationMessage
 
-from ..._config import Config
-from ..._execution_context import ExecutionContext
 from ..._utils import Endpoint, RequestSpec
 from ...tracing import traced
-from ..common._base_service import BaseService
+from ..common import BaseService, Config, ExecutionContext
 
 
 class ConversationsService(BaseService):
diff --git a/src/uipath/platform/chat/__init__.py b/src/uipath/platform/chat/__init__.py
@@ -1,9 +1,12 @@
-"""UiPath LLM Gateway Models.
+"""UiPath Chat Services.
 
-This module contains models related to UiPath LLM Gateway service.
+This module provides services for chat-related functionality including:
+- MCP (Model Context Protocol) servers management
+- LLM Gateway services for chat completions and embeddings
 """
 
 from ._llm_gateway_service import UiPathLlmChatService, UiPathOpenAIService
+from ._mcp_service import McpService
 from .llm_gateway import (
     AutoToolChoice,
     ChatCompletion,
@@ -24,8 +27,12 @@
 )
 
 __all__ = [
+    # MCP Service
+    "McpService",
+    # LLM Gateway Services
     "UiPathLlmChatService",
     "UiPathOpenAIService",
+    # LLM Gateway Models
     "ToolPropertyDefinition",
     "ToolParametersDefinition",
     "ToolFunctionDefinition",
diff --git a/src/uipath/platform/chat/_llm_gateway_service.py b/src/uipath/platform/chat/_llm_gateway_service.py
@@ -20,12 +20,10 @@
 
 from pydantic import BaseModel
 
-from ..._config import Config
-from ..._execution_context import ExecutionContext
 from ..._utils import Endpoint
 from ...tracing import traced
 from ...utils import EndpointManager
-from ..common._base_service import BaseService
+from ..common import BaseService, Config, ExecutionContext
 from .llm_gateway import (
     ChatCompletion,
     SpecificToolChoice,
diff --git a/src/uipath/platform/chat/_mcp_service.py b/src/uipath/platform/chat/_mcp_service.py
@@ -1,11 +1,8 @@
 from typing import List
 
-from ..._config import Config
-from ..._execution_context import ExecutionContext
-from ..._folder_context import FolderContext
 from ..._utils import Endpoint, RequestSpec, header_folder
 from ...tracing import traced
-from ..common._base_service import BaseService
+from ..common import BaseService, Config, ExecutionContext, FolderContext
 from ..orchestrator._folder_service import FolderService
 from ..orchestrator.mcp import McpServer
 
diff --git a/src/uipath/platform/chat/llm_gateway.py b/src/uipath/platform/chat/llm_gateway.py
diff --git a/src/uipath/platform/common/__init__.py b/src/uipath/platform/common/__init__.py
@@ -5,7 +5,10 @@
 
 from ._api_client import ApiClient
 from ._base_service import BaseService
+from ._config import Config, ConfigurationManager, UiPathConfig
+from ._execution_context import ExecutionContext
 from ._external_application_service import ExternalApplicationService
+from ._folder_context import FolderContext
 from .auth import TokenData
 from .interrupt_models import (
     CreateEscalation,
@@ -20,8 +23,13 @@
 __all__ = [
     "ApiClient",
     "BaseService",
+    "Config",
+    "ConfigurationManager",
+    "ExecutionContext",
     "ExternalApplicationService",
+    "FolderContext",
     "TokenData",
+    "UiPathConfig",
     "CreateTask",
     "CreateEscalation",
     "WaitEscalation",
diff --git a/src/uipath/platform/common/_api_client.py b/src/uipath/platform/common/_api_client.py
@@ -2,10 +2,10 @@
 
 from httpx import URL, Response
 
-from ..._config import Config
-from ..._execution_context import ExecutionContext
-from ..._folder_context import FolderContext
 from ._base_service import BaseService
+from ._config import Config
+from ._execution_context import ExecutionContext
+from ._folder_context import FolderContext
 
 
 class ApiClient(FolderContext, BaseService):
diff --git a/src/uipath/platform/common/_base_service.py b/src/uipath/platform/common/_base_service.py
@@ -19,8 +19,8 @@
     wait_exponential,
 )
 
-from ..._config import Config
-from ..._execution_context import ExecutionContext
+from ._config import Config
+from ._execution_context import ExecutionContext
 from ..._utils import UiPathUrl, user_agent_value
 from ..._utils._ssl_context import get_httpx_client_kwargs
 from ..._utils.constants import HEADER_USER_AGENT
diff --git a/src/uipath/platform/common/_config.py b/src/uipath/platform/common/_config.py
diff --git a/src/uipath/platform/common/_execution_context.py b/src/uipath/platform/common/_execution_context.py
@@ -1,6 +1,6 @@
 from os import environ as env
 
-from ._utils.constants import ENV_JOB_ID, ENV_JOB_KEY, ENV_ROBOT_KEY
+from uipath._utils.constants import ENV_JOB_ID, ENV_JOB_KEY, ENV_ROBOT_KEY
 
 
 class ExecutionContext:
diff --git a/src/uipath/platform/common/_folder_context.py b/src/uipath/platform/common/_folder_context.py
@@ -1,7 +1,7 @@
 from os import environ as env
 from typing import Any
 
-from ._utils.constants import (
+from uipath._utils.constants import (
     ENV_FOLDER_KEY,
     ENV_FOLDER_PATH,
     HEADER_FOLDER_KEY,
diff --git a/src/uipath/platform/connections/_connections_service.py b/src/uipath/platform/connections/_connections_service.py
@@ -5,11 +5,9 @@
 
 from httpx import Response
 
-from ..._config import Config
-from ..._execution_context import ExecutionContext
 from ..._utils import Endpoint, RequestSpec, header_folder, resource_override
 from ...tracing import traced
-from ..common._base_service import BaseService
+from ..common import BaseService, Config, ExecutionContext
 from ..orchestrator._folder_service import FolderService
 from .connections import (
     ActivityMetadata,
diff --git a/src/uipath/platform/context_grounding/_context_grounding_service.py b/src/uipath/platform/context_grounding/_context_grounding_service.py
@@ -4,16 +4,13 @@
 from pydantic import TypeAdapter
 from typing_extensions import deprecated
 
-from ..._config import Config
-from ..._execution_context import ExecutionContext
-from ..._folder_context import FolderContext
 from ..._utils import Endpoint, RequestSpec, header_folder, resource_override
 from ..._utils.constants import (
     LLMV4_REQUEST,
     ORCHESTRATOR_STORAGE_BUCKET_DATA_SOURCE,
 )
 from ...tracing import traced
-from ..common._base_service import BaseService
+from ..common import BaseService, Config, ExecutionContext, FolderContext
 from ..errors import (
     IngestionInProgressException,
     UnsupportedDataSourceException,
diff --git a/src/uipath/platform/documents/_documents_service.py b/src/uipath/platform/documents/_documents_service.py
@@ -6,12 +6,9 @@
 from typing import Any, Awaitable, Callable, Dict, List, Optional, Set, Tuple, Union
 from uuid import UUID
 
-from ..._config import Config
-from ..._execution_context import ExecutionContext
-from ..._folder_context import FolderContext
 from ..._utils import Endpoint
 from ...tracing import traced
-from ..common._base_service import BaseService
+from ..common import BaseService, Config, ExecutionContext, FolderContext
 from .documents import (
     ActionPriority,
     ClassificationResponse,
diff --git a/src/uipath/platform/entities/_entities_service.py b/src/uipath/platform/entities/_entities_service.py
@@ -2,11 +2,9 @@
 
 from httpx import Response
 
-from ..._config import Config
-from ..._execution_context import ExecutionContext
 from ..._utils import Endpoint, RequestSpec
 from ...tracing import traced
-from ..common._base_service import BaseService
+from ..common import BaseService, Config, ExecutionContext
 from .entities import (
     Entity,
     EntityRecord,
diff --git a/src/uipath/platform/guardrails/_guardrails_service.py b/src/uipath/platform/guardrails/_guardrails_service.py
@@ -1,10 +1,8 @@
 from typing import Any
 
-from ..._config import Config
-from ..._execution_context import ExecutionContext
 from ..._utils import Endpoint, RequestSpec
 from ...tracing import traced
-from ..common._base_service import BaseService
+from ..common import BaseService, Config, ExecutionContext
 from .guardrails import BuiltInValidatorGuardrail, Guardrail, GuardrailValidationResult
 
 
diff --git a/src/uipath/platform/orchestrator/_assets_service.py b/src/uipath/platform/orchestrator/_assets_service.py
@@ -2,12 +2,9 @@
 
 from httpx import Response
 
-from ..._config import Config
-from ..._execution_context import ExecutionContext
-from ..._folder_context import FolderContext
 from ..._utils import Endpoint, RequestSpec, header_folder, resource_override
 from ...tracing import traced
-from ..common._base_service import BaseService
+from ..common import BaseService, Config, ExecutionContext, FolderContext
 from .assets import Asset, UserAsset
 
 
diff --git a/src/uipath/platform/orchestrator/_attachments_service.py b/src/uipath/platform/orchestrator/_attachments_service.py
@@ -11,15 +11,12 @@
 from httpx import Response
 from httpx._types import RequestContent
 
-from ..._config import Config
-from ..._execution_context import ExecutionContext
-from ..._folder_context import FolderContext
 from ..._utils import Endpoint, RequestSpec, header_folder
 from ..._utils._ssl_context import get_httpx_client_kwargs
 from ..._utils.constants import TEMP_ATTACHMENTS_FOLDER
 from ...tracing import traced
 from ..attachments import Attachment, AttachmentMode
-from ..common._base_service import BaseService
+from ..common import BaseService, Config, ExecutionContext, FolderContext
 
 
 def _upload_attachment_input_processor(inputs: dict[str, Any]) -> dict[str, Any]:
diff --git a/src/uipath/platform/orchestrator/_buckets_service.py b/src/uipath/platform/orchestrator/_buckets_service.py
@@ -6,13 +6,10 @@
 
 import httpx
 
-from ..._config import Config
-from ..._execution_context import ExecutionContext
-from ..._folder_context import FolderContext
 from ..._utils import Endpoint, RequestSpec, header_folder, resource_override
 from ..._utils._ssl_context import get_httpx_client_kwargs
 from ...tracing import traced
-from ..common._base_service import BaseService
+from ..common import BaseService, Config, ExecutionContext, FolderContext
 from ..common.paging import PagedResult
 from .buckets import Bucket, BucketFile
 
diff --git a/src/uipath/platform/orchestrator/_folder_service.py b/src/uipath/platform/orchestrator/_folder_service.py
@@ -2,11 +2,9 @@
 
 from typing_extensions import deprecated
 
-from ..._config import Config
-from ..._execution_context import ExecutionContext
 from ..._utils import Endpoint, RequestSpec
 from ...tracing import traced
-from ..common._base_service import BaseService
+from ..common import BaseService, Config, ExecutionContext
 from ..errors import FolderNotFoundException
 
 
diff --git a/src/uipath/platform/orchestrator/_jobs_service.py b/src/uipath/platform/orchestrator/_jobs_service.py
diff --git a/src/uipath/platform/orchestrator/_processes_service.py b/src/uipath/platform/orchestrator/_processes_service.py
diff --git a/src/uipath/platform/orchestrator/_queues_service.py b/src/uipath/platform/orchestrator/_queues_service.py
diff --git a/src/uipath/platform/resource_catalog/_resource_catalog_service.py b/src/uipath/platform/resource_catalog/_resource_catalog_service.py
diff --git a/testcases/apicalls-testcase/main.py b/testcases/apicalls-testcase/main.py
diff --git a/tests/sdk/services/test_llm_integration.py b/tests/sdk/services/test_llm_integration.py
diff --git a/tests/sdk/services/test_llm_schema_cleanup.py b/tests/sdk/services/test_llm_schema_cleanup.py
diff --git a/tests/sdk/services/test_llm_service.py b/tests/sdk/services/test_llm_service.py
diff --git a/tests/sdk/services/test_mcp_service.py b/tests/sdk/services/test_mcp_service.py
diff --git a/tests/sdk/services/test_uipath_llm_integration.py b/tests/sdk/services/test_uipath_llm_integration.py
