Llama Stack recently moved the API definitions and datatypes in a separate package llama_stack_api. So all imports should use llama_stack_api instead of llama_stack.api.
We are working on publishing llama_stack_api as a package too. It's not 100% needed the provider always runs next to llama-stack which brings llama_stack_api with it. But it's likely useful for your testing.