mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-07-29 15:23:51 +00:00
fix issue w/ enforcing api
This commit is contained in:
parent
21f2e9adf5
commit
07d45f2af3
3 changed files with 14 additions and 7 deletions
|
@ -35,6 +35,7 @@ def builtin_automatically_routed_apis() -> List[AutoRoutedApiInfo]:
|
||||||
routing_table_api=Api.memory_banks,
|
routing_table_api=Api.memory_banks,
|
||||||
router_api=Api.memory,
|
router_api=Api.memory,
|
||||||
),
|
),
|
||||||
|
# TODO: re-enable once we have proper checking on builtin routing: Issue #297
|
||||||
AutoRoutedApiInfo(
|
AutoRoutedApiInfo(
|
||||||
routing_table_api=Api.datasets,
|
routing_table_api=Api.datasets,
|
||||||
router_api=Api.datasetio,
|
router_api=Api.datasetio,
|
||||||
|
|
|
@ -288,6 +288,8 @@ def main(
|
||||||
apis_to_serve = set(impls.keys())
|
apis_to_serve = set(impls.keys())
|
||||||
|
|
||||||
for inf in builtin_automatically_routed_apis():
|
for inf in builtin_automatically_routed_apis():
|
||||||
|
if inf.router_api.value not in apis_to_serve:
|
||||||
|
continue
|
||||||
apis_to_serve.add(inf.routing_table_api.value)
|
apis_to_serve.add(inf.routing_table_api.value)
|
||||||
|
|
||||||
apis_to_serve.add("inspect")
|
apis_to_serve.add("inspect")
|
||||||
|
|
|
@ -13,14 +13,18 @@ apis:
|
||||||
- safety
|
- safety
|
||||||
providers:
|
providers:
|
||||||
inference:
|
inference:
|
||||||
- provider_id: meta-reference
|
- provider_id: tgi0
|
||||||
provider_type: meta-reference
|
provider_type: remote::tgi
|
||||||
config:
|
config:
|
||||||
model: Llama3.1-8B-Instruct
|
url: http://127.0.0.1:5009
|
||||||
quantization: null
|
# - provider_id: meta-reference
|
||||||
torch_seed: null
|
# provider_type: meta-reference
|
||||||
max_seq_len: 4096
|
# config:
|
||||||
max_batch_size: 1
|
# model: Llama3.1-8B-Instruct
|
||||||
|
# quantization: null
|
||||||
|
# torch_seed: null
|
||||||
|
# max_seq_len: 4096
|
||||||
|
# max_batch_size: 1
|
||||||
safety:
|
safety:
|
||||||
- provider_id: meta-reference
|
- provider_id: meta-reference
|
||||||
provider_type: meta-reference
|
provider_type: meta-reference
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue