Upload folder using huggingface_hub
Browse files
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:
|