klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-02 21:19:12 +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
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):
"""
@@ -61,8 +64,8 @@ def understand(text, api_key=None, temperature=0.5, max_tokens=100, verbose=0):
stop=["\n"])
# Extract, Clean Message from GPT's Response
story = response['choices'][0]['text']
return json.loads(story)
story = str(response['choices'][0]['text'])
return json.loads(story.strip(empty_escape_sequences))
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:"])
# Extract, Clean Message from GPT's Response
story = response['choices'][0]['text']
return str(story).strip()
story = str(response['choices'][0]['text'])
return story.strip(empty_escape_sequences)
def message_to_prompt(user_message, conversation_history="", gpt_message=None, start_sequence="\nAI:", restart_sequence="\nHuman:"):