From 1f8adceebc0366df99ae46394906e69af0d8aaac Mon Sep 17 00:00:00 2001 From: pliny <133052465+elder-plinius@users.noreply.github.com> Date: Sat, 18 Nov 2023 14:19:16 -0800 Subject: [PATCH] Create autotemp_example.py --- playground/autotemp_example.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100644 playground/autotemp_example.py diff --git a/playground/autotemp_example.py b/playground/autotemp_example.py new file mode 100644 index 00000000..9047268d --- /dev/null +++ b/playground/autotemp_example.py @@ -0,0 +1,22 @@ +from swarms.models import OpenAIChat +from swarms.models.autotemp import AutoTemp + +# Your OpenAI API key +api_key = "" + +autotemp_agent = AutoTemp( + api_key=api_key, + alt_temps=[0.4, 0.6, 0.8, 1.0, 1.2], + auto_select=False, + # model_version="gpt-3.5-turbo" # Specify the model version if needed +) + +# Define the task and temperature string +task = "Generate a short story about a lost civilization." +temperature_string = "0.4,0.6,0.8,1.0,1.2," + +# Run the AutoTempAgent +result = autotemp_agent.run(task, temperature_string) + +# Print the result +print(result)