diff --git a/cognee/modules/data/extraction/knowledge_graph/extract_content_graph.py b/cognee/modules/data/extraction/knowledge_graph/extract_content_graph.py index 20bd4db4a..f477a4b72 100644 --- a/cognee/modules/data/extraction/knowledge_graph/extract_content_graph.py +++ b/cognee/modules/data/extraction/knowledge_graph/extract_content_graph.py @@ -1,4 +1,5 @@ -from typing import Type, Optional +import os +from typing import Type from pydantic import BaseModel from cognee.infrastructure.llm.get_llm_client import get_llm_client from cognee.infrastructure.llm.prompts import render_prompt @@ -10,7 +11,18 @@ async def extract_content_graph(content: str, response_model: Type[BaseModel]): llm_config = get_llm_config() prompt_path = llm_config.graph_prompt_path - system_prompt = render_prompt(prompt_path, {}) + + # Check if the prompt path is an absolute path or just a filename + if os.path.isabs(prompt_path): + # directory containing the file + base_directory = os.path.dirname(prompt_path) + # just the filename itself + prompt_path = os.path.basename(prompt_path) + else: + base_directory = None + + system_prompt = render_prompt(prompt_path, {}, base_directory=base_directory) + content_graph = await llm_client.acreate_structured_output( content, system_prompt, response_model )