mirror of
https://github.com/aljazceru/dev-gpt.git
synced 2025-12-20 15:14:20 +01:00
29 lines
934 B
Python
29 lines
934 B
Python
# from typing import Dict
|
|
#
|
|
# from dev_gpt.apis.gpt import ask_gpt
|
|
# from dev_gpt.options.generate.chains.question_answering import answer_yes_no_question
|
|
# from dev_gpt.options.generate.parser import identity_parser, boolean_parser
|
|
#
|
|
#
|
|
# def extract_information(text, info_keys) -> Dict[str, str]:
|
|
# extracted_infos = {}
|
|
# for info_key in info_keys:
|
|
# is_information_in_text = answer_yes_no_question(text, f'Is a {info_key} mentioned above?')
|
|
# if is_information_in_text:
|
|
# extracted_info = ask_gpt(
|
|
# extract_information_prompt,
|
|
# identity_parser,
|
|
# text=text,
|
|
# info_key=info_key
|
|
# )
|
|
# extracted_infos[info_key] = extracted_info
|
|
# return extracted_infos
|
|
#
|
|
#
|
|
# extract_information_prompt = '''\
|
|
# {text}
|
|
#
|
|
# Your task:
|
|
# Return all {info_key}s from above.'
|
|
# Note: you must only output your answer.
|
|
# ''' |