mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-06-14 23:35:34 +08:00
47d55b8b45
* add support to read pyproject.toml from custom node * sf * use pydantic instead * sf * use pydantic_settings * remove unnecessary try/catch and handle single-file python node * sf
81 lines
2.3 KiB
Python
81 lines
2.3 KiB
Python
from pydantic import BaseModel, Field
|
|
from pydantic_settings import BaseSettings, SettingsConfigDict
|
|
from typing import List, Optional
|
|
|
|
# IMPORTANT: The type definitions specified in pyproject.toml for custom nodes
|
|
# must remain synchronized with the corresponding files in the https://github.com/Comfy-Org/comfy-cli/blob/main/comfy_cli/registry/types.py.
|
|
# Any changes to one must be reflected in the other to maintain consistency.
|
|
|
|
class NodeVersion(BaseModel):
|
|
changelog: str
|
|
dependencies: List[str]
|
|
deprecated: bool
|
|
id: str
|
|
version: str
|
|
download_url: str
|
|
|
|
|
|
class Node(BaseModel):
|
|
id: str
|
|
name: str
|
|
description: str
|
|
author: Optional[str] = None
|
|
license: Optional[str] = None
|
|
icon: Optional[str] = None
|
|
repository: Optional[str] = None
|
|
tags: List[str] = Field(default_factory=list)
|
|
latest_version: Optional[NodeVersion] = None
|
|
|
|
|
|
class PublishNodeVersionResponse(BaseModel):
|
|
node_version: NodeVersion
|
|
signedUrl: str
|
|
|
|
|
|
class URLs(BaseModel):
|
|
homepage: str = Field(default="", alias="Homepage")
|
|
documentation: str = Field(default="", alias="Documentation")
|
|
repository: str = Field(default="", alias="Repository")
|
|
issues: str = Field(default="", alias="Issues")
|
|
|
|
|
|
class Model(BaseModel):
|
|
location: str
|
|
model_url: str
|
|
|
|
|
|
class ComfyConfig(BaseModel):
|
|
publisher_id: str = Field(default="", alias="PublisherId")
|
|
display_name: str = Field(default="", alias="DisplayName")
|
|
icon: str = Field(default="", alias="Icon")
|
|
models: List[Model] = Field(default_factory=list, alias="Models")
|
|
includes: List[str] = Field(default_factory=list)
|
|
|
|
|
|
class License(BaseModel):
|
|
file: str = ""
|
|
text: str = ""
|
|
|
|
|
|
class ProjectConfig(BaseModel):
|
|
name: str = ""
|
|
description: str = ""
|
|
version: str = "1.0.0"
|
|
requires_python: str = Field(default=">= 3.9", alias="requires-python")
|
|
dependencies: List[str] = Field(default_factory=list)
|
|
license: License = Field(default_factory=License)
|
|
urls: URLs = Field(default_factory=URLs)
|
|
|
|
|
|
class PyProjectConfig(BaseModel):
|
|
project: ProjectConfig = Field(default_factory=ProjectConfig)
|
|
tool_comfy: ComfyConfig = Field(default_factory=ComfyConfig)
|
|
|
|
|
|
class PyProjectSettings(BaseSettings):
|
|
project: dict = Field(default_factory=dict)
|
|
|
|
tool: dict = Field(default_factory=dict)
|
|
|
|
model_config = SettingsConfigDict()
|