mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-15 03:27:24 +08:00
b94941d8d3
* [Partner Nodes] add Claude LLM node Signed-off-by: bigcat88 <bigcat88@icloud.com> * [Partner Nodes] add seed param Signed-off-by: bigcat88 <bigcat88@icloud.com> * [Partner Nodes] use image urls instead of base64 Signed-off-by: bigcat88 <bigcat88@icloud.com> * [Partner Nodes] fixed pricing for the claude 4.7 Signed-off-by: bigcat88 <bigcat88@icloud.com> --------- Signed-off-by: bigcat88 <bigcat88@icloud.com>
76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
from enum import Enum
|
|
from typing import Literal
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class AnthropicRole(str, Enum):
|
|
user = "user"
|
|
assistant = "assistant"
|
|
|
|
|
|
class AnthropicTextContent(BaseModel):
|
|
type: Literal["text"] = "text"
|
|
text: str = Field(...)
|
|
|
|
|
|
class AnthropicImageSourceBase64(BaseModel):
|
|
type: Literal["base64"] = "base64"
|
|
media_type: str = Field(..., description="MIME type of the image, e.g. image/png, image/jpeg")
|
|
data: str = Field(..., description="Base64-encoded image data")
|
|
|
|
|
|
class AnthropicImageSourceUrl(BaseModel):
|
|
type: Literal["url"] = "url"
|
|
url: str = Field(...)
|
|
|
|
|
|
class AnthropicImageContent(BaseModel):
|
|
type: Literal["image"] = "image"
|
|
source: AnthropicImageSourceBase64 | AnthropicImageSourceUrl = Field(...)
|
|
|
|
|
|
class AnthropicMessage(BaseModel):
|
|
role: AnthropicRole = Field(...)
|
|
content: list[AnthropicTextContent | AnthropicImageContent] = Field(...)
|
|
|
|
|
|
class AnthropicMessagesRequest(BaseModel):
|
|
model: str = Field(...)
|
|
messages: list[AnthropicMessage] = Field(...)
|
|
max_tokens: int = Field(..., ge=1)
|
|
system: str | None = Field(None, description="Top-level system prompt")
|
|
temperature: float | None = Field(None, ge=0.0, le=1.0)
|
|
top_p: float | None = Field(None, ge=0.0, le=1.0)
|
|
top_k: int | None = Field(None, ge=0)
|
|
stop_sequences: list[str] | None = Field(None)
|
|
|
|
|
|
class AnthropicResponseTextBlock(BaseModel):
|
|
type: Literal["text"] = "text"
|
|
text: str = Field(...)
|
|
|
|
|
|
class AnthropicCacheCreationUsage(BaseModel):
|
|
ephemeral_5m_input_tokens: int | None = Field(None)
|
|
ephemeral_1h_input_tokens: int | None = Field(None)
|
|
|
|
|
|
class AnthropicMessagesUsage(BaseModel):
|
|
input_tokens: int | None = Field(None)
|
|
output_tokens: int | None = Field(None)
|
|
cache_creation_input_tokens: int | None = Field(None)
|
|
cache_read_input_tokens: int | None = Field(None)
|
|
cache_creation: AnthropicCacheCreationUsage | None = Field(None)
|
|
|
|
|
|
class AnthropicMessagesResponse(BaseModel):
|
|
id: str | None = Field(None)
|
|
type: str | None = Field(None)
|
|
role: str | None = Field(None)
|
|
model: str | None = Field(None)
|
|
content: list[AnthropicResponseTextBlock] | None = Field(None)
|
|
stop_reason: str | None = Field(None)
|
|
stop_sequence: str | None = Field(None)
|
|
usage: AnthropicMessagesUsage | None = Field(None)
|