vonliechti commited on
Commit
800b8b7
·
verified ·
1 Parent(s): 75ccb54

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -11,7 +11,7 @@ openai_role_conversions = {
11
  }
12
 
13
  class OpenAIModel:
14
- def __init__(self, model_name="gpt-4o"):
15
  self.model_name = model_name
16
  self.client = OpenAI(
17
  api_key=os.getenv("OPENAI_API_KEY"),
@@ -33,7 +33,7 @@ def get_agent(
33
  system_prompt=DEFAULT_SQUAD_REACT_CODE_SYSTEM_PROMPT,
34
  toolbox=DEFAULT_TASK_SOLVING_TOOLBOX,
35
  use_openai=False,
36
- openai_model_name="gpt-4o",
37
  ):
38
  DEFAULT_MODEL_NAME = "http://localhost:1234/v1"
39
  if model_name is None:
 
11
  }
12
 
13
  class OpenAIModel:
14
+ def __init__(self, model_name="gpt-4o-mini-2024-07-18"):
15
  self.model_name = model_name
16
  self.client = OpenAI(
17
  api_key=os.getenv("OPENAI_API_KEY"),
 
33
  system_prompt=DEFAULT_SQUAD_REACT_CODE_SYSTEM_PROMPT,
34
  toolbox=DEFAULT_TASK_SOLVING_TOOLBOX,
35
  use_openai=False,
36
+ openai_model_name="gpt-4o-mini-2024-07-18",
37
  ):
38
  DEFAULT_MODEL_NAME = "http://localhost:1234/v1"
39
  if model_name is None: