langgraph-api 0.0.5__tar.gz → 0.0.6__tar.gz

{langgraph_api-0.0.5 → langgraph_api-0.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-api
-Version: 0.0.5
+Version: 0.0.6
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos

{langgraph_api-0.0.5 → langgraph_api-0.0.6}/langgraph_api/cli.py

@@ -94,10 +94,6 @@ class StoreConfig(TypedDict, total=False):
     index: IndexConfig
 
 
-class Config(TypedDict, total=False):
-    store: StoreConfig | None
-
-
 def run_server(
     host: str = "127.0.0.1",
     port: int = 2024,
@@ -107,10 +103,11 @@ def run_server(
     env_file: str | None = None,
     open_browser: bool = False,
     debug_port: int | None = None,
+    wait_for_client: bool = False,
     env: str | pathlib.Path | Mapping[str, str] | None = None,
     reload_includes: Sequence[str] | None = None,
     reload_excludes: Sequence[str] | None = None,
-    config: Config | None = None,
+    store: StoreConfig | None = None,
     **kwargs: Any,
 ):
     """Run the LangGraph API server."""
@@ -148,8 +145,9 @@ def run_server(
         logger.info("   - Host: 0.0.0.0")
         logger.info(f"   - Port: {debug_port}")
         logger.info("3. Start the debugger to connect to the server.")
-        debugpy.wait_for_client()
-        logger.info("Debugger attached. Starting server...")
+        if wait_for_client:
+            debugpy.wait_for_client()
+            logger.info("Debugger attached. Starting server...")
 
     local_url = f"http://{host}:{port}"
     studio_url = f"https://smith.langchain.com/studio/?baseUrl={local_url}"
@@ -213,7 +211,7 @@ For production use, please use LangGraph Cloud.
         DATABASE_URI=":memory:",
         REDIS_URI="fake",
         N_JOBS_PER_WORKER=str(n_jobs_per_worker if n_jobs_per_worker else 1),
-        LANGGRAPH_CONFIG=json.dumps(config) if config else None,
+        LANGGRAPH_STORE=json.dumps(store) if store else None,
         LANGSERVE_GRAPHS=json.dumps(graphs) if graphs else None,
         LANGSMITH_LANGGRAPH_API_VARIANT="local_dev",
         **(env_vars or {}),
@@ -274,6 +272,11 @@ def main():
     parser.add_argument(
         "--debug-port", type=int, help="Port for debugger to listen on (default: none)"
     )
+    parser.add_argument(
+        "--wait-for-client",
+        action="store_true",
+        help="Whether to break and wait for a debugger to attach",
+    )
 
     args = parser.parse_args()
 
@@ -289,6 +292,7 @@ def main():
         n_jobs_per_worker=args.n_jobs_per_worker,
         open_browser=not args.no_browser,
         debug_port=args.debug_port,
+        wait_for_client=args.wait_for_client,
         env=config_data.get("env", None),
     )
 

{langgraph_api-0.0.5 → langgraph_api-0.0.6}/langgraph_storage/database.py

@@ -157,10 +157,9 @@ async def connect(*, __test__: bool = False) -> AsyncIterator[AsyncConnectionPro
 
 async def start_pool() -> None:
     if store._STORE_CONFIG is None:
-        if (config_val := os.getenv("LANGGRAPH_CONFIG")) and config_val.strip():
+        if (config_val := os.getenv("LANGGRAPH_STORE")) and config_val.strip():
             config_ = json.loads(config_val.strip())
-            if "store" in config_:
-                store.set_store_config(config_["store"])
+            store.set_store_config(config_)
 
     if not os.path.exists(".langgraph_api"):
         os.mkdir(".langgraph_api")

{langgraph_api-0.0.5 → langgraph_api-0.0.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-api"
-version = "0.0.5"
+version = "0.0.6"
 description = ""
 authors = [
     "Nuno Campos <nuno@langchain.dev>",