From da05671fd778b2df00206a669042e7f041a9fcff Mon Sep 17 00:00:00 2001 From: Andrej Milicevic Date: Wed, 22 Oct 2025 10:23:34 +0200 Subject: [PATCH 1/7] test: changed the api key of baml tests --- .github/workflows/basic_tests.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/basic_tests.yml b/.github/workflows/basic_tests.yml index 3f3e644a2..a956e31fd 100644 --- a/.github/workflows/basic_tests.yml +++ b/.github/workflows/basic_tests.yml @@ -164,7 +164,7 @@ jobs: BAML_LLM_PROVIDER: azure-openai BAML_LLM_MODEL: ${{ secrets.LLM_MODEL }} BAML_LLM_ENDPOINT: ${{ secrets.LLM_ENDPOINT }} - BAML_LLM_API_KEY: ${{ secrets.LLM_API_KEY }} + BAML_LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }} BAML_LLM_API_VERSION: ${{ secrets.LLM_API_VERSION }} LLM_PROVIDER: openai From 5f2e9bd84b7b70a55a5259405b5af60cc906fe8a Mon Sep 17 00:00:00 2001 From: Andrej Milicevic Date: Wed, 22 Oct 2025 10:37:37 +0200 Subject: [PATCH 2/7] fix: change baml llm provider --- .github/workflows/basic_tests.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/basic_tests.yml b/.github/workflows/basic_tests.yml index a956e31fd..574832dfd 100644 --- a/.github/workflows/basic_tests.yml +++ b/.github/workflows/basic_tests.yml @@ -161,7 +161,7 @@ jobs: runs-on: ubuntu-22.04 env: STRUCTURED_OUTPUT_FRAMEWORK: "BAML" - BAML_LLM_PROVIDER: azure-openai + BAML_LLM_PROVIDER: openai BAML_LLM_MODEL: ${{ secrets.LLM_MODEL }} BAML_LLM_ENDPOINT: ${{ secrets.LLM_ENDPOINT }} BAML_LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }} From 1ecea0a95509a202c1f84f7551b39ca7d0ab0d5c Mon Sep 17 00:00:00 2001 From: Andrej Milicevic Date: Wed, 22 Oct 2025 12:39:31 +0200 Subject: [PATCH 3/7] change endpoint --- .github/workflows/basic_tests.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/basic_tests.yml b/.github/workflows/basic_tests.yml index 574832dfd..e54217273 100644 --- a/.github/workflows/basic_tests.yml +++ b/.github/workflows/basic_tests.yml @@ -163,7 +163,7 @@ jobs: STRUCTURED_OUTPUT_FRAMEWORK: "BAML" BAML_LLM_PROVIDER: openai BAML_LLM_MODEL: ${{ secrets.LLM_MODEL }} - BAML_LLM_ENDPOINT: ${{ secrets.LLM_ENDPOINT }} + BAML_LLM_ENDPOINT: ${{ secrets.OPENAI_ENDPOINT }} BAML_LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }} BAML_LLM_API_VERSION: ${{ secrets.LLM_API_VERSION }} From bb756a0e1906db2ab665d40f328a2337a622c4ed Mon Sep 17 00:00:00 2001 From: Andrej Milicevic Date: Wed, 22 Oct 2025 12:42:23 +0200 Subject: [PATCH 4/7] remove api version --- .github/workflows/basic_tests.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/basic_tests.yml b/.github/workflows/basic_tests.yml index e54217273..4520fc0fc 100644 --- a/.github/workflows/basic_tests.yml +++ b/.github/workflows/basic_tests.yml @@ -165,7 +165,7 @@ jobs: BAML_LLM_MODEL: ${{ secrets.LLM_MODEL }} BAML_LLM_ENDPOINT: ${{ secrets.OPENAI_ENDPOINT }} BAML_LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }} - BAML_LLM_API_VERSION: ${{ secrets.LLM_API_VERSION }} +# BAML_LLM_API_VERSION: ${{ secrets.LLM_API_VERSION }} LLM_PROVIDER: openai LLM_MODEL: ${{ secrets.LLM_MODEL }} From a7c74de208a18d91fd8ea156dcbd7b220cc03bce Mon Sep 17 00:00:00 2001 From: Andrej Milicevic Date: Wed, 22 Oct 2025 13:11:26 +0200 Subject: [PATCH 5/7] changed temperature for baml --- cognee/infrastructure/llm/config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cognee/infrastructure/llm/config.py b/cognee/infrastructure/llm/config.py index 8fd196eaf..e1ac55405 100644 --- a/cognee/infrastructure/llm/config.py +++ b/cognee/infrastructure/llm/config.py @@ -51,7 +51,7 @@ class LLMConfig(BaseSettings): baml_llm_model: str = "gpt-5-mini" baml_llm_endpoint: str = "" baml_llm_api_key: Optional[str] = None - baml_llm_temperature: float = 0.0 + baml_llm_temperature: float = 1.0 baml_llm_api_version: str = "" transcription_model: str = "whisper-1" From 1d1c7d21f7376815cfee1e8448f5adb34a65a466 Mon Sep 17 00:00:00 2001 From: Andrej Milicevic Date: Wed, 22 Oct 2025 14:06:43 +0200 Subject: [PATCH 6/7] change model --- .github/workflows/basic_tests.yml | 2 +- cognee/infrastructure/llm/config.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/basic_tests.yml b/.github/workflows/basic_tests.yml index 4520fc0fc..e1c6a666b 100644 --- a/.github/workflows/basic_tests.yml +++ b/.github/workflows/basic_tests.yml @@ -162,7 +162,7 @@ jobs: env: STRUCTURED_OUTPUT_FRAMEWORK: "BAML" BAML_LLM_PROVIDER: openai - BAML_LLM_MODEL: ${{ secrets.LLM_MODEL }} + BAML_LLM_MODEL: ${{ secrets.OPENAI_MODEL }} BAML_LLM_ENDPOINT: ${{ secrets.OPENAI_ENDPOINT }} BAML_LLM_API_KEY: ${{ secrets.OPENAI_API_KEY }} # BAML_LLM_API_VERSION: ${{ secrets.LLM_API_VERSION }} diff --git a/cognee/infrastructure/llm/config.py b/cognee/infrastructure/llm/config.py index e1ac55405..8fd196eaf 100644 --- a/cognee/infrastructure/llm/config.py +++ b/cognee/infrastructure/llm/config.py @@ -51,7 +51,7 @@ class LLMConfig(BaseSettings): baml_llm_model: str = "gpt-5-mini" baml_llm_endpoint: str = "" baml_llm_api_key: Optional[str] = None - baml_llm_temperature: float = 1.0 + baml_llm_temperature: float = 0.0 baml_llm_api_version: str = "" transcription_model: str = "whisper-1" From 8d9ee07083811b6aa6b126e56678070469123f25 Mon Sep 17 00:00:00 2001 From: hajdul88 <52442977+hajdul88@users.noreply.github.com> Date: Wed, 22 Oct 2025 15:31:57 +0200 Subject: [PATCH 7/7] chore: regenerating lock files --- poetry.lock | 5 +++-- uv.lock | 8 +++++--- 2 files changed, 8 insertions(+), 5 deletions(-) diff --git a/poetry.lock b/poetry.lock index c974a0b43..2ab4d42bf 100644 --- a/poetry.lock +++ b/poetry.lock @@ -1,4 +1,4 @@ -# This file is automatically @generated by Poetry 2.1.2 and should not be changed by hand. +# This file is automatically @generated by Poetry 2.1.3 and should not be changed by hand. [[package]] name = "accelerate" @@ -8532,6 +8532,7 @@ files = [ {file = "psycopg2-2.9.10-cp311-cp311-win_amd64.whl", hash = "sha256:0435034157049f6846e95103bd8f5a668788dd913a7c30162ca9503fdf542cb4"}, {file = "psycopg2-2.9.10-cp312-cp312-win32.whl", hash = "sha256:65a63d7ab0e067e2cdb3cf266de39663203d38d6a8ed97f5ca0cb315c73fe067"}, {file = "psycopg2-2.9.10-cp312-cp312-win_amd64.whl", hash = "sha256:4a579d6243da40a7b3182e0430493dbd55950c493d8c68f4eec0b302f6bbf20e"}, + {file = "psycopg2-2.9.10-cp313-cp313-win_amd64.whl", hash = "sha256:91fd603a2155da8d0cfcdbf8ab24a2d54bca72795b90d2a3ed2b6da8d979dee2"}, {file = "psycopg2-2.9.10-cp39-cp39-win32.whl", hash = "sha256:9d5b3b94b79a844a986d029eee38998232451119ad653aea42bb9220a8c5066b"}, {file = "psycopg2-2.9.10-cp39-cp39-win_amd64.whl", hash = "sha256:88138c8dedcbfa96408023ea2b0c369eda40fe5d75002c0964c78f46f11fa442"}, {file = "psycopg2-2.9.10.tar.gz", hash = "sha256:12ec0b40b0273f95296233e8750441339298e6a572f7039da5b260e3c8b60e11"}, @@ -13531,4 +13532,4 @@ scraping = ["APScheduler", "beautifulsoup4", "lxml", "playwright", "protego", "t [metadata] lock-version = "2.1" python-versions = ">=3.10,<3.14" -content-hash = "bcab5420339473ec08b89cde588899b60999762fb8ca9a011240d47ea86198e3" +content-hash = "9d64a8e91222a3c8108e192368862d9cbc3c38b658e43d5806af6a66561051cd" diff --git a/uv.lock b/uv.lock index 8093df133..af3250a01 100644 --- a/uv.lock +++ b/uv.lock @@ -1,6 +1,6 @@ version = 1 revision = 2 -requires-python = ">=3.10, <=3.13" +requires-python = ">=3.10, <3.14" resolution-markers = [ "python_full_version >= '3.13' and platform_python_implementation != 'PyPy' and sys_platform == 'darwin'", "python_full_version >= '3.13' and platform_machine == 'aarch64' and platform_python_implementation != 'PyPy' and sys_platform == 'linux'", @@ -1027,6 +1027,7 @@ docling = [ { name = "transformers" }, ] docs = [ + { name = "lxml" }, { name = "unstructured", extra = ["csv", "doc", "docx", "epub", "md", "odt", "org", "pdf", "ppt", "pptx", "rst", "rtf", "tsv", "xlsx"] }, ] evals = [ @@ -1136,7 +1137,8 @@ requires-dist = [ { name = "limits", specifier = ">=4.4.1,<5" }, { name = "litellm", specifier = ">=1.76.0" }, { name = "llama-index-core", marker = "extra == 'llama-index'", specifier = ">=0.12.11,<0.13" }, - { name = "lxml", marker = "extra == 'scraping'", specifier = ">=4.9.3,<5.0.0" }, + { name = "lxml", marker = "extra == 'docs'", specifier = "<6.0.0" }, + { name = "lxml", marker = "extra == 'scraping'", specifier = ">=4.9.3" }, { name = "matplotlib", marker = "extra == 'evals'", specifier = ">=3.8.3,<4" }, { name = "mistral-common", marker = "extra == 'mistral'", specifier = ">=1.5.2,<2" }, { name = "mistralai", specifier = ">=1.9.10" }, @@ -1184,7 +1186,7 @@ requires-dist = [ { name = "sentry-sdk", extras = ["fastapi"], marker = "extra == 'monitoring'", specifier = ">=2.9.0,<3" }, { name = "sqlalchemy", specifier = ">=2.0.39,<3.0.0" }, { name = "structlog", specifier = ">=25.2.0,<26" }, - { name = "tavily-python", marker = "extra == 'scraping'", specifier = ">=0.7.0" }, + { name = "tavily-python", marker = "extra == 'scraping'", specifier = ">=0.7.12" }, { name = "tenacity", specifier = ">=9.0.0" }, { name = "tiktoken", specifier = ">=0.8.0,<1.0.0" }, { name = "transformers", marker = "extra == 'codegraph'", specifier = ">=4.46.3,<5" },