Suggest Google image gen model, new Anthropic chat models on first run

- Update default anthropic chat models to latest good models.

- Now that Google supports a good text to image model. Suggest adding
  that if Google AI API is setup on first run.
This commit is contained in:
Debanjum
2025-03-28 19:13:51 +05:30
parent 1e34de69e9
commit 56b63f95ea
2 changed files with 18 additions and 2 deletions

View File

@@ -19,7 +19,7 @@ default_offline_chat_models = [
]
default_openai_chat_models = ["gpt-4o-mini", "gpt-4o"]
default_gemini_chat_models = ["gemini-2.0-flash", "gemini-1.5-pro"]
default_anthropic_chat_models = ["claude-3-5-sonnet-20241022", "claude-3-5-haiku-20241022"]
default_anthropic_chat_models = ["claude-3-7-sonnet-latest", "claude-3-5-haiku-latest"]
empty_config = {
"search-type": {

View File

@@ -104,7 +104,7 @@ def initialization(interactive: bool = True):
)
# Set up Google's Gemini online chat models
_setup_chat_model_provider(
google_ai_configured, google_ai_provider = _setup_chat_model_provider(
ChatModel.ModelType.GOOGLE,
default_gemini_chat_models,
default_api_key=os.getenv("GEMINI_API_KEY"),
@@ -114,6 +114,22 @@ def initialization(interactive: bool = True):
provider_name="Google Gemini",
)
# Setup Google text to image model
if google_ai_configured:
default_text_to_image_model = "imagen-3.0-generate-002"
if interactive:
gemini_text_to_image_model = input(
f"Enter the Google text to image model you want to use (default: {default_text_to_image_model}): "
)
gemini_text_to_image_model = gemini_text_to_image_model or default_text_to_image_model
else:
gemini_text_to_image_model = default_text_to_image_model
TextToImageModelConfig.objects.create(
model_name=gemini_text_to_image_model,
model_type=TextToImageModelConfig.ModelType.GOOGLE,
ai_model_api=google_ai_provider,
)
# Set up Anthropic's online chat models
_setup_chat_model_provider(
ChatModel.ModelType.ANTHROPIC,