diff --git a/dsRag/Start.py b/dsRag/Start.py index f6a3e5b9..05e547b5 100644 --- a/dsRag/Start.py +++ b/dsRag/Start.py @@ -7,10 +7,9 @@ import uvicorn from fastapi import FastAPI, Request, Body from openai import OpenAI from sse_starlette.sse import EventSourceResponse -from starlette.staticfiles import StaticFiles from gensim.models import KeyedVectors from Config import Config -from Milvus.Config.MulvusConfig import * +from Config.Config import * from Milvus.Utils.MilvusCollectionManager import MilvusCollectionManager from Milvus.Utils.MilvusConnectionPool import * from Milvus.Utils.MilvusConnectionPool import MilvusConnectionPool @@ -195,5 +194,6 @@ async def rag_stream(request: Request, query: str = Body(...)): ) ) + if __name__ == "__main__": uvicorn.run(app, host="0.0.0.0", port=8000)