Spaces:
Paused
Paused
File size: 1,520 Bytes
8d1819a |
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 |
import json
from agent import LoopData
from python.helpers.extension import Extension
class InitialMessage(Extension):
async def execute(self, **kwargs):
"""
Add an initial greeting message when first user message is processed.
Called only once per session via _process_chain method.
"""
# Only add initial message for main agent (A0), not subordinate agents
if self.agent.number != 0:
return
# If the context already contains log messages, do not add another initial message
if self.agent.context.log.logs:
return
# Construct the initial message from prompt template
initial_message = self.agent.read_prompt("fw.initial_message.md")
# add initial loop data to agent (for hist_add_ai_response)
self.agent.loop_data = LoopData(user_message=None)
# Add the message to history as an AI response
self.agent.hist_add_ai_response(initial_message)
# json parse the message, get the tool_args text
initial_message_json = json.loads(initial_message)
initial_message_text = initial_message_json.get("tool_args", {}).get("text", "Hello! How can I help you?")
# Add to log (green bubble) for immediate UI display
self.agent.context.log.log(
type="response",
heading=f"{self.agent.agent_name}: Welcome",
content=initial_message_text,
finished=True,
update_progress="none",
)
|