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

Added input/output type declaration.

Browse source
This commit is contained in:
ilya-kolchinsky 2025-03-04 12:34:30 +01:00
parent 224d2d2891
commit 5014de434e
2 changed files with 15 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
llama_stack/apis/preprocessing/preprocessing.py
View file

@ -14,15 +14,19 @@ from llama_stack.apis.vector_io import Chunk
from llama_stack.schema_utils import json_schema_type, webmethod
class PreprocessingInputType(Enum):
document_content = "document_content"
document_path = "document_path"
class PreprocessingDataType(Enum):
document_uri = "document_uri"
document_directory_uri = "document_directory_uri"
binary_document = "binary_document"
raw_text_document = "raw_text_document"
chunks = "chunks"
@json_schema_type
class PreprocessingInput(BaseModel):
preprocessor_input_id: str
preprocessor_input_type: Optional[PreprocessingInputType]
preprocessor_input_type: Optional[PreprocessingDataType]
path_or_content: str | URL