diff --git a/modules/llm/chat.py b/modules/llm/chat.py index 9f899ae..ea729c0 100644 --- a/modules/llm/chat.py +++ b/modules/llm/chat.py @@ -8,29 +8,42 @@ from typing import List, Dict, Union, Optional, Any from ..utils import ensure_package, tensor2pil, pil2base64 -gpt_models = [ - "gpt-3.5-turbo", - "gpt-3.5-turbo-16k", + +gpt_vision_models = [ "gpt-4o", "gpt-4o-mini", + "gpt-4.1", + "gpt-4.1-mini", + "gpt-4.1-nano", "gpt-4-turbo", - "gpt-4-vision-preview", "gpt-4-turbo-preview", - "gpt-4-0125-preview", - "gpt-4-1106-preview", - "gpt-4-0613", - "gpt-4", + "gpt-4-vision-preview", "o1", "o1-mini", "o1-preview", - "o3-mini", + "o1-pro", + "o3", + "o3-mini", + "o3-pro", + "o4-mini", +] + +gpt_models = [ + *gpt_vision_models, + "gpt-4", + "gpt-3.5-turbo", + "gpt-3.5-turbo-16k", ] -gpt_vision_models = ["gpt-4o", "gpt-4o-mini", "gpt-4-turbo", "gpt-4-turbo-preview", "gpt-4-vision-preview"] claude3_models = [ + "claude-sonnet-4-20250514", + "claude-opus-4-20250514", + "claude-3-7-sonnet-latest", + "claude-3-7-sonnet-20250219", "claude-3-5-sonnet-latest", "claude-3-5-sonnet-20241022", + "claude-3-5-haiku-20241022", "claude-3-opus-latest", "claude-3-opus-20240229", "claude-3-sonnet-20240229", @@ -53,7 +66,11 @@ aws_regions = [ bedrock_anthropic_versions = ["bedrock-2023-05-31"] bedrock_claude3_models = [ + "anthropic.claude-opus-4-20250514-v1:0", + "anthropic.claude-sonnet-4-20250514-v1:0", + "anthropic.claude-3-7-sonnet-20250219-v1:0", "anthropic.claude-3-5-sonnet-20241022-v2:0", + "anthropic.claude-3-5-haiku-20241022-v1:0", "anthropic.claude-3-haiku-20240307-v1:0", "anthropic.claude-3-sonnet-20240229-v1:0", "anthropic.claude-3-opus-20240229-v1:0", @@ -219,7 +236,7 @@ class AwsBedrockMistralApi(BaseModel): def __init__(self, **data): super().__init__(**data) - ensure_package("boto3", version="1.34.101") + ensure_package("boto3", required_version=">=1.34.101") import boto3 self.bedrock_runtime = boto3.client( @@ -265,7 +282,7 @@ class AwsBedrockClaudeApi(BaseModel): def __init__(self, **data): super().__init__(**data) - ensure_package("boto3", version="1.34.101") + ensure_package("boto3", required_version=">=1.34.101") import boto3 self.bedrock_runtime = boto3.client(