37 lines
994 B
Python
37 lines
994 B
Python
from comfy_api.latest import ComfyExtension, io
|
|
from .gemini_nodes import GetKeyAPI, SSL_GeminiAPIKeyConfig, SSL_GeminiVideoConfig, SSL_GeminiTextPrompt
|
|
from .interactions_nodes import (
|
|
SSL_GeminiInteractionsAPIKeyConfig,
|
|
SSL_GeminiInteractionsSafetyConfig,
|
|
SSL_GeminiInteractionsStore,
|
|
SSL_GeminiInteractionsTextPrompt,
|
|
)
|
|
from .response_schema import SSL_GeminiResponseSchema
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
|
|
class GeminiExtension(ComfyExtension):
|
|
async def on_load(self) -> None:
|
|
pass
|
|
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
return [
|
|
GetKeyAPI,
|
|
SSL_GeminiAPIKeyConfig,
|
|
SSL_GeminiVideoConfig,
|
|
SSL_GeminiResponseSchema,
|
|
SSL_GeminiTextPrompt,
|
|
SSL_GeminiInteractionsAPIKeyConfig,
|
|
SSL_GeminiInteractionsSafetyConfig,
|
|
SSL_GeminiInteractionsStore,
|
|
SSL_GeminiInteractionsTextPrompt,
|
|
]
|
|
|
|
|
|
async def comfy_entrypoint() -> GeminiExtension:
|
|
return GeminiExtension()
|
|
|
|
|
|
__all__ = ['comfy_entrypoint', 'WEB_DIRECTORY']
|