From 6de36b6a15e2549f4e7c381d4623feea7fe12bc9 Mon Sep 17 00:00:00 2001
From: Hardik Shah <hjshah@fb.com>
Date: Thu, 8 Aug 2024 10:23:54 -0700
Subject: [PATCH] make optional subclasses default to yes for configuration

---
 .gitignore                                  | 1 +
 llama_toolchain/common/prompt_for_config.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/.gitignore b/.gitignore
index 067fe2c6f..db254fa7f 100644
--- a/.gitignore
+++ b/.gitignore
@@ -3,3 +3,4 @@ __pycache__
 dist
 *.egg-info
 dev_requirements.txt
+build
diff --git a/llama_toolchain/common/prompt_for_config.py b/llama_toolchain/common/prompt_for_config.py
index 35e2aaa69..d29180520 100644
--- a/llama_toolchain/common/prompt_for_config.py
+++ b/llama_toolchain/common/prompt_for_config.py
@@ -158,7 +158,7 @@ def prompt_for_config(
             and issubclass(get_non_none_type(field_type), BaseModel)
         ):
             prompt = f"Do you want to configure {field_name}? (y/n): "
-            if input(prompt).lower() != "y":
+            if input(prompt).lower() == "n":
                 config_data[field_name] = None
                 continue
             nested_type = get_non_none_type(field_type)