changes

Author: jonathanalgar

agents/generate_and_validate.py

@@ -34,13 +34,13 @@ def generate_and_validate(
     logger.info(f"Generating solution for: {use_case}")
 
     env_retain = os.getenv("RETAIN_ON_FAILURE")
-    if env_retain is not None:
+    if env_retain and env_retain.strip():
         retain_on_failure = env_retain.lower() in ["true", "1", "yes", "y"]
         logger.info(f"Using RETAIN_ON_FAILURE={retain_on_failure} from .env")
     else:
         retain_input = input("Retain solution even if build fails? (y/n) [default: y]: ")
         retain_on_failure = retain_input.lower() in ["y", "yes"] or retain_input == ""
-        print("(Tip: add RETAIN_ON_FAILURE=true or RETAIN_ON_FAILURE=true to .env to skip this input prompt)")
+        print("(Tip: add RETAIN_ON_FAILURE=true or RETAIN_ON_FAILURE=true to .env to skip this input prompt in future)")
 
     llm_inputs = LLMInputs(
         use_case=use_case,

agents/generate_library.py

@@ -56,9 +56,7 @@ def _get_model_from_env_or_prompt(available_models: dict, model_type: str, env_v
                 print(f"Using {env_model_name} for {model_type} (from environment variable)")
                 return models[env_model_name]
         print(f"Warning: Model '{env_model_name}' specified in {env_var_name} not found. Prompting for selection.")
-
-    print(f"(Tip: add {env_var_name} to .env to skip this input prompt)")
-    return select_model(available_models, model_type)
+    return select_model(available_models, model_type, env_var_name)
 
 
 def main() -> None:
@@ -68,9 +66,10 @@ def main() -> None:
         raise EnvironmentError("No OpenAI API key found in .env file. Please add OPENAI_API_KEY")
 
     available_models = get_available_models(api_keys)
+
     use_case = get_user_input()
 
-    print("\nSelect models for library generation:")
+    print("\nNow let's select the which LLMs to use to generate the library..")
     search_term_llm = _get_model_from_env_or_prompt(available_models, "NuGet package search", "SEARCH_TERM_LLM")
 
     code_generation_llm = _get_model_from_env_or_prompt(available_models, "code generation", "CODE_GENERATION_LLM")

agents/utils/model_definitions.py

@@ -45,17 +45,19 @@ def get_model(model_name: str) -> BaseChatModel:
         raise KeyError(f"Model {model_name} not found in any provider.")
 
 
-def select_model(available_models: Dict[str, Dict[str, Any]], purpose: str) -> Any:
+def select_model(available_models: Dict[str, Dict[str, Any]], purpose: str, env_var_name: str) -> Any:
     """Prompts the user to select a model from the available models based on the given purpose."""
-    print(f"\nSelect model for {purpose}:")
+    print(f"\nSelect model for {purpose} (Tip: add {env_var_name} to .env to skip this input prompt in future):")
     options = _get_model_options(available_models, purpose)
     default_option = options[0]
-
     for idx, (name, _) in enumerate(options, start=1):
         if idx == 1:
             print(f"{idx}. {name} (default - press Enter)")
         else:
             print(f"{idx}. {name}")
+    print(
+        "! If you are using the free tier of the OpenAI API, only gpt4o-mini model will work (see https://platform.openai.com/docs/guides/rate-limits)"
+    )
 
     while True:
         choice = input("Enter your choice (number or press Enter for default): ").strip()