phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-31 11:40:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Simplified the preprocessing interface.

Browse source
This commit is contained in:
ilya-kolchinsky 2025-03-11 13:05:48 +01:00
parent 1eeba2cc8a
commit ad4cf97604
8 changed files with 31 additions and 43 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/distribution/utils/chain.py
View file

@ -8,16 +8,16 @@ from itertools import pairwise
from typing import List
from llama_stack.apis.preprocessing import (
Preprocessing,
PreprocessingDataElement,
PreprocessorChain,
PreprocessorResponse,
)
from llama_stack.providers.datatypes import PreprocessorsProtocolPrivate
log = logging.getLogger(__name__)
def validate_chain(chain_impls: List[Preprocessing]) -> bool:
def validate_chain(chain_impls: List[PreprocessorsProtocolPrivate]) -> bool:
if len(chain_impls) == 0:
log.error("Empty preprocessing chain was provided")
return False
@ -37,7 +37,7 @@ def validate_chain(chain_impls: List[Preprocessing]) -> bool:
async def execute_preprocessor_chain(
preprocessor_chain: PreprocessorChain,
preprocessor_chain_impls: List[Preprocessing],
preprocessor_chain_impls: List[PreprocessorsProtocolPrivate],
preprocessor_inputs: List[PreprocessingDataElement],
) -> PreprocessorResponse:
if not validate_chain(preprocessor_chain_impls):
@ -50,7 +50,7 @@ async def execute_preprocessor_chain(
# TODO: replace with a parallel implementation
for i, current_params in enumerate(preprocessor_chain):
current_impl = preprocessor_chain_impls[i]
response = await current_impl.preprocess(
response = await current_impl.do_preprocess(
preprocessor_id=current_params.preprocessor_id,
preprocessor_inputs=current_inputs,
options=current_params.options,