agentex-sdk 0.4.17__tar.gz → 0.4.19__tar.gz

agentex_sdk-0.4.19/.release-please-manifest.json

@@ -0,0 +1,3 @@
+{
+  ".": "0.4.19"
+}

{agentex_sdk-0.4.17 → agentex_sdk-0.4.19}/CHANGELOG.md

@@ -1,14 +1,24 @@
 # Changelog
 
-## 0.4.17 (2025-09-29)
+## 0.4.19 (2025-10-01)
 
-Full Changelog: [v0.4.16...v0.4.17](https://github.com/scaleapi/agentex-python/compare/v0.4.16...v0.4.17)
+Full Changelog: [v0.4.18...v0.4.19](https://github.com/scaleapi/agentex-python/compare/v0.4.18...v0.4.19)
+
+### Features
+
+* Adds helm config to Agent Environment ([#125](https://github.com/scaleapi/agentex-python/issues/125)) ([e4b39b5](https://github.com/scaleapi/agentex-python/commit/e4b39b5f319452bbc6650a7ef41b3a3179bb3b93))
+
+## 0.4.18 (2025-09-29)
+
+Full Changelog: [v0.4.17...v0.4.18](https://github.com/scaleapi/agentex-python/compare/v0.4.17...v0.4.18)
 
 ### Chores
 
-* **internal:** codegen related update ([2fdc0e7](https://github.com/scaleapi/agentex-python/commit/2fdc0e75ea3874cf896cdbb119b50a4165b2e942))
-* **internal:** version bump ([0a59ad4](https://github.com/scaleapi/agentex-python/commit/0a59ad40b55b3577ef2addcea2fe4c0e4f002d49))
-* **internal:** version bump ([6174ef1](https://github.com/scaleapi/agentex-python/commit/6174ef1573a539f5e0f57bc625a67da31311afb6))
+* **internal:** version bump ([eded756](https://github.com/scaleapi/agentex-python/commit/eded756bde2f3b4cfcf02c7a9cf72e70a82dd9aa))
+
+## 0.4.17 (2025-09-29)
+
+Full Changelog: [v0.4.16...v0.4.17](https://github.com/scaleapi/agentex-python/compare/v0.4.16...v0.4.17)
 
 ## 0.4.16 (2025-09-16)
 

{agentex_sdk-0.4.17 → agentex_sdk-0.4.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentex-sdk
-Version: 0.4.17
+Version: 0.4.19
 Summary: The official Python library for the agentex API
 Project-URL: Homepage, https://github.com/scaleapi/agentex-python
 Project-URL: Repository, https://github.com/scaleapi/agentex-python
@@ -22,11 +22,14 @@ Requires-Python: <4,>=3.12
 Requires-Dist: aiohttp<4,>=3.10.10
 Requires-Dist: anyio<5,>=3.5.0
 Requires-Dist: cloudpickle>=3.1.1
+Requires-Dist: datadog>=0.52.1
+Requires-Dist: ddtrace>=3.13.0
 Requires-Dist: distro<2,>=1.7.0
 Requires-Dist: fastapi<0.116,>=0.115.0
 Requires-Dist: httpx<0.28,>=0.27.2
 Requires-Dist: ipykernel>=6.29.5
 Requires-Dist: jinja2<4,>=3.1.3
+Requires-Dist: json-log-formatter>=1.1.1
 Requires-Dist: jsonref<2,>=1.1.0
 Requires-Dist: jsonschema<5,>=4.23.0
 Requires-Dist: kubernetes<29.0.0,>=25.0.0

{agentex_sdk-0.4.17 → agentex_sdk-0.4.19}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentex-sdk"
-version = "0.4.17"
+version = "0.4.19"
 description = "The official Python library for the agentex API"
 dynamic = ["readme"]
 license = "Apache-2.0"
@@ -36,11 +36,14 @@ dependencies = [
     "tzlocal>=5.3.1",
     "tzdata>=2025.2",
     "pytest>=8.4.0",
+    "json_log_formatter>=1.1.1",
     "pytest-asyncio>=1.0.0",
     "scale-gp-beta==0.1.0a20",
     "ipykernel>=6.29.5",
     "openai==1.99.9", # anything higher than 1.99.9 breaks litellm - https://github.com/BerriAI/litellm/issues/13711
     "cloudpickle>=3.1.1",
+    "datadog>=0.52.1",
+    "ddtrace>=3.13.0"
 ]
 requires-python = ">= 3.12,<4"
 classifiers = [
@@ -86,6 +89,7 @@ dev-dependencies = [
     "dirty-equals>=0.6.0",
     "importlib-metadata>=6.7.0",
     "rich>=13.7.1",
+    "nest_asyncio==1.6.0",
     "pytest-xdist>=3.6.1",
     "debugpy>=1.8.15",
 ]
@@ -221,7 +225,7 @@ warn_unused_ignores = false
 warn_redundant_casts = false
 
 disallow_any_generics = true
-disallow_untyped_defs = true
+# disallow_untyped_defs = true
 disallow_untyped_calls = true
 disallow_subclassing_any = true
 disallow_incomplete_defs = true
@@ -237,7 +241,7 @@ cache_fine_grained = true
 # ```
 # Changing this codegen to make mypy happy would increase complexity
 # and would not be worth it.
-disable_error_code = "func-returns-value,overload-cannot-match"
+disable_error_code = "func-returns-value,overload-cannot-match,no-untyped-def"
 
 # https://github.com/python/mypy/issues/12162
 [[tool.mypy.overrides]]

{agentex_sdk-0.4.17 → agentex_sdk-0.4.19}/requirements-dev.lock

@@ -40,6 +40,8 @@ attrs==25.3.0
     # via aiohttp
     # via jsonschema
     # via referencing
+bytecode==0.17.0
+    # via ddtrace
 cachetools==5.5.2
     # via google-auth
 certifi==2023.7.22
@@ -61,6 +63,10 @@ colorlog==6.7.0
     # via nox
 comm==0.2.3
     # via ipykernel
+datadog==0.52.1
+    # via agentex-sdk
+ddtrace==3.15.0
+    # via agentex-sdk
 debugpy==1.8.16
     # via ipykernel
 decorator==5.2.1
@@ -73,6 +79,8 @@ distro==1.8.0
     # via openai
     # via scale-gp
     # via scale-gp-beta
+envier==0.6.1
+    # via ddtrace
 execnet==2.1.1
     # via pytest-xdist
 executing==2.2.0
@@ -120,6 +128,7 @@ idna==3.4
     # via yarl
 importlib-metadata==7.0.0
     # via litellm
+    # via opentelemetry-api
 iniconfig==2.0.0
     # via pytest
 ipykernel==6.30.1
@@ -135,6 +144,8 @@ jinja2==3.1.6
     # via litellm
 jiter==0.10.0
     # via openai
+json-log-formatter==1.1.1
+    # via agentex-sdk
 jsonref==1.1.0
     # via agentex-sdk
 jsonschema==4.25.0
@@ -186,6 +197,8 @@ openai==1.99.9
     # via openai-agents
 openai-agents==0.2.7
     # via agentex-sdk
+opentelemetry-api==1.37.0
+    # via ddtrace
 packaging==23.2
     # via huggingface-hub
     # via ipykernel
@@ -207,6 +220,7 @@ propcache==0.3.1
     # via aiohttp
     # via yarl
 protobuf==5.29.5
+    # via ddtrace
     # via temporalio
 psutil==7.0.0
     # via ipykernel
@@ -280,6 +294,7 @@ referencing==0.36.2
 regex==2025.7.34
     # via tiktoken
 requests==2.32.4
+    # via datadog
     # via huggingface-hub
     # via kubernetes
     # via openai-agents
@@ -363,6 +378,7 @@ typing-extensions==4.12.2
     # via nexus-rpc
     # via openai
     # via openai-agents
+    # via opentelemetry-api
     # via pydantic
     # via pydantic-core
     # via pyright
@@ -393,6 +409,8 @@ wcwidth==0.2.13
     # via prompt-toolkit
 websocket-client==1.8.0
     # via kubernetes
+wrapt==1.17.3
+    # via ddtrace
 yarl==1.20.0
     # via aiohttp
 zipp==3.17.0

{agentex_sdk-0.4.17 → agentex_sdk-0.4.19}/requirements.lock

@@ -38,6 +38,8 @@ attrs==25.3.0
     # via aiohttp
     # via jsonschema
     # via referencing
+bytecode==0.17.0
+    # via ddtrace
 cachetools==5.5.2
     # via google-auth
 certifi==2023.7.22
@@ -57,6 +59,10 @@ colorama==0.4.6
     # via griffe
 comm==0.2.3
     # via ipykernel
+datadog==0.52.1
+    # via agentex-sdk
+ddtrace==3.15.0
+    # via agentex-sdk
 debugpy==1.8.16
     # via ipykernel
 decorator==5.2.1
@@ -66,6 +72,8 @@ distro==1.8.0
     # via openai
     # via scale-gp
     # via scale-gp-beta
+envier==0.6.1
+    # via ddtrace
 executing==2.2.0
     # via stack-data
 fastapi==0.115.14
@@ -109,6 +117,7 @@ idna==3.4
     # via yarl
 importlib-metadata==8.7.0
     # via litellm
+    # via opentelemetry-api
 iniconfig==2.1.0
     # via pytest
 ipykernel==6.30.1
@@ -124,6 +133,8 @@ jinja2==3.1.6
     # via litellm
 jiter==0.10.0
     # via openai
+json-log-formatter==1.1.1
+    # via agentex-sdk
 jsonref==1.1.0
     # via agentex-sdk
 jsonschema==4.25.0
@@ -169,6 +180,8 @@ openai==1.99.9
     # via openai-agents
 openai-agents==0.2.7
     # via agentex-sdk
+opentelemetry-api==1.37.0
+    # via ddtrace
 packaging==25.0
     # via huggingface-hub
     # via ipykernel
@@ -188,6 +201,7 @@ propcache==0.3.1
     # via aiohttp
     # via yarl
 protobuf==5.29.5
+    # via ddtrace
     # via temporalio
 psutil==7.0.0
     # via ipykernel
@@ -255,6 +269,7 @@ referencing==0.36.2
 regex==2025.7.34
     # via tiktoken
 requests==2.32.4
+    # via datadog
     # via huggingface-hub
     # via kubernetes
     # via openai-agents
@@ -333,6 +348,7 @@ typing-extensions==4.12.2
     # via nexus-rpc
     # via openai
     # via openai-agents
+    # via opentelemetry-api
     # via pydantic
     # via pydantic-core
     # via python-on-whales
@@ -360,6 +376,8 @@ wcwidth==0.2.13
     # via prompt-toolkit
 websocket-client==1.8.0
     # via kubernetes
+wrapt==1.17.3
+    # via ddtrace
 yarl==1.20.0
     # via aiohttp
 zipp==3.23.0

{agentex_sdk-0.4.17 → agentex_sdk-0.4.19}/src/agentex/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "agentex"
-__version__ = "0.4.17"  # x-release-please-version
+__version__ = "0.4.19"  # x-release-please-version

{agentex_sdk-0.4.17 → agentex_sdk-0.4.19}/src/agentex/lib/cli/handlers/deploy_handlers.py

@@ -1,22 +1,21 @@
 import os
-import subprocess
 import tempfile
-from pathlib import Path
+import subprocess
 from typing import Any
+from pathlib import Path
 
 import yaml
-from pydantic import BaseModel, Field
+from pydantic import Field, BaseModel
 from rich.console import Console
 
-from agentex.lib.cli.utils.exceptions import DeploymentError, HelmError
-from agentex.lib.sdk.config.environment_config import AgentEnvironmentConfig
-from agentex.lib.cli.utils.kubectl_utils import check_and_switch_cluster_context
-from agentex.lib.cli.utils.path_utils import calculate_docker_acp_module, PathResolutionError
+from agentex.lib.utils.logging import make_logger
+from agentex.lib.cli.utils.exceptions import HelmError, DeploymentError
+from agentex.lib.cli.utils.path_utils import PathResolutionError, calculate_docker_acp_module
 from agentex.lib.environment_variables import EnvVarKeys
+from agentex.lib.cli.utils.kubectl_utils import check_and_switch_cluster_context
 from agentex.lib.sdk.config.agent_config import AgentConfig
 from agentex.lib.sdk.config.agent_manifest import AgentManifest
-
-from agentex.lib.utils.logging import make_logger
+from agentex.lib.sdk.config.environment_config import AgentEnvironmentConfig
 
 logger = make_logger(__name__)
 console = Console()
@@ -46,7 +45,7 @@ def check_helm_installed() -> bool:
         return False
 
 
-def add_helm_repo() -> None:
+def add_helm_repo(helm_repository_name: str, helm_repository_url: str) -> None:
     """Add the agentex helm repository if not already added"""
     try:
         # Check if repo already exists
@@ -54,15 +53,15 @@ def add_helm_repo() -> None:
             ["helm", "repo", "list"], capture_output=True, text=True, check=True
         )
 
-        if "scale-egp" not in result.stdout:
+        if helm_repository_name not in result.stdout:
             console.print("Adding agentex helm repository...")
             subprocess.run(
                 [
                     "helm",
                     "repo",
                     "add",
-                    "scale-egp",
-                    "https://scale-egp-helm-charts-us-west-2.s3.amazonaws.com/charts",
+                    helm_repository_name,
+                    helm_repository_url,
                 ],
                 check=True,
             )
@@ -265,7 +264,7 @@ def merge_deployment_configs(
     if not helm_overrides_command:
         add_acp_command_to_helm_values(helm_values, manifest, manifest_path)
     
-    print("Deploying with the following helm values: ", helm_values)
+    console.print("Deploying with the following helm values: ", helm_values)
     return helm_values
 
 
@@ -318,8 +317,14 @@ def deploy_agent(
         else:
             console.print(f"[yellow]⚠[/yellow] No environments.yaml found, skipping environment-specific config")
 
+    if agent_env_config:
+        helm_repository_name = agent_env_config.helm_repository_name
+        helm_repository_url = agent_env_config.helm_repository_url
+    else:
+        helm_repository_name = "scale-egp"
+        helm_repository_url = "https://scale-egp-helm-charts-us-west-2.s3.amazonaws.com/charts"
     # Add helm repository/update
-    add_helm_repo()
+    add_helm_repo(helm_repository_name, helm_repository_url)
 
     # Merge configurations
     helm_values = merge_deployment_configs(manifest, agent_env_config, deploy_overrides, manifest_path)
@@ -349,7 +354,7 @@ def deploy_agent(
                 "helm",
                 "upgrade",
                 release_name,
-                "scale-egp/agentex-agent",
+                f"{helm_repository_name}/agentex-agent",
                 "--version",
                 AGENTEX_AGENTS_HELM_CHART_VERSION,
                 "-f",
@@ -371,7 +376,7 @@ def deploy_agent(
                 "helm",
                 "install",
                 release_name,
-                "scale-egp/agentex-agent",
+                f"{helm_repository_name}/agentex-agent",
                 "--version",
                 AGENTEX_AGENTS_HELM_CHART_VERSION,
                 "-f",

{agentex_sdk-0.4.17 → agentex_sdk-0.4.19}/src/agentex/lib/core/services/adk/providers/openai.py

@@ -358,6 +358,7 @@ class OpenAIService:
             },
         ) as span:
             heartbeat_if_in_workflow("run agent auto send")
+
             async with mcp_server_context(mcp_server_params, mcp_timeout_seconds) as servers:
                 tools = [tool.to_oai_function_tool() for tool in tools] if tools else []
                 handoffs = [Agent(**handoff.model_dump()) for handoff in handoffs] if handoffs else []
@@ -395,9 +396,12 @@ class OpenAIService:
                     result = await Runner.run(
                         starting_agent=agent, input=input_list, previous_response_id=previous_response_id
                     )
-                            item.raw_item.model_dump()
-                            if isinstance(item.raw_item, BaseModel)
-                            else item.raw_item
+                else:
+                    result = await Runner.run(starting_agent=agent, input=input_list)
+
+                if span:
+                    span.output = {
+                        "new_items": [
                             item.raw_item.model_dump() if isinstance(item.raw_item, BaseModel) else item.raw_item
                             for item in result.new_items
                         ],
@@ -427,6 +431,7 @@ class OpenAIService:
 
                     elif item.type == "tool_call_item":
                         tool_call_item = item.raw_item
+
                         # Extract tool call information using the helper method
                         call_id, tool_name, tool_arguments = self._extract_tool_call_info(tool_call_item)
                         tool_call_map[call_id] = tool_call_item
@@ -552,15 +557,9 @@ class OpenAIService:
         ) as span:
             heartbeat_if_in_workflow("run agent streamed")
 
-            async with mcp_server_context(
-                mcp_server_params, mcp_timeout_seconds
-            ) as servers:
+            async with mcp_server_context(mcp_server_params, mcp_timeout_seconds) as servers:
                 tools = [tool.to_oai_function_tool() for tool in tools] if tools else []
-                handoffs = (
-                    [Agent(**handoff.model_dump()) for handoff in handoffs]
-                    if handoffs
-                    else []
-                )
+                handoffs = [Agent(**handoff.model_dump()) for handoff in handoffs] if handoffs else []
                 agent_kwargs = {
                     "name": agent_name,
                     "instructions": agent_instructions,
@@ -573,9 +572,7 @@ class OpenAIService:
                     "tool_use_behavior": tool_use_behavior,
                 }
                 if model_settings is not None:
-                    agent_kwargs["model_settings"] = (
-                        model_settings.to_oai_model_settings()
-                    )
+                    agent_kwargs["model_settings"] = model_settings.to_oai_model_settings()
                 if input_guardrails is not None:
                     agent_kwargs["input_guardrails"] = input_guardrails
                 if output_guardrails is not None:
@@ -603,9 +600,7 @@ class OpenAIService:
                 if span:
                     span.output = {
                         "new_items": [
-                            item.raw_item.model_dump()
-                            if isinstance(item.raw_item, BaseModel)
-                            else item.raw_item
+                            item.raw_item.model_dump() if isinstance(item.raw_item, BaseModel) else item.raw_item
                             for item in result.new_items
                         ],
                         "final_output": result.final_output,
@@ -738,6 +733,7 @@ class OpenAIService:
                         if event.type == "run_item_stream_event":
                             if event.item.type == "tool_call_item":
                                 tool_call_item = event.item.raw_item
+
                                 # Extract tool call information using the helper method
                                 call_id, tool_name, tool_arguments = self._extract_tool_call_info(tool_call_item)
                                 tool_call_map[call_id] = tool_call_item
@@ -750,12 +746,10 @@ class OpenAIService:
                                 )
 
                                 # Create tool request using streaming context (immediate completion)
-                                async with (
-                                    self.streaming_service.streaming_task_message_context(
-                                        task_id=task_id,
-                                        initial_content=tool_request_content,
-                                    ) as streaming_context
-                                ):
+                                async with self.streaming_service.streaming_task_message_context(
+                                    task_id=task_id,
+                                    initial_content=tool_request_content,
+                                ) as streaming_context:
                                     # The message has already been persisted, but we still need to send an upda
                                     await streaming_context.stream_update(
                                         update=StreamTaskMessageFull(
@@ -781,12 +775,9 @@ class OpenAIService:
                                 )
 
                                 # Create tool response using streaming context (immediate completion)
-                                async with (
-                                    self.streaming_service.streaming_task_message_context(
-                                        task_id=task_id,
-                                        initial_content=tool_response_content
-                                    ) as streaming_context
-                                ):
+                                async with self.streaming_service.streaming_task_message_context(
+                                    task_id=task_id, initial_content=tool_response_content
+                                ) as streaming_context:
                                     # The message has already been persisted, but we still need to send an update
                                     await streaming_context.stream_update(
                                         update=StreamTaskMessageFull(
@@ -812,14 +803,10 @@ class OpenAIService:
                                         ),
                                     )
                                     # Open the streaming context
-                                    item_id_to_streaming_context[
-                                        item_id
-                                    ] = await streaming_context.open()
+                                    item_id_to_streaming_context[item_id] = await streaming_context.open()
                                     unclosed_item_ids.add(item_id)
                                 else:
-                                    streaming_context = item_id_to_streaming_context[
-                                        item_id
-                                    ]
+                                    streaming_context = item_id_to_streaming_context[item_id]
 
                                 # Stream the delta through the streaming service
                                 await streaming_context.stream_update(
@@ -849,14 +836,10 @@ class OpenAIService:
                                         ),
                                     )
                                     # Open the streaming context
-                                    item_id_to_streaming_context[
-                                        item_id
-                                    ] = await streaming_context.open()
+                                    item_id_to_streaming_context[item_id] = await streaming_context.open()
                                     unclosed_item_ids.add(item_id)
                                 else:
-                                    streaming_context = item_id_to_streaming_context[
-                                        item_id
-                                    ]
+                                    streaming_context = item_id_to_streaming_context[item_id]
 
                                 # Stream the summary delta through the streaming service
                                 await streaming_context.stream_update(
@@ -890,14 +873,10 @@ class OpenAIService:
                                         ),
                                     )
                                     # Open the streaming context
-                                    item_id_to_streaming_context[
-                                        item_id
-                                    ] = await streaming_context.open()
+                                    item_id_to_streaming_context[item_id] = await streaming_context.open()
                                     unclosed_item_ids.add(item_id)
                                 else:
-                                    streaming_context = item_id_to_streaming_context[
-                                        item_id
-                                    ]
+                                    streaming_context = item_id_to_streaming_context[item_id]
 
                                 # Stream the content delta through the streaming service
                                 await streaming_context.stream_update(
@@ -925,6 +904,7 @@ class OpenAIService:
                                 # to close the streaming context, but they do!!!
                                 # They output both a ResponseReasoningSummaryTextDoneEvent and a ResponseReasoningSummaryPartDoneEvent
                                 # I have no idea why they do this.
+
                             elif isinstance(event.data, ResponseReasoningTextDoneEvent):
                                 # Handle reasoning content text completion
                                 item_id = event.data.item_id
@@ -940,9 +920,7 @@ class OpenAIService:
 
                                 # Finish the streaming context (sends DONE event and updates message)
                                 if item_id in item_id_to_streaming_context:
-                                    streaming_context = item_id_to_streaming_context[
-                                        item_id
-                                    ]
+                                    streaming_context = item_id_to_streaming_context[item_id]
                                     await streaming_context.close()
                                     if item_id in unclosed_item_ids:
                                         unclosed_item_ids.remove(item_id)
@@ -952,17 +930,17 @@ class OpenAIService:
                                 # Create a copy to avoid modifying set during iteration
                                 remaining_items = list(unclosed_item_ids)
                                 for item_id in remaining_items:
-                                    if (item_id in unclosed_item_ids and
-                                            item_id in item_id_to_streaming_context):  # Check if still unclosed
-                                        streaming_context = item_id_to_streaming_context[
-                                            item_id
-                                        ]
+                                    if (
+                                        item_id in unclosed_item_ids and item_id in item_id_to_streaming_context
+                                    ):  # Check if still unclosed
+                                        streaming_context = item_id_to_streaming_context[item_id]
                                         await streaming_context.close()
                                         unclosed_item_ids.discard(item_id)
 
                 except InputGuardrailTripwireTriggered as e:
                     # Handle guardrail trigger by sending a rejection message
                     rejection_message = "I'm sorry, but I cannot process this request due to a guardrail. Please try a different question."
+
                     # Try to extract rejection message from the guardrail result
                     if hasattr(e, "guardrail_result") and hasattr(e.guardrail_result, "output"):
                         output_info = getattr(e.guardrail_result.output, "output_info", {})
@@ -993,6 +971,7 @@ class OpenAIService:
                                 type="full",
                             ),
                         )
+
                     # Re-raise to let the activity handle it
                     raise
 
@@ -1030,6 +1009,7 @@ class OpenAIService:
                                 type="full",
                             ),
                         )
+
                     # Re-raise to let the activity handle it
                     raise
 

{agentex_sdk-0.4.17 → agentex_sdk-0.4.19}/src/agentex/lib/environment_variables.py

@@ -39,6 +39,7 @@ class EnvVarKeys(str, Enum):
 
 
 class Environment(str, Enum):
+    LOCAL = "local"
     DEV = "development"
     STAGING = "staging"
     PROD = "production"

{agentex_sdk-0.4.17 → agentex_sdk-0.4.19}/src/agentex/lib/sdk/config/environment_config.py

@@ -7,11 +7,11 @@ configurations that are separate from the main manifest.yaml file.
 
 from __future__ import annotations
 
-from pathlib import Path
 from typing import Any, Dict, override
+from pathlib import Path
 
 import yaml
-from pydantic import BaseModel, Field, field_validator
+from pydantic import Field, BaseModel, field_validator
 
 from agentex.lib.utils.model_utils import BaseModel as UtilsBaseModel
 
@@ -73,6 +73,14 @@ class AgentEnvironmentConfig(BaseModel):
         ..., 
         description="Authentication and authorization configuration"
     )
+    helm_repository_name: str = Field(
+        default="scale-egp", 
+        description="Helm repository name for the environment"
+    )
+    helm_repository_url: str = Field(
+        default="https://scale-egp-helm-charts-us-west-2.s3.amazonaws.com/charts", 
+        description="Helm repository url for the environment"
+    )
     helm_overrides: Dict[str, Any] = Field(
         default_factory=dict, 
         description="Helm chart value overrides for environment-specific tuning"