test: Add CI test for fastembed
This commit is contained in:
parent
e591c53991
commit
a16100f8cd
3 changed files with 38 additions and 0 deletions
|
|
@ -11,6 +11,7 @@ WORKFLOWS=(
|
||||||
"test_multimetric_qa_eval_run.yaml"
|
"test_multimetric_qa_eval_run.yaml"
|
||||||
"test_graphrag_vs_rag_notebook.yml"
|
"test_graphrag_vs_rag_notebook.yml"
|
||||||
"test_gemini.yml"
|
"test_gemini.yml"
|
||||||
|
"test_fastembed.yml"
|
||||||
"test_multimedia_example.yaml"
|
"test_multimedia_example.yaml"
|
||||||
"test_deduplication.yml"
|
"test_deduplication.yml"
|
||||||
"test_eval_framework.yml"
|
"test_eval_framework.yml"
|
||||||
|
|
|
||||||
28
.github/workflows/test_fastembed.yml
vendored
Normal file
28
.github/workflows/test_fastembed.yml
vendored
Normal 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
|
||||||
9
.github/workflows/test_suites.yml
vendored
9
.github/workflows/test_suites.yml
vendored
|
|
@ -121,6 +121,12 @@ jobs:
|
||||||
uses: ./.github/workflows/test_gemini.yml
|
uses: ./.github/workflows/test_gemini.yml
|
||||||
secrets: inherit
|
secrets: inherit
|
||||||
|
|
||||||
|
fastembed-tests:
|
||||||
|
name: Fastembed Tests
|
||||||
|
needs: [basic-tests, e2e-tests, cli-tests]
|
||||||
|
uses: ./.github/workflows/test_fastembed.yml
|
||||||
|
secrets: inherit
|
||||||
|
|
||||||
openrouter-tests:
|
openrouter-tests:
|
||||||
name: OpenRouter Tests
|
name: OpenRouter Tests
|
||||||
needs: [basic-tests, e2e-tests, cli-tests]
|
needs: [basic-tests, e2e-tests, cli-tests]
|
||||||
|
|
@ -140,6 +146,7 @@ jobs:
|
||||||
vector-db-tests,
|
vector-db-tests,
|
||||||
example-tests,
|
example-tests,
|
||||||
gemini-tests,
|
gemini-tests,
|
||||||
|
fastembed-tests,
|
||||||
openrouter-tests,
|
openrouter-tests,
|
||||||
mcp-test,
|
mcp-test,
|
||||||
relational-db-migration-tests,
|
relational-db-migration-tests,
|
||||||
|
|
@ -163,6 +170,7 @@ jobs:
|
||||||
db-examples-tests,
|
db-examples-tests,
|
||||||
mcp-test,
|
mcp-test,
|
||||||
gemini-tests,
|
gemini-tests,
|
||||||
|
fastembed-tests,
|
||||||
openrouter-tests,
|
openrouter-tests,
|
||||||
ollama-tests,
|
ollama-tests,
|
||||||
relational-db-migration-tests,
|
relational-db-migration-tests,
|
||||||
|
|
@ -185,6 +193,7 @@ jobs:
|
||||||
"${{ needs.db-examples-tests.result }}" == "success" &&
|
"${{ needs.db-examples-tests.result }}" == "success" &&
|
||||||
"${{ needs.relational-db-migration-tests.result }}" == "success" &&
|
"${{ needs.relational-db-migration-tests.result }}" == "success" &&
|
||||||
"${{ needs.gemini-tests.result }}" == "success" &&
|
"${{ needs.gemini-tests.result }}" == "success" &&
|
||||||
|
"${{ needs.fastembed-tests.result }}" == "success" &&
|
||||||
"${{ needs.openrouter-tests.result }}" == "success" &&
|
"${{ needs.openrouter-tests.result }}" == "success" &&
|
||||||
"${{ needs.docker-compose-test.result }}" == "success" &&
|
"${{ needs.docker-compose-test.result }}" == "success" &&
|
||||||
"${{ needs.docker-ci-test.result }}" == "success" &&
|
"${{ needs.docker-ci-test.result }}" == "success" &&
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue