4d3acc358a
<!-- .github/pull_request_template.md --> ## Description <!-- Provide a clear description of the changes in this PR --> ## 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 - **Dependency Update** - Downgraded `mcp` package version from 1.2.0 to 1.1.3 - Updated `cognee` dependency to include additional features with `cognee[codegraph]` - **New Features** - Introduced a new tool, "codify", for transforming codebases into knowledge graphs - Enhanced the existing "search" tool to accept a new parameter for search type - **Improvements** - Streamlined search functionality with a new modular approach - Added new asynchronous function for retrieving and formatting code parts - **Documentation** - Updated import paths for `SearchType` in various modules and tests to reflect structural changes - **Code Cleanup** - Removed legacy search module and associated classes/functions - Refined data transfer object classes for consistency and clarity <!-- end of auto-generated comment: release notes by coderabbit.ai --> --------- Co-authored-by: Boris Arzentar <borisarzentar@gmail.com>
18 lines
554 B
Python
18 lines
554 B
Python
from cognee.modules.retrieval.description_to_codepart_search import (
|
|
code_description_to_code_part_search,
|
|
)
|
|
|
|
|
|
async def code_graph_retrieval(query, include_docs=False):
|
|
retrieved_codeparts, __ = await code_description_to_code_part_search(
|
|
query, include_docs=include_docs
|
|
)
|
|
|
|
return [
|
|
{
|
|
"name": codepart.attributes["file_path"],
|
|
"description": codepart.attributes["file_path"],
|
|
"content": codepart.attributes["source_code"],
|
|
}
|
|
for codepart in retrieved_codeparts
|
|
]
|