mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-03 01:48:05 +00:00
fix: more accurate type
https://github.com/llamastack/llama-stack/pull/4191#discussion_r2557389025 Signed-off-by: Sébastien Han <seb@redhat.com>
This commit is contained in:
parent
b0b3034f16
commit
3dc5b5d3a0
1 changed files with 2 additions and 1 deletions
|
|
@ -10,6 +10,7 @@ This module provides utilities to create FastAPI routers from API packages.
|
|||
APIs with routers are explicitly listed here.
|
||||
"""
|
||||
|
||||
from collections.abc import Callable
|
||||
from typing import Any, cast
|
||||
|
||||
from fastapi import APIRouter
|
||||
|
|
@ -19,7 +20,7 @@ from fastapi import APIRouter
|
|||
from llama_stack_api.batches.fastapi_routes import create_router as create_batches_router
|
||||
from llama_stack_api.datatypes import Api
|
||||
|
||||
_ROUTER_FACTORIES: dict[str, Any] = {
|
||||
_ROUTER_FACTORIES: dict[str, Callable[[Any], APIRouter]] = {
|
||||
"batches": create_batches_router,
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue