Commit 0c4e7f03

Filipe de Avila Belbute-Peres <10751509+filipeabperes@users.noreply.github.com>
2022-01-04 02:52:53
Fix CLI tool for generating JSONL for classifications (#62)
Edited line 597 to match condition in line 242, "classifications". Also edited line 243 to match documentation ("label", not "labels").
main
1 parent 26fbacb
Changed files (1)
openai
cli.py
openai/cli.py
@@ -240,7 +240,7 @@ class Search:
         optional_fields = ["metadata"]
 
         if purpose == "classifications":
-            required_fields = ["text", "labels"]
+            required_fields = ["text", "label"]
         else:
             required_fields = ["text"]
 
@@ -594,7 +594,7 @@ def tools_register(parser):
         action="store_true",
         help="Auto accepts all suggestions, without asking for user input. To be used within scripts.",
     )
-    sub.set_defaults(func=partial(Search.prepare_data, purpose="classification"))
+    sub.set_defaults(func=partial(Search.prepare_data, purpose="classifications"))
 
     sub = subparsers.add_parser("answers.prepare_data")
     sub.add_argument(