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

Use empty_escape_sequence constant to strip response text from gpt

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2022-02-27 19:53:53 -05:00
parent 37bfc956c9
commit 1eba7b1c6f
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/processor/conversation/gpt.py
View File

@@ -6,6 +6,9 @@ from datetime import datetime
# External Packages # External Packages
import openai import openai
# Internal Packages
from src.utils.constants import empty_escape_sequences
def summarize(text, summary_type, user_query=None, api_key=None, temperature=0.5, max_tokens=100): def summarize(text, summary_type, user_query=None, api_key=None, temperature=0.5, max_tokens=100):
""" """
@@ -61,8 +64,8 @@ def understand(text, api_key=None, temperature=0.5, max_tokens=100, verbose=0):
stop=["\n"]) stop=["\n"])
# Extract, Clean Message from GPT's Response # Extract, Clean Message from GPT's Response
story = response['choices'][0]['text'] story = str(response['choices'][0]['text'])
return json.loads(story) return json.loads(story.strip(empty_escape_sequences))
def converse(text, conversation_history=None, api_key=None, temperature=0.9, max_tokens=150): def converse(text, conversation_history=None, api_key=None, temperature=0.9, max_tokens=150):
@@ -91,8 +94,8 @@ def converse(text, conversation_history=None, api_key=None, temperature=0.9, max
stop=["\n", "Human:", "AI:"]) stop=["\n", "Human:", "AI:"])
# Extract, Clean Message from GPT's Response # Extract, Clean Message from GPT's Response
story = response['choices'][0]['text'] story = str(response['choices'][0]['text'])
return str(story).strip() return story.strip(empty_escape_sequences)
def message_to_prompt(user_message, conversation_history="", gpt_message=None, start_sequence="\nAI:", restart_sequence="\nHuman:"): def message_to_prompt(user_message, conversation_history="", gpt_message=None, start_sequence="\nAI:", restart_sequence="\nHuman:"):