mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-09-09 08:37:04 +08:00
46 lines
959 B
Python
46 lines
959 B
Python
|
|
|
|
class GetChoices:
|
|
|
|
def __init__(self, resources, configs):
|
|
self.resources = resources
|
|
self.configs = configs
|
|
|
|
def get_choices(self, source: str):
|
|
"""
|
|
Get choices of things from certain websites
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
# Create a list of available models for the API
|
|
AVAILABLE_MODELS: list[str] = []
|
|
ANTHROPIC_MODELS = [
|
|
"claude-3-5-sonnet-latest",
|
|
"claude-3-5-haiku-latest",
|
|
"claude-3-opus-latest",
|
|
]
|
|
OPEN_AI_MODELS = [
|
|
"gpt-4o",
|
|
"chatgpt-4o-latest",
|
|
"gpt-4o-mini",
|
|
"o1",
|
|
"o1-mini",
|
|
"o1-preview",
|
|
"gpt-4o-realtime-preview",
|
|
"gpt-4o-mini-realtime-preview",
|
|
"gpt-4o-audio-preview",
|
|
]
|
|
AVAILABLE_MODELS.extend(ANTHROPIC_MODELS)
|
|
AVAILABLE_MODELS.extend(OPEN_AI_MODELS)
|
|
|
|
|
|
OPEN_AI_EMBEDDING_MODELS = [
|
|
"text-embedding-3-small",
|
|
"text-embedding-3-large",
|
|
"text-embedding-ada-002",
|
|
]
|
|
TEXT_EMBEDDING_MODELS = []
|
|
TEXT_EMBEDDING_MODELS.append(OPEN_AI_EMBEDDING_MODELS) |