File size: 893 Bytes
5f685fd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
from shortGPT.gpt import gpt_utils
import json
def generateFacts(facts_type):
chat, system = gpt_utils.load_local_yaml_prompt('prompt_templates/facts_generator.yaml')
chat = chat.replace("<<FACTS_TYPE>>", facts_type)
result = gpt_utils.gpt3Turbo_completion(chat_prompt=chat, system=system, temp=1.3)
return result
def generateFactSubjects(n):
out = []
chat, system = gpt_utils.load_local_yaml_prompt('prompt_templates/facts_subjects_generation.yaml')
chat = chat.replace("<<N>>", f"{n}")
count = 0
while len(out) != n:
result = gpt_utils.gpt3Turbo_completion(chat_prompt=chat, system=system, temp=1.69)
count+=1
try:
out = json.loads(result.replace("'", '"'))
except Exception as e:
print(f"INFO - Failed generating {n} fact subjects after {count} trials", e)
pass
return out |