Mirrors/talemate
1
0
Fork 0
mirror of https://github.com/vegu-ai/talemate.git synced 2025-12-16 19:57:47 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add methods for caching model data in Backend class: implement _get_cache_data and _apply_cache_data to enhance data sharing between backend instances.

Browse Source
This commit is contained in:
vegu-ai-tools
2025-11-29 21:24:07 +02:00
parent 647e2f277c
commit f455a11096
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/talemate/agents/visual/backends/sdnext.py
View File

@@ -88,7 +88,16 @@ class Backend(backends.Backend):
choices_changed_edit = await visual_agent.sdnext_update_model_choices("sdnext_image_edit", backend=self) choices_changed_edit = await visual_agent.sdnext_update_model_choices("sdnext_image_edit", backend=self)
if choices_changed_create or choices_changed_edit: if choices_changed_create or choices_changed_edit:
await super().on_status_change() await super().on_status_change()
def _get_cache_data(self) -> dict:
"""Return models list to cache for sharing with other backends."""
return {"models": self.models}
def _apply_cache_data(self, data: dict):
"""Apply cached models list to this backend instance."""
if "models" in data:
self.models = data["models"]
async def test_connection(self, timeout: int = 2) -> backends.BackendStatus: async def test_connection(self, timeout: int = 2) -> backends.BackendStatus:
try: try:
auth = self._get_auth() auth = self._get_auth()