mirror of
https://github.com/khoaliber/khoj.git
synced 2026-03-08 05:39:13 +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_openai_chat_models = ["gpt-4o-mini", "gpt-4o"]
|
||||||
default_gemini_chat_models = ["gemini-2.0-flash", "gemini-1.5-pro"]
|
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 = {
|
empty_config = {
|
||||||
"search-type": {
|
"search-type": {
|
||||||
|
|||||||
@@ -104,7 +104,7 @@ def initialization(interactive: bool = True):
|
|||||||
)
|
)
|
||||||
|
|
||||||
# Set up Google's Gemini online chat models
|
# 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,
|
ChatModel.ModelType.GOOGLE,
|
||||||
default_gemini_chat_models,
|
default_gemini_chat_models,
|
||||||
default_api_key=os.getenv("GEMINI_API_KEY"),
|
default_api_key=os.getenv("GEMINI_API_KEY"),
|
||||||
@@ -114,6 +114,22 @@ def initialization(interactive: bool = True):
|
|||||||
provider_name="Google Gemini",
|
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
|
# Set up Anthropic's online chat models
|
||||||
_setup_chat_model_provider(
|
_setup_chat_model_provider(
|
||||||
ChatModel.ModelType.ANTHROPIC,
|
ChatModel.ModelType.ANTHROPIC,
|
||||||
|
|||||||
Reference in New Issue
Block a user