Merge pull request #6 from silveroxides/dev

Dev
2025-11-25 21:29:45 +01:00
parent 0a3e58d19f caa97c15f1
commit 32478bd056
2 changed files with 481 additions and 552 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']