fix: uipath new

Author: cosmyo

packages/uipath-openai-agents/src/uipath_openai_agents/__init__.py

@@ -1,7 +1,28 @@
-"""UiPath OpenAI Agents SDK."""
+"""
+UiPath OpenAI Agents SDK.
+
+NOTE: This module uses lazy imports via __getattr__ to avoid loading heavy
+dependencies (openai SDK) at import time. This significantly improves CLI
+startup performance.
+
+Do NOT add eager imports like:
+    from .chat import UiPathChatOpenAI  # BAD - loads openai SDK immediately
+
+Instead, all exports are loaded on-demand when first accessed.
+"""
+
+
+def __getattr__(name):
+    if name == "UiPathChatOpenAI":
+        from .chat import UiPathChatOpenAI
+
+        return UiPathChatOpenAI
+    if name == "register_middleware":
+        from .middlewares import register_middleware
+
+        return register_middleware
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
 
-from .chat import UiPathChatOpenAI
-from .middlewares import register_middleware
 
 __version__ = "0.1.0"
 __all__ = ["register_middleware", "UiPathChatOpenAI"]

packages/uipath-openai-agents/src/uipath_openai_agents/_cli/_templates/main.py.template

@@ -1,7 +1,11 @@
-from agents import Agent
-from openai import OpenAI
+from agents import Agent, function_tool
+from agents.models import _openai_shared
 
+from uipath_openai_agents.chat import UiPathChatOpenAI
+from uipath_openai_agents.chat.supported_models import OpenAIModels
 
+
+@function_tool
 def get_weather(location: str) -> str:
     """Get the current weather for a location.
 
@@ -12,17 +16,19 @@ def get_weather(location: str) -> str:
         Weather information for the location
     """
     # This is a mock implementation
-    return f"The weather in {location} is sunny and 72�F"
+    return f"The weather in {location} is sunny and 32 degrees Celsius."
+
 
+MODEL = OpenAIModels.gpt_5_2_2025_12_11
 
-# Initialize the OpenAI client
-client = OpenAI()
+# Initialize the UiPath OpenAI client
+uipath_openai_client = UiPathChatOpenAI(model_name=MODEL)
+_openai_shared.set_default_openai_client(uipath_openai_client.async_client)
 
 # Create an agent with tools
 agent = Agent(
     name="weather_agent",
     instructions="You are a helpful weather assistant. Use the get_weather tool to provide weather information.",
-    model="gpt-4o-mini",
+    model=MODEL,
     tools=[get_weather],
-    client=client,
 )

packages/uipath-openai-agents/src/uipath_openai_agents/_cli/cli_new.py

@@ -39,8 +39,7 @@ def generate_pyproject(target_directory, project_name):
 description = "{project_name}"
 authors = [{{ name = "John Doe", email = "john.doe@myemail.com" }}]
 dependencies = [
-    "uipath-openai-agents>=0.1.0",
-    "openai>=1.0.0"
+    "uipath-openai-agents>=0.0.1, <0.1.0",
 ]
 requires-python = ">=3.11"
 """
@@ -63,9 +62,6 @@ def openai_agents_new_middleware(name: str) -> MiddlewareResult:
             console.success("Created 'AGENTS.md' file.")
             generate_pyproject(directory, name)
             console.success("Created 'pyproject.toml' file.")
-            console.config(
-                f""" Please ensure to define {click.style("OPENAI_API_KEY", fg="bright_yellow")} in your .env file. """
-            )
             init_command = """uipath init"""
             run_command = """uipath run agent '{"message": "What is the weather in San Francisco?"}'"""
             console.hint(

packages/uipath-openai-agents/src/uipath_openai_agents/chat/__init__.py

@@ -1,5 +1,32 @@
-"""UiPath OpenAI Chat models."""
+"""
+UiPath OpenAI Chat module.
 
-from .openai import UiPathChatOpenAI
+NOTE: This module uses lazy imports via __getattr__ to avoid loading heavy
+dependencies (openai SDK, httpx) at import time. This significantly
+improves CLI startup performance.
 
