From 370b190c0ea8a74682de20dab6c1870ade33cc73 Mon Sep 17 00:00:00 2001 From: frdel <38891707+frdel@users.noreply.github.com> Date: Mon, 26 May 2025 18:07:20 +0200 Subject: [PATCH] extensions update --- agent.py | 3 +++ models.py | 2 +- .../{monologue_end => monologue_start}/_60_rename_chat.py | 2 +- 3 files changed, 5 insertions(+), 2 deletions(-) rename python/extensions/{monologue_end => monologue_start}/_60_rename_chat.py (98%) diff --git a/agent.py b/agent.py index 3b3e8d9f6..869bd9fec 100644 --- a/agent.py +++ b/agent.py @@ -282,6 +282,9 @@ class Agent: self.context.streaming_agent = self # mark self as current streamer self.loop_data.iteration += 1 + # call message_loop_start extensions + await self.call_extensions("message_loop_start", loop_data=self.loop_data) + try: # prepare LLM chain (model, system, history) prompt = await self.prepare_prompt(loop_data=self.loop_data) diff --git a/models.py b/models.py index 76855f31a..ec270cb9c 100644 --- a/models.py +++ b/models.py @@ -343,7 +343,7 @@ def get_openrouter_chat( base_url=base_url, stream_usage=True, model_kwargs={ - "headers": { + "extra_headers": { "HTTP-Referer": "https://agent-zero.ai", "X-Title": "Agent Zero", } diff --git a/python/extensions/monologue_end/_60_rename_chat.py b/python/extensions/monologue_start/_60_rename_chat.py similarity index 98% rename from python/extensions/monologue_end/_60_rename_chat.py rename to python/extensions/monologue_start/_60_rename_chat.py index 745b5bc0a..280d73e74 100644 --- a/python/extensions/monologue_end/_60_rename_chat.py +++ b/python/extensions/monologue_start/_60_rename_chat.py @@ -13,7 +13,7 @@ class RenameChat(Extension): try: # prepare history history_text = self.agent.history.output_text() - ctx_length = int(self.agent.config.utility_model.ctx_length * 0.7) + 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")