Formatting (#2526)

This commit is contained in:
Dev Khant
2025-04-10 11:42:25 +05:30
committed by GitHub
parent 616313b8b5
commit 07462adc9a
2 changed files with 79 additions and 85 deletions

View File

@@ -165,7 +165,7 @@ class MemoryGraph:
try:
for tool_call in search_results["tool_calls"]:
if tool_call['name'] != "extract_entities":
if tool_call["name"] != "extract_entities":
continue
for item in tool_call["arguments"]["entities"]:
entity_type_map[item["entity"]] = item["entity_type"]