From 5422f1f62dc90d59a4df974a185a66159af120a4 Mon Sep 17 00:00:00 2001 From: Baptiste Pasquier Date: Thu, 21 Mar 2024 11:28:13 +0100 Subject: [PATCH] keep only rag 3 --- app/server.py | 37 +++++++++++++++++++------------------ 1 file changed, 19 insertions(+), 18 deletions(-) diff --git a/app/server.py b/app/server.py index c012fb1..87aec05 100644 --- 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)