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_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
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
|
||||
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" &&
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue