Merge pull request #57 from robwilkes/entity_map_string_fix

Fix. Issue #55. Treat entities and synonyms as strings when building entity_map
treethought · Jan 25, 2018 · 89996aa · 89996aa
2 parents edaaf1d + afc18c7
commit 89996aa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api_ai/schema_handlers.py b/api_ai/schema_handlers.py
@@ -194,9 +194,9 @@ def build_user_says(self, intent):
             mapping = {}
             for a in [a for a in annotations if a]:
                 for annotation, entity in a.items():
-                    mapping.update({annotation:entity})
+                    mapping.update({str(annotation):str(entity)})
                     for synonym in self.get_synonyms(annotation, entity):
-                        mapping.update({synonym:entity})
+                        mapping.update({str(synonym):str(entity)})
 
             for phrase in [p for p in phrases if p]:
                 if phrase != '':