Skip to content

Commit 5d399ea

Browse files
committed
refactor: simplify client initialization and enhance content generation configuration
1 parent 95f6c2f commit 5d399ea

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

  • apps/models_provider/impl/gemini_model_provider/model

apps/models_provider/impl/gemini_model_provider/model/tti.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def generate_image(self, prompt: str, negative_prompt: str = None):
5252
from google.genai import types
5353
from PIL import Image
5454
file_urls = []
55-
client = genai.Client(api_key=self.api_key, http_options={"base_url": self.base_url}, **self.params)
55+
client = genai.Client(api_key=self.api_key, http_options={"base_url": self.base_url})
5656
if self.model.startswith('imagen'):
5757
config = types.GenerateImagesConfig(**self.params)
5858

@@ -68,7 +68,9 @@ def generate_image(self, prompt: str, negative_prompt: str = None):
6868
img_base64 = base64.b64encode(generated_image.image.image_bytes).decode("utf-8")
6969
file_urls.append(f'data:{generated_image.image.mime_type};base64,{img_base64}')
7070
else:
71-
config = types.GenerateContentConfig(**self.params)
71+
config = types.GenerateContentConfig(image_config=types.ImageConfig(
72+
**self.params
73+
))
7274
if negative_prompt:
7375
config.negative_prompt = negative_prompt
7476
response = client.models.generate_content(

0 commit comments

Comments
 (0)