Update floor for langgraph-api

libs/cli/langgraph_cli/cli.py

@@ -511,19 +511,6 @@ def dockerfile(save_path: str, config: pathlib.Path, add_docker_compose: bool) -
     )
 
 
-@click.argument("path", required=False)
-@click.option(
-    "--template",
-    type=str,
-    help=TEMPLATE_HELP_STRING,
-)
-@cli.command("new", help="🌱 Create a new LangGraph project from a template.")
-@log_command
-def new(path: Optional[str], template: Optional[str]) -> None:
-    """Create a new LangGraph project from a template."""
-    return create_new(path, template)
-
-
 @click.option(
     "--host",
     default="127.0.0.1",
@@ -566,6 +553,7 @@ def new(path: Optional[str], template: Optional[str]) -> None:
 @cli.command(
     "dev",
     help="🏃‍♀️‍➡️ Run LangGraph API server in development mode with hot reloading and debugging support",
+    context_settings={"ignore_unknown_options": True},
 )
 @log_command
 def dev(
@@ -576,6 +564,7 @@ def dev(
     n_jobs_per_worker: Optional[int],
     no_browser: bool,
     debug_port: Optional[int],
+    uvicorn_flags: tuple[str, ...],
 ):
     """CLI entrypoint for running the LangGraph API server."""
     try:
@@ -608,6 +597,10 @@ def dev(
             sys.path.append(str(dep_path))
 
     graphs = config_json.get("graphs", {})
+    additional_config = {}
+    if config_json.get("store"):
+        additional_config["store"] = config_json["store"]
+
     run_server(
         host,
         port,
@@ -617,10 +610,23 @@ def dev(
         open_browser=not no_browser,
         debug_port=debug_port,
         env=config_json.get("env", None),
-        store=config_json.get("store", None),
+        config=additional_config,
     )
 
 
+@click.argument("path", required=False)
+@click.option(
+    "--template",
+    type=str,
+    help=TEMPLATE_HELP_STRING,
+)
+@cli.command("new", help="🌱 Create a new LangGraph project from a template.")
+@log_command
+def new(path: Optional[str], template: Optional[str]) -> None:
+    """Create a new LangGraph project from a template."""
+    return create_new(path, template)
+
+
 def prepare_args_and_stdin(
     *,
     capabilities: DockerCapabilities,

libs/cli/langgraph_cli/config.py

@@ -10,8 +10,8 @@
 MIN_PYTHON_VERSION = "3.11"
 
 
-class EmbeddingConfig(TypedDict, total=False):
-    """Configuration for vector embeddings in PostgreSQL store."""
+class IndexConfig(TypedDict, total=False):
+    """Configuration for indexing documents for semantic search in the store."""
 
     dims: int
     """Number of dimensions in the embedding vectors.
@@ -32,25 +32,18 @@ class EmbeddingConfig(TypedDict, total=False):
     Examples:
         - "openai:text-embedding-3-large"
         - "cohere:embed-multilingual-v3.0"
-        - "src/app.py::embeddings
-    """
-    aembed: Optional[str]
-    """Optional asynchronous function to generate embeddings from text.
-    
-    Provide for asynchronous embedding generation. This should be a string
-    representing the path to an async function, e.g., 'src/embeddings.py::aembed'.
-    Use this if you don't provide an Embeddings object through the 'embed' field.
+        - "src/app.py:embeddings
     """
 
-    text_fields: Optional[list[str]]
+    fields: Optional[list[str]]
     """Fields to extract text from for embedding generation.
     
-    Defaults to ["__root__"], which embeds the json object as a whole.
+    Defaults to the root ["$"], which embeds the json object as a whole.
     """
 
 
 class StoreConfig(TypedDict, total=False):
-    embed: Optional[EmbeddingConfig]
+    embed: Optional[IndexConfig]
     """Configuration for vector embeddings in store."""
 
 
@@ -399,16 +392,24 @@ def python_config_to_docker(config_path: pathlib.Path, config: Config, base_imag
             ],
         )
     )
-
+    additional_config = {}
+    if config.get("store"):
+        additional_config["store"] = config["store"]
+    env_additional_config = (
+        ""
+        if not additional_config
+        else f"""
+ENV LANGGRAPH_CONFIG='{json.dumps(additional_config)}'
+"""
+    )
     return f"""FROM {base_image}:{config['python_version']}
 
 {os.linesep.join(config["dockerfile_lines"])}
 
 {installs}
 
 RUN {pip_install} -e /deps/*
-
-ENV LANGGRAPH_STORE='{json.dumps(config["store"])}'
+{env_additional_config}
 ENV LANGSERVE_GRAPHS='{json.dumps(config["graphs"])}'
 
 {f"WORKDIR {local_deps.working_dir}" if local_deps.working_dir else ""}"""
@@ -438,16 +439,24 @@ def test_file(file_name):
         install_cmd = "npm ci"
     else:
         install_cmd = "npm i"
-
+    additional_config = {}
+    if config.get("store"):
+        additional_config["store"] = config["store"]
+    env_additional_config = (
+        ""
+        if not additional_config
+        else f"""
+ENV LANGGRAPH_CONFIG='{json.dumps(additional_config)}'
+"""
+    )
     return f"""FROM {base_image}:{config['node_version']}
 
 {os.linesep.join(config["dockerfile_lines"])}
 
 ADD . {faux_path}
 
 RUN cd {faux_path} && {install_cmd}
-
-ENV LANGGRAPH_STORE='{json.dumps(config["store"])}'
+{env_additional_config}
 ENV LANGSERVE_GRAPHS='{json.dumps(config["graphs"])}'
 
 WORKDIR {faux_path}

libs/cli/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-cli"
-version = "0.1.59"
+version = "0.1.60"
 description = "CLI for interacting with LangGraph API"
 authors = []
 license = "MIT"
@@ -14,7 +14,7 @@ langgraph = "langgraph_cli.cli:cli"
 [tool.poetry.dependencies]
 python = "^3.9.0,<4.0"
 click = "^8.1.7"
-langgraph-api = { version = ">=0.0.2,<0.1.0", optional = true, python = ">=3.11,<4.0" }
+langgraph-api = { version = ">=0.0.5,<0.1.0", optional = true, python = ">=3.11,<4.0" }
 python-dotenv = { version = ">=0.8.0", optional = true }
 
 [tool.poetry.group.dev.dependencies]

libs/cli/tests/unit_tests/test_config.py

@@ -30,6 +30,7 @@ def test_validate_config():
         "pip_config_file": None,
         "dockerfile_lines": [],
         "env": {},
+        "store": None,
         **expected_config,
     }
     actual_config = validate_config(expected_config)
@@ -46,6 +47,7 @@ def test_validate_config():
             "agent": "./agent.py:graph",
         },
         "env": env,
+        "store": None,
     }
     actual_config = validate_config(expected_config)
     assert actual_config == expected_config