diff --git a/.github/workflows/disable_independent_workflows.sh b/.github/workflows/disable_independent_workflows.sh index 693c3092d..9ec4a711e 100755 --- a/.github/workflows/disable_independent_workflows.sh +++ b/.github/workflows/disable_independent_workflows.sh @@ -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" diff --git a/.github/workflows/test_fastembed.yml b/.github/workflows/test_fastembed.yml new file mode 100644 index 000000000..5d37c1118 --- /dev/null +++ b/.github/workflows/test_fastembed.yml @@ -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 diff --git a/.github/workflows/test_suites.yml b/.github/workflows/test_suites.yml index b6e84ac1e..06e3e10da 100644 --- a/.github/workflows/test_suites.yml +++ b/.github/workflows/test_suites.yml @@ -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" &&