From 3342e03a2cb75bcbcc36060b5204bee89ba773af Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 6 Jun 2023 09:59:37 -0300
Subject: [PATCH] =?UTF-8?q?=F0=9F=94=80=20refactor(langflow):=20move=20rou?=
 =?UTF-8?q?ters=20to=20a=20single=20file=20and=20add=20health=20check=20en?=
 =?UTF-8?q?dpoint=20The=20routers=20for=20the=20langflow=20API=20have=20be?=
 =?UTF-8?q?en=20moved=20to=20a=20single=20file=20for=20better=20organizati?=
 =?UTF-8?q?on=20and=20maintainability.=20The=20routers=20have=20been=20imp?=
 =?UTF-8?q?orted=20and=20included=20in=20the=20main.py=20file=20using=20th?=
 =?UTF-8?q?e=20new=20file.=20A=20new=20health=20check=20endpoint=20has=20b?=
 =?UTF-8?q?een=20added=20to=20the=20API=20to=20check=20the=20status=20of?=
 =?UTF-8?q?=20the=20application.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/__init__.py |  2 +-
 src/backend/langflow/main.py     | 12 ++++++------
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/backend/langflow/__init__.py b/src/backend/langflow/__init__.py
index 35fe814d2..17b1d940c 100644
--- a/src/backend/langflow/__init__.py
+++ b/src/backend/langflow/__init__.py
@@ -1,4 +1,4 @@
 from langflow.cache import cache_manager
-from langflow.interface.loading import load_flow_from_json
+from langflow.processing.process import load_flow_from_json
 
 __all__ = ["load_flow_from_json", "cache_manager"]
diff --git a/src/backend/langflow/main.py b/src/backend/langflow/main.py
index 56cc32e46..de39d8750 100644
--- a/src/backend/langflow/main.py
+++ b/src/backend/langflow/main.py
@@ -1,9 +1,7 @@
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 
-from langflow.api.chat import router as chat_router
-from langflow.api.endpoints import router as endpoints_router
-from langflow.api.validate import router as validate_router
+from langflow.api import router
 
 
 def create_app():
@@ -14,6 +12,10 @@ def create_app():
         "*",
     ]
 
+    @app.get("/health")
+    def get_health():
+        return {"status": "OK"}
+
     app.add_middleware(
         CORSMiddleware,
         allow_origins=origins,
@@ -22,9 +24,7 @@ def create_app():
         allow_headers=["*"],
     )
 
-    app.include_router(endpoints_router)
-    app.include_router(validate_router)
-    app.include_router(chat_router)
+    app.include_router(router)
     return app