phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 23:29:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added remaining files

Browse source
This commit is contained in:
Sarthak Deshpande 2024-10-21 20:33:08 +05:30
parent 46795b9bbf
commit 8e213e049c
4 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/tests/inference/test_inference.py
View file

@ -71,8 +71,8 @@ async def inference_settings(request):
)
return {
"impl": impls['impls'][Api.inference],
"models_impl": impls['impls'][Api.models],
"impl": impls[Api.inference],
"models_impl": impls[Api.models],
"common_params": {
"model": model,
"tool_choice": ToolChoice.auto,

4
llama_stack/providers/tests/memory/test_memory.py
View file

@ -36,8 +36,8 @@ async def memory_settings():
Api.memory,
)
return {
"memory_impl": impls['impls'][Api.memory],
"memory_banks_impl": impls['impls'][Api.memory_banks],
"memory_impl": impls[Api.memory],
"memory_banks_impl": impls[Api.memory_banks],
}

2
llama_stack/providers/tests/resolver.py
View file

@ -46,7 +46,7 @@ async def resolve_impls_for_test(api: Api, deps: List[Api] = None):
{"X-LlamaStack-ProviderData": json.dumps(provider_data)}
)
return {"impls": impls, "persistence": config_dict['providers']['agents'][0]['config']['persistence_store']}
return impls
def read_providers(api: Api, config_dict: Dict[str, Any]) -> Dict[str, Any]:

4
llama_stack/providers/tests/safety/test_safety.py
View file

@ -36,8 +36,8 @@ async def safety_settings():
impls = await resolve_impls_for_test(Api.safety, deps=[Api.inference])
return {
"impl": impls['impls'][Api.safety],
"shields_impl": impls['impls'][Api.shields],
"impl": impls[Api.safety],
"shields_impl": impls[Api.shields],
}