klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-06 13:22:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Log prompts being passed to GPT for debugging

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2023-03-08 19:17:52 -06:00
parent 2739a492b4
commit f7b8cdd02e
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/khoj/processor/conversation/gpt.py
View File

@@ -1,6 +1,7 @@
# Standard Packages # Standard Packages
import os import os
import json import json
import logging
from datetime import datetime from datetime import datetime
# External Packages # External Packages
@@ -10,6 +11,8 @@ import openai
from khoj.utils.constants import empty_escape_sequences from khoj.utils.constants import empty_escape_sequences
from khoj.utils.helpers import merge_dicts from khoj.utils.helpers import merge_dicts
logger = logging.getLogger(__name__)
def answer(text, user_query, model, api_key=None, temperature=0.5, max_tokens=500): def answer(text, user_query, model, api_key=None, temperature=0.5, max_tokens=500):
""" """
@@ -30,6 +33,7 @@ Question: {user_query}
Answer (in second person):""" Answer (in second person):"""
# Get Response from GPT # Get Response from GPT
logger.debug(f"Prompt for GPT: {prompt}")
response = openai.Completion.create( response = openai.Completion.create(
prompt=prompt, model=model, temperature=temperature, max_tokens=max_tokens, stop='"""' prompt=prompt, model=model, temperature=temperature, max_tokens=max_tokens, stop='"""'
) )
@@ -63,6 +67,7 @@ Summarize the below notes about {user_query}:
Summarize the notes in second person perspective:""" Summarize the notes in second person perspective:"""
# Get Response from GPT # Get Response from GPT
logger.debug(f"Prompt for GPT: {prompt}")
response = openai.Completion.create( response = openai.Completion.create(
prompt=prompt, model=model, temperature=temperature, max_tokens=max_tokens, frequency_penalty=0.2, stop='"""' prompt=prompt, model=model, temperature=temperature, max_tokens=max_tokens, frequency_penalty=0.2, stop='"""'
) )
@@ -106,6 +111,7 @@ A:{ "search-type": "notes" }"""
print(f"Message -> Prompt: {text} -> {prompt}") print(f"Message -> Prompt: {text} -> {prompt}")
# Get Response from GPT # Get Response from GPT
logger.debug(f"Prompt for GPT: {prompt}")
response = openai.Completion.create( response = openai.Completion.create(
prompt=prompt, model=model, temperature=temperature, max_tokens=max_tokens, frequency_penalty=0.2, stop=["\n"] prompt=prompt, model=model, temperature=temperature, max_tokens=max_tokens, frequency_penalty=0.2, stop=["\n"]
) )
@@ -142,6 +148,7 @@ Question: {user_query}"""
) )
# Get Response from GPT # Get Response from GPT
logger.debug(f"Conversation Context for GPT: {messages}")
response = openai.ChatCompletion.create( response = openai.ChatCompletion.create(
messages=messages, messages=messages,
model=model, model=model,