File size: 1,492 Bytes
301a0ac |
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 |
from python.helpers import persist_chat, tokens
from python.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):
try:
# prepare history
history_text = self.agent.history.output_text()
ctx_length = int(self.agent.config.utility_model.ctx_length * 0.3)
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
|