mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-09-10 19:46:38 +00:00
47f4db3e84
* 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
20 lines
761 B
Python
20 lines
761 B
Python
from __future__ import annotations
|
|
|
|
from typing import List, Optional
|
|
|
|
from comfy_api_nodes.apis import GeminiGenerationConfig, GeminiContent, GeminiSafetySetting, GeminiSystemInstructionContent, GeminiTool, GeminiVideoMetadata
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class GeminiImageGenerationConfig(GeminiGenerationConfig):
|
|
responseModalities: Optional[List[str]] = None
|
|
|
|
|
|
class GeminiImageGenerateContentRequest(BaseModel):
|
|
contents: List[GeminiContent]
|
|
generationConfig: Optional[GeminiImageGenerationConfig] = None
|
|
safetySettings: Optional[List[GeminiSafetySetting]] = None
|
|
systemInstruction: Optional[GeminiSystemInstructionContent] = None
|
|
tools: Optional[List[GeminiTool]] = None
|
|
videoMetadata: Optional[GeminiVideoMetadata] = None
|