chore: places retrievers under modules directory
This commit is contained in:
parent
163bdc527c
commit
c66c43e717
4 changed files with 2 additions and 2 deletions
|
|
@ -1,6 +1,6 @@
|
||||||
import cognee
|
import cognee
|
||||||
import asyncio
|
import asyncio
|
||||||
from cognee.pipelines.retriever.brute_force_triplet_search import brute_force_triplet_search
|
from cognee.modules.retrieval.brute_force_triplet_search import brute_force_triplet_search
|
||||||
|
|
||||||
job_1 = """
|
job_1 = """
|
||||||
CV 1: Relevant
|
CV 1: Relevant
|
||||||
|
|
@ -187,7 +187,7 @@ async def main(enable_steps):
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
# Flags to enable/disable steps
|
# Flags to enable/disable steps
|
||||||
|
|
||||||
rebuild_kg = False
|
rebuild_kg = True
|
||||||
retrieve = True
|
retrieve = True
|
||||||
steps_to_enable = {
|
steps_to_enable = {
|
||||||
"prune_data": rebuild_kg,
|
"prune_data": rebuild_kg,
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue