uipath-langchain 0.0.94__py3-none-any.whl → 0.0.95__py3-none-any.whl

uipath_langchain/_cli/_templates/langgraph.json.template

@@ -0,0 +1,7 @@
+{
+    "dependencies": ["."],
+    "graphs": {
+      "agent": "./main.py:graph"
+    },
+    "env": ".env"
+  }

uipath_langchain/_cli/_templates/main.py.template

@@ -0,0 +1,33 @@
+from langchain_anthropic import ChatAnthropic
+from langchain_core.messages import HumanMessage, SystemMessage
+from langchain_openai import ChatOpenAI
+from langgraph.graph import START, StateGraph, END
+from pydantic import BaseModel
+import os
+
+class GraphInput(BaseModel):
+    topic: str
+
+class GraphOutput(BaseModel):
+    report: str
+
+async def generate_report(state: GraphInput) -> GraphOutput:
+    if os.getenv("ANTHROPIC_API_KEY"):
+        llm_model = ChatAnthropic(model="claude-3-5-sonnet-latest")
+    elif os.getenv("OPENAI_API_KEY"):
+        llm_model = ChatOpenAI(model="gpt-4o-mini", temperature=0)
+    else:
+        raise Exception("Error: Missing API Key. Please define either ANTHROPIC_API_KEY or OPENAI_API_KEY.")
+
+    system_prompt = "You are a report generator. Please provide a brief report based on the given topic."
+    output = await llm_model.ainvoke([SystemMessage(system_prompt), HumanMessage(state.topic)])
+    return GraphOutput(report=output.content)
+
+builder = StateGraph(input=GraphInput, output=GraphOutput)
+
+builder.add_node("generate_report", generate_report)
+
+builder.add_edge(START, "generate_report")
+builder.add_edge("generate_report", END)
+
+graph = builder.compile()

uipath_langchain/_cli/cli_new.py

@@ -0,0 +1,76 @@
+import os
+import shutil
+
+import click
+from uipath._cli.middlewares import MiddlewareResult
+from uipath._cli.spinner import Spinner
+
+
+def generate_script(target_directory):
+    template_script_path = os.path.join(
+        os.path.dirname(__file__), "_templates/main.py.template"
+    )
+    target_path = os.path.join(target_directory, "main.py")
+
+    shutil.copyfile(template_script_path, target_path)
+
+    template_langgraph_json_path = os.path.join(
+        os.path.dirname(__file__), "_templates/langgraph.json.template"
+    )
+    target_path = os.path.join(target_directory, "langgraph.json")
+    shutil.copyfile(template_langgraph_json_path, target_path)
+
+
+def generate_pyproject(target_directory, project_name):
+    project_toml_path = os.path.join(target_directory, "pyproject.toml")
+    toml_content = f"""[project]
+                        name = "{project_name}"
+                        version = "0.0.1"
+                        description = "{project_name}"
+                        authors = [{{ name = "John Doe", email = "john.doe@myemail.com" }}]
+                        dependencies = [
+                            "uipath-langchain>=0.0.95",
+                            "langchain-anthropic>=0.3.8",
+                        ]
+                        requires-python = ">=3.10"
+                    """
+
+    with open(project_toml_path, "w") as f:
+        f.write(toml_content)
+
+
+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"}' `""",
+        )
+    except Exception as e:
+        spinner.stop()
+        return MiddlewareResult(
+            should_continue=False,
+            error_message=f"❌ Error creating demo agent {str(e)}",
+            should_include_stacktrace=True,
+        )

uipath_langchain/middlewares.py

@@ -1,6 +1,7 @@
 from uipath._cli.middlewares import Middlewares
 
 from ._cli.cli_init import langgraph_init_middleware
+from ._cli.cli_new import langgraph_new_middleware
 from ._cli.cli_run import langgraph_run_middleware
 
 
@@ -8,3 +9,4 @@ def register_middleware():
     """This function will be called by the entry point system when uipath_langchain is installed"""
     Middlewares.register("init", langgraph_init_middleware)
     Middlewares.register("run", langgraph_run_middleware)
+    Middlewares.register("new", langgraph_new_middleware)

