silveroxides
8dbc870407
Add a DOM schema builder and optional response formatting before image Autogrow. Return thoughts separately from final text and retain all outputs in schema-aware caches.
27 lines
655 B
Python
27 lines
655 B
Python
from comfy_api.latest import ComfyExtension, io
|
|
from .gemini_nodes import GetKeyAPI, SSL_GeminiAPIKeyConfig, SSL_GeminiVideoConfig, SSL_GeminiTextPrompt
|
|
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,
|
|
]
|
|
|
|
|
|
async def comfy_entrypoint() -> GeminiExtension:
|
|
return GeminiExtension()
|
|
|
|
|
|
__all__ = ['comfy_entrypoint', 'WEB_DIRECTORY']
|