This commit is contained in:
Eugene Yurtsev
2025-05-02 11:12:37 -04:00
parent d497907164
commit 01526ef665
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -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)
@@ -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