From eadf54b2e5b9070204d7a3f723f1534cf61ddb94 Mon Sep 17 00:00:00 2001
From: Filipe de Avila Belbute-Peres
 <10751509+filipeabperes@users.noreply.github.com>
Date: Tue, 28 Dec 2021 10:32:25 -0300
Subject: [PATCH] Fix CLI tool for generating JSONL for classifications

Edited line 597 to match condition in line 242, "classifications". Also edited line 243 to match documentation ("label", not "labels").
---
 openai/cli.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/openai/cli.py b/openai/cli.py
index 6c07b49135..4c5c8559cf 100644
--- a/openai/cli.py
+++ b/openai/cli.py
@@ -240,7 +240,7 @@ def prepare_data(cls, args, purpose):
         optional_fields = ["metadata"]
 
         if purpose == "classifications":
-            required_fields = ["text", "labels"]
+            required_fields = ["text", "label"]
         else:
             required_fields = ["text"]
 
@@ -594,7 +594,7 @@ def help(args):
         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(