mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-10-04 04:04:14 +00:00
Merge 49b729b30a
into 8422bd102a
This commit is contained in:
commit
6d68ece4ef
4 changed files with 433 additions and 0 deletions
|
@ -80,6 +80,7 @@ from llama_stack.providers.utils.telemetry.tracing import (
|
|||
)
|
||||
|
||||
from .auth import AuthenticationMiddleware
|
||||
from .metrics import RequestMetricsMiddleware
|
||||
from .quota import QuotaMiddleware
|
||||
|
||||
REPO_ROOT = Path(__file__).parent.parent.parent.parent
|
||||
|
@ -556,6 +557,10 @@ def main(args: argparse.Namespace | None = None):
|
|||
app.__llama_stack_impls__ = impls
|
||||
app.add_middleware(TracingMiddleware, impls=impls, external_apis=external_apis)
|
||||
|
||||
# Add request metrics middleware
|
||||
telemetry_impl = impls.get(Api.telemetry) if Api.telemetry in impls else None
|
||||
app.add_middleware(RequestMetricsMiddleware, telemetry=telemetry_impl)
|
||||
|
||||
import uvicorn
|
||||
|
||||
# Configure SSL if certificates are provided
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue