uipath-langchain 0.0.95__tar.gz → 0.0.97__tar.gz

{uipath_langchain-0.0.95 → uipath_langchain-0.0.97}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-langchain
-Version: 0.0.95
+Version: 0.0.97
 Summary: UiPath Langchain
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-langchain-python
@@ -25,7 +25,7 @@ Requires-Dist: pydantic-settings>=2.6.0
 Requires-Dist: python-dotenv>=1.0.1
 Requires-Dist: requests>=2.23.3
 Requires-Dist: types-requests>=2.32.0.20241016
-Requires-Dist: uipath<2.1.0,>=2.0.23
+Requires-Dist: uipath<2.1.0,>=2.0.28
 Provides-Extra: langchain
 Description-Content-Type: text/markdown
 

{uipath_langchain-0.0.95 → uipath_langchain-0.0.97}/docs/chat_models.md

@@ -1,6 +1,6 @@
 # UiPath chat models
 
-UiPath provides two chat models `UiPathAzureChatOpenAI` and `UiPathNormalizedChatModel`. These are compatible with langgraph as drop in replacements. You do not need to add tokens from OpenAI or Anthropic, usage of these chat models will consume `AI Units` on your account.
+UiPath provides two chat models `UiPathAzureChatOpenAI` and `UiPathChat`. These are compatible with langgraph as drop in replacements. You do not need to add tokens from OpenAI or Anthropic, usage of these chat models will consume `AI Units` on your account.
 
 ## UiPathAzureChatOpenAI
 
