Commit graph

4974 commits

Author SHA1 Message Date
hajdul88
3aadb91a6f ruff ruff 2026-01-16 16:13:22 +01:00
hajdul88
707269e8b8 feat: adds integration test for usage logger 2026-01-16 16:11:53 +01:00
Milenko Gavrić
17dbf2137b
Merge branch 'dev' into feature/cog-3725-add-guide-example-codes-to-main-repo 2026-01-16 15:32:10 +01:00
Vasilije
da35f028df
Merge branch 'dev' into feature/cog-3502-tool-logging-with-redis 2026-01-16 15:26:50 +01:00
Vasilije
1d674d459f
feat: Configurable batch size (#1941)
<!-- .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.
-->

## Acceptance Criteria
<!--
* Key requirements to the new feature or modification;
* Proof that the changes work and meet the requirements;
* Include instructions on how to verify the changes. Describe how to
test it locally;
* Proof that it's sufficiently tested.
-->

## 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.


<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit

* **New Features**
* Added configurable chunks-per-batch to control per-batch processing
size via CLI flag, API payload, and configuration; defaults are now
driven by config with an automatic fallback.

* **Style / Documentation**
* Updated contribution/style guidelines (formatting, line length,
string-quote rule, pre-commit note).

* **Tests**
* Updated CLI tests to verify propagation of the new chunks-per-batch
parameter.

<sub>✏️ Tip: You can customize this high-level summary in your review
settings.</sub>
<!-- end of auto-generated comment: release notes by coderabbit.ai -->
2026-01-16 15:20:10 +01:00
Milenko Gavrić
70f84162e0
Merge branch 'dev' into feature/cog-3725-add-guide-example-codes-to-main-repo 2026-01-16 15:16:03 +01:00
Milenko Gavric
7ae44a4565 fix formatting in search_basics_core guide example 2026-01-16 15:03:03 +01:00
Igor Ilic
e13877cb65 refactor: Update cache clear calls 2026-01-16 14:51:30 +01:00
Igor Ilic
12560c391e refactor: update alembic upgrade calls to be done from cognee folder 2026-01-16 14:48:00 +01:00
Igor Ilic
9a9eacf8c8 refactor: update test with cache change 2026-01-16 14:11:06 +01:00
Igor Ilic
e24707c8da fix: Resolve issue with parameter caching for engine creation 2026-01-16 13:43:01 +01:00
Igor Ilic
5d412ed19b refactor: Add batch search support for leaving out payload 2026-01-16 13:21:42 +01:00
Milenko Gavric
6ab127c85d fixed indentation in search_basics_core 2026-01-16 13:11:46 +01:00
hajdul88
1c96e3b469 ruff ruff 2026-01-16 13:01:46 +01:00
hajdul88
2e9f646edd
Merge branch 'dev' into feature/cog-3502-tool-logging-with-redis 2026-01-16 13:00:11 +01:00
hajdul88
2ad8bcf6e9 feat: adds unit test for usage logging 2026-01-16 12:59:45 +01:00
Milenko Gavric
be829dcc3f created search_basics_core example and renamed search_basics to search_basics_additional 2026-01-16 12:54:17 +01:00
Igor Ilic
2c29868f9a
Neo4j multiuser delete (#1985)
<!-- .github/pull_request_template.md -->

## Description
- Add delete ability for Neo4j Aura 
- Refactor Neo4j Aura to use aiohttp to make async requests and perform
better

## Acceptance Criteria
<!--
* Key requirements to the new feature or modification;
* Proof that the changes work and meet the requirements;
* Include instructions on how to verify the changes. Describe how to
test it locally;
* Proof that it's sufficiently tested.
-->

## 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.


<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->

## Summary by CodeRabbit

* **Performance Improvements**
* Neo4j Aura database operations are now asynchronous, eliminating
blocking requests and improving system responsiveness during dataset
management.
* Token retrieval and database provisioning workflows now use
non-blocking asynchronous calls.
  * Enhanced error handling for database API interactions.

<sub>✏️ Tip: You can customize this high-level summary in your review
settings.</sub>

<!-- end of auto-generated comment: release notes by coderabbit.ai -->
2026-01-16 12:43:09 +01:00
Andrej Milicevic
98e8d226eb test: add tests for batch query graph completions 2026-01-16 11:57:03 +01:00
Andrej Milicevic
b05c93bf5f fix: fix sessions tests 2026-01-15 20:40:33 +01:00
Andrej Milicevic
2655df9b21 fix: fix failing tests 2026-01-15 19:29:21 +01:00
hajdul88
abb45c65d7 feat: adding and fixing mcp tool logging 2026-01-15 18:03:47 +01:00
hajdul88
e17ca5ac59 feat: adds logging to memify endpoint 2026-01-15 17:35:06 +01:00
Igor Ilic
e51149c3a2 refactor: Update temporal tests 2026-01-15 17:25:59 +01:00
hajdul88
7773e811b0 chore: cleaning and adding correct defaults 2026-01-15 16:31:38 +01:00
Milenko Gavric
8761eab0c0 move guides to examples 2026-01-15 16:20:48 +01:00
hajdul88
c0a7b14ff3 feat: adds log usage decorator to main api endpoints 2026-01-15 16:04:09 +01:00
hajdul88
bc8c6e8bae Update usage_logger.py 2026-01-15 16:02:22 +01:00
hajdul88
2d5e74ced0 fix: fixes codebunny suggestion 2026-01-15 16:01:21 +01:00
Milenko Gavric
4ae64f05a3 delete sessions since it needs to be improved 2026-01-15 16:00:32 +01:00
hajdul88
b83af5f63f feat: adds new exception to shared usage logger 2026-01-15 15:58:40 +01:00
Milenko Gavric
ae6b2d196e add codes from guides to a separate folder 2026-01-15 15:25:26 +01:00
hajdul88
7ebe8563c5 ruff 2026-01-15 15:21:24 +01:00
hajdul88
8dc358da39 feat: adds default param logging 2026-01-15 15:21:15 +01:00
hajdul88
bf2357e7bf chore: cleaning usage logger logic 2026-01-15 15:16:26 +01:00
Igor Ilic
3635bda6cd refactor: Use id instead of payload id for temporal retriever 2026-01-15 14:51:15 +01:00
hajdul88
e803f10417 feat: implements first version of usage_logger decorator 2026-01-15 14:22:01 +01:00
Andrej Milicevic
40667e63c9 feat: enable batch queries for graph completion retrievers 2026-01-15 14:01:10 +01:00
hajdul88
34513f2c10 fix: fixes unit test for cacheconfig params 2026-01-15 12:46:19 +01:00
Igor Ilic
f356369708 refactor: Update unit tests 2026-01-15 12:00:41 +01:00
hajdul88
ace34b9a91 feat: adds log key to RedisAdapter 2026-01-15 11:56:01 +01:00
hajdul88
8b49f892ce ruff fix 2026-01-15 11:53:54 +01:00
hajdul88
e8edf4482d feat: adds usage logging and log key to the cache engine factory 2026-01-15 11:53:36 +01:00
hajdul88
97fcc15af5 feat: updates constructor params in base class 2026-01-15 11:49:37 +01:00
hajdul88
f4c2365c23 Adds default methods to satisfy base class (FSCache does not support the logging for now) 2026-01-15 11:48:10 +01:00
Igor Ilic
e65a6a1679 refactor: include payload for summaries retriever 2026-01-15 11:43:19 +01:00
Igor Ilic
4f7ab87683 refactor: Use include_payload where necessary 2026-01-15 11:32:04 +01:00
hajdul88
8f0705359a feat: adds log usage and get logs operations to RedisAdapter 2026-01-15 11:13:03 +01:00
hajdul88
90cf79b420 feat: adds new config params to Cacheconfig 2026-01-15 11:10:13 +01:00
hajdul88
eb8996dd81 feat: extends CacheDBInterface base class with the logging related methods 2026-01-15 11:08:30 +01:00