phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-03 14:02:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Restrict the changes to the new preprocessing API only.

Browse source
This commit is contained in:
ilya-kolchinsky 2025-04-03 12:19:08 +02:00
parent 2008cd7921
commit 863f87aa15
90 changed files with 104 additions and 1138 deletions
Download patch file Download diff file Expand all files Collapse all files

21
llama_stack/providers/registry/preprocessing.py
View file

@ -7,28 +7,9 @@
from typing import List
from llama_stack.providers.datatypes import (
Api,
InlineProviderSpec,
ProviderSpec,
)
def available_providers() -> List[ProviderSpec]:
return [
InlineProviderSpec(
api=Api.preprocessing,
provider_type="inline::basic",
pip_packages=["httpx", "pypdf"],
module="llama_stack.providers.inline.preprocessing.basic",
config_class="llama_stack.providers.inline.preprocessing.basic.InlineBasicPreprocessorConfig",
api_dependencies=[],
),
InlineProviderSpec(
api=Api.preprocessing,
provider_type="inline::simple_chunking",
pip_packages=[],
module="llama_stack.providers.inline.preprocessing.simple_chunking",
config_class="llama_stack.providers.inline.preprocessing.simple_chunking.InclineSimpleChunkingConfig",
api_dependencies=[],
),
]
return []

2
llama_stack/providers/registry/tool_runtime.py
View file

@ -34,7 +34,7 @@ def available_providers() -> List[ProviderSpec]:
],
module="llama_stack.providers.inline.tool_runtime.rag",
config_class="llama_stack.providers.inline.tool_runtime.rag.config.RagToolRuntimeConfig",
api_dependencies=[Api.vector_io, Api.inference, Api.preprocessing],
api_dependencies=[Api.vector_io, Api.inference],
),
InlineProviderSpec(
api=Api.tool_runtime,