Update entities_checker_util.py

GoogleCloudPlatform · Oct 16, 2023 · 72ef9e7 · 72ef9e7
1 parent a5b3377
commit 72ef9e7
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/src/dfcx_scrapi/tools/entities_checker_util.py b/src/dfcx_scrapi/tools/entities_checker_util.py
@@ -46,12 +46,17 @@ def __init__(
             scope=scope,
         )
         self.agent_id = agent_id
-
         if creds_path:
             self.creds_path = creds_path
 
-        self._intents = Intents(agent_id=self.agent_id, creds_path=self.creds_path)
-        self._entity_types = EntityTypes(agent_id=self.agent_id, creds_path=self.creds_path)
+        self._intents = Intents(
+            agent_id=self.agent_id, 
+            creds_path=self.creds_path
+        )
+        self._entity_types = EntityTypes(
+            agent_id=self.agent_id, 
+            creds_path=self.creds_path
+        )
         self.intents_df = pd.DataFrame()
         self.entity_types_df = pd.DataFrame()
         self._intents_list = []
@@ -82,7 +87,9 @@ def _set_intents_df(self) -> pd.DataFrame:
                 })
 
         if not self._intents_list:
-            self._intents_list = self._intents.list_intents(agent_id=self.agent_id)
+            self._intents_list = self._intents.list_intents(
+                agent_id=self.agent_id
+            )
 
         for intent in self._intents_list:
             if 'parameters' in intent:
@@ -95,7 +102,9 @@ def _set_intents_df(self) -> pd.DataFrame:
                             text=part.text
                             params = intent.parameters
                             param_id = part.parameter_id
-                            entity_type = self._get_entity_by_param_id(params,param_id)
+                            entity_type = self._get_entity_by_param_id(
+                                params, param_id
+                            )
                             tag_texts_set = (text, param_id, entity_type)
                             tag_texts.append(tag_texts_set)
                     if tag_texts: