From 57e4656d4bd2d500cbd95287f5bff83a066b78cc Mon Sep 17 00:00:00 2001
From: pliny <133052465+elder-plinius@users.noreply.github.com>
Date: Wed, 1 Nov 2023 14:09:07 -0700
Subject: [PATCH] Create autotemp.py
---
playground/models/autotemp.py | 64 +++++++++++++++++++++++++++++++++++
1 file changed, 64 insertions(+)
create mode 100644 playground/models/autotemp.py
diff --git a/playground/models/autotemp.py b/playground/models/autotemp.py
new file mode 100644
index 00000000..b133e98e
--- /dev/null
+++ b/playground/models/autotemp.py
@@ -0,0 +1,64 @@
+from swarms.models import OpenAIChat # Replace with your actual OpenAIChat import
+from termcolor import colored
+
+class MultiTempAgent:
+ def __init__(self, api_key, default_temp=0.5, alt_temps=None, auto_select=True):
+ self.api_key = api_key
+ self.default_temp = default_temp
+ self.alt_temps = alt_temps if alt_temps else [0.1, 0.3, 0.7, 0.9] # Default alternative temperatures
+ self.auto_select = auto_select
+
+ def ask_user_feedback(self, text):
+ print(colored("Generated text:", "green"))
+ print(colored(text, "white"))
+ feedback = input(colored("Are you satisfied with this output? (yes/no): ", "green"))
+ return feedback.lower() == 'yes'
+
+ def present_options_to_user(self, outputs):
+ print(colored("Alternative outputs:", "green"))
+ for temp, output in outputs.items():
+ print(colored(f"Temperature {temp}:", "green") + colored(f" {output}", "blue"))
+ chosen_temp = float(input(colored("Choose the temperature of the output you like: ", "green")))
+ return outputs.get(chosen_temp, "Invalid temperature chosen.")
+
+ def run(self, prompt):
+ try:
+ llm = OpenAIChat(openai_api_key=self.api_key, temperature=self.default_temp)
+ initial_output = llm(prompt)
+ except Exception as e:
+ print(colored(f"Error generating initial output: {e}", "red"))
+ initial_output = None
+
+ user_satisfied = self.ask_user_feedback(initial_output)
+
+ if user_satisfied:
+ return initial_output
+ else:
+ outputs = {}
+ scores = {}
+ for temp in self.alt_temps:
+ try:
+ llm = OpenAIChat(openai_api_key=self.api_key, temperature=temp)
+ outputs[temp] = llm(prompt)
+ eval_prompt = f"Rate the quality of the following output for our specific task on a scale from 1 to 10. Output only an integer. The output is: {outputs[temp]}"
+ score_str = llm(eval_prompt)
+ scores[temp] = int(score_str.strip())
+ except Exception as e:
+ print(colored(f"Error generating text at temperature {temp}: {e}", "red"))
+ outputs[temp] = None
+ scores[temp] = 0
+
+ if self.auto_select:
+ best_temp = max(scores, key=scores.get)
+ print(colored(f"Automatically selected output from Temperature {best_temp}:", "green"))
+ print(colored(outputs[best_temp], "white"))
+ return outputs[best_temp]
+ else:
+ chosen_output = self.present_options_to_user(outputs)
+ return chosen_output
+
+if __name__ == "__main__":
+ api_key = "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" # Your OpenAI API key here
+ agent = MultiTempAgent(api_key, auto_select=True) # Set auto_select to False if you want manual selection
+ prompt = "Write a creative short story about a purple dragon"
+ final_output = agent.run(prompt)