mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-17 06:12:36 +00:00
refactor: use DEFAULT_VECTOR_IO_DEPS variable for chardet and pypdf
Consolidate the common vector IO dependencies (chardet, pypdf) into a single DEFAULT_VECTOR_IO_DEPS variable to reduce duplication across all vector IO provider registries and the RAG tool runtime.
This commit is contained in:
parent
da6e55b136
commit
6c422cadab
3 changed files with 61 additions and 74 deletions
|
|
@ -11,6 +11,7 @@ from llama_stack.providers.datatypes import (
|
|||
ProviderSpec,
|
||||
RemoteProviderSpec,
|
||||
)
|
||||
from llama_stack.providers.registry.vector_io import DEFAULT_VECTOR_IO_DEPS
|
||||
|
||||
|
||||
def available_providers() -> list[ProviderSpec]:
|
||||
|
|
@ -18,9 +19,8 @@ def available_providers() -> list[ProviderSpec]:
|
|||
InlineProviderSpec(
|
||||
api=Api.tool_runtime,
|
||||
provider_type="inline::rag-runtime",
|
||||
pip_packages=[
|
||||
"chardet",
|
||||
"pypdf",
|
||||
pip_packages=DEFAULT_VECTOR_IO_DEPS
|
||||
+ [
|
||||
"tqdm",
|
||||
"numpy",
|
||||
"scikit-learn",
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue