klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-09 21:29:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
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 # Setup Prompt based on Summary Type
if summary_type == "chat": 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": 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 # Get Response from GPT
response = openai.Completion.create( response = openai.Completion.create(