update installers

install.bat

@@ -5,6 +5,7 @@ setlocal enabledelayedexpansion
 if not exist %~dp0\yolo.py ( echo `yolo.py` missing in %~dp0 cannot install & goto :choice_default_3 )
 if not exist %~dp0\prompt.txt ( echo `prompt.txt` missing in %~dp0 cannot install & goto :choice_default_3 ) 
 if not exist %~dp0\yolo.yaml ( echo `yolo.yaml` missing in %~dp0 cannot install & goto :choice_default_3 ) 
+if not exist %~dp0\ai_model.py ( echo `ai_model.py` missing in %~dp0 cannot install & goto :choice_default_3 ) 
 
 
 :: Note: "~" or %HOME% is equivalent to "%HOMEDRIVE%%HOMEPATH%\" but the latter is set in VM environments (from what I can tell)
@@ -137,6 +138,7 @@ mkdir !TARGET_DIR!
 copy  %~dp0\yolo.py !TARGET_DIR!
 copy  %~dp0\prompt.txt !TARGET_DIR!
 copy  %~dp0\yolo.yaml !TARGET_DIR!
+copy  %~dp0\ai_model.py !TARGET_DIR!
 goto :EOF
 
 :: Create yolo.bat and input code linking to created directory
@@ -312,6 +314,6 @@ echo   -If you run PowerShell as administrator you can then run `setx OPENAI_API
 echo   -Go to `Start` and search `edit environment variables for your account` and manually create the variable with name `OPENAI_API_KEY` and value `[yourkey]`
 echo (4) Another option is to put the API key in the yolo.yaml configuration file (since v.0.2)
 echo.
-echo Yolo also supports Azure OpenAI now. Configure settings in yolo.yaml accordingly.
+echo Yolo also supports Azure OpenAI, and many other LLMs now. Configure settings in yolo.yaml accordingly.
 echo.
 goto :EOF

install.sh

@@ -7,7 +7,7 @@ TARGET_FULLPATH=$TARGET_DIR/yolo.py
 mkdir -p $TARGET_DIR
 
 echo "- Copying files..."
-cp yolo.py prompt.txt yolo.yaml $TARGET_DIR
+cp yolo.py prompt.txt yolo.yaml ai_model.py $TARGET_DIR
 chmod +x $TARGET_FULLPATH
 
 # Creates two aliases for use
@@ -34,11 +34,11 @@ fi
 echo
 echo "Done."
 echo
-echo "Make sure you have the OpenAI API key set via one of these options:" 
+echo "Make sure you have your LLM key (e.g. OpenAI API) set via one of these options:" 
 echo "  - environment variable"
-echo "  - .env or an ~/.openai.apikey file or in"
+echo "  - .env or in"
 echo "  - yolo.yaml"
 echo
-echo "Yolo also supports Azure OpenAI now. Change settings in yolo.yaml accordingly."
+echo "Yolo also supports Azure OpenAI, Ollama, groq, Claude now. Change settings in yolo.yaml accordingly."
 echo
 echo "Have fun!"