From f91d48aff85cd39589f5944d5059ccda6878374f Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Tue, 21 May 2024 14:53:35 -0700 Subject: [PATCH] refac --- main.py | 2 +- rag_pipeline.py => pipeline.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) rename rag_pipeline.py => pipeline.py (60%) diff --git a/main.py b/main.py index 1116422..048b602 100644 --- a/main.py +++ b/main.py @@ -14,7 +14,7 @@ from utils import get_last_user_message, stream_message_template from schemas import OpenAIChatCompletionForm from config import MODEL_ID, MODEL_NAME -from rag_pipeline import get_response +from pipeline import get_response app = FastAPI(docs_url="/docs", redoc_url=None) diff --git a/rag_pipeline.py b/pipeline.py similarity index 60% rename from rag_pipeline.py rename to pipeline.py index 1be77ee..aeb2a85 100644 --- a/rag_pipeline.py +++ b/pipeline.py @@ -5,8 +5,7 @@ from schemas import OpenAIChatMessage def get_response( user_message: str, messages: List[OpenAIChatMessage] ) -> Union[str, Generator]: - # This is where you can add your custom RAG pipeline. - # Typically, you would retrieve relevant information from your knowledge base and synthesize it to generate a response. + # This is where you can add your custom pipelines like RAG. print(messages) print(user_message)