phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-04 20:14:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

faiss provider implementation

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-23 20:58:27 -07:00
parent 14637bea66
commit a08958c000
9 changed files with 401 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
llama_toolchain/distribution/registry.py
View file

@ -32,6 +32,7 @@ def available_distribution_specs() -> List[DistributionSpec]:
description="Use code from `llama_toolchain` itself to serve all llama stack APIs",
provider_specs={
Api.inference: providers[Api.inference]["meta-reference"],
Api.memory: providers[Api.memory]["meta-reference-faiss"],
Api.safety: providers[Api.safety]["meta-reference"],
Api.agentic_system: providers[Api.agentic_system]["meta-reference"],
},
@ -50,6 +51,12 @@ def available_distribution_specs() -> List[DistributionSpec]:
Api.agentic_system: providers[Api.agentic_system]["meta-reference"],
},
),
DistributionSpec(
spec_id="test-memory",
provider_specs={
Api.memory: providers[Api.memory]["meta-reference-faiss"],
},
),
]