Merge pull request #153 from UiPath/feat/dev_terminal

feat: dev terminal

Author: cristipufu

pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "uipath-langchain"
-version = "0.0.123"
+version = "0.0.124"
 description = "UiPath Langchain"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
 dependencies = [
-    "uipath>=2.1.10, <2.2.0",
+    "uipath>=2.1.30, <2.2.0",
     "langgraph>=0.5.0, <0.7.0",
     "langchain-core>=0.3.34",
     "langgraph-checkpoint-sqlite>=2.0.3",
@@ -16,6 +16,7 @@ dependencies = [
     "python-dotenv>=1.0.1",
     "httpx>=0.27.0",
     "openai>=1.65.5",
+    "openinference-instrumentation-langchain>=0.1.50",
 ]
 classifiers = [
     "Development Status :: 3 - Alpha",

src/uipath_langchain/_cli/_runtime/_runtime.py

@@ -18,6 +18,7 @@
 
 from ..._utils import _instrument_traceable_attributes
 from ...tracers import AsyncUiPathTracer
+from .._utils._graph import LangGraphConfig
 from ._context import LangGraphRuntimeContext
 from ._exception import LangGraphRuntimeError
 from ._input import LangGraphInputProcessor
@@ -190,12 +191,14 @@ async def validate(self) -> None:
             ) from e
 
         if self.context.langgraph_config is None:
-            raise LangGraphRuntimeError(
-                "CONFIG_MISSING",
-                "Invalid configuration",
-                "Failed to load configuration",
-                UiPathErrorCategory.DEPLOYMENT,
-            )
+            self.context.langgraph_config = LangGraphConfig()
+            if not self.context.langgraph_config.exists:
+                raise LangGraphRuntimeError(
+                    "CONFIG_MISSING",
+                    "Invalid configuration",
+                    "Failed to load configuration",
+                    UiPathErrorCategory.DEPLOYMENT,
+                )
 
         try:
             self.context.langgraph_config.load_config()

src/uipath_langchain/_cli/cli_dev.py

@@ -0,0 +1,41 @@
+import asyncio
+from typing import Optional
+
+from openinference.instrumentation.langchain import (
+    LangChainInstrumentor,
+    get_current_span,
+)
+from uipath._cli._dev._terminal import UiPathDevTerminal
+from uipath._cli._runtime._contracts import UiPathRuntimeFactory
+from uipath._cli._utils._console import ConsoleLogger
+from uipath._cli.middlewares import MiddlewareResult
+
+from ._runtime._context import LangGraphRuntimeContext
+from ._runtime._runtime import LangGraphRuntime
+
+console = ConsoleLogger()
+
+
+def langgraph_dev_middleware(interface: Optional[str]) -> MiddlewareResult:
+    """Middleware to launch the developer terminal"""
+
+    try:
+        if interface == "terminal":
+            runtime_factory = UiPathRuntimeFactory(
+                LangGraphRuntime, LangGraphRuntimeContext
+            )
+            runtime_factory.add_instrumentor(LangChainInstrumentor, get_current_span)
+            app = UiPathDevTerminal(runtime_factory)
+            asyncio.run(app.run_async())
+        else:
+            console.error(f"Unknown interface: {interface}")
+    except KeyboardInterrupt:
+        console.info("Debug session interrupted by user")
+    except Exception as e:
+        console.error(f"Error occurred: {e}")
+        return MiddlewareResult(
+            should_continue=False,
+            should_include_stacktrace=True,
+        )
+
+    return MiddlewareResult(should_continue=False)

src/uipath_langchain/middlewares.py

@@ -1,5 +1,6 @@
 from uipath._cli.middlewares import Middlewares
 
+from ._cli.cli_dev import langgraph_dev_middleware
 from ._cli.cli_init import langgraph_init_middleware
 from ._cli.cli_new import langgraph_new_middleware
 from ._cli.cli_run import langgraph_run_middleware
@@ -10,3 +11,4 @@ def register_middleware():
     Middlewares.register("init", langgraph_init_middleware)
     Middlewares.register("run", langgraph_run_middleware)
     Middlewares.register("new", langgraph_new_middleware)
+    Middlewares.register("dev", langgraph_dev_middleware)