gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

chore: adds extract kg from events and changes temporal tasks call

Browse source
This commit is contained in:
hajdul88 2025-08-27 18:02:57 +02:00
parent 2ec22567c3
commit 3482f353a9
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
cognee/api/v1/cognify/cognify.py
View file

@ -22,7 +22,7 @@ from cognee.tasks.graph import extract_graph_from_data
from cognee.tasks.storage import add_data_points from cognee.tasks.storage import add_data_points
from cognee.tasks.summarization import summarize_text from cognee.tasks.summarization import summarize_text
from cognee.modules.pipelines.layers.pipeline_execution_mode import get_pipeline_executor from cognee.modules.pipelines.layers.pipeline_execution_mode import get_pipeline_executor
from cognee.tasks.temporal_graph import extract_events_and_entities from cognee.tasks.temporal_graph import extract_events_and_timestamps, extract_knowledge_graph_from_events
logger = get_logger("cognify") logger = get_logger("cognify")
@ -180,7 +180,7 @@ async def cognify(
- LLM_RATE_LIMIT_REQUESTS: Max requests per interval (default: 60) - LLM_RATE_LIMIT_REQUESTS: Max requests per interval (default: 60)
""" """
if temporal_cognify: if temporal_cognify:
tasks = await get_temporal_tasks(user, graph_model, chunker, chunk_size, ontology_file_path) tasks = await get_temporal_tasks(user, chunker, chunk_size)
else: else:
tasks = await get_default_tasks(user, graph_model, chunker, chunk_size, ontology_file_path) tasks = await get_default_tasks(user, graph_model, chunker, chunk_size, ontology_file_path)
@ -241,7 +241,8 @@ async def get_temporal_tasks(
max_chunk_size=chunk_size or get_max_chunk_tokens(), max_chunk_size=chunk_size or get_max_chunk_tokens(),
chunker=chunker, chunker=chunker,
), ),
Task(extract_events_and_entities, task_config={"chunk_size": 10}), Task(extract_events_and_timestamps, task_config={"chunk_size": 10}),
Task(extract_knowledge_graph_from_events),
Task(add_data_points, task_config={"batch_size": 10}), Task(add_data_points, task_config={"batch_size": 10}),
] ]