Skip to content

Commit 3646712

Browse files
authored
Feat/add time step (#331)
* fix time * add timed step for llm * fix:ci code * fix:test * remove * update * feat: remove request body * feat: remove trace id * feat: remove trace id * feat: add workers * feat: update load fast api
1 parent 312223c commit 3646712

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/memos/api/product_api.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
from fastapi import FastAPI
44

55
from memos.api.exceptions import APIExceptionHandler
6+
from memos.api.middleware.request_context import RequestContextMiddleware
67
from memos.api.routers.product_router import router as product_router
78

89

@@ -16,7 +17,7 @@
1617
version="1.0.1",
1718
)
1819

19-
20+
app.add_middleware(RequestContextMiddleware)
2021
# Include routers
2122
app.include_router(product_router)
2223

@@ -32,5 +33,6 @@
3233

3334
parser = argparse.ArgumentParser()
3435
parser.add_argument("--port", type=int, default=8001)
36+
parser.add_argument("--workers", type=int, default=32)
3537
args = parser.parse_args()
36-
uvicorn.run(app, host="0.0.0.0", port=args.port)
38+
uvicorn.run(app, host="0.0.0.0", port=args.port, workers=args.workers)

0 commit comments

Comments
 (0)