From e79a55f2e686f929e46c1357fd72b0bf09c37107 Mon Sep 17 00:00:00 2001 From: pliny <133052465+elder-plinius@users.noreply.github.com> Date: Fri, 3 Nov 2023 17:51:29 -0700 Subject: [PATCH] Update autotemp.py --- autotemp.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/autotemp.py b/autotemp.py index b5f9ff6..bbac0a8 100644 --- a/autotemp.py +++ b/autotemp.py @@ -1,7 +1,7 @@ from swarms.models import OpenAIChat # Replace with your actual OpenAIChat import from termcolor import colored -class MultiTempAgent: +class AutoTemp: 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 @@ -60,6 +60,6 @@ class MultiTempAgent: if __name__ == "__main__": api_key = "" # Your OpenAI API key here - agent = MultiTempAgent(api_key, auto_select=True) # Set auto_select to False if you want manual selection + agent = AutoTemp(api_key, auto_select=True) # Set auto_select to False if you want manual selection prompt = "code a simple new innovative video game that i can play in browser" final_output = agent.run(prompt)