{uipath_langchain-0.0.94.dist-info → uipath_langchain-0.0.95.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-langchain
-Version: 0.0.94
+Version: 0.0.95
 Summary: UiPath Langchain
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-langchain-python

{uipath_langchain-0.0.94.dist-info → uipath_langchain-0.0.95.dist-info}/RECORD

@@ -1,7 +1,8 @@
 uipath_langchain/__init__.py,sha256=VBrvQn7d3nuOdN7zEnV2_S-uhmkjgEIlXiFVeZxZakQ,80
-uipath_langchain/middlewares.py,sha256=AkGsj-zqyh9u5AyX79MCL0yKL4YArF-hudAHZs_0K0Y,400
+uipath_langchain/middlewares.py,sha256=tre7o9DFMgWk1DJiEEUmT6_wiP-PPkWtKmG0iOyvr9c,509
 uipath_langchain/_cli/__init__.py,sha256=juqd9PbXs4yg45zMJ7BHAOPQjb7sgEbWE9InBtGZhfo,24
 uipath_langchain/_cli/cli_init.py,sha256=PKkebO0P4Qh1N-LAx_HdypbffbGq5_JSquJY1OMsK6E,7700
+uipath_langchain/_cli/cli_new.py,sha256=JjnDbns8ok2Bj2M6Tg-FKOfm1YxpTIq6ReLeCAl6vKM,2745
 uipath_langchain/_cli/cli_run.py,sha256=zMn2P1gB1ogM8_EmZsIhqNpeI9Oc02Z5Gj-oni7eUV4,2915
 uipath_langchain/_cli/_runtime/_context.py,sha256=wr4aNn06ReIXmetEZ6b6AnpAt64p13anQ2trZ5Bzgio,807
 uipath_langchain/_cli/_runtime/_escalation.py,sha256=oA5NvZvCo8ngELFJRyhZNM69DxVHrshhMY6CUk_cukQ,8055
@@ -9,6 +10,8 @@ uipath_langchain/_cli/_runtime/_exception.py,sha256=USKkLYkG-dzjX3fEiMMOHnVUpiXJ
 uipath_langchain/_cli/_runtime/_input.py,sha256=gKzPaGW-EzgeAskWJjbCWnfZRLu_BM7lCXkq0XkVGLU,5614
 uipath_langchain/_cli/_runtime/_output.py,sha256=WQSrsvGaaclZ6GLWEh6Nk1Mz1iGaIB45PgIX3DS3AN4,16130
 uipath_langchain/_cli/_runtime/_runtime.py,sha256=wj6e6tQdP2GjuSM-6G6Ug-FMqwOHwfUq-DGbd1gMfIg,11825
+uipath_langchain/_cli/_templates/langgraph.json.template,sha256=eeh391Gta_hoRgaNaZ58nW1LNvCVXA7hlAH6l7Veous,107
+uipath_langchain/_cli/_templates/main.py.template,sha256=FQIldd8PKYAxL9oAS6Y1CB0mKtY4LroXdYW8Hg_ZnHo,1202
 uipath_langchain/_cli/_utils/_graph.py,sha256=WLBSJfPc3_C07SqJhePRe17JIc5wcBvEqLviMcNOdTA,6950
 uipath_langchain/_utils/__init__.py,sha256=WoY66enCygRXTh6v5B1UrRcFCnQYuPJ8oqDkwomXzLc,194
 uipath_langchain/_utils/_request_mixin.py,sha256=t_1HWBxqEl-wsSk9ubmIM-8vs9BlNy4ZVBxtDxktn6U,18489
@@ -33,8 +36,8 @@ uipath_langchain/utils/_request_mixin.py,sha256=WFyTDyAthSci1DRwUwS21I3hLntD7HdV
 uipath_langchain/utils/_settings.py,sha256=MhwEVj4gVRSar0RBf2w2hTjO-5Qm-HpCuufqN3gSWjA,3390
 uipath_langchain/utils/_sleep_policy.py,sha256=e9pHdjmcCj4CVoFM1jMyZFelH11YatsgWfpyrfXzKBQ,1251
 uipath_langchain/vectorstores/context_grounding_vectorstore.py,sha256=eTa5sX43-ydB1pj9VNHUPbB-hC36fZK_CGrNe5U2Nrw,9393
-uipath_langchain-0.0.94.dist-info/METADATA,sha256=0K_IaP1-pgZHmR0vDZZa3yhIy1oCbPR-z7mOYMWSepM,3819
-uipath_langchain-0.0.94.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-uipath_langchain-0.0.94.dist-info/entry_points.txt,sha256=FUtzqGOEntlJKMJIXhQUfT7ZTbQmGhke1iCmDWZaQZI,81
-uipath_langchain-0.0.94.dist-info/licenses/LICENSE,sha256=JDpt-uotAkHFmxpwxi6gwx6HQ25e-lG4U_Gzcvgp7JY,1063
-uipath_langchain-0.0.94.dist-info/RECORD,,
+uipath_langchain-0.0.95.dist-info/METADATA,sha256=WZEMEDLVkErrg3hoVdG8qdcT97yNSYvWtQO17CK2IHs,3819
+uipath_langchain-0.0.95.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+uipath_langchain-0.0.95.dist-info/entry_points.txt,sha256=FUtzqGOEntlJKMJIXhQUfT7ZTbQmGhke1iCmDWZaQZI,81
+uipath_langchain-0.0.95.dist-info/licenses/LICENSE,sha256=JDpt-uotAkHFmxpwxi6gwx6HQ25e-lG4U_Gzcvgp7JY,1063
+uipath_langchain-0.0.95.dist-info/RECORD,,