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

fixes to cognee

Browse source
This commit is contained in:
Vasilije 2024-05-17 14:21:16 +02:00
parent db0b19bb30
commit 4e6fcdec25
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -32,7 +32,7 @@ from cognee.modules.data.get_cognitive_layers import get_cognitive_layers
from cognee.modules.data.get_layer_graphs import get_layer_graphs
from cognee.modules.topology.topology import TopologyEngine
from cognee.shared.GithubClassification import CodeContentPrediction
from cognee.shared.data_models import ChunkStrategy
from cognee.shared.data_models import ChunkStrategy, DefaultGraphModel
from cognee.utils import send_telemetry
@ -178,8 +178,10 @@ async def process_text(chunk_collection: str, chunk_id: str, input_text: str, fi
graph_topology = infrastructure_config.get_config()["graph_topology"]
print("got here")
if graph_topology == "default":
parent_node_id = f"{file_metadata['name']}.{file_metadata['extension']}"
elif graph_topology == DefaultGraphModel:
parent_node_id = f"DefaultGraphModel__{USER_ID}"
document_id = await add_document_node(
graph_client,