Spaces:
Paused
Paused
File size: 1,710 Bytes
3dfb537 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 | from helpers import persist_chat, tokens
from helpers.extension import Extension
from agent import LoopData
import asyncio
class RenameChat(Extension):
async def execute(self, loop_data: LoopData = LoopData(), **kwargs):
asyncio.create_task(self.change_name())
async def change_name(self):
if not self.agent:
return
try:
# prepare history
from plugins._model_config.helpers.model_config import get_utility_model_config
util_cfg = get_utility_model_config(self.agent)
history_text = self.agent.history.output_text()
ctx_length = min(
int(util_cfg.get("ctx_length", 128000) * 0.7), 5000
)
history_text = tokens.trim_to_tokens(history_text, ctx_length, "start")
# prepare system and user prompt
system = self.agent.read_prompt("fw.rename_chat.sys.md")
current_name = self.agent.context.name
message = self.agent.read_prompt(
"fw.rename_chat.msg.md", current_name=current_name, history=history_text
)
# call utility model
new_name = await self.agent.call_utility_model(
system=system, message=message, background=True
)
# update name
if new_name:
# trim name to max length if needed
if len(new_name) > 40:
new_name = new_name[:40] + "..."
# apply to context and save
self.agent.context.name = new_name
persist_chat.save_tmp_chat(self.agent.context)
except Exception as e:
pass # non-critical
|