Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions cogsol/management/commands/importagent.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}
Expand Down Expand Up @@ -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],
Expand Down
3 changes: 3 additions & 0 deletions cogsol/management/commands/startagent.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}"
Expand Down
Loading