phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-05 10:23:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

chore: removed impl_getter from router function

Browse source 
Refactored the router to accept the implementation directly instead of
using the impl_getter pattern.
The caller already knows which API it's building a router for.for

Signed-off-by: Sébastien Han <seb@redhat.com>
This commit is contained in:
Sébastien Han 2025-11-21 12:02:09 +01:00
parent 8a21d8debe
commit 95e9455335
No known key found for this signature in database
4 changed files with 12 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/llama_stack_api/batches/fastapi_routes.py
View file

@ -11,7 +11,6 @@ FastAPI route decorators. The router is defined in the API package to keep
all API-related code together.
"""
from collections.abc import Callable
from typing import Annotated
from fastapi import APIRouter, Body, Depends, Path, Query
@ -23,16 +22,15 @@ from llama_stack_api.batches.models import (
ListBatchesRequest,
RetrieveBatchRequest,
)
from llama_stack_api.datatypes import Api
from llama_stack_api.router_utils import standard_responses
from llama_stack_api.version import LLAMA_STACK_API_V1
def create_router(impl_getter: Callable[[Api], Batches]) -> APIRouter:
def create_router(impl: Batches) -> APIRouter:
"""Create a FastAPI router for the Batches API.
Args:
impl_getter: Function that returns the Batches implementation for the batches API
impl: The Batches implementation instance
Returns:
APIRouter configured for the Batches API
@ -45,7 +43,7 @@ def create_router(impl_getter: Callable[[Api], Batches]) -> APIRouter:
def get_batch_service() -> Batches:
"""Dependency function to get the batch service implementation."""
return impl_getter(Api.batches)
return impl
@router.post(
"/batches",