main
HuangHai 4 weeks ago
parent 1c473d8b6e
commit 0f7fa471b5

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

Loading…
Cancel
Save