mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-04-15 04:52:31 +08:00
30 lines
1.7 KiB
Python
30 lines
1.7 KiB
Python
"""Research Workbench custom nodes for ComfyUI."""
|
|
from typing import List
|
|
from typing_extensions import override
|
|
from comfy_api.latest import ComfyExtension, io
|
|
|
|
|
|
class ResearchExtension(ComfyExtension):
|
|
@override
|
|
async def get_node_list(self) -> List[type[io.ComfyNode]]:
|
|
from custom_nodes.research.paper_search import PaperSearch
|
|
from custom_nodes.research.claim_extract import PaperClaimExtract
|
|
from custom_nodes.research.evidence_assemble import ClaimEvidenceAssemble
|
|
from custom_nodes.research.style_profile import StyleProfileExtract
|
|
from custom_nodes.research.reference_paper_select import ReferencePaperSelect
|
|
from custom_nodes.research.section_plan import SectionPlan
|
|
from custom_nodes.research.abstract_draft import AbstractDraft
|
|
from custom_nodes.research.introduction_draft import IntroductionDraft
|
|
from custom_nodes.research.methods_draft import MethodsDraft
|
|
from custom_nodes.research.consistency_check import ConsistencyCheck
|
|
from custom_nodes.research.export_manuscript import ExportManuscript
|
|
from custom_nodes.research.review_import import ReviewImport
|
|
from custom_nodes.research.review_atomize import ReviewAtomize
|
|
from custom_nodes.research.review_classify import ReviewClassify
|
|
from custom_nodes.research.review_map import ReviewMap
|
|
return [PaperSearch, PaperClaimExtract, ClaimEvidenceAssemble, StyleProfileExtract, ReferencePaperSelect, SectionPlan, AbstractDraft, IntroductionDraft, MethodsDraft, ConsistencyCheck, ExportManuscript, ReviewImport, ReviewAtomize, ReviewClassify, ReviewMap]
|
|
|
|
|
|
async def comfy_entrypoint() -> ComfyExtension:
|
|
return ResearchExtension()
|