fix: add constructor input for lang_code

GoogleCloudPlatform · Oct 29, 2024 · 0822934 · 0822934
1 parent dfdf230
commit 0822934
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/dfcx_scrapi/core/agents.py b/src/dfcx_scrapi/core/agents.py
@@ -41,6 +41,7 @@ def __init__(
         creds=None,
         scope=False,
         agent_id: str = None,
+        language_code: str = "en"
     ):
         super().__init__(
             creds_path=creds_path,
@@ -53,6 +54,8 @@ def __init__(
             self.agent_id = agent_id
             self.client_options = self._set_region(agent_id)
 
+        self.langauge_code = language_code
+
     @scrapi_base.api_call_counter_decorator
     def _list_agents_client_request(self, location_id) -> List[
         types.agent.Agent

diff --git a/src/dfcx_scrapi/core/intents.py b/src/dfcx_scrapi/core/intents.py
@@ -44,6 +44,7 @@ def __init__(
         scope=False,
         intent_id: str = None,
         agent_id: str = None,
+        language_code: str = "en"
     ):
         super().__init__(
             creds_path=creds_path,
@@ -59,6 +60,8 @@ def __init__(
         if agent_id:
             self.agent_id = agent_id
 
+        self.language_code = language_code
+
     @staticmethod
     def concat_dict_and_df(
         intent_df: pd.DataFrame,