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

chore: fixes ruff

Browse source
This commit is contained in:
hajdul88 2025-08-18 18:36:04 +02:00
parent d4ace62664
commit 0fbe218eef
3 changed files with 18 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
cognee/modules/retrieval/user_qa_feedback.py
View file

@ -26,7 +26,6 @@ class UserQAFeedback(BaseFeedback):
self.last_k = last_k self.last_k = last_k
async def add_feedback(self, feedback_text: str) -> List[str]: async def add_feedback(self, feedback_text: str) -> List[str]:
feedback_sentiment = await LLMGateway.acreate_structured_output( feedback_sentiment = await LLMGateway.acreate_structured_output(
text_input=feedback_text, text_input=feedback_text,
system_prompt="You are a sentiment analysis assistant. For each piece of user feedback you receive, return exactly one of: Positive, Negative, or Neutral classification", system_prompt="You are a sentiment analysis assistant. For each piece of user feedback you receive, return exactly one of: Positive, Negative, or Neutral classification",

21
cognee/modules/search/methods/search.py
View file

@ -68,7 +68,7 @@ async def search(
system_prompt_path=system_prompt_path, system_prompt_path=system_prompt_path,
top_k=top_k, top_k=top_k,
save_interaction=save_interaction, save_interaction=save_interaction,
last_k=last_k last_k=last_k,
) )
query = await log_query(query_text, query_type.value, user.id) query = await log_query(query_text, query_type.value, user.id)
@ -82,7 +82,7 @@ async def search(
node_type=node_type, node_type=node_type,
node_name=node_name, node_name=node_name,
save_interaction=save_interaction, save_interaction=save_interaction,
last_k=last_k last_k=last_k,
) )
await log_result( await log_result(
@ -188,7 +188,14 @@ async def authorized_search(
# Searches all provided datasets and handles setting up of appropriate database context based on permissions # Searches all provided datasets and handles setting up of appropriate database context based on permissions
search_results = await specific_search_by_context( search_results = await specific_search_by_context(
search_datasets, query_text, query_type, user, system_prompt_path, top_k, save_interaction, last_k=last_k search_datasets,
query_text,
query_type,
user,
system_prompt_path,
top_k,
save_interaction,
last_k=last_k,
) )
await log_result(query.id, json.dumps(search_results, cls=JSONEncoder), user.id) await log_result(query.id, json.dumps(search_results, cls=JSONEncoder), user.id)
@ -211,7 +218,9 @@ async def specific_search_by_context(
Not to be used outside of active access control mode. Not to be used outside of active access control mode.
""" """
async def _search_by_context(dataset, user, query_type, query_text, system_prompt_path, top_k, last_k): async def _search_by_context(
dataset, user, query_type, query_text, system_prompt_path, top_k, last_k
):
# Set database configuration in async context for each dataset user has access for # Set database configuration in async context for each dataset user has access for
await set_database_global_context_variables(dataset.id, dataset.owner_id) await set_database_global_context_variables(dataset.id, dataset.owner_id)
search_results = await specific_search( search_results = await specific_search(
@ -233,7 +242,9 @@ async def specific_search_by_context(
tasks = [] tasks = []
for dataset in search_datasets: for dataset in search_datasets:
tasks.append( tasks.append(
_search_by_context(dataset, user, query_type, query_text, system_prompt_path, top_k, last_k) _search_by_context(
dataset, user, query_type, query_text, system_prompt_path, top_k, last_k
)
) )
return await asyncio.gather(*tasks) return await asyncio.gather(*tasks)