embedchain

1
import yaml
2

3
from embedchain.utils.misc import validate_config
4

5
CONFIG_YAMLS = [
6
    "configs/anthropic.yaml",
7
    "configs/azure_openai.yaml",
8
    "configs/chroma.yaml",
9
    "configs/chunker.yaml",
10
    "configs/cohere.yaml",
11
    "configs/together.yaml",
12
    "configs/ollama.yaml",
13
    "configs/full-stack.yaml",
14
    "configs/gpt4.yaml",
15
    "configs/gpt4all.yaml",
16
    "configs/huggingface.yaml",
17
    "configs/jina.yaml",
18
    "configs/llama2.yaml",
19
    "configs/opensearch.yaml",
20
    "configs/opensource.yaml",
21
    "configs/pinecone.yaml",
22
    "configs/vertexai.yaml",
23
    "configs/weaviate.yaml",
24
]
25

26

27
def test_all_config_yamls():
28
    """Test that all config yamls are valid."""
29
    for config_yaml in CONFIG_YAMLS:
30
        with open(config_yaml, "r") as f:
31
            config = yaml.safe_load(f)
32
        assert config is not None
33

34
        try:
35
            validate_config(config)
36
        except Exception as e:
37
            print(f"Error in {config_yaml}: {e}")
38
            raise e
39