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

refactor: Remove unused argument

Browse source
This commit is contained in:
Igor Ilic 2025-08-29 12:24:15 +02:00
parent 5bfae7a36b
commit c3f5840bff
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
cognee/modules/retrieval/utils/completion.py
View file

@ -7,13 +7,12 @@ async def generate_completion(
context: str, context: str,
user_prompt_path: str, user_prompt_path: str,
system_prompt_path: str, system_prompt_path: str,
user_prompt: Optional[str] = None,
system_prompt: Optional[str] = None, system_prompt: Optional[str] = None,
only_context: bool = False, only_context: bool = False,
) -> str: ) -> str:
"""Generates a completion using LLM with given context and prompts.""" """Generates a completion using LLM with given context and prompts."""
args = {"question": query, "context": context} args = {"question": query, "context": context}
user_prompt = user_prompt if user_prompt else LLMGateway.render_prompt(user_prompt_path, args) user_prompt = LLMGateway.render_prompt(user_prompt_path, args)
system_prompt = ( system_prompt = (
system_prompt if system_prompt else LLMGateway.read_query_prompt(system_prompt_path) system_prompt if system_prompt else LLMGateway.read_query_prompt(system_prompt_path)
) )