sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2025-02-17 08:04:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Improve readability of GPT prompt strings in conversation processor

Browse source
This commit is contained in:
Debanjum Singh Solanky 2023-01-08 12:04:56 -03:00
parent 1c091e509b
commit f2436039a0
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/processor/conversation/gpt.py
View file

@ -19,9 +19,19 @@ def summarize(text, summary_type, user_query=None, api_key=None, temperature=0.5
# Setup Prompt based on Summary Type
if summary_type == "chat":
prompt = f"You are an AI. Summarize the conversation below from your perspective:\n\n{text}\n\nSummarize the conversation from the AI's first-person perspective:"
prompt = f'''
You are an AI. Summarize the conversation below from your perspective:
{text}
Summarize the conversation from the AI's first-person perspective:'''
elif summary_type == "notes":
prompt = f"Summarize the below notes about {user_query}:\n\n{text}\n\nSummarize the notes in second person perspective and use past tense:"
prompt = f'''
Summarize the below notes about {user_query}:
{text}
Summarize the notes in second person perspective and use past tense:'''
# Get Response from GPT
response = openai.Completion.create(