langgraph-cli 0.1.4__tar.gz → 0.1.6__tar.gz

{langgraph_cli-0.1.4 → langgraph_cli-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-cli
-Version: 0.1.4
+Version: 0.1.6
 Summary: 
 Author: Nuno Campos
 Author-email: nuno@langchain.dev

{langgraph_cli-0.1.4 → langgraph_cli-0.1.6}/langgraph_cli/cli.py

@@ -179,6 +179,70 @@ def up(
         runner.run(exec("docker", "compose", *args, input=stdin))
 
 
+@OPT_C
+@click.option("--tag", "-t", help="Tag for the image", required=True)
+@cli.command(help="Build langgraph API server image")
+def build(
+    config: pathlib.Path,
+    tag: str,
+):
+    with asyncio.Runner() as runner:
+        # check docker available
+        try:
+            runner.run(exec("docker", "--version"))
+            runner.run(exec("docker", "compose", "version"))
+        except click.exceptions.Exit:
+            click.echo("Docker not installed or not running")
+            return
+        # apply options
+        args = [
+            "-f",
+            "-",  # stdin
+            "-t",
+            tag,
+        ]
+        with open(config) as f:
+            stdin = langgraph_cli.config.config_to_docker(config, json.load(f))
+        # run docker build
+        runner.run(exec("docker", "build", *args, config.parent, input=stdin))
+
+
+@OPT_DEBUGGER_PORT
+@OPT_PORT
+@OPT_O
+@OPT_C
+@cli.command(help="Write k8s files", hidden=True)
+def k8s(
+    config: pathlib.Path,
+    docker_compose: Optional[pathlib.Path],
+    port: int,
+    debugger_port: Optional[int],
+):
+    with asyncio.Runner() as runner:
+        # check docker available
+        try:
+            runner.run(exec("docker", "--version"))
+            runner.run(exec("docker", "compose", "version"))
+        except click.exceptions.Exit:
+            click.echo("Docker not installed or not running")
+            return
+        # prepare args
+        stdin = langgraph_cli.docker.compose(port=port, debugger_port=debugger_port)
+        args = [
+            "-f",
+            "-",  # stdin
+        ]
+        # apply options
+        if docker_compose:
+            args.extend(["-f", str(docker_compose)])
+        args.append("convert")
+        if config:
+            with open(config) as f:
+                stdin += langgraph_cli.config.config_to_compose(config, json.load(f))
+        # run kompose convert
+        runner.run(exec("kompose", *args, input=stdin))
+
+
 @OPT_O
 @OPT_PORT
 @cli.command()

{langgraph_cli-0.1.4 → langgraph_cli-0.1.6}/langgraph_cli/config.py

@@ -1,6 +1,7 @@
 import json
 import os
 import pathlib
+import textwrap
 from typing import TypedDict, Union
 
 
@@ -10,7 +11,7 @@ class Config(TypedDict):
     env: Union[dict[str, str], str]
 
 
-def config_to_compose(config_path: pathlib.Path, config: Config):
+def config_to_docker(config_path: pathlib.Path, config: Config):
     pypi_deps = [dep for dep in config["dependencies"] if not dep.startswith(".")]
     local_pkgs: dict[pathlib.Path, str] = {}
     faux_pkgs: dict[pathlib.Path, str] = {}
@@ -83,7 +84,23 @@ def config_to_compose(config_path: pathlib.Path, config: Config):
         f"ADD {relpath} /tmp/{fullpath.name}/{fullpath.name}\n                RUN touch /tmp/{fullpath.name}/pyproject.toml"
         for fullpath, relpath in faux_pkgs.items()
     )
-    local_pkgs_str = f"ADD {' '.join(local_pkgs.values())} /tmp/" if local_pkgs else ""
+    local_pkgs_str = "\n".join(
+        f"ADD {relpath} /tmp/{fullpath.name}"
+        for fullpath, relpath in local_pkgs.items()
+    )
+
+    return f"""FROM langchain/langgraph-api
+
+ENV LANGSERVE_GRAPHS='{json.dumps(config["graphs"])}'
+
+{local_pkgs_str}
+
+{faux_pkgs_str}
+
+RUN pip install {' '.join(pypi_deps)} /tmp/*"""
+
+
+def config_to_compose(config_path: pathlib.Path, config: Config):
     env_vars_str = (
         "\n".join(f"            {k}: {v}" for k, v in config["env"].items())
         if isinstance(config["env"], dict)
@@ -94,17 +111,10 @@ def config_to_compose(config_path: pathlib.Path, config: Config):
     )
 
     return f"""
-            LANGSERVE_GRAPHS: '{json.dumps(config["graphs"])}'
             {env_vars_str}
         {env_file_str}
         pull_policy: build
         build:
             dockerfile_inline: |
-                FROM langchain/langserve
-
-                {local_pkgs_str}
-
-                {faux_pkgs_str}
-
-                RUN pip install {' '.join(pypi_deps)} /tmp/*
+{textwrap.indent(config_to_docker(config_path, config), "                ")}
 """

{langgraph_cli-0.1.4 → langgraph_cli-0.1.6}/langgraph_cli/docker.py

@@ -6,7 +6,7 @@ ROOT = pathlib.Path(__file__).parent.resolve()
 
 
 DB = f"""
-    postgres:
+    langgraph-postgres:
         image: postgres:16
         restart: on-failure
         healthcheck:
@@ -22,18 +22,18 @@ DB = f"""
             POSTGRES_USER: postgres
             POSTGRES_PASSWORD: postgres
         volumes:
-            - ./.langserve-data:/var/lib/postgresql/data
+            - ./.langgraph-data:/var/lib/postgresql/data
             - {ROOT}/initdb:/docker-entrypoint-initdb.d
 """
 
 DEBUGGER = """
-    debugger:
+    langgraph-debugger:
         image: langchain/langserve-debugger
-        restart: on-failure:3
+        restart: on-failure
         ports:
             - "{debugger_port}:5173"
         depends_on:
-            langserve:
+            langgraph-api:
                 condition: service_healthy
         environment:
             VITE_API_BASE_URL: http://localhost:{port}
@@ -49,7 +49,7 @@ def compose(
 ) -> str:
     if postgres_uri is None:
         include_db = True
-        postgres_uri = "postgres://postgres:postgres@postgres:5432/postgres?sslmode=disable&search_path=langserve"
+        postgres_uri = "postgres://postgres:postgres@langgraph-postgres:5432/postgres?sslmode=disable&search_path=langgraph"
     else:
         include_db = False
 
@@ -57,22 +57,14 @@ def compose(
 services:
 {DB if include_db else ""}
 {DEBUGGER.format(port=port, debugger_port=debugger_port) if debugger_port else ""}
-    migrate:
-        image: langchain/langserve-migrate
-        pull_policy: always
-        {'''depends_on:
-            postgres:
-                condition: service_healthy''' if include_db else ""}
-        environment:
-            POSTGRES_URI: {postgres_uri}
-    langserve:
-        image: langchain/langserve
-        restart: on-failure:3
+    langgraph-api:
+        image: langchain/langgraph-api
+        restart: on-failure
         ports:
             - "{port}:8000"
         depends_on:
-            migrate:
-                condition: service_completed_successfully
+            langgraph-postgres:
+                condition: service_healthy
         environment:
             POSTGRES_URI: {postgres_uri}
 """

langgraph_cli-0.1.6/langgraph_cli/initdb/init.sql

@@ -0,0 +1 @@
+create schema if not exists langgraph;

{langgraph_cli-0.1.4 → langgraph_cli-0.1.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-cli"
-version = "0.1.4"
+version = "0.1.6"
 description = ""
 authors = ["Nuno Campos <nuno@langchain.dev>"]
 readme = "README.md"

langgraph_cli-0.1.4/langgraph_cli/initdb/init.sql

@@ -1 +0,0 @@
-create schema if not exists langserve;