Added tests for persistence

This commit is contained in:
Sarthak Deshpande 2024-10-21 15:56:54 +05:30
parent cae5b0708b
commit 862ff6c653
7 changed files with 157 additions and 11 deletions

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
return {"impls": impls, "persistence": config_dict['providers']['agents'][0]['config']['persistence_store']}
def read_providers(api: Api, config_dict: Dict[str, Any]) -> Dict[str, Any]: