Class TestEmbeddingModelConfig

tests/test_long_term_memory.py:19–48

测试嵌入模型配置

class TestEmbeddingModelConfig:
    """测试嵌入模型配置"""

    def test_embedding_model_config_creation(self):
        """测试嵌入模型配置创建"""
        config = EmbeddingModelConfig(
            model_name="text-embedding-v2",
            model_type="dashscope",
            api_key="test_key",
            dimensions=1536,
        )

        assert config.model_name == "text-embedding-v2"
        assert config.model_type == "dashscope"
        assert config.api_key == "test_key"
        assert config.dimensions == 1536

    def test_embedding_model_config_env_var(self):
        """测试嵌入模型配置环境变量解析"""
        os.environ["TEST_EMBEDDING_KEY"] = "test_embedding_key"

        config = EmbeddingModelConfig(
            model_name="text-embedding-v2",
            model_type="dashscope",
            api_key="${TEST_EMBEDDING_KEY}",
        )

        assert config.api_key == "test_embedding_key"

        del os.environ["TEST_EMBEDDING_KEY"]


class TestAgentConfigWithLongTermMemory: