vrr/talemate
2
0
Fork 0
mirror of https://github.com/vegu-ai/talemate.git synced 2025-09-02 02:19:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

disable revision during image prompt generation
Some checks failed
Python Tests / test (3.10) (push) Has been cancelled
Details
Python Tests / test (3.11) (push) Has been cancelled
Details
Python Tests / test (3.12) (push) Has been cancelled
Details
Python Tests / test (3.13) (push) Has been cancelled
Details

Browse source
This commit is contained in:
vegu-ai-tools 2025-05-05 00:44:08 +03:00
parent 5e9e89d452
commit f7c61aff02
2 changed files with 43 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
src/talemate/agents/editor/revision.py
View file

@ -28,8 +28,9 @@ from talemate.util import count_tokens
from talemate.prompts import Prompt
from talemate.exceptions import GenerationCancelled
import talemate.game.focal as focal
from talemate.status import LoadingStatus, set_loading
from talemate.status import LoadingStatus
from talemate.world_state.templates.content import PhraseDetection
from contextvars import ContextVar
if TYPE_CHECKING:
from talemate.tale_mate import Character, Scene
@ -51,6 +52,16 @@ detect_bad_prose_condition = AgentActionConditional(
value=True,
)
revision_disabled_context = ContextVar("revision_disabled", default=False)
class RevisionDisabled:
def __enter__(self):
self.token = revision_disabled_context.set(True)
def __exit__(self, exc_type, exc_value, traceback):
revision_disabled_context.reset(self.token)
class RevisionMixin:
"""
@ -226,6 +237,13 @@ class RevisionMixin:
if not self.revision_enabled:
return
try:
if revision_disabled_context.get():
log.debug("revision_on_generation: revision disabled through context", emission=emission)
return
except LookupError:
pass
log.info("revise generation", emission=emission)
edited = []

3
src/talemate/agents/visual/__init__.py
View file

@ -15,6 +15,7 @@ from talemate.agents.base import (
set_processing,
)
from talemate.agents.registry import register
from talemate.agents.editor.revision import RevisionDisabled
from talemate.client.base import ClientBase
from talemate.config import load_config
from talemate.emit import emit
@ -504,6 +505,7 @@ class VisualBase(Agent):
@set_processing
async def generate_environment_prompt(self, instructions: str = None):
with RevisionDisabled():
response = await Prompt.request(
"visual.generate-environment-prompt",
self.client,
@ -523,6 +525,7 @@ class VisualBase(Agent):
character = self.scene.get_character(character_name)
with RevisionDisabled():
response = await Prompt.request(
"visual.generate-character-prompt",
self.client,