Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

make default model of a constant class variable #1780

Merged
merged 4 commits into from
Feb 28, 2024
Merged

make default model of a constant class variable #1780

Changes from 3 commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
b205ca4
make default model of a constant class variable
RohitRathore1 Feb 24, 2024
79a3bbd
Merge branch 'main' into default_model
RohitRathore1 Feb 26, 2024
8b44b35
make default model of a constant class variable
RohitRathore1 Feb 26, 2024
fcdac9d
Merge branch 'main' into default_model
sonichi Feb 27, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 8 additions & 5 deletions autogen/agentchat/contrib/gpt_assistant_agent.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,8 @@ class GPTAssistantAgent(ConversableAgent):
This agent is unique in its reliance on the OpenAI Assistant for state management, differing from other agents like ConversableAgent.
"""

DEFAULT_MODEL_NAME = "gpt-4-0125-preview"

def __init__(
self,
name="GPT Assistant",
Expand Down Expand Up @@ -61,16 +63,17 @@ def __init__(

if llm_config is False:
raise ValueError("llm_config=False is not supported for GPTAssistantAgent.")

# Use AutoGen OpenAIWrapper to create a client
model_name = "gpt-4-0125-preview"
# Use AutooGen OpenAIWrapper to create a client
openai_client_cfg = copy.deepcopy(llm_config)
# Use the class variable
model_name = GPTAssistantAgent.DEFAULT_MODEL_NAME
sonichi marked this conversation as resolved.
Show resolved Hide resolved

# GPTAssistantAgent's azure_deployment param may cause NotFoundError (404) in client.beta.assistants.list()
# See: https://github.com/microsoft/autogen/pull/1721
if openai_client_cfg.get("config_list") is not None and len(openai_client_cfg["config_list"]) > 0:
model_name = openai_client_cfg["config_list"][0].pop("model", "gpt-4-0125-preview")
model_name = openai_client_cfg["config_list"][0].pop("model", GPTAssistantAgent.DEFAULT_MODEL_NAME)
else:
model_name = openai_client_cfg.pop("model", "gpt-4-0125-preview")
model_name = openai_client_cfg.pop("model", GPTAssistantAgent.DEFAULT_MODEL_NAME)

logger.warning("OpenAI client config of GPTAssistantAgent(%s) - model: %s", name, model_name)

Expand Down
Loading