From db1fbe74cc0920d91c9b10200efb0b86be964c3e Mon Sep 17 00:00:00 2001
From: Tat Dat Duong <david@duong.cz>
Date: Thu, 17 Apr 2025 22:22:43 +0200
Subject: [PATCH] Revert args

---
 libs/cli/langgraph_cli/cli.py    | 17 +++--------------
 libs/cli/langgraph_cli/config.py |  5 ++---
 2 files changed, 5 insertions(+), 17 deletions(-)

diff --git a/libs/cli/langgraph_cli/cli.py b/libs/cli/langgraph_cli/cli.py
index 4e56bb8cc..1f4b73d43 100644
--- a/libs/cli/langgraph_cli/cli.py
+++ b/libs/cli/langgraph_cli/cli.py
@@ -273,7 +273,6 @@ def _build(
     pull: bool,
     tag: str,
     passthrough: Sequence[str] = (),
-    base_docker_tag: Optional[str] = None,
 ):
     # pull latest images
     if pull:
@@ -281,9 +280,7 @@ def _build(
             subp_exec(
                 "docker",
                 "pull",
-                langgraph_cli.config.docker_tag(
-                    config_json, base_image, base_docker_tag
-                ),
+                langgraph_cli.config.docker_tag(config_json, base_image),
                 verbose=True,
             )
         )
@@ -298,7 +295,7 @@ def _build(
     ]
     # apply config
     stdin, additional_contexts = langgraph_cli.config.config_to_docker(
-        config, config_json, base_image, base_docker_tag
+        config, config_json, base_image
     )
     # add additional_contexts
     if additional_contexts:
@@ -359,15 +356,7 @@ def build(
             raise click.UsageError("Docker not installed") from None
         config_json = langgraph_cli.config.validate_config_file(config)
         _build(
-            runner,
-            set,
-            config,
-            config_json,
-            base_image,
-            pull,
-            tag,
-            docker_build_args,
-            config_json.get("_INTERNAL_docker_tag"),
+            runner, set, config, config_json, base_image, pull, tag, docker_build_args
         )
 
 
diff --git a/libs/cli/langgraph_cli/config.py b/libs/cli/langgraph_cli/config.py
index 8170062fe..5a363ea42 100644
--- a/libs/cli/langgraph_cli/config.py
+++ b/libs/cli/langgraph_cli/config.py
@@ -1262,11 +1262,10 @@ def default_base_image(config: Config) -> str:
 def docker_tag(
     config: Config,
     base_image: Optional[str] = None,
-    base_docker_tag: Optional[str] = None,
 ) -> str:
     base_image = base_image or default_base_image(config)
-    if base_docker_tag:
-        return f"{base_image}:{base_docker_tag}"
+    if config.get("_INTERNAL_docker_tag"):
+        return f"{base_image}:{config['_INTERNAL_docker_tag']}"
 
     if config.get("node_version") and not config.get("python_version"):
         return f"{base_image}:{config['node_version']}"