diff --git a/langgraph_tutorials/customer_support/db.py b/langgraph_tutorials/customer_support/db.py index 79ad092..76e7fb8 100644 --- a/langgraph_tutorials/customer_support/db.py +++ b/langgraph_tutorials/customer_support/db.py @@ -48,7 +48,7 @@ class DatabaseManager: def _download_original(self) -> None: """Download the original database from the remote URL.""" print("Downloading original database...") - response: requests.Response = requests.get(self.db_url) + response: requests.Response = requests.get(self.db_url, timeout=10) response.raise_for_status() with open(self.original_file, "wb") as f: f.write(response.content) diff --git a/langgraph_tutorials/customer_support/policy.py b/langgraph_tutorials/customer_support/policy.py index 8639873..9440705 100644 --- a/langgraph_tutorials/customer_support/policy.py +++ b/langgraph_tutorials/customer_support/policy.py @@ -41,7 +41,7 @@ class PolicyRetriever: def initialize(self) -> None: """Download FAQ content, split into sections, embed, and store in memory.""" print(f"Fetching FAQ content from {self.FAQ_URL}...") - response = requests.get(self.FAQ_URL) + response = requests.get(self.FAQ_URL, timeout=10) response.raise_for_status() faq_text = response.text