Skip to content

Commit 47f4db3

Browse files
authored
Adding Google Gemini Image API node (Comfy-Org#9566)
* bigcat88's progress on adding Google Gemini Image node * Made Google Gemini Image node functional * Bump frontend version to get static pricing badge on Gemini Image node
1 parent 5352abc commit 47f4db3

File tree

3 files changed

+314
-95
lines changed

3 files changed

+314
-95
lines changed

comfy_api_nodes/apis/gemini_api.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
from __future__ import annotations
2+
3+
from typing import List, Optional
4+
5+
from comfy_api_nodes.apis import GeminiGenerationConfig, GeminiContent, GeminiSafetySetting, GeminiSystemInstructionContent, GeminiTool, GeminiVideoMetadata
6+
from pydantic import BaseModel
7+
8+
9+
class GeminiImageGenerationConfig(GeminiGenerationConfig):
10+
responseModalities: Optional[List[str]] = None
11+
12+
13+
class GeminiImageGenerateContentRequest(BaseModel):
14+
contents: List[GeminiContent]
15+
generationConfig: Optional[GeminiImageGenerationConfig] = None
16+
safetySettings: Optional[List[GeminiSafetySetting]] = None
17+
systemInstruction: Optional[GeminiSystemInstructionContent] = None
18+
tools: Optional[List[GeminiTool]] = None
19+
videoMetadata: Optional[GeminiVideoMetadata] = None

0 commit comments

Comments
 (0)