chore: clarify function and log about which router

It's FastAPI

Signed-off-by: Sébastien Han <seb@redhat.com>
This commit is contained in:
Sébastien Han 2025-11-25 10:51:52 +01:00
parent 3770963130
commit 9a2b4efabd
No known key found for this signature in database
5 changed files with 11 additions and 11 deletions

View file

@ -14,7 +14,7 @@ from typing import Any
from fastapi import FastAPI
from llama_stack.core.resolver import api_protocol_map
from llama_stack.core.server.fastapi_router_registry import build_router
from llama_stack.core.server.fastapi_router_registry import build_fastapi_router
from llama_stack_api import Api
from .state import _protocol_methods_cache
@ -81,7 +81,7 @@ def create_llama_stack_app() -> FastAPI:
protocols = api_protocol_map()
for api in protocols.keys():
# For OpenAPI generation, we don't need a real implementation
router = build_router(api, None)
router = build_fastapi_router(api, None)
if router:
app.include_router(router)
@ -95,7 +95,7 @@ def create_llama_stack_app() -> FastAPI:
for api, routes in api_routes.items():
# Skip APIs that have routers - they're already included above
if build_router(api, None) is not None:
if build_fastapi_router(api, None) is not None:
continue
for route, webmethod in routes: