Feature/rate limit

.continue/rules/python_best_practices.md

@@ -4,8 +4,8 @@
 
 name: Python Best Practices
 description: A collection of best practices
-globs: ["**/*.py"]
-rules:
+
+---
 
 - Use snake_case for variable names.
 - Avoid using magic numbers, use constants instead.

.continue/rules/python_rules.md

@@ -7,4 +7,8 @@ description: A description of your rule
 
 ---
 
-- Use `uv add` to install python dependencies.
+- Always use `uv add` to install python dependencies.
+- Always use `uv run` to run python scripts or commands. Example: `uv run python src/main.py` or `uv run pytest src/tests/`.
+- Always use `uv sync` to syncronize dependencies from pyproject.toml.
+- Always use `uv remove` to remove dependencies.
+- Always use `.venv/bin/python` to run python.

commands.py

@@ -5,7 +5,9 @@
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="Crear base de datos vectorial")
     parser.add_argument("--create", action="store_true", help="Crear la base de datos vectorial")
-    parser.add_argument("--download", action="store_true", help="Descargar datos desde Google Drive")
+    parser.add_argument(
+        "--download", action="store_true", help="Descargar datos desde Google Drive"
+    )
     parser.add_argument("--fill", action="store_true", help="Rellenar la base de datos con datos")
 
     args = parser.parse_args()
@@ -20,4 +22,6 @@
         fill_database.fill_database()
         print("Base de datos rellena exitosamente.")
     else:
-        print("Por favor, usa --create para crear la base de datos vectorial, --download para descargar datos o --fill para rellenar la base de datos.")
+        print(
+            "Por favor, usa --create para crear la base de datos vectorial, --download para descargar datos o --fill para rellenar la base de datos."
+        )

notebooks/test_agent.ipynb

@@ -28,39 +28,33 @@
     "    get_topic_prompt_builder,\n",
     "    get_embedding_model,\n",
     "    get_agent,\n",
+    "    get_election_searcher,\n",
     ")\n",
     "from src.mongo import get_mongo_db\n",
     "from src import ENV\n",
-    "from src.agent.schemas import Topic\n",
+    "from src.agent.schemas import Platform, Topic\n",
     "from langchain_chroma import Chroma\n",
     "from langchain_openai import OpenAIEmbeddings, ChatOpenAI\n",
     "from langchain_core.chat_history import InMemoryChatMessageHistory\n",
     "from IPython.display import display, Markdown\n",
-    "from langchain_core.messages import AIMessage, HumanMessage\n"
+    "from langchain_core.messages import AIMessage, HumanMessage"
    ]
   },
   {
    "cell_type": "code",
    "execution_count": 3,
    "id": "fdad5438",
    "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "MongoDB connected successfully\n"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "chat_model = get_chat_model()\n",
     "emb_model = get_embedding_model()\n",
     "topic_selector = get_topic_selector()\n",
     "db = get_mongo_db()\n",
     "vector_db = get_mongo_vdb(emb_model, db)\n",
     "rag_engine = get_rag_engine(vector_db)\n",
-    "topic_prompt_builder = get_topic_prompt_builder(db)"
+    "search_election = get_election_searcher(chat_model, vector_db)\n",
+    "topic_prompt_builder = get_topic_prompt_builder(db, vector_db, search_election)"
    ]
   },
   {
@@ -92,16 +86,21 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 6,
    "id": "51be3c0b",
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/markdown": [
-       "No se encontró información específica sobre los candidatos en las elecciones en la información recuperada.  \n",
+       "La información recuperada no incluye detalles específicos sobre cómo programar en Python para consultar candidatos. Sin embargo, puedo indicarte que en las elecciones generales de Bolivia de 2025, las candidaturas son las siguientes:\n",
        "\n",
-       "Puedes encontrar más detalles en el siguiente enlace: [programas de gobierno](https://www.chequeatuvoto.chequeabolivia.bo/#parties)"
+       "- **Alianza Libertad y Democracia (LIBRE)**: Jorge \"Tuto\" Quiroga - Presidente, Juan Pablo Velasco - Vice-presidente.\n",
+       "- **Partido Demócrata Cristiano (PDC)**: Rodrigo Paz - Presidente, Edman Lara - Vice-presidente.\n",
+       "\n",
+       "Para consultar candidatos, generalmente se puede acceder a la fuente oficial o a bases de datos públicas, pero en el CONTEXTO no se proporciona un API o una fuente específica para realizar consultas programáticas. \n",
+       "\n",
+       "¿Quieres que te ayude a crear un ejemplo de cómo estructurar un script en Python para consultar datos de candidatos si tuvieras una API o un archivo de datos?"
       ],
       "text/plain": [
        "<IPython.core.display.Markdown object>"
@@ -114,27 +113,43 @@
    "source": [
     "if query and result:\n",
     "    history.extend([HumanMessage(query), AIMessage(result)])\n",
-    "query = \"y quienes son los candidatos?\"\n",
+    "query = \"entonces programa en python un script para consultar los candidatos\"\n",
     "result = \"\"\n",
     "async for token in agent.stream([*history, HumanMessage(content=query)]):\n",
     "    if token.type == \"info\" or token.type == \"error\":\n",
     "        display(Markdown(token.content), clear=True)\n",
     "        continue\n",
     "    result += token.content\n",
     "    display(Markdown(result), clear=True)\n",
-    "display(Markdown(result), clear=True)"
+    "# display(Markdown(result), clear=False)"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 7,
    "id": "6b905a9a",
    "metadata": {},
    "outputs": [],
    "source": [
     "%rm -rf chroma_db\n",
-    "%cp -r ../chroma_db .\n"
+    "%cp -r ../chroma_db ."
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "f4a8cdc7",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "75bae7f3",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {

pyproject.toml

@@ -18,6 +18,7 @@ dependencies = [
     "langchain-mongodb>=0.7.0",
     "langchain-nebius>=0.1.3",
     "langchain-openai>=0.3.28",
+    "pydantic>=2.11.7",
     "pydantic-settings>=2.10.1",
     "pymongo>=4.15.1",
     "pytest-asyncio>=1.1.0",
@@ -28,13 +29,17 @@ dependencies = [
 
 [dependency-groups]
 dev = [
+    "bs4>=0.0.2",
     "ipykernel>=6.30.1",
     "ipywidgets>=8.1.7",
     "mypy>=1.17.1",
+    "pandas>=2.3.3",
+    "pandas-stubs>=2.3.2.250926",
     "pytest>=8.4.1",
     "ruff>=0.12.5",
+    "types-pytz>=2025.2.0.20250809",
 ]
 
 [tool.ruff]
-line-length = 120
+line-length = 100
 include = ["./**/*.py", "./**/*.ipynb"]