Drop image generation support for Stability AI models

Reduces maintenance burden by dropping support for old ai model
providers

documentation/docs/advanced/admin.md

@@ -52,8 +52,8 @@ Search models are used to generate vector embeddings of your documents for natur
 <img src="/img/example_search_model_admin_settings.png" alt="Example Search Model Settings" style={{width: 500}} />
 
 ### Text to Image Model Options
-Add text to image generation models with these settings. Khoj currently supports text to image models available via OpenAI, Stability or Replicate API
-- `api-key`: Set to your OpenAI, Stability or Replicate API key
+Add text to image generation models with these settings. Khoj currently supports text to image models available via OpenAI, Google or Replicate API
+- `api-key`: Set to your OpenAI, Google AI or Replicate API key
 - `model`: Set the model name available over the selected model provider
 - `model-type`: Set to the appropriate model provider
 - `openai-config`: For image generation models available via OpenAI (compatible) API you can set the appropriate OpenAI Processor Conversation Settings instead of specifying the `api-key` field above

src/khoj/database/migrations/0098_alter_texttoimagemodelconfig_model_type.py

@@ -0,0 +1,31 @@
+# Generated by Django 5.1.14 on 2025-11-30 00:47
+
+from django.db import migrations, models
+
+
+def delete_stability_configs(apps, schema_editor):
+    TextToImageModelConfig = apps.get_model("database", "TextToImageModelConfig")
+    TextToImageModelConfig.objects.filter(model_type="stability-ai").delete()
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0097_serverchatsettings_priority"),
+    ]
+
+    operations = [
+        migrations.RunPython(delete_stability_configs, migrations.RunPython.noop),
+        migrations.AlterField(
+            model_name="texttoimagemodelconfig",
+            name="model_type",
+            field=models.CharField(
+                choices=[
+                    ("openai", "Openai"),
+                    ("replicate", "Replicate"),
+                    ("google", "Google"),
+                ],
+                default="openai",
+                max_length=200,
+            ),
+        ),
+    ]

src/khoj/database/models/__init__.py

@@ -575,7 +575,6 @@ class SearchModelConfig(DbBaseModel):
 class TextToImageModelConfig(DbBaseModel):
     class ModelType(models.TextChoices):
         OPENAI = "openai"
-        STABILITYAI = "stability-ai"
         REPLICATE = "replicate"
         GOOGLE = "google"
 

src/khoj/processor/image/generate.py

@@ -112,8 +112,6 @@ async def text_to_image(
                 webp_image_bytes = generate_image_with_openai(
                     image_prompt, text_to_image_config, text2image_model, image_shape
                 )
-            elif text_to_image_config.model_type == TextToImageModelConfig.ModelType.STABILITYAI:
-                webp_image_bytes = generate_image_with_stability(image_prompt, text_to_image_config, text2image_model)
             elif text_to_image_config.model_type == TextToImageModelConfig.ModelType.REPLICATE:
                 webp_image_bytes = generate_image_with_replicate(
                     image_prompt, text_to_image_config, text2image_model, image_shape
@@ -220,35 +218,6 @@ def generate_image_with_openai(
     return convert_image_to_webp(base64.b64decode(image))
 
 
-@retry(
-    retry=retry_if_exception_type(requests.RequestException),
-    wait=wait_random_exponential(min=1, max=10),
-    stop=stop_after_attempt(3),
-    before_sleep=before_sleep_log(logger, logging.DEBUG),
-    reraise=True,
-)
-def generate_image_with_stability(
-    improved_image_prompt: str, text_to_image_config: TextToImageModelConfig, text2image_model: str
-):
-    "Generate image using Stability AI"
-
-    # Call Stability AI API to generate image
-    response = requests.post(
-        "https://api.stability.ai/v2beta/stable-image/generate/sd3",
-        headers={"authorization": f"Bearer {text_to_image_config.api_key}", "accept": "image/*"},
-        files={"none": ""},
-        data={
-            "prompt": improved_image_prompt,
-            "model": text2image_model,
-            "mode": "text-to-image",
-            "output_format": "png",
-            "aspect_ratio": "1:1",
-        },
-    )
-    # Convert png to webp for faster loading
-    return convert_image_to_webp(response.content)
-
-
 @retry(
     retry=retry_if_exception_type(requests.RequestException),
     wait=wait_random_exponential(min=1, max=10),