personal-consumption-service/tests/TestRunner.py

# @description: 
# @author: licanglong
# @date: 2025/12/19 16:19
import os

from tests.TestEmbeddingStoreApp import EmbeddingStoreApp
from tests.store_embed.case_embed_store import case_embed_update
from tests.store_embed.edges_embed_store import edges_embed_update
from tests.store_embed.merchants_embed_store import merchants_embed_update
from tests.store_embed.rule_embed_store import rule_embed_update

# 设置环境变量
os.environ["APP_PATH"] = os.getenv('APP_PATH') or os.path.abspath(os.path.dirname("../run.py"))  # noqa
from app.App import AppContext  # noqa

AppContext.DEFAULT_LOG_FILE = "logs/test.log"  # noqa

if __name__ == '__main__':
    from tests.TestEmbeddingStoreQueryApp import TestEmbeddingStoreQueryApp

    # TestEmbeddingApp().start()
    EmbeddingStoreApp().start(collection_name="case_embed_store", vector_data=case_embed_update)
    EmbeddingStoreApp().start(collection_name="edges_embed_store", vector_data=edges_embed_update)
    EmbeddingStoreApp().start(collection_name="merchants_embed_store", vector_data=merchants_embed_update)
    EmbeddingStoreApp().start(collection_name="rule_embed_store", vector_data=rule_embed_update)
    # TestEmbeddingStoreQueryApp().start()