mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-03 09:53:45 +00:00
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:
parent
3770963130
commit
9a2b4efabd
5 changed files with 11 additions and 11 deletions
|
|
@ -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:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue