mem0/tests/test_utils.py

import yaml

from embedchain.utils import validate_yaml_config

CONFIG_YAMLS = [
    "configs/anthropic.yaml",
    "configs/azure_openai.yaml",
    "configs/chroma.yaml",
    "configs/chunker.yaml",
    "configs/cohere.yaml",
    "configs/full-stack.yaml",
    "configs/gpt4all.yaml",
    "configs/huggingface.yaml",
    "configs/jina.yaml",
    "configs/llama2.yaml",
    "configs/opensearch.yaml",
    "configs/opensource.yaml",
    "configs/pinecone.yaml",
    "configs/vertexai.yaml",
    "configs/weaviate.yaml",
]


class TestAllConfigYamls:
    def test_all_config_yamls(self):
        """Test that all config yamls are valid."""
        for config_yaml in CONFIG_YAMLS:
            with open(config_yaml, "r") as f:
                config = yaml.safe_load(f)
            assert config is not None

            try:
                validate_yaml_config(config)
            except Exception as e:
                print(f"Error in {config_yaml}: {e}")
                raise e