mirror of
https://github.com/vegu-ai/talemate.git
synced 2025-09-02 18:39:09 +00:00
add4893939
* linting * improve prompt devtools: test changes, show more information * some more polish for the new promp devtools * up default conversation gen length to 128 * openai client tweaks, talemate sets max_tokens on gpt-3.5 generations * support new openai embeddings (and default to text-embedding-3-small) * ux polish for character sheet and character state ux * actor instructions * experiment using # for context / instructions * fix bug where regenerating history would mess up time stamps * remove trailing ] * prevent client ctx from being unset * fix issue where sometimes you'd need to delete a client twice for it to disappear * upgrade dependencies * set 0.19.0 * fix performance degradation caused by circular loading animation * remove coqui studio support * fix issue when switching from unsaved creative mode to loading a scene * third party client / agent support * edit dialogue examples through character / actor editor * remove "edit dialogue" action from editor - replaced by character actor instructions * different icon for delete * prompt adjustment for acting instructions * adhoc context generation for character attributes and details * add adhoc generation for character description * contextual generation tweaks * contextual generation for dialogue examples fix some formatting issues * contextual generation for world entries * prepopulate initial recen scenarios with demo scenes add experimental holodeck scenario * scene info scene experimental * assortment of fixes for holodeck improvements * more holodeck fixes * refactor holodeck instructions * rename holodeck to simulation suite * better scene status messages * add new gpt-3.5-turbo model, better json response coercion for older models * allow exclusion of characters when persisting based on world state * better error handling of world state response * better error handling of world state response * more simulation suite fixes * progress color * world state character name mapping support * if neither quote nor asterisk is in message default to quotes * fix rerun of new paraphrase op * sim suite ping that ensure's characters are not aware of sim * fixes for better character name assessment simulation suite can now give the player character a proper name * fix bug with new status notifications * sim suite adjustments and fixes and tuning * sim suite tweaks * impl scene restore from file * prompting tweaks for reinforcement messages and acting instructions * more tweaks * dialogue prompt tweaks for rerun + rewrite * fix bug with character entry / exit with narration * linting * simsuite screenshots * screenshots
48 lines
No EOL
1.2 KiB
Python
48 lines
No EOL
1.2 KiB
Python
from talemate.agents.base import Agent, AgentAction
|
|
from talemate.agents.registry import register
|
|
from talemate.events import GameLoopEvent
|
|
import talemate.emit.async_signals
|
|
from talemate.emit import emit
|
|
|
|
@register()
|
|
class TestAgent(Agent):
|
|
|
|
agent_type = "test"
|
|
verbose_name = "Test"
|
|
|
|
def __init__(self, client):
|
|
self.client = client
|
|
self.is_enabled = True
|
|
self.actions = {
|
|
"test": AgentAction(
|
|
enabled=True,
|
|
label="Test",
|
|
description="Test",
|
|
),
|
|
}
|
|
|
|
@property
|
|
def enabled(self):
|
|
return self.is_enabled
|
|
|
|
@property
|
|
def has_toggle(self):
|
|
return True
|
|
|
|
@property
|
|
def experimental(self):
|
|
return True
|
|
|
|
def connect(self, scene):
|
|
super().connect(scene)
|
|
talemate.emit.async_signals.get("game_loop").connect(self.on_game_loop)
|
|
|
|
async def on_game_loop(self, emission: GameLoopEvent):
|
|
"""
|
|
Called on the beginning of every game loop
|
|
"""
|
|
|
|
if not self.enabled:
|
|
return
|
|
|
|
emit("status", status="info", message="Annoying you with a test message every game loop.") |