From f81405052b2f637bc909c8bd683b04d6bfd0ca00 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)