File size: 1,229 Bytes
0c51b93 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
import sys
sys.path.append("../../")
import argparse
from sotopia_rl.prompting.key_utterance_prompting import (
generate_key_utterance_recognition,
)
def main(
data_dir: str, llm_name: str, input_file: str, output_file: str
) -> None:
generate_key_utterance_recognition(
data_dir, llm_name, input_file, output_file
)
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Process some integers.")
parser.add_argument(
"--data_dir",
type=str,
required=True,
help="Directory containing data files",
)
parser.add_argument(
"--llm_name",
type=str,
required=True,
help="Name of the language model",
)
parser.add_argument(
"--input_file",
type=str,
required=False,
default="example_episodes_with_scores.jsonl",
help="Input file",
)
parser.add_argument(
"--output_file",
type=str,
required=False,
default="openai_log_key_utterance.jsonl",
help="Output file",
)
args = parser.parse_args()
print(args.data_dir, args.llm_name)
main(args.data_dir, args.llm_name, args.input_file, args.output_file)
|