feat: add query knowledge graph tool version configuration to agents#2319
Open
feat: add query knowledge graph tool version configuration to agents#2319
Conversation
Co-authored-by: anders.hafreager <anders.hafreager@cognite.com>
|
Cursor Agent can help with this pull request. Just |
Co-authored-by: anders.hafreager <anders.hafreager@cognite.com>
- Revert qkg_example to original format to avoid breaking existing tests - Add separate qkg_example_with_v2 for version-specific testing - Update configuration comparison logic to account for automatic version field addition - Fix long test method name that was causing linting failure
andeplane
changed the title
Co-authored-by: anders.hafreager <anders.hafreager@cognite.com>
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Closes AAA-3112
Description
This PR introduces a
versionconfiguration option for theQueryKnowledgeGraphAgentTool.Why this change?
Users need the ability to specify which version of the QKG tool (
v1orv2) their agent deployment should use. This change allows for explicit selection while settingv2as the default, aligning with the latest QKG tool capabilities.What was changed?
version: Literal["v1", "v2"] = "v2"field toQueryKnowledgeGraphAgentToolConfiguration._loadanddumpmethods to handle the newversionfield, ensuring proper serialization and deserialization withv2as the default.Checklist:
If a new method has been added it should be referenced in cognite.rst in order to generate docs based on its docstring.