nodes/sipherxyz-comfyui-art-venture
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor(llm): reorganize model lists

Browse Source
This commit is contained in:
Tung Nguyen
2025-07-09 14:04:00 +00:00
parent 27c0905ba4
commit e4510faffb
1 changed files with 29 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/llm/chat.py
+29 -12
View File
@@ -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(