Merge pull request #2 from mel-seto/handle-429-error

src/verification/wiktionary_client.py

@@ -5,7 +5,7 @@ _WIKTIONARY_CACHE = {}
 
 class WiktionaryClient:
     """Simple wrapper for Wiktionary API queries."""
-    BASE_URL = "https://zh.wiktionary.org/w/api.php"
+    BASE_URL = "https://en.wiktionary.org/w/api.php"
     HEADERS = {"User-Agent": "MyChineseIdiomApp/1.0 (email@example.com)"}
 
     def exists(self, term: str) -> bool:
@@ -19,6 +19,7 @@ class WiktionaryClient:
             response = requests.get(self.BASE_URL, params=params, headers=self.HEADERS, timeout=5)
             response.raise_for_status()
             data = response.json()
+            print(f'data: {data}')
             pages = data.get("query", {}).get("pages", {})
             exists = "-1" not in pages
         except Exception:

tests/test_retriever.py

@@ -23,6 +23,7 @@ def test_empty_query():
         assert isinstance(item, str)
         assert len(item) > 0
 
+## TODO: fix embedding logic so that this test passes
 @pytest.mark.parametrize(
     "situation, idiom",
     [
@@ -34,4 +35,5 @@ def test_definition_returns_correct_idiom(situation, idiom):
     """if input situation is the same as the embedded English definition, 
     RAG implementation should return the correct Chinese idiom"""
     top_k_idioms = retrieve_idiom(situation, top_k=3)
+    print(top_k_idioms)
     assert any(idiom in s for s in top_k_idioms)