In progress migration to V3 and update of code.

2025-11-23 19:32:23 +01:00
parent 0a3e58d19f
commit d3adb48c2c
2 changed files with 423 additions and 554 deletions
@@ -1,5 +1,18 @@
-import os
-import folder_paths
-from .gemini_nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+from comfy_api.latest import ComfyExtension, io
+from .gemini_nodes import GetKeyAPI, SSL_GeminiAPIKeyConfig, SSL_GeminiTextPrompt

-__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
+
+class GeminiExtension(ComfyExtension):
+	async def get_node_list(self) -> list[type[io.ComfyNode]]:
+		return [
+			GetKeyAPI,
+			SSL_GeminiAPIKeyConfig,
+			SSL_GeminiTextPrompt,
+		]
+
+
+async def comfy_entrypoint() -> GeminiExtension:
+	return GeminiExtension()
+
+
+__all__ = ['comfy_entrypoint']