From c54c3546fc91fb0abbe7822fc797749a83ab8951 Mon Sep 17 00:00:00 2001
From: Daulet Amirkhanov <damirkhanov01@gmail.com>
Date: Wed, 8 Oct 2025 18:18:43 +0100
Subject: [PATCH] chore: add Docker image pull for MCP in start_ui function

---
 cognee/api/v1/ui/ui.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/cognee/api/v1/ui/ui.py b/cognee/api/v1/ui/ui.py
index 108806ba2..7711cb6ac 100644
--- a/cognee/api/v1/ui/ui.py
+++ b/cognee/api/v1/ui/ui.py
@@ -505,6 +505,8 @@ def start_ui(
         cwd = os.getcwd()
         env_file = os.path.join(cwd, ".env")
         try:
+            image = "cognee/cognee-mcp:main"
+            subprocess.run(["docker", "pull", image], check=True)
             mcp_process = subprocess.Popen(
                 [
                     "docker",