diff --git a/cogsol/management/commands/importagent.py b/cogsol/management/commands/importagent.py index 17c7736..ee72b9a 100644 --- a/cogsol/management/commands/importagent.py +++ b/cogsol/management/commands/importagent.py @@ -362,6 +362,9 @@ class {class_name}(BaseAgent): max_msg_length = {assistant.get("max_msg_length") or 0} max_consecutive_tool_calls = {assistant.get("max_consecutive_tool_calls") or 0} temperature = {assistant.get("temperature") or 0.0} + initial_message = {assistant.get("initial_message")!r} + forced_termination_message = {assistant.get("end_message")!r} + no_information_message = {assistant.get("not_info_message")!r} class Meta: name = {class_name!r} @@ -559,6 +562,9 @@ def _tool_name_for_id(tool_id: int) -> str | None: "max_responses": assistant.get("max_responses"), "max_msg_length": assistant.get("max_msg_length"), "max_consecutive_tool_calls": assistant.get("max_consecutive_tool_calls"), + "initial_message": assistant.get("initial_message"), + "forced_termination_message": assistant.get("end_message"), + "no_information_message": assistant.get("not_info_message"), "streaming": assistant.get("streaming_available"), "realtime": assistant.get("realtime_available"), "tools": [n for n in (_tool_name_for_id(sid) for sid in tools_ids) if n], diff --git a/cogsol/management/commands/startagent.py b/cogsol/management/commands/startagent.py index ad57b3c..9d2129a 100644 --- a/cogsol/management/commands/startagent.py +++ b/cogsol/management/commands/startagent.py @@ -20,6 +20,9 @@ class {class_name}(BaseAgent): max_msg_length = 2048 max_consecutive_tool_calls = 3 temperature = 0.3 + # initial_message = "Hi! how can I help you?" + # forced_termination_message = "We need to end this chat." + # no_information_message = "I don't have that information." class Meta: name = "{class_name}"