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)