Update role.py

由于提示词里要求llm选择的状态码是 0 到 状态数-1,所以我认为这里往states里添加动作对应状态时,也应当从0开始。
This commit is contained in:
lixiangtiandashen 2024-02-22 04:31:24 +08:00 committed by GitHub
parent 06d72698bb
commit dfa33fcdaa
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -281,7 +281,7 @@ class Role(SerializationMixin, ContextMixin, BaseModel):
i = action
self._init_action(i)
self.actions.append(i)
self.states.append(f"{len(self.actions)}. {action}")
self.states.append(f"{len(self.actions) - 1}. {action}")
def _set_react_mode(self, react_mode: str, max_react_loop: int = 1, auto_run: bool = True, use_tools: bool = False):
"""Set strategy of the Role reacting to observed Message. Variation lies in how