uipath-langchain 0.0.112__py3-none-any.whl → 0.0.114__py3-none-any.whl

uipath_langchain/_cli/_templates/main.py.template

@@ -5,13 +5,13 @@ from langgraph.graph import START, StateGraph, END
 from pydantic import BaseModel
 import os
 
-class GraphInput(BaseModel):
+class GraphState(BaseModel):
     topic: str
 
 class GraphOutput(BaseModel):
     report: str
 
-async def generate_report(state: GraphInput) -> GraphOutput:
+async def generate_report(state: GraphState) -> GraphOutput:
     if os.getenv("ANTHROPIC_API_KEY"):
         llm_model = ChatAnthropic(model="claude-3-5-sonnet-latest")
     elif os.getenv("OPENAI_API_KEY"):
@@ -23,7 +23,7 @@ async def generate_report(state: GraphInput) -> GraphOutput:
     output = await llm_model.ainvoke([SystemMessage(system_prompt), HumanMessage(state.topic)])
     return GraphOutput(report=output.content)
 
-builder = StateGraph(input=GraphInput, output=GraphOutput)
+builder = StateGraph(GraphState, output=GraphOutput)
 
 builder.add_node("generate_report", generate_report)
 

uipath_langchain/_cli/cli_init.py

@@ -2,7 +2,7 @@ import asyncio
 import json
 import os
 import uuid
-from typing import Any, Dict
+from typing import Any, Callable, Dict, overload
 
 from langgraph.graph.state import CompiledStateGraph
 from uipath._cli._utils._console import ConsoleLogger
@@ -94,8 +94,14 @@ def generate_schema_from_graph(graph: CompiledStateGraph) -> Dict[str, Any]:
     return schema
 
 
-async def langgraph_init_middleware_async(entrypoint: str) -> MiddlewareResult:
+async def langgraph_init_middleware_async(
+    entrypoint: str,
+    options: dict[str, Any] | None = None,
+    write_config: Callable[[Any], str] | None = None,
+) -> MiddlewareResult:
     """Middleware to check for langgraph.json and create uipath.json with schemas"""
+    options = options or {}
+
     config = LangGraphConfig()
     if not config.exists:
         return MiddlewareResult(
@@ -125,8 +131,9 @@ async def langgraph_init_middleware_async(entrypoint: str) -> MiddlewareResult:
                 mermaids[graph.name] = compiled_graph.get_graph(xray=1).draw_mermaid()
 
                 try:
+                    should_infer_bindings = options.get("infer_bindings", True)
                     # Make sure the file path exists
-                    if os.path.exists(graph.file_path):
+                    if os.path.exists(graph.file_path) and should_infer_bindings:
                         file_bindings = generate_bindings_json(graph.file_path)
 
                         # Merge bindings
@@ -163,10 +170,13 @@ async def langgraph_init_middleware_async(entrypoint: str) -> MiddlewareResult:
 
         uipath_config = {"entryPoints": entrypoints, "bindings": all_bindings}
 
-        # Save the uipath.json file
-        config_path = "uipath.json"
-        with open(config_path, "w") as f:
-            json.dump(uipath_config, f, indent=2)
+        if write_config:
+            config_path = write_config(uipath_config)
+        else:
+            # Save the uipath.json file
+            config_path = "uipath.json"
+            with open(config_path, "w") as f:
+                json.dump(uipath_config, f, indent=4)
 
         for graph_name, mermaid_content in mermaids.items():
             mermaid_file_path = f"{graph_name}.mermaid"
@@ -193,6 +203,26 @@ async def langgraph_init_middleware_async(entrypoint: str) -> MiddlewareResult:
         )
 
 
-def langgraph_init_middleware(entrypoint: str) -> MiddlewareResult:
+@overload
+def langgraph_init_middleware(
+    entrypoint: str,
+) -> MiddlewareResult: ...
+
+
+@overload
+def langgraph_init_middleware(
+    entrypoint: str,
+    options: dict[str, Any],
+    write_config: Callable[[Any], str],
+) -> MiddlewareResult: ...
+
+
+def langgraph_init_middleware(
+    entrypoint: str,
+    options: dict[str, Any] | None = None,
+    write_config: Callable[[Any], str] | None = None,
+) -> MiddlewareResult:
     """Middleware to check for langgraph.json and create uipath.json with schemas"""
-    return asyncio.run(langgraph_init_middleware_async(entrypoint))
+    return asyncio.run(
+        langgraph_init_middleware_async(entrypoint, options, write_config)
+    )

