fix: instantiate llm inside node in langgraph scaffold template (#897)

Author: radu-mocanu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-langchain"
-version = "0.11.15"
+version = "0.11.16"
 description = "Python SDK that enables developers to build and deploy LangGraph agents to the UiPath Cloud Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

src/uipath_langchain/_cli/_templates/main.py.template

@@ -1,10 +1,12 @@
 from langchain_core.messages import HumanMessage, SystemMessage
 from langgraph.graph import START, StateGraph, END
-from uipath_langchain.chat import UiPathChat
+from uipath_langchain.chat import (
+    UiPathAzureChatOpenAI,
+    UiPathChatAnthropicBedrock,
+    UiPathChatGoogleGenerativeAI,
+)
 from pydantic import BaseModel
 
-llm = UiPathChat(model="gpt-4o-mini-2024-07-18")
-
 
 class GraphState(BaseModel):
     topic: str
@@ -15,6 +17,10 @@ class GraphOutput(BaseModel):
 
 
 async def generate_report(state: GraphState) -> GraphOutput:
+    # Choose your LLM provider by uncommenting one of the following:
+    llm = UiPathChatAnthropicBedrock(model="anthropic.claude-haiku-4-5-20251001-v1:0")
+    # llm = UiPathAzureChatOpenAI(model="gpt-4.1-mini-2025-04-14")
+    # llm = UiPathChatGoogleGenerativeAI(model="gemini-2.5-flash")
     system_prompt = "You are a report generator. Please provide a brief report based on the given topic."
     output = await llm.ainvoke(
         [SystemMessage(system_prompt), HumanMessage(state.topic)]

src/uipath_langchain/_cli/cli_new.py

@@ -31,7 +31,7 @@ def generate_pyproject(target_directory, project_name):
 description = "{project_name}"
 authors = [{{ name = "John Doe", email = "john.doe@myemail.com" }}]
 dependencies = [
-    "uipath-langchain>=0.10.0, <0.11.0",
+    "uipath-langchain[bedrock,vertex]>=0.10.0, <0.11.0",
 ]
 requires-python = ">=3.11"
 """

testcases/init-flow/expected_traces.json

@@ -14,11 +14,16 @@
       }
     },
     {
-      "name": "UiPathChat",
+      "name": [
+        "UiPathChatAnthropicBedrock",
+        "UiPathChatBedrockConverse",
+        "UiPathChatBedrock",
+        "UiPathAzureChatOpenAI",
+        "UiPathChatGoogleGenerativeAI",
+        "UiPathChat"
+      ],
       "attributes": {
-        "openinference.span.kind": "LLM",
-        "llm.provider": "uipath",
-        "llm.model_name": "*"
+        "openinference.span.kind": "LLM"
       }
     }
   ]

testcases/init-flow/pyproject.toml

@@ -5,7 +5,7 @@ description = "agent"
 authors = [{ name = "John Doe", email = "john.doe@myemail.com" }]
 dependencies = [
     "langchain-anthropic>=0.3.8",
-    "uipath-langchain",
+    "uipath-langchain[bedrock,vertex]",
 ]
 requires-python = ">=3.11"