Refactor mix prompt construction in mix_kg_vector_query function #1200
+13
−6
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.
Description
This PR enhances the way LightRAG manages custom system prompts in the "mix" query! Before, when you provided a custom system prompt, it missed out on including the retrieval information. The only option was to stick with a predefined prompt. But now, it also takes into account situations where the custom system prompt might not have all the key parameters. In such cases, it seamlessly extends the prompt with theflexible and user-friendly. retrieved data, making it more
Changes Made
mix_kg_vector_query
- handles both knowledge graph and vector contextskg_query
- handles knowledge graph contextnaive_query
- handles vector search resultsUse Cases Addressed
Checklist
Additional Notes
This change allows users to provide custom prompts with and without having to include specific formatting placeholders. The system will now intelligently append any available context information to the end of the prompt with proper section headers when formatting keys aren't present.