Commit graph

694 commits

Author SHA1 Message Date
hajdul88
2bf1619e8f chore: renaming acquire and release methods 2025-10-16 11:47:32 +02:00
hajdul88
9bb47e4231 Merge branch 'dev' into feature/cog-3142-agentic-use-case-kuzu-lock-fix-redis-integration 2025-10-13 09:56:42 +02:00
vasilije
5527b6e0b3 format done 2025-10-12 11:47:40 +02:00
vasilije
ca88200b7c Merge remote-tracking branch 'origin/pr-1449' into pr-1449
# Conflicts:
#	poetry.lock
#	uv.lock
2025-10-12 11:46:43 +02:00
Aniruddha Mandal
06dca5bf26 Enhance MistralAdapter error handling and response validation in chat completion method 2025-10-12 11:44:36 +02:00
Aniruddha Mandal
617c1f0d71 Added Mistral support as LLM provider using litellm 2025-10-12 11:44:33 +02:00
Vasilije
fb8e6ae47c
Fix exception handling across cognee - lost exception stacktraces and exception metadata (#1518)
<!-- .github/pull_request_template.md -->

## Description
<!--
Please provide a clear, human-generated description of the changes in
this PR.
DO NOT use AI-generated descriptions. We want to understand your thought
process and reasoning.
-->

In several places, our exception handling discards valuable debugging
context - either by losing the original stack trace or omitting metadata
from the underlying error. This makes diagnosing root causes
significantly harder.

This PR updates exception throwing to use [`raise ...
from`](https://docs.python.org/3/reference/simple_stmts.html#raise),
which:
1. The original stack trace is always preserved, even when wrapping
exceptions
2. Custom exceptions explicitly reference their underlying cause, making
error chains clearer



## Type of Change
<!-- Please check the relevant option -->
- [ ] Bug fix (non-breaking change that fixes an issue)
- [ ] New feature (non-breaking change that adds functionality)
- [ ] Breaking change (fix or feature that would cause existing
functionality to change)
- [ ] Documentation update
- [ ] Code refactoring
- [ ] Performance improvement
- [ ] Other (please specify):

## Screenshots/Videos (if applicable)
<!-- Add screenshots or videos to help explain your changes -->

## Pre-submission Checklist
<!-- Please check all boxes that apply before submitting your PR -->
- [ ] **I have tested my changes thoroughly before submitting this PR**
- [ ] **This PR contains minimal changes necessary to address the
issue/feature**
- [ ] My code follows the project's coding standards and style
guidelines
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [ ] I have added necessary documentation (if applicable)
- [ ] All new and existing tests pass
- [ ] I have searched existing PRs to ensure this change hasn't been
submitted already
- [ ] I have linked any relevant issues in the description
- [ ] My commits have clear and descriptive messages

## DCO Affirmation
I affirm that all code in every commit of this pull request conforms to
the terms of the Topoteretes Developer Certificate of Origin.
2025-10-12 10:29:26 +02:00
Vasilije
8576f18a9f
Change default pdf loader (#1525)
<!-- .github/pull_request_template.md -->

## Description
Change default loader from Advanced to light weight PyPDF loader

## Type of Change
<!-- Please check the relevant option -->
- [ ] Bug fix (non-breaking change that fixes an issue)
- [ ] New feature (non-breaking change that adds functionality)
- [ ] Breaking change (fix or feature that would cause existing
functionality to change)
- [ ] Documentation update
- [x] Code refactoring
- [ ] Performance improvement
- [ ] Other (please specify):

## Screenshots/Videos (if applicable)
<!-- Add screenshots or videos to help explain your changes -->

## Pre-submission Checklist
<!-- Please check all boxes that apply before submitting your PR -->
- [ ] **I have tested my changes thoroughly before submitting this PR**
- [ ] **This PR contains minimal changes necessary to address the
issue/feature**
- [ ] My code follows the project's coding standards and style
guidelines
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [ ] I have added necessary documentation (if applicable)
- [ ] All new and existing tests pass
- [ ] I have searched existing PRs to ensure this change hasn't been
submitted already
- [ ] I have linked any relevant issues in the description
- [ ] My commits have clear and descriptive messages

## DCO Affirmation
I affirm that all code in every commit of this pull request conforms to
the terms of the Topoteretes Developer Certificate of Origin.
2025-10-11 22:02:15 +02:00
Vasilije
fa7aa38b8f
COG-3050 - remove insights search (#1506)
<!-- .github/pull_request_template.md -->

## Description
<!--
Please provide a clear, human-generated description of the changes in
this PR.
DO NOT use AI-generated descriptions. We want to understand your thought
process and reasoning.
-->

As per COG-3050:
1. Remove insights search type and clean up any orphaned code
2. Replace callsites with default search type - `GRAPH_COMPLETION` -
where applicable

## Type of Change
<!-- Please check the relevant option -->
- [ ] Bug fix (non-breaking change that fixes an issue)
- [ ] New feature (non-breaking change that adds functionality)
- [ ] Breaking change (fix or feature that would cause existing
functionality to change)
- [ ] Documentation update
- [x] Code refactoring
- [ ] Performance improvement
- [ ] Other (please specify):

## Screenshots/Videos (if applicable)
<!-- Add screenshots or videos to help explain your changes -->

## Pre-submission Checklist
<!-- Please check all boxes that apply before submitting your PR -->
- [ ] **I have tested my changes thoroughly before submitting this PR**
- [ ] **This PR contains minimal changes necessary to address the
issue/feature**
- [ ] My code follows the project's coding standards and style
guidelines
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [ ] I have added necessary documentation (if applicable)
- [ ] All new and existing tests pass
- [ ] I have searched existing PRs to ensure this change hasn't been
submitted already
- [ ] I have linked any relevant issues in the description
- [ ] My commits have clear and descriptive messages

## DCO Affirmation
I affirm that all code in every commit of this pull request conforms to
the terms of the Topoteretes Developer Certificate of Origin.
2025-10-11 09:09:56 +02:00
Igor Ilic
9a33439876 refactor: Change default PDF loader 2025-10-09 19:10:31 +02:00
hajdul88
d449c8cfb7 chore: deletes dead code that exists only in kuzuadapter and uses connection.execute directly 2025-10-09 14:44:17 +02:00
hajdul88
f5954415a6
Merge branch 'dev' into feature/cog-3142-agentic-use-case-kuzu-lock-fix-redis-integration 2025-10-09 14:08:09 +02:00
hajdul88
faeca138d9
fix: fixes distributed pipeline (#1454)
<!-- .github/pull_request_template.md -->

## Description
This PR fixes distributed pipeline + updates core changes in distr
logic.

## Type of Change
<!-- Please check the relevant option -->
- [x] Bug fix (non-breaking change that fixes an issue)
- [x] New feature (non-breaking change that adds functionality)
- [ ] Breaking change (fix or feature that would cause existing
functionality to change)
- [ ] Documentation update
- [x] Code refactoring
- [x] Performance improvement
- [ ] Other (please specify):

## Changes Made
Fixes distributed pipeline:
-Changed spawning logic + adds incremental loading to
run_tasks_diistributed
-Adds batching to consumer nodes
-Fixes consumer stopping criteria by adding stop signal + handling
-Changed edge embedding solution to avoid huge network load in a case of
a multicontainer environment

## Testing
Tested it by running 1GB on modal + manually

## Screenshots/Videos (if applicable)
None

## Pre-submission Checklist
<!-- Please check all boxes that apply before submitting your PR -->
- [x] **I have tested my changes thoroughly before submitting this PR**
- [x] **This PR contains minimal changes necessary to address the
issue/feature**
- [ ] My code follows the project's coding standards and style
guidelines
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [ ] I have added necessary documentation (if applicable)
- [ ] All new and existing tests pass
- [ ] I have searched existing PRs to ensure this change hasn't been
submitted already
- [ ] I have linked any relevant issues in the description
- [ ] My commits have clear and descriptive messages

## Related Issues
None

## Additional Notes
None

## DCO Affirmation
I affirm that all code in every commit of this pull request conforms to
the terms of the Topoteretes Developer Certificate of Origin.

---------

Co-authored-by: Boris <boris@topoteretes.com>
Co-authored-by: Boris Arzentar <borisarzentar@gmail.com>
2025-10-09 14:06:25 +02:00
hajdul88
b58bd60bb6 ruff fix 2025-10-08 17:04:30 +02:00
hajdul88
d657e1b311 feat: adds new separate parameter for shared kuzu lock 2025-10-08 16:27:17 +02:00
hajdul88
fc2f27f21e feat: adds redis double locking mechanism (Thread + Process) 2025-10-08 15:23:38 +02:00
Daulet Amirkhanov
97f90c95ed Remove all references to SearchType.INSIGHTS across the codebase, meaningfully replacing it with SearchType.GRAPH_COMPLETION where applicable. 2025-10-08 12:13:59 +01:00
Daulet Amirkhanov
6acc44de16 Improve exception handling to provide clearer error context 2025-10-08 12:09:21 +01:00
hajdul88
57439ae8df testing on demand connection with ci/cd 2025-10-08 10:40:30 +02:00
hajdul88
93416cccc2 linting fix 2025-10-08 10:24:42 +02:00
hajdul88
e230488e97 Merge branch 'dev' into feature/cog-3142-agentic-use-case-kuzu-lock-fix-redis-integration 2025-10-08 10:20:49 +02:00
hajdul88
a170355f94 feat: adds no cache case 2025-10-08 10:19:34 +02:00
Daulet Amirkhanov
ee079604f4 chore: keep exception handling changes to instructor adapters 2025-10-07 22:03:16 +01:00
hajdul88
88fb4cef05 Merge branch 'dev' into feature/cog-3142-agentic-use-case-kuzu-lock-fix-redis-integration 2025-10-07 20:44:39 +02:00
Igor Ilic
840991c36f
refactor: Don't use async lock if not needed (#1510)
<!-- .github/pull_request_template.md -->

## Description
Reduce time PGVector is using async lock, by checking if collection
exists before taking async lock

## Type of Change
<!-- Please check the relevant option -->
- [ ] Bug fix (non-breaking change that fixes an issue)
- [ ] New feature (non-breaking change that adds functionality)
- [ ] Breaking change (fix or feature that would cause existing
functionality to change)
- [ ] Documentation update
- [ ] Code refactoring
- [ ] Performance improvement
- [ ] Other (please specify):

## Screenshots/Videos (if applicable)
<!-- Add screenshots or videos to help explain your changes -->

## Pre-submission Checklist
<!-- Please check all boxes that apply before submitting your PR -->
- [ ] **I have tested my changes thoroughly before submitting this PR**
- [ ] **This PR contains minimal changes necessary to address the
issue/feature**
- [ ] My code follows the project's coding standards and style
guidelines
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [ ] I have added necessary documentation (if applicable)
- [ ] All new and existing tests pass
- [ ] I have searched existing PRs to ensure this change hasn't been
submitted already
- [ ] I have linked any relevant issues in the description
- [ ] My commits have clear and descriptive messages

## DCO Affirmation
I affirm that all code in every commit of this pull request conforms to
the terms of the Topoteretes Developer Certificate of Origin.
2025-10-07 20:44:10 +02:00
hajdul88
b468a13005 Update adapter.py 2025-10-07 20:43:50 +02:00
Igor Ilic
e7709f7bd0
fix: Resolve issue with Kuzu graph database persistence on our local … (#1490)
…Docker

<!-- .github/pull_request_template.md -->

## Description
Resolves issue with kuzu graph data persistence for Cognee backend
docker

## Type of Change
<!-- Please check the relevant option -->
- [x] Bug fix (non-breaking change that fixes an issue)
- [ ] New feature (non-breaking change that adds functionality)
- [ ] Breaking change (fix or feature that would cause existing
functionality to change)
- [ ] Documentation update
- [ ] Code refactoring
- [ ] Performance improvement
- [ ] Other (please specify):

## Pre-submission Checklist
<!-- Please check all boxes that apply before submitting your PR -->
- [ ] **I have tested my changes thoroughly before submitting this PR**
- [ ] **This PR contains minimal changes necessary to address the
issue/feature**
- [ ] My code follows the project's coding standards and style
guidelines
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [ ] I have added necessary documentation (if applicable)
- [ ] All new and existing tests pass
- [ ] I have searched existing PRs to ensure this change hasn't been
submitted already
- [ ] I have linked any relevant issues in the description
- [ ] My commits have clear and descriptive messages

## DCO Affirmation
I affirm that all code in every commit of this pull request conforms to
the terms of the Topoteretes Developer Certificate of Origin.

---------

Co-authored-by: Boris <boris@topoteretes.com>
2025-10-07 20:38:43 +02:00
Igor Ilic
f81d427466
Main merge vol7 (#1509)
<!-- .github/pull_request_template.md -->

## Description
Merge main to dev

## Type of Change
<!-- Please check the relevant option -->
- [ ] Bug fix (non-breaking change that fixes an issue)
- [ ] New feature (non-breaking change that adds functionality)
- [ ] Breaking change (fix or feature that would cause existing
functionality to change)
- [ ] Documentation update
- [ ] Code refactoring
- [ ] Performance improvement
- [ ] Other (please specify):

## Screenshots/Videos (if applicable)
<!-- Add screenshots or videos to help explain your changes -->

## Pre-submission Checklist
<!-- Please check all boxes that apply before submitting your PR -->
- [ ] **I have tested my changes thoroughly before submitting this PR**
- [ ] **This PR contains minimal changes necessary to address the
issue/feature**
- [ ] My code follows the project's coding standards and style
guidelines
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [ ] I have added necessary documentation (if applicable)
- [ ] All new and existing tests pass
- [ ] I have searched existing PRs to ensure this change hasn't been
submitted already
- [ ] I have linked any relevant issues in the description
- [ ] My commits have clear and descriptive messages

## DCO Affirmation
I affirm that all code in every commit of this pull request conforms to
the terms of the Topoteretes Developer Certificate of Origin.
2025-10-07 20:17:30 +02:00
Igor Ilic
0ef2623ac6
refactor: use gpt-5-mini by default (#1505)
<!-- .github/pull_request_template.md -->

## Description
Set gpt-5-mini back to default LLM as issues with it have been resolved

## Type of Change
<!-- Please check the relevant option -->
- [ ] Bug fix (non-breaking change that fixes an issue)
- [ ] New feature (non-breaking change that adds functionality)
- [ ] Breaking change (fix or feature that would cause existing
functionality to change)
- [ ] Documentation update
- [ ] Code refactoring
- [ ] Performance improvement
- [ ] Other (please specify):

## Pre-submission Checklist
<!-- Please check all boxes that apply before submitting your PR -->
- [ ] **I have tested my changes thoroughly before submitting this PR**
- [ ] **This PR contains minimal changes necessary to address the
issue/feature**
- [ ] My code follows the project's coding standards and style
guidelines
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [ ] I have added necessary documentation (if applicable)
- [ ] All new and existing tests pass
- [ ] I have searched existing PRs to ensure this change hasn't been
submitted already
- [ ] I have linked any relevant issues in the description
- [ ] My commits have clear and descriptive messages

## DCO Affirmation
I affirm that all code in every commit of this pull request conforms to
the terms of the Topoteretes Developer Certificate of Origin.
2025-10-07 20:17:10 +02:00
Igor Ilic
b4eaea2133 Merge branch 'main' into main-merge-vol7 2025-10-07 19:56:38 +02:00
Igor Ilic
3b8b839057
fix: Resolve pydantic issue regarding deprecation (#1508)
<!-- .github/pull_request_template.md -->

## Description
Fix latest pydantic version issues

## Type of Change
<!-- Please check the relevant option -->
- [ ] Bug fix (non-breaking change that fixes an issue)
- [ ] New feature (non-breaking change that adds functionality)
- [ ] Breaking change (fix or feature that would cause existing
functionality to change)
- [ ] Documentation update
- [ ] Code refactoring
- [ ] Performance improvement
- [ ] Other (please specify):

## Screenshots/Videos (if applicable)
<!-- Add screenshots or videos to help explain your changes -->

## Pre-submission Checklist
<!-- Please check all boxes that apply before submitting your PR -->
- [ ] **I have tested my changes thoroughly before submitting this PR**
- [ ] **This PR contains minimal changes necessary to address the
issue/feature**
- [ ] My code follows the project's coding standards and style
guidelines
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [ ] I have added necessary documentation (if applicable)
- [ ] All new and existing tests pass
- [ ] I have searched existing PRs to ensure this change hasn't been
submitted already
- [ ] I have linked any relevant issues in the description
- [ ] My commits have clear and descriptive messages

## DCO Affirmation
I affirm that all code in every commit of this pull request conforms to
the terms of the Topoteretes Developer Certificate of Origin.
2025-10-07 19:54:59 +02:00
hajdul88
466707b003 feat: adds redis locks 2025-10-07 19:45:48 +02:00
hajdul88
948dde0e41 Merge branch 'dev' into feature/cog-3142-agentic-use-case-kuzu-lock-fix-redis-integration 2025-10-07 19:24:31 +02:00
hajdul88
31280471c9 ruff 2025-10-07 19:20:36 +02:00
hajdul88
9a5af434e4 trying out connection open/close on ci/cd 2025-10-07 19:20:27 +02:00
hajdul88
68a8977301 feat: adds connection locks 2025-10-07 19:19:57 +02:00
hajdul88
b288e5d7a0 fix: fixes pgvector deadlock 2025-10-07 19:19:32 +02:00
hajdul88
7ec1c75bee Update adapter.py 2025-10-07 18:56:08 +02:00
hajdul88
03eedddf29 Update get_cache_engine.py 2025-10-07 18:35:41 +02:00
hajdul88
9adb89b718 Revert "feat: adds connection execture wrapper to keep track open executions"
This reverts commit e3964cb208.
2025-10-07 18:21:59 +02:00
Igor Ilic
38cdacbcb6
fix: Resolve issue with Gemini adapter (#1494)
<!-- .github/pull_request_template.md -->

## Description
Resolve Gemini Adapter issues:
 1. resolve embedding batch issue,
2. Resolve slowness because gemini tokenizer was sending word per word
to Googles API to count tokens (using OpenAI's local tokenizer to count
tokens for Gemini now)
 3. Update deprecated library and move to instructor

## Type of Change
<!-- Please check the relevant option -->
- [x] Bug fix (non-breaking change that fixes an issue)
- [ ] New feature (non-breaking change that adds functionality)
- [ ] Breaking change (fix or feature that would cause existing
functionality to change)
- [ ] Documentation update
- [ ] Code refactoring
- [ ] Performance improvement
- [ ] Other (please specify):

## Pre-submission Checklist
<!-- Please check all boxes that apply before submitting your PR -->
- [ ] **I have tested my changes thoroughly before submitting this PR**
- [ ] **This PR contains minimal changes necessary to address the
issue/feature**
- [ ] My code follows the project's coding standards and style
guidelines
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [ ] I have added necessary documentation (if applicable)
- [ ] All new and existing tests pass
- [ ] I have searched existing PRs to ensure this change hasn't been
submitted already
- [ ] I have linked any relevant issues in the description
- [ ] My commits have clear and descriptive messages

## DCO Affirmation
I affirm that all code in every commit of this pull request conforms to
the terms of the Topoteretes Developer Certificate of Origin.
2025-10-07 18:04:18 +02:00
hajdul88
395b1c289a feat: makes redis import optional 2025-10-07 15:56:47 +02:00
hajdul88
e3964cb208 feat: adds connection execture wrapper to keep track open executions 2025-10-07 15:47:54 +02:00
hajdul88
76c4a4bd4c back to zero 2025-10-07 14:21:48 +02:00
hajdul88
ac4dd8776a feat: adds redis adapter and factory 2025-10-07 13:49:09 +02:00
hajdul88
21c203d1eb feat: adds pydantic settings to set caching + some locks to Kuzu adapter 2025-10-07 10:29:31 +02:00
hajdul88
c51787c987 ruff format 2025-10-06 18:05:48 +02:00
hajdul88
e7e8ec3bfd feat: adds redis locking mechanism to KuzuAdapter 2025-10-06 18:05:17 +02:00
hajdul88
2848b03bcc feat: adds opening and closing kuzuconnection to avoid full lock due to lru_cache 2025-10-06 15:24:15 +02:00
EricXiao
d868912df5
Merge branch 'dev' into feat/add-pdfproloader 2025-09-30 23:24:14 +08:00