{uipath_langchain-0.0.112.dist-info → uipath_langchain-0.0.114.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-langchain
-Version: 0.0.112
+Version: 0.0.114
 Summary: UiPath Langchain
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-langchain-python
@@ -24,7 +24,7 @@ Requires-Dist: langgraph>=0.2.70
 Requires-Dist: openai>=1.65.5
 Requires-Dist: pydantic-settings>=2.6.0
 Requires-Dist: python-dotenv>=1.0.1
-Requires-Dist: uipath<2.1.0,>=2.0.65
+Requires-Dist: uipath<2.1.0,>=2.0.71
 Provides-Extra: langchain
 Description-Content-Type: text/markdown
 

{uipath_langchain-0.0.112.dist-info → uipath_langchain-0.0.114.dist-info}/RECORD

@@ -1,7 +1,7 @@
 uipath_langchain/__init__.py,sha256=VBrvQn7d3nuOdN7zEnV2_S-uhmkjgEIlXiFVeZxZakQ,80
 uipath_langchain/middlewares.py,sha256=tre7o9DFMgWk1DJiEEUmT6_wiP-PPkWtKmG0iOyvr9c,509
 uipath_langchain/_cli/__init__.py,sha256=juqd9PbXs4yg45zMJ7BHAOPQjb7sgEbWE9InBtGZhfo,24
-uipath_langchain/_cli/cli_init.py,sha256=B5BVUA7pDvKyRRGEx5mgmeE5SJvLPM3cnLGt6a9iixY,7417
+uipath_langchain/_cli/cli_init.py,sha256=aGG0YVsR4_nDgdor4kaNS8oKe263ZOIN27sHfNiCk0c,8177
 uipath_langchain/_cli/cli_new.py,sha256=dL8-Rri6u67ZZdbb4nT38A5xD_Q3fVnG0UK9VSeKaqg,2563
 uipath_langchain/_cli/cli_run.py,sha256=rGuYp9AEJdhHclOCFPX100Mo0kaDYt6e6pIKBgbS6ek,3023
 uipath_langchain/_cli/_runtime/_context.py,sha256=wr4aNn06ReIXmetEZ6b6AnpAt64p13anQ2trZ5Bzgio,807
@@ -10,7 +10,7 @@ uipath_langchain/_cli/_runtime/_input.py,sha256=vZ8vfVxvPSaPWmIPghvNx1VRKzbalHsK
 uipath_langchain/_cli/_runtime/_output.py,sha256=yJOZPWv2FRUJWv1NRs9JmpB4QMTDXu8jrxoaKrfJvzw,9078
 uipath_langchain/_cli/_runtime/_runtime.py,sha256=SjOZkal6c1ZeZNFzW23DoAnF6LNQ2nN0dISSu2cSFhQ,13881
 uipath_langchain/_cli/_templates/langgraph.json.template,sha256=eeh391Gta_hoRgaNaZ58nW1LNvCVXA7hlAH6l7Veous,107
-uipath_langchain/_cli/_templates/main.py.template,sha256=9JEyPxwc4Ce8Dfd2UAgHgpwkkjuXwWXOQZ-65P53QuM,1195
+uipath_langchain/_cli/_templates/main.py.template,sha256=nMJQIYPlRk90iANfNVpkJ2EQX20Dxsyq92-BucEz_UM,1189
 uipath_langchain/_cli/_utils/_graph.py,sha256=P7m03i6kcLda8XVpVtppcM8GOrSW62zWcv3rCR1H5zs,7086
 uipath_langchain/_utils/__init__.py,sha256=WoY66enCygRXTh6v5B1UrRcFCnQYuPJ8oqDkwomXzLc,194
 uipath_langchain/_utils/_request_mixin.py,sha256=t_1HWBxqEl-wsSk9ubmIM-8vs9BlNy4ZVBxtDxktn6U,18489
@@ -29,8 +29,8 @@ uipath_langchain/tracers/_instrument_traceable.py,sha256=0e841zVzcPWjOGtmBx0GeHb
 uipath_langchain/tracers/_utils.py,sha256=JOT1tKMdvqjMDtj2WbmbOWMeMlTXBWavxWpogX7KlRA,1543
 uipath_langchain/vectorstores/__init__.py,sha256=w8qs1P548ud1aIcVA_QhBgf_jZDrRMK5Lono78yA8cs,114
 uipath_langchain/vectorstores/context_grounding_vectorstore.py,sha256=TncIXG-YsUlO0R5ZYzWsM-Dj1SVCZbzmo2LraVxXelc,9559
-uipath_langchain-0.0.112.dist-info/METADATA,sha256=MSb7h7hBN5BjhDchfFQYu2X4_OTt2hki8Zr5a-t0TDg,4166
-uipath_langchain-0.0.112.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-uipath_langchain-0.0.112.dist-info/entry_points.txt,sha256=FUtzqGOEntlJKMJIXhQUfT7ZTbQmGhke1iCmDWZaQZI,81
-uipath_langchain-0.0.112.dist-info/licenses/LICENSE,sha256=JDpt-uotAkHFmxpwxi6gwx6HQ25e-lG4U_Gzcvgp7JY,1063
-uipath_langchain-0.0.112.dist-info/RECORD,,
+uipath_langchain-0.0.114.dist-info/METADATA,sha256=BTxHYf-8_z61F3QMH8WQFXiaqbQtMJpgYB7x8BiZKZQ,4166
+uipath_langchain-0.0.114.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+uipath_langchain-0.0.114.dist-info/entry_points.txt,sha256=FUtzqGOEntlJKMJIXhQUfT7ZTbQmGhke1iCmDWZaQZI,81
+uipath_langchain-0.0.114.dist-info/licenses/LICENSE,sha256=JDpt-uotAkHFmxpwxi6gwx6HQ25e-lG4U_Gzcvgp7JY,1063
+uipath_langchain-0.0.114.dist-info/RECORD,,