diff --git a/.github/workflows/examples_tests.yml b/.github/workflows/examples_tests.yml index 809e87e0b..370b79ad2 100644 --- a/.github/workflows/examples_tests.yml +++ b/.github/workflows/examples_tests.yml @@ -109,3 +109,27 @@ jobs: EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} run: poetry run python ./examples/python/dynamic_steps_example.py + + test-starter-pipelines: + name: Run Starter Kit Pipeline Tests + runs-on: ubuntu-22.04 + steps: + - name: Check out repository + uses: actions/checkout@v4 + + - name: Cognee Setup + uses: ./.github/actions/cognee_setup + with: + python-version: '3.11.x' + + - name: Run Starter Kit Pipeline Tests + env: + LLM_API_KEY: ${{ secrets.LLM_API_KEY }} + EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} + LLM_MODEL: ${{ secrets.LLM_MODEL }} + LLM_ENDPOINT: ${{ secrets.LLM_ENDPOINT }} + LLM_API_VERSION: ${{ secrets.LLM_API_VERSION }} + EMBEDDING_MODEL: ${{ secrets.EMBEDDING_MODEL }} + EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} + EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} + run: poetry run python -m pytest cognee/tests/test_starter_pipelines.py -v diff --git a/cognee/tests/test_starter_pipelines.py b/cognee/tests/test_starter_pipelines.py index 97e9d0881..5d1078ec8 100644 --- a/cognee/tests/test_starter_pipelines.py +++ b/cognee/tests/test_starter_pipelines.py @@ -10,7 +10,7 @@ class TestPipelines(unittest.TestCase): def setUp(self): # Ensure we're in the correct directory self.project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../..")) - self.pipelines_dir = os.path.join(self.project_root, "src", "pipelines") + self.pipelines_dir = os.path.join(self.project_root, "cognee-starter-kit", "src", "pipelines") # Required environment variables self.required_env_vars = ["LLM_API_KEY", "EMBEDDING_API_KEY"]