Further bug fixes

This commit is contained in:
Ashwin Bharambe 2024-09-20 15:15:57 -07:00
parent 9e16b0948b
commit e5a7001874
3 changed files with 30 additions and 18 deletions

View file

@ -248,7 +248,7 @@ class ChatAgent(ShieldRunnerMixin):
self,
turn_id: str,
messages: List[Message],
shields: List[ShieldDefinition],
shields: List[str],
touchpoint: str,
) -> AsyncGenerator:
if len(shields) == 0:
@ -608,7 +608,6 @@ class ChatAgent(ShieldRunnerMixin):
else:
return True
print(f"{enabled_tools=}")
return AgentTool.memory.value in enabled_tools
def _memory_tool_definition(self) -> Optional[MemoryToolDefinition]: