uipath-langchain 0.0.120__py3-none-any.whl → 0.0.121__py3-none-any.whl

uipath_langchain/_cli/_runtime/_runtime.py

@@ -101,7 +101,7 @@ class LangGraphRuntime(UiPathBaseRuntime):
                     graph_config["max_concurrency"] = int(max_concurrency)
 
                 # Stream the output at debug time
-                if self.context.job_id is None:
+                if self.is_debug_run():
                     # Get final chunk while streaming
                     final_chunk = None
                     async for stream_chunk in graph.astream(
@@ -115,7 +115,7 @@ class LangGraphRuntime(UiPathBaseRuntime):
 
                     self.context.output = self._extract_graph_result(final_chunk, graph)
                 else:
-                    # Execute the graph normally at runtime
+                    # Execute the graph normally at runtime or eval
                     self.context.output = await graph.ainvoke(
                         processed_input, graph_config
                     )
@@ -310,9 +310,23 @@ class LangGraphRuntime(UiPathBaseRuntime):
             # Check which channels are present
             available_channels = [ch for ch in output_channels if ch in final_chunk]
 
+            # if no available channels, output may contain the last_node name as key
+            unwrapped_final_chunk = {}
+            if not available_channels:
+                if len(final_chunk) == 1 and isinstance(
+                    unwrapped_final_chunk := next(iter(final_chunk.values())), dict
+                ):
+                    available_channels = [
+                        ch for ch in output_channels if ch in unwrapped_final_chunk
+                    ]
+
             if available_channels:
                 # Create a dict with the available channels
-                return {channel: final_chunk[channel] for channel in available_channels}
+                return {
+                    channel: final_chunk.get(channel, None)
+                    or unwrapped_final_chunk[channel]
+                    for channel in available_channels
+                }
 
         # Fallback for any other case
         return final_chunk

uipath_langchain/_cli/cli_run.py

@@ -31,8 +31,8 @@ def langgraph_run_middleware(
             tracing = bool_map[tracing.lower()]
 
         async def execute():
-            context = LangGraphRuntimeContext.from_config(
-                env.get("UIPATH_CONFIG_PATH", "uipath.json")
+            context: LangGraphRuntimeContext = LangGraphRuntimeContext.from_config(
+                env.get("UIPATH_CONFIG_PATH", "uipath.json"), **kwargs
             )
             context.entrypoint = entrypoint
             context.input = input
@@ -42,6 +42,7 @@ def langgraph_run_middleware(
             context.logs_min_level = env.get("LOG_LEVEL", "INFO")
             context.job_id = env.get("UIPATH_JOB_KEY")
             context.trace_id = env.get("UIPATH_TRACE_ID")
+            context.is_eval_run = kwargs.get("is_eval_run", False)
             context.tracing_enabled = tracing
             context.input_file = kwargs.get("input_file", None)
             context.execution_output_file = kwargs.get("execution_output_file", None)

{uipath_langchain-0.0.120.dist-info → uipath_langchain-0.0.121.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-langchain
-Version: 0.0.120
+Version: 0.0.121
 Summary: UiPath Langchain
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-langchain-python
@@ -20,11 +20,11 @@ Requires-Dist: langchain-core>=0.3.34
 Requires-Dist: langchain-openai>=0.3.3
 Requires-Dist: langchain>=0.3.4
 Requires-Dist: langgraph-checkpoint-sqlite>=2.0.3
-Requires-Dist: langgraph>=0.2.70
+Requires-Dist: langgraph<0.7.0,>=0.5.0
 Requires-Dist: openai>=1.65.5
 Requires-Dist: pydantic-settings>=2.6.0
 Requires-Dist: python-dotenv>=1.0.1
-Requires-Dist: uipath<2.2.0,>=2.1.0
+Requires-Dist: uipath<2.2.0,>=2.1.10
 Provides-Extra: langchain
 Description-Content-Type: text/markdown
 

{uipath_langchain-0.0.120.dist-info → uipath_langchain-0.0.121.dist-info}/RECORD

@@ -3,12 +3,12 @@ uipath_langchain/middlewares.py,sha256=tre7o9DFMgWk1DJiEEUmT6_wiP-PPkWtKmG0iOyvr
 uipath_langchain/_cli/__init__.py,sha256=juqd9PbXs4yg45zMJ7BHAOPQjb7sgEbWE9InBtGZhfo,24
 uipath_langchain/_cli/cli_init.py,sha256=xhxJ8tuMSrVUNHvltgyPpOrvgMA-wq9shHeYYwvHILs,8199
 uipath_langchain/_cli/cli_new.py,sha256=dL8-Rri6u67ZZdbb4nT38A5xD_Q3fVnG0UK9VSeKaqg,2563
-uipath_langchain/_cli/cli_run.py,sha256=pi_3GdRxcDenO6wXa0e0zdh-_hm7nTKpqnS7LKf4gTs,3273
+uipath_langchain/_cli/cli_run.py,sha256=Jr-nxzBaSPQoVrdV1nQBSDVFElFV3N0a0uNDDrdSc8I,3375
 uipath_langchain/_cli/_runtime/_context.py,sha256=yyzYJDmk2fkH4T5gm4cLGRyXtjLESrpzHBT9euqluTA,817
 uipath_langchain/_cli/_runtime/_exception.py,sha256=USKkLYkG-dzjX3fEiMMOHnVUpiXJs_xF0OQXCCOvbYM,546
 uipath_langchain/_cli/_runtime/_input.py,sha256=vZ8vfVxvPSaPWmIPghvNx1VRKzbalHsKUMBPiKDvJWM,5492
 uipath_langchain/_cli/_runtime/_output.py,sha256=yJOZPWv2FRUJWv1NRs9JmpB4QMTDXu8jrxoaKrfJvzw,9078
-uipath_langchain/_cli/_runtime/_runtime.py,sha256=y12VlDl6wC53458acFvI9zikGV5epEfOm2YEW9ezSQg,13910
+uipath_langchain/_cli/_runtime/_runtime.py,sha256=XTlPaMuHyv-4DjQMoEh76D2Z18M3S5wm4tDP-UAiE5Y,14505
 uipath_langchain/_cli/_templates/langgraph.json.template,sha256=eeh391Gta_hoRgaNaZ58nW1LNvCVXA7hlAH6l7Veous,107
 uipath_langchain/_cli/_templates/main.py.template,sha256=nMJQIYPlRk90iANfNVpkJ2EQX20Dxsyq92-BucEz_UM,1189
 uipath_langchain/_cli/_utils/_graph.py,sha256=2FrlU4bMuEzcHglXzIC54IB8jw55SlENF-SVcpwWsS0,7167
@@ -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.120.dist-info/METADATA,sha256=_oQM5qOV7W-PPTcScseI-BCMMpy-8JDifrP_yCMnjxA,4165
-uipath_langchain-0.0.120.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-uipath_langchain-0.0.120.dist-info/entry_points.txt,sha256=FUtzqGOEntlJKMJIXhQUfT7ZTbQmGhke1iCmDWZaQZI,81
-uipath_langchain-0.0.120.dist-info/licenses/LICENSE,sha256=JDpt-uotAkHFmxpwxi6gwx6HQ25e-lG4U_Gzcvgp7JY,1063
-uipath_langchain-0.0.120.dist-info/RECORD,,
+uipath_langchain-0.0.121.dist-info/METADATA,sha256=gmDInqkPI5vgkYkMORrf0FaSQmU2ABppifipcvZDEkc,4172
+uipath_langchain-0.0.121.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+uipath_langchain-0.0.121.dist-info/entry_points.txt,sha256=FUtzqGOEntlJKMJIXhQUfT7ZTbQmGhke1iCmDWZaQZI,81
+uipath_langchain-0.0.121.dist-info/licenses/LICENSE,sha256=JDpt-uotAkHFmxpwxi6gwx6HQ25e-lG4U_Gzcvgp7JY,1063
+uipath_langchain-0.0.121.dist-info/RECORD,,