apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-06-30 21:59:46 +02:00
Code Issues Projects Releases Packages Wiki Activity

chore: made generate_image more agnostic

Browse source
This commit is contained in:
DESKTOP-RTLN3BA\$punk 2026-02-05 17:18:27 -08:00
parent 19e2857343
commit f85adefe5e
15 changed files with 176 additions and 112 deletions
Download patch file Download diff file Expand all files Collapse all files

9
surfsense_backend/app/db.py
View file

@ -100,7 +100,6 @@ class PodcastStatus(str, Enum):
FAILED = "failed"
class LiteLLMProvider(str, Enum):
"""
Enum for LLM providers supported by LiteLLM.
@ -941,7 +940,9 @@ class ImageGenerationConfig(BaseModel, TimestampMixin):
search_space_id = Column(
Integer, ForeignKey("searchspaces.id", ondelete="CASCADE"), nullable=False
)
search_space = relationship("SearchSpace", back_populates="image_generation_configs")
search_space = relationship(
"SearchSpace", back_populates="image_generation_configs"
)
class ImageGeneration(BaseModel, TimestampMixin):
@ -973,9 +974,7 @@ class ImageGeneration(BaseModel, TimestampMixin):
String(50), nullable=True
) # "1024x1024", "1536x1024", "1024x1536", etc.
style = Column(String(50), nullable=True) # Model-specific style parameter
response_format = Column(
String(50), nullable=True
) # "url" or "b64_json"
response_format = Column(String(50), nullable=True) # "url" or "b64_json"
# Image generation config reference
# 0 = Auto mode (router), negative IDs = global configs from YAML,