mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-17 22:14:28 +01:00
Use gpt-4 by default for the main thought process
Allow specifying the llm through dotenv Move more things into config
This commit is contained in:
@@ -1,3 +1,9 @@
|
||||
import os
|
||||
|
||||
from dotenv import load_dotenv
|
||||
# Load environment variables from .env file
|
||||
load_dotenv()
|
||||
|
||||
class Singleton(type):
|
||||
"""
|
||||
Singleton metaclass for ensuring only one instance of a class.
|
||||
@@ -21,9 +27,25 @@ class Config(metaclass=Singleton):
|
||||
def __init__(self):
|
||||
self.continuous_mode = False
|
||||
self.speak_mode = False
|
||||
self.fast_llm_model = "gpt-3.5-turbo"
|
||||
self.smart_llm_model = "gpt-3.5-turbo"
|
||||
# TODO - make these models be self-contained, using langchain, so we can configure them once and call it good
|
||||
self.fast_llm_model = os.getenv("FAST_LLM_MODEL", "gpt-3.5-turbo")
|
||||
self.smart_llm_model = os.getenv("SMART_LLM_MODEL", "gpt-4")
|
||||
self.thinking_token_limit = 4000
|
||||
# Initialize the OpenAI API client
|
||||
self.openai_api_key = os.getenv("OPENAI_API_KEY")
|
||||
self.elevenlabs_api_key = os.getenv("ELEVENLABS_API_KEY")
|
||||
# Print values:
|
||||
print("Config values:")
|
||||
print(f"continuous_mode: {self.continuous_mode}")
|
||||
print(f"speak_mode: {self.speak_mode}")
|
||||
print(f"fast_llm_model: {self.fast_llm_model}")
|
||||
print(f"smart_llm_model: {self.smart_llm_model}")
|
||||
print(f"thinking_token_limit: {self.thinking_token_limit}")
|
||||
print(f"openai_api_key: {self.openai_api_key}")
|
||||
print(f"elevenlabs_api_key: {self.elevenlabs_api_key}")
|
||||
|
||||
|
||||
|
||||
|
||||
def set_continuous_mode(self, value: bool):
|
||||
self.continuous_mode = value
|
||||
@@ -39,3 +61,12 @@ class Config(metaclass=Singleton):
|
||||
|
||||
def set_thinking_token_limit(self, value: int):
|
||||
self.thinking_token_limit = value
|
||||
|
||||
def set_openai_api_key(self, value: str):
|
||||
self.apiopenai_api_key_key = value
|
||||
|
||||
def set_elevenlabs_api_key(self, value: str):
|
||||
self.elevenlabs_api_key = value
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user