From 3459cf40085ad696db6cc84b48a76ed31e7aad41 Mon Sep 17 00:00:00 2001
From: "pixeebot[bot]" <104101892+pixeebot[bot]@users.noreply.github.com>
Date: Sat, 1 Feb 2025 03:05:58 +0000
Subject: [PATCH] Add timeout to `requests` calls

---
 src/core/embeddings.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/core/embeddings.py b/src/core/embeddings.py
index adab305..70c90ee 100644
--- a/src/core/embeddings.py
+++ b/src/core/embeddings.py
@@ -57,10 +57,10 @@ async def _get_hf_embedding(self, text: str) -> Tuple[np.ndarray, float]:
         response = requests.post(
             Settings.HF_API_URL,
             headers=headers,
-            json={"inputs": text}
-        )
+            json={"inputs": text}, 
+        timeout=60)
         
         if response.status_code == 200:
             return np.array(response.json()), 0.0  # TODO: Add timing
         else:
-            raise Exception(f"HF API error: {response.text}")
\ No newline at end of file
+            raise Exception(f"HF API error: {response.text}")