keep only rag 3

artefactory · Mar 25, 2024 · 5422f1f · 5422f1f
1 parent f7cd508
commit 5422f1f
Showing 1 changed file with 19 additions and 18 deletions.
diff --git a/app/server.py b/app/server.py
@@ -7,10 +7,11 @@
 
 from backend import ENABLE_AUTHENTICATION
 from backend.api_plugins import authentication_routes, session_routes
-from backend.rag_1.chain import get_chain as get_chain_rag_1
-from backend.rag_1.config import validate_config as validate_config_1
-from backend.rag_2.chain import get_chain as get_chain_rag_2
-from backend.rag_2.config import validate_config as validate_config_2
+
+# from backend.rag_1.chain import get_chain as get_chain_rag_1
+# from backend.rag_1.config import validate_config as validate_config_1
+# from backend.rag_2.chain import get_chain as get_chain_rag_2
+# from backend.rag_2.config import validate_config as validate_config_2
 from backend.rag_3.chain import get_chain as get_chain_rag_3
 from backend.rag_3.config import validate_config as validate_config_3
 
@@ -23,19 +24,19 @@ async def redirect_root_to_docs() -> RedirectResponse:
     return RedirectResponse("/docs")
 
 
-with initialize(config_path="../backend/rag_1", version_base=None):
-    config_1 = compose(config_name="config")
-    print(config_1)
+# with initialize(config_path="../backend/rag_1", version_base=None):
+#     config_1 = compose(config_name="config")
+#     print(config_1)
 
-    # validate config
-    _ = validate_config_1(config_1)
+#     # validate config
+#     _ = validate_config_1(config_1)
 
-with initialize(config_path="../backend/rag_2", version_base=None):
-    config_2 = compose(config_name="config")
-    print(config_2)
+# with initialize(config_path="../backend/rag_2", version_base=None):
+#     config_2 = compose(config_name="config")
+#     print(config_2)
 
-    # validate config
-    _ = validate_config_2(config_2)
+#     # validate config
+#     _ = validate_config_2(config_2)
 
 with initialize(config_path="../backend/rag_3", version_base=None):
     config_3 = compose(config_name="config")
@@ -51,11 +52,11 @@ async def redirect_root_to_docs() -> RedirectResponse:
 else:
     dependencies = None
 
-chain_rag_1 = get_chain_rag_1(config_1)
-add_routes(app, chain_rag_1, path="/rag-1", dependencies=dependencies)
+# chain_rag_1 = get_chain_rag_1(config_1)
+# add_routes(app, chain_rag_1, path="/rag-1", dependencies=dependencies)
 
-chain_rag_2 = get_chain_rag_2(config_2)
-add_routes(app, chain_rag_2, path="/rag-2", dependencies=dependencies)
+# chain_rag_2 = get_chain_rag_2(config_2)
+# add_routes(app, chain_rag_2, path="/rag-2", dependencies=dependencies)
 
 chain_rag_3 = get_chain_rag_3(config_3)
 add_routes(app, chain_rag_3, path="/rag-3", dependencies=dependencies)