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

feat: Add simple instruction for system prompt

Browse source 
Add simple instruction for system prompt

Feature COG-656
This commit is contained in:
Igor Ilic 2024-12-13 15:30:24 +01:00
parent c90d90a86a
commit 67585d0ab1
2 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
cognee/infrastructure/llm/prompts/answer_simple_question.txt Normal file
View file

@ -0,0 +1 @@
Answer the question using the provided context. Be as brief as possible.

2
cognee/tasks/compute/query_compute.py
View file

@ -24,7 +24,7 @@ async def query_compute(query: str) -> list:
"context": found_chunks[0].payload["text"], "context": found_chunks[0].payload["text"],
} }
user_prompt = render_prompt("context_for_question.txt", args) user_prompt = render_prompt("context_for_question.txt", args)
system_prompt = read_query_prompt("answer_hotpot_using_cognee_search.txt") system_prompt = read_query_prompt("answer_simple_question.txt")
llm_client = get_llm_client() llm_client = get_llm_client()
computed_answer = await llm_client.acreate_structured_output( computed_answer = await llm_client.acreate_structured_output(