added gemini 2.5

2025-03-25 19:06:24 +01:00
parent 82d1ef55a9
commit cbfcfac12c
1 changed files with 23 additions and 8 deletions
@@ -514,14 +514,29 @@ class Polymath:
                contents = prompt_text

            # Call the model with a configuration that asks for both text and image outputs.
-            response = client.models.generate_content(
-                model=model_value,
-                contents=contents,
-                config=types.GenerateContentConfig(
-                    response_modalities=["Text", "Image"],
-                    seed=seed,
-                ),
-            )
+            try:
+                response = client.models.generate_content(
+                    model=model_value,
+                    contents=contents,
+                    config=types.GenerateContentConfig(
+                        response_modalities=["Text", "Image"],
+                        seed=seed,
+                    ),
+                )
+            except Exception as e:
+                if "only supports text output" in str(e):
+                    if console_log:
+                        print("Gemini model does not support image output. Falling back to text-only.")
+                    response = client.models.generate_content(
+                        model=model_value,
+                        contents=contents,
+                        config=types.GenerateContentConfig(
+                            response_modalities=["Text"],
+                            seed=seed,
+                        ),
+                    )
+                else:
+                    raise e

            # Parse the response: concatenate all text parts and load any image part.
            output_text = ""