test: Add CI test for fastembed

This commit is contained in:
Andrej Milicevic 2025-09-11 17:50:27 +02:00
parent e591c53991
commit a16100f8cd
3 changed files with 38 additions and 0 deletions

View file

@ -11,6 +11,7 @@ WORKFLOWS=(
"test_multimetric_qa_eval_run.yaml"
"test_graphrag_vs_rag_notebook.yml"
"test_gemini.yml"
"test_fastembed.yml"
"test_multimedia_example.yaml"
"test_deduplication.yml"
"test_eval_framework.yml"

28
.github/workflows/test_fastembed.yml vendored Normal file
View file

@ -0,0 +1,28 @@
name: test | fastembed
on:
workflow_call:
jobs:
test-gemini:
name: Run Fastembed Test
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 Fastembed Simple Example
env:
LLM_PROVIDER: "openai"
LLM_API_KEY: ${{ secrets.LLM_API_KEY }}
LLM_MODEL: "gpt-4o-mini"
EMBEDDING_PROVIDER: "fastembed"
EMBEDDING_MODEL: "sentence-transformers/all-MiniLM-L6-v2"
EMBEDDING_DIMENSIONS: "384"
EMBEDDING_MAX_TOKENS: "256"
run: uv run python ./examples/python/simple_example.py

View file

@ -121,6 +121,12 @@ jobs:
uses: ./.github/workflows/test_gemini.yml
secrets: inherit
fastembed-tests:
name: Fastembed Tests
needs: [basic-tests, e2e-tests, cli-tests]
uses: ./.github/workflows/test_fastembed.yml
secrets: inherit
openrouter-tests:
name: OpenRouter Tests
needs: [basic-tests, e2e-tests, cli-tests]
@ -140,6 +146,7 @@ jobs:
vector-db-tests,
example-tests,
gemini-tests,
fastembed-tests,
openrouter-tests,
mcp-test,
relational-db-migration-tests,
@ -163,6 +170,7 @@ jobs:
db-examples-tests,
mcp-test,
gemini-tests,
fastembed-tests,
openrouter-tests,
ollama-tests,
relational-db-migration-tests,
@ -185,6 +193,7 @@ jobs:
"${{ needs.db-examples-tests.result }}" == "success" &&
"${{ needs.relational-db-migration-tests.result }}" == "success" &&
"${{ needs.gemini-tests.result }}" == "success" &&
"${{ needs.fastembed-tests.result }}" == "success" &&
"${{ needs.openrouter-tests.result }}" == "success" &&
"${{ needs.docker-compose-test.result }}" == "success" &&
"${{ needs.docker-ci-test.result }}" == "success" &&