mirror of
https://github.com/khoaliber/khoj.git
synced 2026-03-02 21:19:12 +00:00
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:
@@ -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": {
|
||||
|
||||
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user