From e4f99b797189ee0f5b68e27b7222e5477103e102 Mon Sep 17 00:00:00 2001
From: Deepankar Mahapatro <deepankar.mahapatro@jina.ai>
Date: Mon, 15 May 2023 17:56:26 +0530
Subject: [PATCH] feat: deploy langflow using langchain-serve

---
 Makefile                         | 3 +--
 pyproject.toml                   | 1 +
 src/backend/langflow/__main__.py | 5 +++--
 3 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/Makefile b/Makefile
index be87989ef..4b634e90c 100644
--- a/Makefile
+++ b/Makefile
@@ -50,8 +50,7 @@ lcserve_push:
 	make build_frontend
 	@version=$$(poetry version --short); \
 	lc-serve push --app langflow.lcserve:app --app-dir . \
-		--image-name langflow --image-tag $${version} \
-		--version deep-0.0.1 --verbose
+		--image-name langflow --image-tag $${version} --verbose
 
 lcserve_deploy:
 	@:$(if $(uses),,$(error `uses` is not set. Please run `make uses=... lcserve_deploy`))
diff --git a/pyproject.toml b/pyproject.toml
index 0cc181617..0e97c56cf 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -51,6 +51,7 @@ websockets = "^11.0.2"
 tiktoken = "^0.3.3"
 wikipedia = "^1.4.0"
 gptcache = "^0.1.23"
+langchain-serve = "^0.0.28"
 
 [tool.poetry.group.dev.dependencies]
 black = "^23.1.0"
diff --git a/src/backend/langflow/__main__.py b/src/backend/langflow/__main__.py
index 8da0062bf..46225e977 100644
--- a/src/backend/langflow/__main__.py
+++ b/src/backend/langflow/__main__.py
@@ -78,9 +78,10 @@ def jcloud():
     from importlib.metadata import version as mod_version
 
     app_name = "langflow.lcserve:app"
+    app_dir = str(Path(__file__).parent)
     version = mod_version("langflow")
-    uses = f"jinaai+docker://langflow:{version}"
-    os.system(f"lc-serve deploy jcloud --app {app_name} --app-dir . --uses {uses}")
+    base_image = "jinaai+docker://deepankarm/langflow"
+    os.system(f"lc-serve deploy jcloud --app {app_name} --app-dir {app_dir} --uses {base_image}:{version}")
 
 
 def main():