phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-10 19:43:16 +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 { return {
"impl": impls['impls'][Api.inference], "impl": impls[Api.inference],
"models_impl": impls['impls'][Api.models], "models_impl": impls[Api.models],
"common_params": { "common_params": {
"model": model, "model": model,
"tool_choice": ToolChoice.auto, "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, Api.memory,
) )
return { return {
"memory_impl": impls['impls'][Api.memory], "memory_impl": impls[Api.memory],
"memory_banks_impl": impls['impls'][Api.memory_banks], "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)} {"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]: 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]) impls = await resolve_impls_for_test(Api.safety, deps=[Api.inference])
return { return {
"impl": impls['impls'][Api.safety], "impl": impls[Api.safety],
"shields_impl": impls['impls'][Api.shields], "shields_impl": impls[Api.shields],
} }