diff --git a/pyproject.toml b/pyproject.toml
index a283ba95..15b84634 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "uipath"
-version = "2.0.3"
+version = "2.0.4"
 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"
@@ -76,6 +76,8 @@ ignore-decorators = []
 "*_test.py" = ["D"]
 # TODO: Remove this once the documentation for CLI is updated
 "uipath/_cli/**" = ["D"]
+# TODO: Remove this once model documentation is added
+"src/uipath/models/**" = ["D101","D100","D104","D102","D107"]
 
 [tool.ruff.format]
 quote-style = "double"
diff --git a/src/uipath/_services/actions_service.py b/src/uipath/_services/actions_service.py
index 6ec8dbaa..24dd4dc8 100644
--- a/src/uipath/_services/actions_service.py
+++ b/src/uipath/_services/actions_service.py
@@ -6,9 +6,9 @@
 from .._config import Config
 from .._execution_context import ExecutionContext
 from .._folder_context import FolderContext
-from .._models import Action, ActionSchema
 from .._utils import Endpoint, RequestSpec
 from .._utils.constants import ENV_TENANT_ID, HEADER_TENANT_ID
+from ..models import Action, ActionSchema
 from ._base_service import BaseService
 
 
diff --git a/src/uipath/_services/assets_service.py b/src/uipath/_services/assets_service.py
index 92492c6c..43d9149c 100644
--- a/src/uipath/_services/assets_service.py
+++ b/src/uipath/_services/assets_service.py
@@ -5,8 +5,8 @@
 from .._config import Config
 from .._execution_context import ExecutionContext
 from .._folder_context import FolderContext
-from .._models import UserAsset
 from .._utils import Endpoint, RequestSpec, header_folder, infer_bindings
+from ..models import UserAsset
 from ._base_service import BaseService
 
 
diff --git a/src/uipath/_services/connections_service.py b/src/uipath/_services/connections_service.py
index d6eed36c..5bb8740a 100644
--- a/src/uipath/_services/connections_service.py
+++ b/src/uipath/_services/connections_service.py
@@ -4,9 +4,9 @@
 
 from .._config import Config
 from .._execution_context import ExecutionContext
-from .._models import Connection, ConnectionToken
 from .._utils import Endpoint, RequestSpec
 from .._utils.constants import ENTRYPOINT
+from ..models import Connection, ConnectionToken
 from ._base_service import BaseService
 
 T_co = TypeVar("T_co", covariant=True)
diff --git a/src/uipath/_services/connections_service.pyi b/src/uipath/_services/connections_service.pyi
index 653ad307..c3362aab 100644
--- a/src/uipath/_services/connections_service.pyi
+++ b/src/uipath/_services/connections_service.pyi
@@ -3,11 +3,11 @@ from typing import TYPE_CHECKING, Any, Protocol, TypeVar
 
 from .._config import Config as Config
 from .._execution_context import ExecutionContext as ExecutionContext
-from .._models import Connection as Connection
-from .._models import ConnectionToken as ConnectionToken
 from .._utils import Endpoint as Endpoint
 from .._utils import RequestSpec as RequestSpec
 from .._utils.constants import ENTRYPOINT as ENTRYPOINT
+from ..models import Connection as Connection
+from ..models import ConnectionToken as ConnectionToken
 from ._base_service import BaseService as BaseService
 
 if TYPE_CHECKING:
diff --git a/src/uipath/_services/context_grounding_service.py b/src/uipath/_services/context_grounding_service.py
index 9f36308a..80aa83d1 100644
--- a/src/uipath/_services/context_grounding_service.py
+++ b/src/uipath/_services/context_grounding_service.py
@@ -6,14 +6,14 @@
 from .._config import Config
 from .._execution_context import ExecutionContext
 from .._folder_context import FolderContext
-from .._models import IngestionInProgressException
-from .._models.context_grounding import ContextGroundingQueryResponse
-from .._models.context_grounding_index import ContextGroundingIndex
 from .._utils import Endpoint, RequestSpec
 from .._utils.constants import (
     HEADER_FOLDER_KEY,
     ORCHESTRATOR_STORAGE_BUCKET_DATA_SOURCE,
 )
+from ..models import IngestionInProgressException
+from ..models.context_grounding import ContextGroundingQueryResponse
+from ..models.context_grounding_index import ContextGroundingIndex
 from ._base_service import BaseService
 from .folder_service import FolderService
 
diff --git a/src/uipath/_services/jobs_service.py b/src/uipath/_services/jobs_service.py
index a700de0e..a09b69b1 100644
--- a/src/uipath/_services/jobs_service.py
+++ b/src/uipath/_services/jobs_service.py
@@ -4,8 +4,8 @@
 from .._config import Config
 from .._execution_context import ExecutionContext
 from .._folder_context import FolderContext
-from .._models.job import Job
 from .._utils import Endpoint, RequestSpec, header_folder
+from ..models.job import Job
 from ._base_service import BaseService
 
 
diff --git a/src/uipath/_services/llm_gateway_service.py b/src/uipath/_services/llm_gateway_service.py
index 79b651d9..e5c75d41 100644
--- a/src/uipath/_services/llm_gateway_service.py
+++ b/src/uipath/_services/llm_gateway_service.py
@@ -3,7 +3,8 @@
 
 from .._config import Config
 from .._execution_context import ExecutionContext
-from .._models.llm_gateway import (
+from .._utils import Endpoint
+from ..models.llm_gateway import (
     ChatCompletion,
     SpecificToolChoice,
     TextEmbedding,
@@ -11,7 +12,6 @@
     ToolDefinition,
     UsageInfo,
 )
-from .._utils import Endpoint
 from ._base_service import BaseService
 
 # Common constants
diff --git a/src/uipath/_services/processes_service.py b/src/uipath/_services/processes_service.py
index bd43b721..7dbceffe 100644
--- a/src/uipath/_services/processes_service.py
+++ b/src/uipath/_services/processes_service.py
@@ -5,9 +5,9 @@
 from .._config import Config
 from .._execution_context import ExecutionContext
 from .._folder_context import FolderContext
-from .._models.job import Job
 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 ._base_service import BaseService
 
 
diff --git a/src/uipath/_services/queues_service.py b/src/uipath/_services/queues_service.py
index 5bfa02e0..66c5c306 100644
--- a/src/uipath/_services/queues_service.py
+++ b/src/uipath/_services/queues_service.py
@@ -5,8 +5,8 @@
 from .._config import Config
 from .._execution_context import ExecutionContext
 from .._folder_context import FolderContext
-from .._models import CommitType, QueueItem, TransactionItem, TransactionItemResult
 from .._utils import Endpoint, RequestSpec
+from ..models import CommitType, QueueItem, TransactionItem, TransactionItemResult
 from ._base_service import BaseService
 
 
diff --git a/src/uipath/_models/__init__.py b/src/uipath/models/__init__.py
similarity index 100%
rename from src/uipath/_models/__init__.py
rename to src/uipath/models/__init__.py
diff --git a/src/uipath/_models/action_schema.py b/src/uipath/models/action_schema.py
similarity index 100%
rename from src/uipath/_models/action_schema.py
rename to src/uipath/models/action_schema.py
diff --git a/src/uipath/_models/actions.py b/src/uipath/models/actions.py
similarity index 100%
rename from src/uipath/_models/actions.py
rename to src/uipath/models/actions.py
diff --git a/src/uipath/_models/assets.py b/src/uipath/models/assets.py
similarity index 100%
rename from src/uipath/_models/assets.py
rename to src/uipath/models/assets.py
diff --git a/src/uipath/_models/connections.py b/src/uipath/models/connections.py
similarity index 100%
rename from src/uipath/_models/connections.py
rename to src/uipath/models/connections.py
diff --git a/src/uipath/_models/context_grounding.py b/src/uipath/models/context_grounding.py
similarity index 100%
rename from src/uipath/_models/context_grounding.py
rename to src/uipath/models/context_grounding.py
diff --git a/src/uipath/_models/context_grounding_index.py b/src/uipath/models/context_grounding_index.py
similarity index 100%
rename from src/uipath/_models/context_grounding_index.py
rename to src/uipath/models/context_grounding_index.py
diff --git a/src/uipath/_models/exceptions.py b/src/uipath/models/exceptions.py
similarity index 100%
rename from src/uipath/_models/exceptions.py
rename to src/uipath/models/exceptions.py
diff --git a/src/uipath/_models/interrupt_models.py b/src/uipath/models/interrupt_models.py
similarity index 100%
rename from src/uipath/_models/interrupt_models.py
rename to src/uipath/models/interrupt_models.py
diff --git a/src/uipath/_models/job.py b/src/uipath/models/job.py
similarity index 100%
rename from src/uipath/_models/job.py
rename to src/uipath/models/job.py
diff --git a/src/uipath/_models/llm_gateway.py b/src/uipath/models/llm_gateway.py
similarity index 100%
rename from src/uipath/_models/llm_gateway.py
rename to src/uipath/models/llm_gateway.py
diff --git a/src/uipath/_models/processes.py b/src/uipath/models/processes.py
similarity index 100%
rename from src/uipath/_models/processes.py
rename to src/uipath/models/processes.py
diff --git a/src/uipath/_models/queues.py b/src/uipath/models/queues.py
similarity index 100%
rename from src/uipath/_models/queues.py
rename to src/uipath/models/queues.py
diff --git a/tests/sdk/services/test_llm_service.py b/tests/sdk/services/test_llm_service.py
index 67de6d14..51ec279d 100644
--- a/tests/sdk/services/test_llm_service.py
+++ b/tests/sdk/services/test_llm_service.py
@@ -5,14 +5,14 @@
 
 from uipath._config import Config
 from uipath._execution_context import ExecutionContext
-from uipath._models.llm_gateway import (
-    TextEmbedding,
-    UsageInfo,
-)
 from uipath._services.llm_gateway_service import (
     EmbeddingModels,
     UiPathOpenAIService,
 )
+from uipath.models.llm_gateway import (
+    TextEmbedding,
+    UsageInfo,
+)
 
 
 class TestOpenAIService:
diff --git a/tests/sdk/services/test_uipath_llm_integration.py b/tests/sdk/services/test_uipath_llm_integration.py
index 91a509f0..3601dd3f 100644
--- a/tests/sdk/services/test_uipath_llm_integration.py
+++ b/tests/sdk/services/test_uipath_llm_integration.py
@@ -6,7 +6,11 @@
 
 from uipath._config import Config
 from uipath._execution_context import ExecutionContext
-from uipath._models.llm_gateway import (
+from uipath._services.llm_gateway_service import (
+    ChatModels,
+    UiPathLlmChatService,
+)
+from uipath.models.llm_gateway import (
     AutoToolChoice,
     SpecificToolChoice,
     ToolDefinition,
@@ -14,10 +18,6 @@
     ToolParametersDefinition,
     ToolPropertyDefinition,
 )
-from uipath._services.llm_gateway_service import (
-    ChatModels,
-    UiPathLlmChatService,
-)
 
 
 def get_env_var(name: str) -> str:
diff --git a/uv.lock b/uv.lock
index 2bc9614a..56fd1807 100644
--- a/uv.lock
+++ b/uv.lock
@@ -1,4 +1,5 @@
 version = 1
+revision = 1
 requires-python = ">=3.10"
 resolution-markers = [
     "python_full_version >= '3.12.4'",
@@ -2319,7 +2320,7 @@ wheels = [
 
 [[package]]
 name = "uipath"
-version = "2.0.2.dev1"
+version = "2.0.4"
 source = { editable = "." }
 dependencies = [
     { name = "click" },
@@ -2367,6 +2368,7 @@ requires-dist = [
     { name = "types-requests", specifier = ">=2.32.0.20250306" },
     { name = "uipath-langchain", marker = "extra == 'langchain'", specifier = "==0.0.85" },
 ]
+provides-extras = ["langchain"]
 
 [package.metadata.requires-dev]
 dev = [