mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-16 15:19:29 +00:00
661985e240
Some checks failed
SqlStore Integration Tests / test-postgres (3.13) (push) Failing after 0s
Integration Auth Tests / test-matrix (oauth2_token) (push) Failing after 1s
Integration Tests (Replay) / generate-matrix (push) Successful in 3s
SqlStore Integration Tests / test-postgres (3.12) (push) Failing after 4s
Test Llama Stack Build / generate-matrix (push) Failing after 3s
Test Llama Stack Build / build (push) Has been skipped
Test External Providers Installed via Module / test-external-providers-from-module (venv) (push) Has been skipped
Test llama stack list-deps / generate-matrix (push) Failing after 3s
Test llama stack list-deps / list-deps (push) Has been skipped
API Conformance Tests / check-schema-compatibility (push) Successful in 11s
Python Package Build Test / build (3.13) (push) Successful in 19s
Python Package Build Test / build (3.12) (push) Successful in 23s
Test Llama Stack Build / build-single-provider (push) Successful in 33s
Test llama stack list-deps / show-single-provider (push) Successful in 36s
Test llama stack list-deps / list-deps-from-config (push) Successful in 44s
Vector IO Integration Tests / test-matrix (push) Failing after 57s
Test External API and Providers / test-external (venv) (push) Failing after 1m37s
Unit Tests / unit-tests (3.12) (push) Failing after 1m56s
UI Tests / ui-tests (22) (push) Successful in 2m2s
Unit Tests / unit-tests (3.13) (push) Failing after 2m35s
Pre-commit / pre-commit (22) (push) Successful in 3m16s
Test Llama Stack Build / build-custom-container-distribution (push) Successful in 3m34s
Test Llama Stack Build / build-ubi9-container-distribution (push) Successful in 3m59s
Integration Tests (Replay) / Integration Tests (, , , client=, ) (push) Failing after 4m30s
# What does this PR do? the build.yaml is only used in the following ways: 1. list-deps 2. distribution code-gen since `llama stack build` no longer exists, I found myself asking "why do we need two different files for list-deps and run"? Removing the BuildConfig and altering the usage of the DistributionTemplate in llama stack list-deps is the first step in removing the build yaml entirely. Removing the BuildConfig and build.yaml cuts the files users need to maintain in half, and allows us to focus on the stability of _just_ the run.yaml This PR removes the build.yaml, BuildConfig datatype, and its usage throughout the codebase. Users are now expected to point to run.yaml files when running list-deps, and our codebase automatically uses these types now for things like `get_provider_registry`. **Additionally, two renames: `StackRunConfig` -> `StackConfig` and `run.yaml` -> `config.yaml`.** The build.yaml made sense for when we were managing the build process for the user and actually _producing_ a run.yaml _from_ the build.yaml, but now that we are simply just getting the provider registry and listing the deps, switching to config.yaml simplifies the scope here greatly. ## Test Plan existing list-deps usage should work in the tests. --------- Signed-off-by: Charlie Doern <cdoern@redhat.com>
71 lines
2.7 KiB
Python
71 lines
2.7 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.
|
|
|
|
import yaml
|
|
|
|
from llama_stack.core.datatypes import StackConfig
|
|
from llama_stack.core.storage.datatypes import (
|
|
PostgresKVStoreConfig,
|
|
PostgresSqlStoreConfig,
|
|
SqliteKVStoreConfig,
|
|
SqliteSqlStoreConfig,
|
|
)
|
|
|
|
|
|
def test_starter_distribution_config_loads_and_resolves():
|
|
"""Integration: Actual starter config should parse and have correct storage structure."""
|
|
with open("llama_stack/distributions/starter/config.yaml") as f:
|
|
config_dict = yaml.safe_load(f)
|
|
|
|
config = StackConfig(**config_dict)
|
|
|
|
# Config should have named backends and explicit store references
|
|
assert config.storage is not None
|
|
assert "kv_default" in config.storage.backends
|
|
assert "sql_default" in config.storage.backends
|
|
assert isinstance(config.storage.backends["kv_default"], SqliteKVStoreConfig)
|
|
assert isinstance(config.storage.backends["sql_default"], SqliteSqlStoreConfig)
|
|
|
|
stores = config.storage.stores
|
|
assert stores.metadata is not None
|
|
assert stores.metadata.backend == "kv_default"
|
|
assert stores.metadata.namespace == "registry"
|
|
|
|
assert stores.inference is not None
|
|
assert stores.inference.backend == "sql_default"
|
|
assert stores.inference.table_name == "inference_store"
|
|
assert stores.inference.max_write_queue_size > 0
|
|
assert stores.inference.num_writers > 0
|
|
|
|
assert stores.conversations is not None
|
|
assert stores.conversations.backend == "sql_default"
|
|
assert stores.conversations.table_name == "openai_conversations"
|
|
|
|
|
|
def test_postgres_demo_distribution_config_loads():
|
|
"""Integration: Postgres demo should use Postgres backend for all stores."""
|
|
with open("llama_stack/distributions/postgres-demo/config.yaml") as f:
|
|
config_dict = yaml.safe_load(f)
|
|
|
|
config = StackConfig(**config_dict)
|
|
|
|
# Should have postgres backend
|
|
assert config.storage is not None
|
|
assert "kv_default" in config.storage.backends
|
|
assert "sql_default" in config.storage.backends
|
|
postgres_backend = config.storage.backends["sql_default"]
|
|
assert isinstance(postgres_backend, PostgresSqlStoreConfig)
|
|
assert postgres_backend.host == "${env.POSTGRES_HOST:=localhost}"
|
|
|
|
kv_backend = config.storage.backends["kv_default"]
|
|
assert isinstance(kv_backend, PostgresKVStoreConfig)
|
|
|
|
stores = config.storage.stores
|
|
# Stores target the Postgres backends explicitly
|
|
assert stores.metadata is not None
|
|
assert stores.metadata.backend == "kv_default"
|
|
assert stores.inference is not None
|
|
assert stores.inference.backend == "sql_default"
|