DH-4761 [ai][server] Increase timeout for GET table-descriptions endp…

…oint (#188)
Dataherald · May 7, 2024 · ea1bf7a · ea1bf7a
1 parent 3755652
commit ea1bf7a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/apps/ai/server/modules/table_description/service.py b/apps/ai/server/modules/table_description/service.py
@@ -25,7 +25,7 @@ async def get_table_descriptions(
             response = await client.get(
                 settings.k2_core_url + "/table-descriptions",
                 params={"db_connection_id": db_connection_id, "table_name": table_name},
-                timeout=60,
+                timeout=settings.default_k2_core_timeout,
             )
             raise_for_status(response.status_code, response.text)
             return [TableDescriptionResponse(**td) for td in response.json()]
@@ -38,7 +38,7 @@ async def get_database_table_descriptions(self, db_connection_id: str):
             response = await client.get(
                 settings.k2_core_url + "/table-descriptions",
                 params={"db_connection_id": db_connection_id, "table_name": ""},
-                timeout=60,
+                timeout=settings.default_k2_core_timeout,
             )
             raise_for_status(response.status_code, response.text)
             db_connection = self.db_connection_service.get_db_connection(