phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-01 22:50:00 +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

7
llama_stack/providers/inline/preprocessing/docling/docling.py
View file

@ -11,6 +11,7 @@ from docling_core.transforms.chunker.hybrid_chunker import HybridChunker
from llama_stack.apis.common.content_types import URL
from llama_stack.apis.preprocessing import (
Preprocessing,
PreprocessingDataType,
PreprocessingInput,
PreprocessingResponse,
Preprocessor,
@ -22,6 +23,12 @@ from llama_stack.providers.inline.preprocessing.docling import InlineDoclingConf
class InclineDoclingPreprocessorImpl(Preprocessing, PreprocessorsProtocolPrivate):
# this preprocessor receives URLs / paths to documents as input
INPUT_TYPES = [PreprocessingDataType.document_uri]
# this preprocessor either only converts the documents into a text format, or also chunks them
OUTPUT_TYPES = [PreprocessingDataType.raw_text_document, PreprocessingDataType.chunks]
def __init__(self, config: InlineDoclingConfig) -> None:
self.config = config
self.converter = DocumentConverter()