diff --git a/.github/actions/cognee_setup/action.yml b/.github/actions/cognee_setup/action.yml index 8a3bbb7ef..e79619221 100644 --- a/.github/actions/cognee_setup/action.yml +++ b/.github/actions/cognee_setup/action.yml @@ -1,5 +1,5 @@ name: cognee-setup -description: "Sets up Python, installs Poetry, loads venv from cache, and installs dependencies for Cognee." +description: "Sets up Python, installs uv, and installs dependencies for Cognee." inputs: python-version: @@ -16,18 +16,11 @@ runs: with: python-version: ${{ inputs.python-version }} - - name: Install Poetry - shell: bash - run: | - python -m pip install --upgrade pip - pip install poetry - - - name: Rebuild Poetry lock file - shell: bash - run: | - rm poetry.lock - poetry lock + - name: Install uv + uses: astral-sh/setup-uv@v4 + with: + enable-cache: true - name: Install dependencies shell: bash - run: poetry install --no-interaction -E api -E docs -E evals -E gemini -E codegraph -E ollama -E dev -E neo4j + run: uv sync --extra api --extra docs --extra evals --extra gemini --extra codegraph --extra ollama --extra dev --extra neo4j diff --git a/.github/workflows/basic_tests.yml b/.github/workflows/basic_tests.yml index 28c708ddb..2173b228e 100644 --- a/.github/workflows/basic_tests.yml +++ b/.github/workflows/basic_tests.yml @@ -98,7 +98,7 @@ jobs: python-version: ${{ inputs.python-version }} - name: Run Unit Tests - run: poetry run pytest cognee/tests/unit/ + run: uv run pytest cognee/tests/unit/ integration-tests: name: Run Integration Tests @@ -115,7 +115,7 @@ jobs: python-version: ${{ inputs.python-version }} - name: Run Integration Tests - run: poetry run pytest cognee/tests/integration/ + run: uv run pytest cognee/tests/integration/ simple-examples: name: Run Simple Examples @@ -144,7 +144,7 @@ jobs: python-version: ${{ inputs.python-version }} - name: Run Simple Examples - run: poetry run python ./examples/python/simple_example.py + run: uv run python ./examples/python/simple_example.py simple-examples-baml: name: Run Simple Examples BAML @@ -180,7 +180,7 @@ jobs: python-version: ${{ inputs.python-version }} - name: Run Simple Examples - run: poetry run python ./examples/python/simple_example.py + run: uv run python ./examples/python/simple_example.py graph-tests: name: Run Basic Graph Tests @@ -209,4 +209,4 @@ jobs: python-version: ${{ inputs.python-version }} - name: Run Graph Tests - run: poetry run python ./examples/python/code_graph_example.py --repo_path ./cognee/tasks/graph + run: uv run python ./examples/python/code_graph_example.py --repo_path ./cognee/tasks/graph diff --git a/.github/workflows/db_examples_tests.yml b/.github/workflows/db_examples_tests.yml index f585e8200..552395e6a 100644 --- a/.github/workflows/db_examples_tests.yml +++ b/.github/workflows/db_examples_tests.yml @@ -56,7 +56,7 @@ jobs: - name: Install Neo4j extra run: | - poetry install -E neo4j + uv sync --extra neo4j - name: Run Neo4j Example env: @@ -74,7 +74,7 @@ jobs: GRAPH_DATABASE_USERNAME: "neo4j" GRAPH_DATABASE_PASSWORD: ${{ secrets.NEO4J_API_KEY }} run: | - poetry run python examples/database_examples/neo4j_example.py + uv run python examples/database_examples/neo4j_example.py run-db-example-kuzu: name: "Kuzu DB Example Test" @@ -92,7 +92,7 @@ jobs: - name: Install Kuzu extra run: | - poetry install + uv sync - name: Run Kuzu Example env: @@ -107,7 +107,7 @@ jobs: EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} GRAPH_DATABASE_PROVIDER: "kuzu" run: | - poetry run python examples/database_examples/kuzu_example.py + uv run python examples/database_examples/kuzu_example.py run-db-example-pgvector: name: "PostgreSQL PGVector DB Example Test" @@ -141,7 +141,7 @@ jobs: - name: Install PGVector extra run: | - poetry install -E postgres + uv sync --extra postgres - name: Run PGVector Example env: @@ -155,4 +155,4 @@ jobs: EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} run: | - poetry run python examples/database_examples/pgvector_example.py + uv run python examples/database_examples/pgvector_example.py diff --git a/.github/workflows/e2e_tests.yml b/.github/workflows/e2e_tests.yml index 99eb2a72c..4f7c67314 100644 --- a/.github/workflows/e2e_tests.yml +++ b/.github/workflows/e2e_tests.yml @@ -57,7 +57,7 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run python ./cognee/tests/test_cognee_server_start.py + run: uv run python ./cognee/tests/test_cognee_server_start.py run-telemetry-test: name: Run Telemetry Test @@ -81,7 +81,7 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run python ./cognee/tests/test_telemetry.py + run: uv run python ./cognee/tests/test_telemetry.py run-telemetry-pipeline-test: name: Run Telemetry Pipeline Test @@ -110,7 +110,7 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run python ./cognee/tests/test_library.py + run: uv run python ./cognee/tests/test_library.py run-deduplication-test: name: Deduplication Test @@ -143,7 +143,7 @@ jobs: - name: Install specific db dependency run: | - poetry install -E postgres + uv sync --extra postgres - name: Run Deduplication Example env: @@ -154,7 +154,7 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run python ./cognee/tests/test_deduplication.py + run: uv run python ./cognee/tests/test_deduplication.py run-deletion-test: name: Deletion Test @@ -179,7 +179,7 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run python ./cognee/tests/test_deletion.py + run: uv run python ./cognee/tests/test_deletion.py run-s3-bucket-test: name: S3 Bucket Test @@ -195,7 +195,7 @@ jobs: - name: Install specific S3 dependency run: | - poetry install -E aws + uv sync --extra aws - name: Run S3 Bucket Test env: @@ -210,7 +210,7 @@ jobs: EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} AWS_ACCESS_KEY_ID: ${{ secrets.AWS_ACCESS_KEY_ID }} AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_SECRET_ACCESS_KEY }} - run: poetry run python ./cognee/tests/test_s3.py + run: uv run python ./cognee/tests/test_s3.py test-parallel-databases: name: Test using different async databases in parallel in Cognee @@ -226,7 +226,7 @@ jobs: - name: Install specific graph db dependency run: | - poetry install + uv sync - name: Run parallel databases test env: @@ -239,7 +239,7 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run python ./cognee/tests/test_parallel_databases.py + run: uv run python ./cognee/tests/test_parallel_databases.py test-permissions: name: Test permissions with different situations in Cognee @@ -255,7 +255,7 @@ jobs: - name: Install specific graph db dependency run: | - poetry install + uv sync - name: Run parallel databases test env: @@ -268,7 +268,7 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run python ./cognee/tests/test_permissions.py + run: uv run python ./cognee/tests/test_permissions.py test-graph-edges: name: Test graph edge ingestion @@ -284,7 +284,7 @@ jobs: - name: Install specific graph db dependency run: | - poetry install + uv sync - name: Run graph edges test env: @@ -297,4 +297,4 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run python ./cognee/tests/test_edge_ingestion.py + run: uv run python ./cognee/tests/test_edge_ingestion.py diff --git a/.github/workflows/examples_tests.yml b/.github/workflows/examples_tests.yml index 796faf2e0..f0ae430ec 100644 --- a/.github/workflows/examples_tests.yml +++ b/.github/workflows/examples_tests.yml @@ -20,7 +20,7 @@ jobs: env: LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }} OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }} - run: poetry run python ./examples/python/multimedia_example.py + run: uv run python ./examples/python/multimedia_example.py test-eval-example: name: Run Eval Example @@ -36,7 +36,7 @@ jobs: - name: Install specific eval dependency run: | - poetry install -E deepeval + uv sync --extra deepeval - name: Run Evaluation Framework Example env: @@ -49,7 +49,7 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run python ./cognee/eval_framework/run_eval.py + run: uv run python ./cognee/eval_framework/run_eval.py test-descriptive-metrics: name: Run Descriptive Metrics Example @@ -74,7 +74,7 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run python ./cognee/tests/tasks/descriptive_metrics/networkx_metrics_test.py + run: uv run python ./cognee/tests/tasks/descriptive_metrics/networkx_metrics_test.py test-dynamic-steps-metrics: @@ -100,4 +100,4 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run python ./examples/python/dynamic_steps_example.py + run: uv run python ./examples/python/dynamic_steps_example.py diff --git a/.github/workflows/graph_db_tests.yml b/.github/workflows/graph_db_tests.yml index 379539e14..91a611168 100644 --- a/.github/workflows/graph_db_tests.yml +++ b/.github/workflows/graph_db_tests.yml @@ -30,7 +30,7 @@ jobs: - name: Install specific db dependency run: | - poetry install + uv sync - name: Run Kuzu Tests env: @@ -43,7 +43,7 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run python ./cognee/tests/test_kuzu.py + run: uv run python ./cognee/tests/test_kuzu.py - name: Run Weighted Edges Tests with Kuzu env: @@ -57,7 +57,7 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run pytest cognee/tests/unit/interfaces/graph/test_weighted_edges.py -v + run: uv run pytest cognee/tests/unit/interfaces/graph/test_weighted_edges.py -v run-neo4j-tests: name: Neo4j Tests @@ -74,7 +74,7 @@ jobs: - name: Install specific db dependency run: | - poetry install -E neo4j + uv sync --extra neo4j - name: Run default Neo4j env: @@ -91,7 +91,7 @@ jobs: GRAPH_DATABASE_URL: ${{ secrets.NEO4J_API_URL }} GRAPH_DATABASE_PASSWORD: ${{ secrets.NEO4J_API_KEY }} GRAPH_DATABASE_USERNAME: "neo4j" - run: poetry run python ./cognee/tests/test_neo4j.py + run: uv run python ./cognee/tests/test_neo4j.py - name: Run Weighted Edges Tests with Neo4j env: @@ -108,4 +108,4 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run pytest cognee/tests/unit/interfaces/graph/test_weighted_edges.py -v + run: uv run pytest cognee/tests/unit/interfaces/graph/test_weighted_edges.py -v diff --git a/.github/workflows/python_version_tests.yml b/.github/workflows/python_version_tests.yml index 6db67d115..d7a7da050 100644 --- a/.github/workflows/python_version_tests.yml +++ b/.github/workflows/python_version_tests.yml @@ -55,7 +55,7 @@ jobs: - name: Run unit tests shell: bash - run: poetry run pytest cognee/tests/unit/ + run: uv run pytest cognee/tests/unit/ env: PYTHONUTF8: 1 LLM_PROVIDER: openai @@ -73,7 +73,7 @@ jobs: - name: Run integration tests if: ${{ !contains(matrix.os, 'windows') }} shell: bash - run: poetry run pytest cognee/tests/integration/ + run: uv run pytest cognee/tests/integration/ env: PYTHONUTF8: 1 LLM_PROVIDER: openai @@ -103,11 +103,11 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run python ./cognee/tests/test_library.py + run: uv run python ./cognee/tests/test_library.py - - name: Build with Poetry + - name: Build with uv shell: bash - run: poetry build + run: uv build - name: Install Package if: ${{ !contains(matrix.os, 'windows') }} diff --git a/.github/workflows/relational_db_migration_tests.yml b/.github/workflows/relational_db_migration_tests.yml index e44d54d02..604964512 100644 --- a/.github/workflows/relational_db_migration_tests.yml +++ b/.github/workflows/relational_db_migration_tests.yml @@ -61,7 +61,7 @@ jobs: - name: Install specific db dependency run: | - poetry install -E postgres + uv sync --extra postgres - name: Run PostgreSQL Script to create test data (Chinook_PostgreSql.sql) env: @@ -90,7 +90,7 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run python ./cognee/tests/test_relational_db_migration.py + run: uv run python ./cognee/tests/test_relational_db_migration.py run-relational-db-migration-test-kuzu: name: Kuzu Relational DB Migration Test @@ -123,7 +123,7 @@ jobs: - name: Install specific db dependency run: | - poetry install -E postgres + uv sync --extra postgres - name: Run PostgreSQL Script to create test data (Chinook_PostgreSql.sql) env: @@ -154,7 +154,7 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run python ./cognee/tests/test_relational_db_migration.py + run: uv run python ./cognee/tests/test_relational_db_migration.py run-relational-db-migration-test-neo4j: name: Neo4j Relational DB Migration Test @@ -187,7 +187,7 @@ jobs: - name: Install specific db dependency run: | - poetry install -E postgres -E neo4j + uv sync --extra postgres --extra neo4j - name: Run PostgreSQL Script to create test data (Chinook_PostgreSql.sql) env: @@ -221,4 +221,4 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run python ./cognee/tests/test_relational_db_migration.py + run: uv run python ./cognee/tests/test_relational_db_migration.py diff --git a/.github/workflows/reusable_notebook.yml b/.github/workflows/reusable_notebook.yml index 4f97476ef..7ee7f61f3 100644 --- a/.github/workflows/reusable_notebook.yml +++ b/.github/workflows/reusable_notebook.yml @@ -49,7 +49,7 @@ jobs: - name: Install specific db dependency run: | - poetry install -E notebook + uv sync --extra notebook - name: Execute Jupyter Notebook env: @@ -64,7 +64,7 @@ jobs: EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} run: | - poetry run jupyter nbconvert \ + uv run jupyter nbconvert \ --to notebook \ --execute ${{ inputs.notebook-location }} \ --output executed_notebook.ipynb \ diff --git a/.github/workflows/search_db_tests.yml b/.github/workflows/search_db_tests.yml index 3f4bbf1ca..a2ab00754 100644 --- a/.github/workflows/search_db_tests.yml +++ b/.github/workflows/search_db_tests.yml @@ -30,7 +30,7 @@ jobs: - name: Install specific db dependency run: | - poetry install + uv sync - name: Run Kuzu search Tests env: @@ -46,7 +46,7 @@ jobs: GRAPH_DATABASE_PROVIDER: 'kuzu' VECTOR_DB_PROVIDER: 'lancedb' DB_PROVIDER: 'sqlite' - run: poetry run python ./cognee/tests/test_search_db.py + run: uv run python ./cognee/tests/test_search_db.py run-neo4j-lance-sqlite-search-tests: name: Search test for Neo4j/LanceDB/Sqlite @@ -80,7 +80,7 @@ jobs: - name: Install specific db dependency run: | - poetry install -E neo4j + uv sync --extra neo4j - name: Run Neo4j search Tests env: @@ -99,7 +99,7 @@ jobs: GRAPH_DATABASE_URL: bolt://localhost:7687 GRAPH_DATABASE_USERNAME: neo4j GRAPH_DATABASE_PASSWORD: pleaseletmein - run: poetry run python ./cognee/tests/test_search_db.py + run: uv run python ./cognee/tests/test_search_db.py run-kuzu-pgvector-postgres-search-tests: name: Search test for Kuzu/PGVector/Postgres @@ -131,7 +131,7 @@ jobs: python-version: ${{ inputs.python-version }} - name: Install dependencies - run: poetry install -E postgres + run: uv sync --extra postgres - name: Run Kuzu/PGVector/Postgres Tests env: @@ -152,7 +152,7 @@ jobs: DB_PORT: 5432 DB_USERNAME: cognee DB_PASSWORD: cognee - run: poetry run python ./cognee/tests/test_search_db.py + run: uv run python ./cognee/tests/test_search_db.py run-neo4j-pgvector-postgres-search-tests: name: Search test for Neo4j/PGVector/Postgres @@ -198,7 +198,7 @@ jobs: - name: Install dependencies run: | - poetry install -E neo4j -E postgres + uv sync --extra neo4j --extra postgres - name: Run Neo4j + PGVector + Postgres search Tests env: @@ -222,4 +222,4 @@ jobs: DB_PORT: 5432 DB_USERNAME: cognee DB_PASSWORD: cognee - run: poetry run python ./cognee/tests/test_search_db.py + run: uv run python ./cognee/tests/test_search_db.py diff --git a/.github/workflows/test_gemini.yml b/.github/workflows/test_gemini.yml index 41fd954a2..544e15a5e 100644 --- a/.github/workflows/test_gemini.yml +++ b/.github/workflows/test_gemini.yml @@ -26,4 +26,4 @@ jobs: EMBEDDING_MODEL: "gemini/text-embedding-004" EMBEDDING_DIMENSIONS: "768" EMBEDDING_MAX_TOKENS: "8076" - run: poetry run python ./examples/python/simple_example.py + run: uv run python ./examples/python/simple_example.py diff --git a/.github/workflows/test_ollama.yml b/.github/workflows/test_ollama.yml index 8f85efadc..fbd687319 100644 --- a/.github/workflows/test_ollama.yml +++ b/.github/workflows/test_ollama.yml @@ -26,7 +26,7 @@ jobs: - name: Install torch dependency run: | - poetry add torch + uv add torch # - name: Install ollama # run: curl -fsSL https://ollama.com/install.sh | sh @@ -101,4 +101,4 @@ jobs: EMBEDDING_ENDPOINT: "http://localhost:11434/api/embeddings" EMBEDDING_DIMENSIONS: "4096" HUGGINGFACE_TOKENIZER: "Salesforce/SFR-Embedding-Mistral" - run: poetry run python ./examples/python/simple_example.py + run: uv run python ./examples/python/simple_example.py diff --git a/.github/workflows/test_s3_file_storage.yml b/.github/workflows/test_s3_file_storage.yml index 8035a9bc3..c4866ec2d 100644 --- a/.github/workflows/test_s3_file_storage.yml +++ b/.github/workflows/test_s3_file_storage.yml @@ -36,4 +36,4 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run python ./cognee/tests/test_s3_file_storage.py + run: uv run python ./cognee/tests/test_s3_file_storage.py diff --git a/.github/workflows/vector_db_tests.yml b/.github/workflows/vector_db_tests.yml index 00ce21cbf..faa570868 100644 --- a/.github/workflows/vector_db_tests.yml +++ b/.github/workflows/vector_db_tests.yml @@ -56,7 +56,7 @@ jobs: # EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} # EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} # EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - # run: poetry run python ./cognee/tests/test_chromadb.py + # run: uv run python ./cognee/tests/test_chromadb.py run-postgres-tests: @@ -90,7 +90,7 @@ jobs: - name: Install specific db dependency run: | - poetry install -E postgres + uv sync --extra postgres - name: Run PGVector Tests env: @@ -103,4 +103,4 @@ jobs: EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }} EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }} EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }} - run: poetry run python ./cognee/tests/test_pgvector.py + run: uv run python ./cognee/tests/test_pgvector.py diff --git a/.github/workflows/weighted_edges_tests.yml b/.github/workflows/weighted_edges_tests.yml index d33bd1ceb..6d33082a9 100644 --- a/.github/workflows/weighted_edges_tests.yml +++ b/.github/workflows/weighted_edges_tests.yml @@ -47,11 +47,11 @@ jobs: - name: Run Weighted Edges Unit Tests run: | - poetry run pytest cognee/tests/unit/interfaces/graph/test_weighted_edges.py -v --tb=short + uv run pytest cognee/tests/unit/interfaces/graph/test_weighted_edges.py -v --tb=short - name: Run Standard Graph Tests (Regression) run: | - poetry run pytest cognee/tests/unit/interfaces/graph/get_graph_from_model_unit_test.py -v --tb=short + uv run pytest cognee/tests/unit/interfaces/graph/get_graph_from_model_unit_test.py -v --tb=short test-with-different-databases: name: Test Weighted Edges with Different Graph Databases @@ -64,7 +64,7 @@ jobs: install_extra: "" graph_db_provider: "kuzu" - database: neo4j - install_extra: "-E neo4j" + install_extra: "--extra neo4j" graph_db_provider: "neo4j" env: LLM_PROVIDER: openai @@ -90,13 +90,13 @@ jobs: - name: Install Database Dependencies run: | - poetry install ${{ matrix.install_extra }} + uv sync ${{ matrix.install_extra }} - name: Run Weighted Edges Tests env: GRAPH_DATABASE_PROVIDER: ${{ matrix.graph_db_provider }} run: | - poetry run pytest cognee/tests/unit/interfaces/graph/test_weighted_edges.py -v --tb=short + uv run pytest cognee/tests/unit/interfaces/graph/test_weighted_edges.py -v --tb=short test-examples: name: Test Weighted Edges Examples @@ -125,7 +125,7 @@ jobs: - name: Test Weighted Edges Example run: | - poetry run python examples/python/weighted_edges_example.py + uv run python examples/python/weighted_edges_example.py - name: Verify Visualization File Created run: |