33 lines
812 B
Python
33 lines
812 B
Python
|
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
"""
|
||
|
|
AIGC 引擎模块
|
||
|
|
所有引擎在此自动注册
|
||
|
|
"""
|
||
|
|
|
||
|
|
from .base import BaseAIGCEngine, EngineResult
|
||
|
|
|
||
|
|
# V1 引擎 (兼容旧接口,使用数据库)
|
||
|
|
from .topic_generate import TopicGenerateEngine
|
||
|
|
from .content_generate import ContentGenerateEngine
|
||
|
|
from .poster_generate import PosterGenerateEngine
|
||
|
|
|
||
|
|
# V2 引擎 (新接口,无数据库依赖)
|
||
|
|
from .topic_generate_v2 import TopicGenerateEngineV2
|
||
|
|
from .content_generate_v2 import ContentGenerateEngineV2
|
||
|
|
from .poster_generate_v2 import PosterGenerateEngineV2
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
'BaseAIGCEngine',
|
||
|
|
'EngineResult',
|
||
|
|
# V1
|
||
|
|
'TopicGenerateEngine',
|
||
|
|
'ContentGenerateEngine',
|
||
|
|
'PosterGenerateEngine',
|
||
|
|
# V2
|
||
|
|
'TopicGenerateEngineV2',
|
||
|
|
'ContentGenerateEngineV2',
|
||
|
|
'PosterGenerateEngineV2',
|
||
|
|
]
|