mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-03 09:53:45 +00:00
f2306b645c
- Add inline::dana agent provider implementation (stub) - Register provider in agents registry with proper dependencies - Add Dana provider to starter distribution (build.yaml, run.yaml, starter.py) - Add unit tests for provider registration and config validation - Implementation follows meta-reference pattern with NotImplementedError stubs
46 lines
1.8 KiB
Python
46 lines
1.8 KiB
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the terms described in the LICENSE file in
|
|
# the root directory of this source tree.
|
|
|
|
"""
|
|
Test suite for the Dana agent provider implementation (stub).
|
|
|
|
TODO: Add tests when implementation is complete.
|
|
"""
|
|
|
|
from llama_stack.core.datatypes import Api
|
|
from llama_stack.core.distribution import get_provider_registry
|
|
from llama_stack.providers.inline.agents.dana.agents import DanaAgentsImpl
|
|
from llama_stack.providers.inline.agents.dana.config import DanaAgentsImplConfig
|
|
|
|
|
|
def test_dana_provider_in_registry():
|
|
"""Test that the Dana provider is registered and can be found in the registry."""
|
|
provider_registry = get_provider_registry()
|
|
agents_providers = provider_registry.get(Api.agents, {})
|
|
|
|
# Verify the provider is in the registry
|
|
assert "inline::dana" in agents_providers, "Dana provider should be registered"
|
|
|
|
provider_spec = agents_providers["inline::dana"]
|
|
assert provider_spec.provider_type == "inline::dana"
|
|
assert provider_spec.api == Api.agents
|
|
assert provider_spec.module == "llama_stack.providers.inline.agents.dana"
|
|
assert provider_spec.config_class == "llama_stack.providers.inline.agents.dana.DanaAgentsImplConfig"
|
|
|
|
|
|
def test_dana_provider_config():
|
|
"""Test that the Dana provider config can be instantiated."""
|
|
config = DanaAgentsImplConfig.sample_run_config(__distro_dir__="test")
|
|
assert isinstance(config, dict)
|
|
assert "persistence" in config
|
|
assert "agent_state" in config["persistence"]
|
|
assert "responses" in config["persistence"]
|
|
|
|
|
|
def test_dana_provider_class_exists():
|
|
"""Test that Dana provider class exists."""
|
|
assert DanaAgentsImpl is not None
|
|
# TODO: Add actual tests when the provider is implemented
|