@@ -15,8 +15,8 @@ from langchain_openai import ChatOpenAI
 llm = ChatOpenAI(
     model="gpt-4o",
     temperature=0,
-    max_tokens=None,
-    timeout=None,
+    max_tokens=4000,
+    timeout=30,
     max_retries=2,
     # api_key="...",  # if you prefer to pass api key in directly instaed of using env vars
     # base_url="...",
@@ -32,10 +32,10 @@ You can simply change `ChatOpenAi` with `UiPathAzureChatOpenAI`, you don't have
 from uipath_langchain.chat.models import UiPathAzureChatOpenAI
 
 llm = UiPathAzureChatOpenAI(
-    model="gpt-4o",
+    model="gpt-4o-2024-08-06",
     temperature=0,
-    max_tokens=None,
-    timeout=None,
+    max_tokens=4000,
+    timeout=30,
     max_retries=2,
     # other params...
 )
@@ -45,9 +45,9 @@ Currently the following models can be used with `UiPathAzureChatOpenAI` (this li
 - `gpt-4`, `gpt-4-1106-Preview`, `gpt-4-32k`, `gpt-4-turbo-2024-04-09`, `gpt-4-vision-preview`, `gpt-4o-2024-05-13`, `gpt-4o-2024-08-06`, `gpt-4o-mini-2024-07-18`, `o3-mini-2025-01-31`
 
 
-## UiPathNormalizedChatModel
+## UiPathChat
 
-`UiPathNormalizedChatModel` is a more versatile clas that can suport models from diferent vendors including OpenAI.
+`UiPathChat` is a more versatile clas that can suport models from diferent vendors including OpenAI.
 
 ### Example usage
 
@@ -66,12 +66,12 @@ llm = ChatAnthropic(
 )
 ```
 
-You can replace it with `UiPathNormalizedChatModel` like so:
+You can replace it with `UiPathChat` like so:
 
 ```python
-from uipath_langchain.chat.models import UiPathNormalizedChatModel
+from uipath_langchain.chat.models import UiPathChat
 
-llm = UiPathNormalizedChatModel(
+llm = UiPathChat(
     model="anthropic.claude-3-opus-20240229-v1:0",
     temperature=0,
     max_tokens=1024,
@@ -81,7 +81,7 @@ llm = UiPathNormalizedChatModel(
 )
 ```
 
-Currently the following models can be used with `UiPathNormalizedChatModel` (this list can be updated in the future): 
+Currently the following models can be used with `UiPathChat` (this list can be updated in the future): 
 - `anthropic.claude-3-5-sonnet-20240620-v1:0`, `anthropic.claude-3-5-sonnet-20241022-v2:0`, `anthropic.claude-3-7-sonnet-20250219-v1:0`, `anthropic.claude-3-haiku-20240307-v1:0`, `gemini-1.5-pro-001`, `gemini-2.0-flash-001`, `gpt-4o-2024-05-13`, `gpt-4o-2024-08-06`, `gpt-4o-2024-11-20`, `gpt-4o-mini-2024-07-18`, `o3-mini-2025-01-31`
 
 ### Note 

{uipath_langchain-0.0.95 → uipath_langchain-0.0.97}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "uipath-langchain"
-version = "0.0.95"
+version = "0.0.97"
 description = "UiPath Langchain"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
 dependencies = [
-    "uipath>=2.0.23, <2.1.0",
+    "uipath>=2.0.28, <2.1.0",
     "langgraph>=0.2.70",
     "langchain-core>=0.3.34",
     "langgraph-checkpoint-sqlite>=2.0.3",

{uipath_langchain-0.0.95 → uipath_langchain-0.0.97}/src/uipath_langchain/_cli/cli_init.py

@@ -4,14 +4,15 @@ import os
 import uuid
 from typing import Any, Dict
 
-import click
 from langgraph.graph.state import CompiledStateGraph
+from uipath._cli._utils._console import ConsoleLogger
 from uipath._cli._utils._parse_ast import generate_bindings_json  # type: ignore
 from uipath._cli.middlewares import MiddlewareResult
-from uipath._cli.spinner import Spinner
 
 from ._utils._graph import LangGraphConfig
 
+console = ConsoleLogger()
+
 
 def resolve_refs(schema, root=None):
     """Recursively resolves $ref references in a JSON schema."""
@@ -95,7 +96,6 @@ def generate_schema_from_graph(graph: CompiledStateGraph) -> Dict[str, Any]:
 
 async def langgraph_init_middleware_async(entrypoint: str) -> MiddlewareResult:
     """Middleware to check for langgraph.json and create uipath.json with schemas"""
-    spinner = Spinner("Initializing UiPath project...")
     config = LangGraphConfig()
     if not config.exists:
         return MiddlewareResult(
@@ -103,7 +103,6 @@ async def langgraph_init_middleware_async(entrypoint: str) -> MiddlewareResult:
         )  # Continue with normal flow if no langgraph.json
 
     try:
-        spinner.start()
         config.load_config()
         entrypoints = []
         all_bindings = {"version": "2.0", "resources": []}
@@ -134,8 +133,8 @@ async def langgraph_init_middleware_async(entrypoint: str) -> MiddlewareResult:
                         if "resources" in file_bindings:
                             all_bindings["resources"] = file_bindings["resources"]
                 except Exception as e:
-                    print(
-                        f"⚠️ Warning: Could not generate bindings for {graph.file_path}: {str(e)}"
+                    console.warning(
+                        f"Warning: Could not generate bindings for {graph.file_path}: {str(e)}"
                     )
 
                 new_entrypoint: dict[str, Any] = {
@@ -148,21 +147,18 @@ async def langgraph_init_middleware_async(entrypoint: str) -> MiddlewareResult:
                 entrypoints.append(new_entrypoint)
 
             except Exception as e:
-                spinner.stop()
-                print(f"Error during graph load: {e}")
+                console.error(f"Error during graph load: {e}")
                 return MiddlewareResult(
                     should_continue=False,
-                    error_message=f"❌ Failed to load graph '{graph.name}': {str(e)}",
                     should_include_stacktrace=True,
                 )
             finally:
                 await graph.cleanup()
 
         if entrypoint and not entrypoints:
-            spinner.stop()
+            console.error(f"Error: No graph found with name '{entrypoint}'")
             return MiddlewareResult(
                 should_continue=False,
-                error_message=f"❌ Error: No graph found with name '{entrypoint}'",
             )
 
         uipath_config = {"entryPoints": entrypoints, "bindings": all_bindings}
@@ -177,25 +173,22 @@ async def langgraph_init_middleware_async(entrypoint: str) -> MiddlewareResult:
             try:
                 with open(mermaid_file_path, "w") as f:
                     f.write(mermaid_content)
+                console.success(f" Created '{mermaid_file_path}' file.")
             except Exception as write_error:
-                spinner.stop()
+                console.error(
+                    f"Error writing mermaid file for '{graph_name}': {str(write_error)}"
+                )
                 return MiddlewareResult(
                     should_continue=False,
-                    error_message=f"❌ Error writing mermaid file for '{graph_name}': {str(write_error)}",
                     should_include_stacktrace=True,
                 )
-        spinner.stop()
-        return MiddlewareResult(
-            should_continue=False,
-            info_message=click.style("✓ ", fg="green", bold=True)
-            + f" Configuration file {config_path} created successfully.",
-        )
+        console.success(f" Created '{config_path}' file.")
+        return MiddlewareResult(should_continue=False)
 
     except Exception as e:
-        spinner.stop()
+        console.error(f"Error processing langgraph configuration: {str(e)}")
         return MiddlewareResult(
             should_continue=False,
-            error_message=f"❌ Error processing langgraph configuration: {str(e)}",
             should_include_stacktrace=True,
         )
 

{uipath_langchain-0.0.95 → uipath_langchain-0.0.97}/src/uipath_langchain/_cli/cli_new.py

@@ -2,8 +2,10 @@ import os
 import shutil
 
 import click
+from uipath._cli._utils._console import ConsoleLogger
 from uipath._cli.middlewares import MiddlewareResult
-from uipath._cli.spinner import Spinner
+
+console = ConsoleLogger()
 
 
 def generate_script(target_directory):
@@ -41,36 +43,29 @@ def generate_pyproject(target_directory, project_name):
 
 def langgraph_new_middleware(name: str) -> MiddlewareResult:
     """Middleware to create demo langchain agent"""
-    spinner = Spinner("Creating demo agent...")
+
     directory = os.getcwd()
 
     try:
-        generate_script(directory)
-        click.echo(click.style("✓ ", fg="green", bold=True) + "Created main.py file")
-        click.echo(
-            click.style("✓ ", fg="green", bold=True) + "Created langgraph.json file"
-        )
-        generate_pyproject(directory, name)
-        click.echo(
-            click.style("✓ ", fg="green", bold=True) + "Created pyproject.toml file"
-        )
-        os.system("uv sync")
-        spinner.start()
-        ctx = click.get_current_context()
-        init_cmd = ctx.parent.command.get_command(ctx, "init")  # type: ignore
-        ctx.invoke(init_cmd)
-        spinner.stop()
-        click.echo(
-            click.style("✓ ", fg="green", bold=True) + " Agent created successfully."
-        )
-        return MiddlewareResult(
-            should_continue=False,
-            info_message="""Usage example: ` uipath run agent '{"topic": "UiPath"}' `""",
-        )
+        with console.spinner(f"Creating new agent {name} in current directory ..."):
+            generate_pyproject(directory, name)
+            generate_script(directory)
+            console.success("Created 'main.py' file.")
+            console.success("Created 'langgraph.json' file.")
+            generate_pyproject(directory, name)
+            console.success("Created 'pyproject.toml' file.")
+            os.system("uv sync")
+            ctx = click.get_current_context()
+            init_cmd = ctx.parent.command.get_command(ctx, "init")  # type: ignore
+            ctx.invoke(init_cmd)
+            console.config(
+                " Please ensure to define either ANTHROPIC_API_KEY or OPENAI_API_KEY in your .env file."
+            )
+            console.hint(""" Run agent: uipath run agent '{"topic": "UiPath"}'""")
+        return MiddlewareResult(should_continue=False)
     except Exception as e:
-        spinner.stop()
+        console.error(f"Error creating demo agent {str(e)}")
         return MiddlewareResult(
             should_continue=False,
-            error_message=f"❌ Error creating demo agent {str(e)}",
             should_include_stacktrace=True,
         )

uipath_langchain-0.0.97/src/uipath_langchain/chat/__init__.py

@@ -0,0 +1,6 @@
+from .models import UiPathAzureChatOpenAI, UiPathChat
+
+__all__ = [
+    "UiPathChat",
+    "UiPathAzureChatOpenAI",
+]

{uipath_langchain-0.0.95 → uipath_langchain-0.0.97}/src/uipath_langchain/chat/models.py

@@ -76,7 +76,7 @@ class UiPathAzureChatOpenAI(UiPathRequestMixin, AzureChatOpenAI):
         )
 
 
-class UiPathNormalizedChatModel(UiPathRequestMixin, AzureChatOpenAI):
+class UiPathChat(UiPathRequestMixin, AzureChatOpenAI):
     """Custom LLM connector for LangChain integration with UiPath Normalized."""
 
     def _create_chat_result(

{uipath_langchain-0.0.95 → uipath_langchain-0.0.97}/tests/test_langchain_client.py

@@ -18,7 +18,7 @@ from pydantic import BaseModel, ValidationError
 
 from uipath_langchain.chat import (
     UiPathAzureChatOpenAI,
-    UiPathNormalizedChatModel,
+    UiPathChat,
 )
 from uipath_langchain.embeddings import (
     UiPathAzureOpenAIEmbeddings,
@@ -131,7 +131,7 @@ def test_cached_call_tokens(cached_llmgw_calls: Optional[SQLiteCache]):
 
 
 def test_normalized_cached_call(cached_llmgw_calls: Optional[SQLiteCache]):
-    model = UiPathNormalizedChatModel(
+    model = UiPathChat(
         model="anthropic.claude-3-5-sonnet-20240620-v1:0", cache=cached_llmgw_calls
     )
     messages = [
@@ -143,7 +143,7 @@ def test_normalized_cached_call(cached_llmgw_calls: Optional[SQLiteCache]):
 
 
 def test_normalized_cached_call_tokens(cached_llmgw_calls: Optional[SQLiteCache]):
-    model = UiPathNormalizedChatModel(
+    model = UiPathChat(
         model="anthropic.claude-3-5-sonnet-20240620-v1:0", cache=cached_llmgw_calls
     )
     messages = [