-__all__ = ["UiPathChatOpenAI"]
+Do NOT add eager imports like:
+    from .openai import UiPathChatOpenAI  # BAD - loads openai SDK immediately
+
+Instead, all exports are loaded on-demand when first accessed.
+"""
+
+
+def __getattr__(name):
+    if name == "UiPathChatOpenAI":
+        from .openai import UiPathChatOpenAI
+
+        return UiPathChatOpenAI
+    if name in ("OpenAIModels", "GeminiModels", "BedrockModels"):
+        from . import supported_models
+
+        return getattr(supported_models, name)
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
+__all__ = [
+    "UiPathChatOpenAI",
+    "OpenAIModels",
+    "GeminiModels",
+    "BedrockModels",
+]

packages/uipath-openai-agents/src/uipath_openai_agents/chat/supported_models.py

@@ -1,78 +1,60 @@
-"""Supported OpenAI model definitions for UiPath LLM Gateway."""
+"""Supported model definitions for UiPath LLM Gateway.
 
+OpenAI Agents SDK supports other LLM providers (Bedrock, Gemini, etc.)
+through LiteLLM integration, so we provide model definitions for all supported providers.
+"""
 
-class OpenAIModels:
-    """OpenAI model names supported by UiPath LLM Gateway.
+from enum import StrEnum
 
-    These are specific model versions required by UiPath.
-    Generic names like "gpt-4o" are not supported - use specific versions.
-    """
 
-    # GPT-4o Models (recommended)
-    gpt_4o_2024_11_20 = "gpt-4o-2024-11-20"
-    gpt_4o_2024_08_06 = "gpt-4o-2024-08-06"
+class OpenAIModels(StrEnum):
+    """Supported OpenAI model identifiers."""
+
+    # GPT-4o models
     gpt_4o_2024_05_13 = "gpt-4o-2024-05-13"
+    gpt_4o_2024_08_06 = "gpt-4o-2024-08-06"
+    gpt_4o_2024_11_20 = "gpt-4o-2024-11-20"
     gpt_4o_mini_2024_07_18 = "gpt-4o-mini-2024-07-18"
 
-    # GPT-4.1 Models
+    # GPT-4.1 models
     gpt_4_1_2025_04_14 = "gpt-4.1-2025-04-14"
     gpt_4_1_mini_2025_04_14 = "gpt-4.1-mini-2025-04-14"
     gpt_4_1_nano_2025_04_14 = "gpt-4.1-nano-2025-04-14"
 
-    # GPT-4 Models
-    gpt_4 = "gpt-4"
-    gpt_4_32k = "gpt-4-32k"
-    gpt_4_turbo_2024_04_09 = "gpt-4-turbo-2024-04-09"
-    gpt_4_1106_preview = "gpt-4-1106-Preview"
-    gpt_4_vision_preview = "gpt-4-vision-preview"
-
-    # GPT-3.5 Models
-    gpt_35_turbo = "gpt-35-turbo"
-    gpt_35_turbo_0125 = "gpt-35-turbo-0125"
-    gpt_35_turbo_1106 = "gpt-35-turbo-1106"
-    gpt_35_turbo_16k = "gpt-35-turbo-16k"
-
-    # GPT-5 Models
+    # GPT-5 models
     gpt_5_2025_08_07 = "gpt-5-2025-08-07"
     gpt_5_chat_2025_08_07 = "gpt-5-chat-2025-08-07"
     gpt_5_mini_2025_08_07 = "gpt-5-mini-2025-08-07"
     gpt_5_nano_2025_08_07 = "gpt-5-nano-2025-08-07"
+
+    # GPT-5.1 models
     gpt_5_1_2025_11_13 = "gpt-5.1-2025-11-13"
+
+    # GPT-5.2 models
     gpt_5_2_2025_12_11 = "gpt-5.2-2025-12-11"
 
-    # o3 Models
-    o3_mini_2025_01_31 = "o3-mini-2025-01-31"
-
-    # Other Models
-    computer_use_preview_2025_03_11 = "computer-use-preview-2025-03-11"
-    text_davinci_003 = "text-davinci-003"
-
-    # Embedding Models
-    text_embedding_3_large = "text-embedding-3-large"
-    text_embedding_3_large_community_ecs = "text-embedding-3-large-community-ecs"
-    text_embedding_ada_002 = "text-embedding-ada-002"
-
-    # Model aliases - maps generic names to specific versions
-    MODEL_ALIASES = {
-        # Map gpt-4.1 variants to gpt-4o (most capable available model)
-        "gpt-4.1": gpt_4o_2024_11_20,
-        "gpt-4.1-mini": gpt_4o_mini_2024_07_18,
-        "gpt-4.1-nano": gpt_4o_mini_2024_07_18,
-        "gpt-4.1-2025-04-14": gpt_4o_2024_11_20,  # Map invalid model to valid one
-        "gpt-4.1-mini-2025-04-14": gpt_4o_mini_2024_07_18,
-        "gpt-4.1-nano-2025-04-14": gpt_4o_mini_2024_07_18,
-        # Generic model mappings
-        "gpt-4o": gpt_4o_2024_11_20,
-        "gpt-4o-mini": gpt_4o_mini_2024_07_18,
-        "gpt-5": gpt_5_2025_08_07,
-        "gpt-5-mini": gpt_5_mini_2025_08_07,
-        "gpt-5-nano": gpt_5_nano_2025_08_07,
-        "gpt-5.1": gpt_5_1_2025_11_13,
-        "gpt-5.2": gpt_5_2_2025_12_11,
-        "o3-mini": o3_mini_2025_01_31,
-    }
-
-    @classmethod
-    def normalize_model_name(cls, model_name: str) -> str:
-        """Normalize a model name to UiPath-specific version."""
-        return cls.MODEL_ALIASES.get(model_name, model_name)
+
+class GeminiModels(StrEnum):
+    """Supported Google Gemini model identifiers."""
+
+    # Gemini 2 models
+    gemini_2_5_pro = "gemini-2.5-pro"
+    gemini_2_5_flash = "gemini-2.5-flash"
+    gemini_2_0_flash_001 = "gemini-2.0-flash-001"
+
+    # Gemini 3 models
+    gemini_3_pro_preview = "gemini-3-pro-preview"
+
+
+class BedrockModels(StrEnum):
+    """Supported AWS Bedrock model identifiers."""
+
+    # Claude 3.7 models
+    anthropic_claude_3_7_sonnet = "anthropic.claude-3-7-sonnet-20250219-v1:0"
+
+    # Claude 4 models
+    anthropic_claude_sonnet_4 = "anthropic.claude-sonnet-4-20250514-v1:0"
+
+    # Claude 4.5 models
+    anthropic_claude_sonnet_4_5 = "anthropic.claude-sonnet-4-5-20250929-v1:0"
+    anthropic_claude_haiku_4_5 = "anthropic.claude-haiku-4-5-20251001-v1:0"

packages/uipath-openai-agents/src/uipath_openai_agents/runtime/__init__.py

@@ -1,25 +1,29 @@
-"""UiPath OpenAI Agents Runtime."""
+"""
+UiPath OpenAI Agents Runtime.
+
+NOTE: This module uses lazy imports for most exports to avoid loading heavy
+dependencies (openai SDK) at import time. However, the runtime factory
+registration must happen eagerly for CLI discovery.
+"""
 
 from uipath.runtime import (
     UiPathRuntimeContext,
     UiPathRuntimeFactoryProtocol,
     UiPathRuntimeFactoryRegistry,
 )
 
-from uipath_openai_agents.runtime.factory import UiPathOpenAIAgentRuntimeFactory
-from uipath_openai_agents.runtime.runtime import UiPathOpenAIAgentRuntime
-from uipath_openai_agents.runtime.schema import (
-    get_agent_schema,
-    get_entrypoints_schema,
-)
-
 
 def register_runtime_factory() -> None:
     """Register the OpenAI Agents factory. Called automatically via entry point."""
 
     def create_factory(
         context: UiPathRuntimeContext | None = None,
     ) -> UiPathRuntimeFactoryProtocol:
+        # Import lazily when factory is actually created
+        from uipath_openai_agents.runtime.factory import (
+            UiPathOpenAIAgentRuntimeFactory,
+        )
+
         return UiPathOpenAIAgentRuntimeFactory(
             context=context if context else UiPathRuntimeContext(),
         )
@@ -29,8 +33,30 @@ def create_factory(
     )
 
 
+# Register factory eagerly (required for CLI discovery)
 register_runtime_factory()
 
+
+def __getattr__(name):
+    if name == "get_entrypoints_schema":
+        from .schema import get_entrypoints_schema
+
+        return get_entrypoints_schema
+    if name == "get_agent_schema":
+        from .schema import get_agent_schema
+
+        return get_agent_schema
+    if name == "UiPathOpenAIAgentRuntimeFactory":
+        from .factory import UiPathOpenAIAgentRuntimeFactory
+
+        return UiPathOpenAIAgentRuntimeFactory
+    if name == "UiPathOpenAIAgentRuntime":
+        from .runtime import UiPathOpenAIAgentRuntime
+
+        return UiPathOpenAIAgentRuntime
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
 __all__ = [
     "register_runtime_factory",
     "get_entrypoints_schema",