调整入口代码结构
This commit is contained in:
+14
-13
@@ -1,4 +1,5 @@
|
|||||||
from dotenv import load_dotenv
|
from dotenv import load_dotenv
|
||||||
|
from llama_index.core.node_parser import SentenceSplitter
|
||||||
|
|
||||||
load_dotenv()
|
load_dotenv()
|
||||||
|
|
||||||
@@ -13,14 +14,18 @@ from app.api.routers.upload import file_upload_router
|
|||||||
from app.settings import init_settings
|
from app.settings import init_settings
|
||||||
from app.observability import init_observability
|
from app.observability import init_observability
|
||||||
from fastapi.staticfiles import StaticFiles
|
from fastapi.staticfiles import StaticFiles
|
||||||
|
from phoenix.trace import using_project
|
||||||
|
|
||||||
logger = logging.getLogger("uvicorn")
|
logger = logging.getLogger("uvicorn")
|
||||||
app = None
|
|
||||||
|
|
||||||
def init_webserver():
|
usPrj = using_project(os.getenv("PHOENIX_PROJECT_NAME"))
|
||||||
global app
|
usPrj.__enter__()
|
||||||
|
|
||||||
|
init_settings()
|
||||||
|
init_observability()
|
||||||
|
|
||||||
app = FastAPI()
|
app = FastAPI()
|
||||||
|
|
||||||
environment = os.getenv("ENVIRONMENT", "dev") # Default to 'development' if not set
|
environment = os.getenv("ENVIRONMENT", "dev") # Default to 'development' if not set
|
||||||
if environment == "dev":
|
if environment == "dev":
|
||||||
logger.warning("Running in development mode - allowing CORS for all origins")
|
logger.warning("Running in development mode - allowing CORS for all origins")
|
||||||
@@ -52,16 +57,12 @@ def init_webserver():
|
|||||||
async def redirect_to_docs():
|
async def redirect_to_docs():
|
||||||
return RedirectResponse(url="/docs")
|
return RedirectResponse(url="/docs")
|
||||||
|
|
||||||
|
SentenceSplitter
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
from phoenix.trace import using_project
|
|
||||||
with using_project(os.getenv("PHOENIX_PROJECT_NAME")) as obj:
|
|
||||||
|
|
||||||
init_settings()
|
|
||||||
init_observability()
|
|
||||||
init_webserver()
|
|
||||||
|
|
||||||
app_host = os.getenv("APP_HOST", "0.0.0.0")
|
app_host = os.getenv("APP_HOST", "0.0.0.0")
|
||||||
app_port = int(os.getenv("APP_PORT", "8000"))
|
app_port = int(os.getenv("APP_PORT", "8000"))
|
||||||
#reload = True if environment == "dev" else False
|
reload = True if environment == "dev" else False
|
||||||
reload = False
|
reload = False
|
||||||
uvicorn.run(app=app, host=app_host, port=app_port, reload=reload)
|
uvicorn.run(app="main:app", host=app_host, port=app_port, reload=reload)
|
||||||
|
|
||||||
|
#usPrj.__exit__()
|
||||||
|
|||||||
Reference in New Issue
Block a user