added new tests

This commit is contained in:
vasilije 2025-06-24 21:30:59 +02:00
parent 456f3b58c0
commit f8707a43c6
2 changed files with 25 additions and 1 deletions

View file

@ -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

View file

@ -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"]