{ "cells": [ { "cell_type": "code", "execution_count": 1, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Requirement already satisfied: datasets in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from -r requirements.txt (line 1)) (3.2.0)\n", "Requirement already satisfied: transformers in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from -r requirements.txt (line 2)) (4.48.1)\n", "Requirement already satisfied: langchain in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from -r requirements.txt (line 3)) (0.3.15)\n", "Requirement already satisfied: langchain-community in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from -r requirements.txt (line 4)) (0.3.15)\n", "Requirement already satisfied: smolagents in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from -r requirements.txt (line 5)) (1.8.1)\n", "Requirement already satisfied: rank_bm25 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from -r requirements.txt (line 6)) (0.2.2)\n", "Requirement already satisfied: litellm in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from -r requirements.txt (line 7)) (1.59.5)\n", "Requirement already satisfied: chromadb in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from -r requirements.txt (line 8)) (0.6.3)\n", "Requirement already satisfied: reportlab in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from -r requirements.txt (line 9)) (4.2.5)\n", "Requirement already satisfied: gradio in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from -r requirements.txt (line 10)) (5.16.1)\n", "Requirement already satisfied: sentence-transformers in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from -r requirements.txt (line 11)) (3.4.0)\n", "Requirement already satisfied: filelock in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from datasets->-r requirements.txt (line 1)) (3.17.0)\n", "Requirement already satisfied: numpy>=1.17 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from datasets->-r requirements.txt (line 1)) (1.26.4)\n", "Requirement already satisfied: pyarrow>=15.0.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from datasets->-r requirements.txt (line 1)) (19.0.0)\n", "Requirement already satisfied: dill<0.3.9,>=0.3.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from datasets->-r requirements.txt (line 1)) (0.3.8)\n", "Requirement already satisfied: pandas in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from datasets->-r requirements.txt (line 1)) (2.2.3)\n", "Requirement already satisfied: requests>=2.32.2 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from datasets->-r requirements.txt (line 1)) (2.32.3)\n", "Requirement already satisfied: tqdm>=4.66.3 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from datasets->-r requirements.txt (line 1)) (4.67.1)\n", "Requirement already satisfied: xxhash in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from datasets->-r requirements.txt (line 1)) (3.5.0)\n", "Requirement already satisfied: multiprocess<0.70.17 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from datasets->-r requirements.txt (line 1)) (0.70.16)\n", "Requirement already satisfied: fsspec<=2024.9.0,>=2023.1.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from fsspec[http]<=2024.9.0,>=2023.1.0->datasets->-r requirements.txt (line 1)) (2024.9.0)\n", "Requirement already satisfied: aiohttp in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from datasets->-r requirements.txt (line 1)) (3.11.11)\n", "Requirement already satisfied: huggingface-hub>=0.23.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from datasets->-r requirements.txt (line 1)) (0.28.1)\n", "Requirement already satisfied: packaging in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from datasets->-r requirements.txt (line 1)) (24.2)\n", "Requirement already satisfied: pyyaml>=5.1 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from datasets->-r requirements.txt (line 1)) (6.0.2)\n", "Requirement already satisfied: regex!=2019.12.17 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from transformers->-r requirements.txt (line 2)) (2024.11.6)\n", "Requirement already satisfied: tokenizers<0.22,>=0.21 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from transformers->-r requirements.txt (line 2)) (0.21.0)\n", "Requirement already satisfied: safetensors>=0.4.1 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from transformers->-r requirements.txt (line 2)) (0.5.2)\n", "Requirement already satisfied: SQLAlchemy<3,>=1.4 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from langchain->-r requirements.txt (line 3)) (2.0.37)\n", "Requirement already satisfied: async-timeout<5.0.0,>=4.0.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from langchain->-r requirements.txt (line 3)) (4.0.3)\n", "Requirement already satisfied: langchain-core<0.4.0,>=0.3.31 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from langchain->-r requirements.txt (line 3)) (0.3.31)\n", "Requirement already satisfied: langchain-text-splitters<0.4.0,>=0.3.3 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from langchain->-r requirements.txt (line 3)) (0.3.5)\n", "Requirement already satisfied: langsmith<0.4,>=0.1.17 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from langchain->-r requirements.txt (line 3)) (0.3.1)\n", "Requirement already satisfied: pydantic<3.0.0,>=2.7.4 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from langchain->-r requirements.txt (line 3)) (2.10.5)\n", "Requirement already satisfied: tenacity!=8.4.0,<10,>=8.1.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from langchain->-r requirements.txt (line 3)) (9.0.0)\n", "Requirement already satisfied: dataclasses-json<0.7,>=0.5.7 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from langchain-community->-r requirements.txt (line 4)) (0.6.7)\n", "Requirement already satisfied: httpx-sse<0.5.0,>=0.4.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from langchain-community->-r requirements.txt (line 4)) (0.4.0)\n", "Requirement already satisfied: pydantic-settings<3.0.0,>=2.4.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from langchain-community->-r requirements.txt (line 4)) (2.7.1)\n", "Requirement already satisfied: rich>=13.9.4 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from smolagents->-r requirements.txt (line 5)) (13.9.4)\n", "Requirement already satisfied: jinja2>=3.1.4 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from smolagents->-r requirements.txt (line 5)) (3.1.5)\n", "Requirement already satisfied: pillow>=11.0.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from smolagents->-r requirements.txt (line 5)) (11.1.0)\n", "Requirement already satisfied: markdownify>=0.14.1 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from smolagents->-r requirements.txt (line 5)) (0.14.1)\n", "Requirement already satisfied: duckduckgo-search>=6.3.7 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from smolagents->-r requirements.txt (line 5)) (7.2.1)\n", "Requirement already satisfied: python-dotenv in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from smolagents->-r requirements.txt (line 5)) (1.0.1)\n", "Requirement already satisfied: click in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from litellm->-r requirements.txt (line 7)) (8.1.8)\n", "Requirement already satisfied: httpx<0.28.0,>=0.23.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from litellm->-r requirements.txt (line 7)) (0.27.2)\n", "Requirement already satisfied: importlib-metadata>=6.8.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from litellm->-r requirements.txt (line 7)) (8.5.0)\n", "Requirement already satisfied: jsonschema<5.0.0,>=4.22.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from litellm->-r requirements.txt (line 7)) (4.23.0)\n", "Requirement already satisfied: openai>=1.55.3 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from litellm->-r requirements.txt (line 7)) (1.60.0)\n", "Requirement already satisfied: tiktoken>=0.7.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from litellm->-r requirements.txt (line 7)) (0.8.0)\n", "Requirement already satisfied: build>=1.0.3 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (1.2.2.post1)\n", "Requirement already satisfied: chroma-hnswlib==0.7.6 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (0.7.6)\n", "Requirement already satisfied: fastapi>=0.95.2 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (0.115.6)\n", "Requirement already satisfied: uvicorn>=0.18.3 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from uvicorn[standard]>=0.18.3->chromadb->-r requirements.txt (line 8)) (0.34.0)\n", "Requirement already satisfied: posthog>=2.4.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (3.9.2)\n", "Requirement already satisfied: typing_extensions>=4.5.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (4.12.2)\n", "Requirement already satisfied: onnxruntime>=1.14.1 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (1.20.1)\n", "Requirement already satisfied: opentelemetry-api>=1.2.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (1.29.0)\n", "Requirement already satisfied: opentelemetry-exporter-otlp-proto-grpc>=1.2.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (1.29.0)\n", "Requirement already satisfied: opentelemetry-instrumentation-fastapi>=0.41b0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (0.50b0)\n", "Requirement already satisfied: opentelemetry-sdk>=1.2.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (1.29.0)\n", "Requirement already satisfied: pypika>=0.48.9 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (0.48.9)\n", "Requirement already satisfied: overrides>=7.3.1 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (7.7.0)\n", "Requirement already satisfied: importlib-resources in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (6.5.2)\n", "Requirement already satisfied: grpcio>=1.58.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (1.69.0)\n", "Requirement already satisfied: bcrypt>=4.0.1 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (4.2.1)\n", "Requirement already satisfied: typer>=0.9.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (0.12.5)\n", "Requirement already satisfied: kubernetes>=28.1.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (31.0.0)\n", "Requirement already satisfied: mmh3>=4.0.1 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (5.0.1)\n", "Requirement already satisfied: orjson>=3.9.12 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from chromadb->-r requirements.txt (line 8)) (3.10.15)\n", "Requirement already satisfied: chardet in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from reportlab->-r requirements.txt (line 9)) (5.2.0)\n", "Requirement already satisfied: aiofiles<24.0,>=22.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from gradio->-r requirements.txt (line 10)) (23.2.1)\n", "Requirement already satisfied: anyio<5.0,>=3.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from gradio->-r requirements.txt (line 10)) (4.8.0)\n", "Requirement already satisfied: ffmpy in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from gradio->-r requirements.txt (line 10)) (0.5.0)\n", "Requirement already satisfied: gradio-client==1.7.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from gradio->-r requirements.txt (line 10)) (1.7.0)\n", "Requirement already satisfied: markupsafe~=2.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from gradio->-r requirements.txt (line 10)) (2.1.5)\n", "Requirement already satisfied: pydub in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from gradio->-r requirements.txt (line 10)) (0.25.1)\n", "Requirement already satisfied: python-multipart>=0.0.18 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from gradio->-r requirements.txt (line 10)) (0.0.18)\n", "Requirement already satisfied: ruff>=0.9.3 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from gradio->-r requirements.txt (line 10)) (0.9.5)\n", "Requirement already satisfied: safehttpx<0.2.0,>=0.1.6 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from gradio->-r requirements.txt (line 10)) (0.1.6)\n", "Requirement already satisfied: semantic-version~=2.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from gradio->-r requirements.txt (line 10)) (2.10.0)\n", "Requirement already satisfied: starlette<1.0,>=0.40.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from gradio->-r requirements.txt (line 10)) (0.41.3)\n", "Requirement already satisfied: tomlkit<0.14.0,>=0.12.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from gradio->-r requirements.txt (line 10)) (0.13.2)\n", "Requirement already satisfied: websockets<15.0,>=10.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from gradio-client==1.7.0->gradio->-r requirements.txt (line 10)) (14.2)\n", "Requirement already satisfied: torch>=1.11.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from sentence-transformers->-r requirements.txt (line 11)) (2.6.0)\n", "Requirement already satisfied: scikit-learn in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from sentence-transformers->-r requirements.txt (line 11)) (1.6.1)\n", "Requirement already satisfied: scipy in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from sentence-transformers->-r requirements.txt (line 11)) (1.11.4)\n", "Requirement already satisfied: aiohappyeyeballs>=2.3.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from aiohttp->datasets->-r requirements.txt (line 1)) (2.4.4)\n", "Requirement already satisfied: aiosignal>=1.1.2 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from aiohttp->datasets->-r requirements.txt (line 1)) (1.3.2)\n", "Requirement already satisfied: attrs>=17.3.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from aiohttp->datasets->-r requirements.txt (line 1)) (24.3.0)\n", "Requirement already satisfied: frozenlist>=1.1.1 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from aiohttp->datasets->-r requirements.txt (line 1)) (1.5.0)\n", "Requirement already satisfied: multidict<7.0,>=4.5 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from aiohttp->datasets->-r requirements.txt (line 1)) (6.1.0)\n", "Requirement already satisfied: propcache>=0.2.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from aiohttp->datasets->-r requirements.txt (line 1)) (0.2.1)\n", "Requirement already satisfied: yarl<2.0,>=1.17.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from aiohttp->datasets->-r requirements.txt (line 1)) (1.18.3)\n", "Requirement already satisfied: exceptiongroup>=1.0.2 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from anyio<5.0,>=3.0->gradio->-r requirements.txt (line 10)) (1.2.2)\n", "Requirement already satisfied: idna>=2.8 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from anyio<5.0,>=3.0->gradio->-r requirements.txt (line 10)) (3.10)\n", "Requirement already satisfied: sniffio>=1.1 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from anyio<5.0,>=3.0->gradio->-r requirements.txt (line 10)) (1.3.1)\n", "Requirement already satisfied: pyproject_hooks in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from build>=1.0.3->chromadb->-r requirements.txt (line 8)) (1.2.0)\n", "Requirement already satisfied: tomli>=1.1.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from build>=1.0.3->chromadb->-r requirements.txt (line 8)) (2.2.1)\n", "Requirement already satisfied: marshmallow<4.0.0,>=3.18.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from dataclasses-json<0.7,>=0.5.7->langchain-community->-r requirements.txt (line 4)) (3.26.0)\n", "Requirement already satisfied: typing-inspect<1,>=0.4.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from dataclasses-json<0.7,>=0.5.7->langchain-community->-r requirements.txt (line 4)) (0.9.0)\n", "Requirement already satisfied: primp>=0.10.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from duckduckgo-search>=6.3.7->smolagents->-r requirements.txt (line 5)) (0.10.1)\n", "Requirement already satisfied: lxml>=5.3.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from duckduckgo-search>=6.3.7->smolagents->-r requirements.txt (line 5)) (5.3.0)\n", "Requirement already satisfied: certifi in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from httpx<0.28.0,>=0.23.0->litellm->-r requirements.txt (line 7)) (2024.12.14)\n", "Requirement already satisfied: httpcore==1.* in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from httpx<0.28.0,>=0.23.0->litellm->-r requirements.txt (line 7)) (1.0.7)\n", "Requirement already satisfied: h11<0.15,>=0.13 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from httpcore==1.*->httpx<0.28.0,>=0.23.0->litellm->-r requirements.txt (line 7)) (0.14.0)\n", "Requirement already satisfied: zipp>=3.20 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from importlib-metadata>=6.8.0->litellm->-r requirements.txt (line 7)) (3.21.0)\n", "Requirement already satisfied: jsonschema-specifications>=2023.03.6 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from jsonschema<5.0.0,>=4.22.0->litellm->-r requirements.txt (line 7)) (2024.10.1)\n", "Requirement already satisfied: referencing>=0.28.4 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from jsonschema<5.0.0,>=4.22.0->litellm->-r requirements.txt (line 7)) (0.36.1)\n", "Requirement already satisfied: rpds-py>=0.7.1 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from jsonschema<5.0.0,>=4.22.0->litellm->-r requirements.txt (line 7)) (0.22.3)\n", "Requirement already satisfied: six>=1.9.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from kubernetes>=28.1.0->chromadb->-r requirements.txt (line 8)) (1.17.0)\n", "Requirement already satisfied: python-dateutil>=2.5.3 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from kubernetes>=28.1.0->chromadb->-r requirements.txt (line 8)) (2.9.0.post0)\n", "Requirement already satisfied: google-auth>=1.0.1 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from kubernetes>=28.1.0->chromadb->-r requirements.txt (line 8)) (2.37.0)\n", "Requirement already satisfied: websocket-client!=0.40.0,!=0.41.*,!=0.42.*,>=0.32.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from kubernetes>=28.1.0->chromadb->-r requirements.txt (line 8)) (1.8.0)\n", "Requirement already satisfied: requests-oauthlib in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from kubernetes>=28.1.0->chromadb->-r requirements.txt (line 8)) (2.0.0)\n", "Requirement already satisfied: oauthlib>=3.2.2 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from kubernetes>=28.1.0->chromadb->-r requirements.txt (line 8)) (3.2.2)\n", "Requirement already satisfied: urllib3>=1.24.2 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from kubernetes>=28.1.0->chromadb->-r requirements.txt (line 8)) (2.3.0)\n", "Requirement already satisfied: durationpy>=0.7 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from kubernetes>=28.1.0->chromadb->-r requirements.txt (line 8)) (0.9)\n", "Requirement already satisfied: jsonpatch<2.0,>=1.33 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from langchain-core<0.4.0,>=0.3.31->langchain->-r requirements.txt (line 3)) (1.33)\n", "Requirement already satisfied: requests-toolbelt<2.0.0,>=1.0.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from langsmith<0.4,>=0.1.17->langchain->-r requirements.txt (line 3)) (1.0.0)\n", "Requirement already satisfied: zstandard<0.24.0,>=0.23.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from langsmith<0.4,>=0.1.17->langchain->-r requirements.txt (line 3)) (0.23.0)\n", "Requirement already satisfied: beautifulsoup4<5,>=4.9 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from markdownify>=0.14.1->smolagents->-r requirements.txt (line 5)) (4.12.3)\n", "Requirement already satisfied: coloredlogs in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from onnxruntime>=1.14.1->chromadb->-r requirements.txt (line 8)) (15.0.1)\n", "Requirement already satisfied: flatbuffers in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from onnxruntime>=1.14.1->chromadb->-r requirements.txt (line 8)) (25.1.21)\n", "Requirement already satisfied: protobuf in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from onnxruntime>=1.14.1->chromadb->-r requirements.txt (line 8)) (5.29.3)\n", "Requirement already satisfied: sympy in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from onnxruntime>=1.14.1->chromadb->-r requirements.txt (line 8)) (1.13.1)\n", "Requirement already satisfied: distro<2,>=1.7.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from openai>=1.55.3->litellm->-r requirements.txt (line 7)) (1.9.0)\n", "Requirement already satisfied: jiter<1,>=0.4.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from openai>=1.55.3->litellm->-r requirements.txt (line 7)) (0.8.2)\n", "Requirement already satisfied: deprecated>=1.2.6 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from opentelemetry-api>=1.2.0->chromadb->-r requirements.txt (line 8)) (1.2.15)\n", "Requirement already satisfied: googleapis-common-protos~=1.52 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from opentelemetry-exporter-otlp-proto-grpc>=1.2.0->chromadb->-r requirements.txt (line 8)) (1.66.0)\n", "Requirement already satisfied: opentelemetry-exporter-otlp-proto-common==1.29.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from opentelemetry-exporter-otlp-proto-grpc>=1.2.0->chromadb->-r requirements.txt (line 8)) (1.29.0)\n", "Requirement already satisfied: opentelemetry-proto==1.29.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from opentelemetry-exporter-otlp-proto-grpc>=1.2.0->chromadb->-r requirements.txt (line 8)) (1.29.0)\n", "Requirement already satisfied: opentelemetry-instrumentation-asgi==0.50b0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from opentelemetry-instrumentation-fastapi>=0.41b0->chromadb->-r requirements.txt (line 8)) (0.50b0)\n", "Requirement already satisfied: opentelemetry-instrumentation==0.50b0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from opentelemetry-instrumentation-fastapi>=0.41b0->chromadb->-r requirements.txt (line 8)) (0.50b0)\n", "Requirement already satisfied: opentelemetry-semantic-conventions==0.50b0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from opentelemetry-instrumentation-fastapi>=0.41b0->chromadb->-r requirements.txt (line 8)) (0.50b0)\n", "Requirement already satisfied: opentelemetry-util-http==0.50b0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from opentelemetry-instrumentation-fastapi>=0.41b0->chromadb->-r requirements.txt (line 8)) (0.50b0)\n", "Requirement already satisfied: wrapt<2.0.0,>=1.0.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from opentelemetry-instrumentation==0.50b0->opentelemetry-instrumentation-fastapi>=0.41b0->chromadb->-r requirements.txt (line 8)) (1.17.2)\n", "Requirement already satisfied: asgiref~=3.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from opentelemetry-instrumentation-asgi==0.50b0->opentelemetry-instrumentation-fastapi>=0.41b0->chromadb->-r requirements.txt (line 8)) (3.8.1)\n", "Requirement already satisfied: pytz>=2020.1 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from pandas->datasets->-r requirements.txt (line 1)) (2024.2)\n", "Requirement already satisfied: tzdata>=2022.7 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from pandas->datasets->-r requirements.txt (line 1)) (2025.1)\n", "Requirement already satisfied: monotonic>=1.5 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from posthog>=2.4.0->chromadb->-r requirements.txt (line 8)) (1.6)\n", "Requirement already satisfied: backoff>=1.10.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from posthog>=2.4.0->chromadb->-r requirements.txt (line 8)) (2.2.1)\n", "Requirement already satisfied: annotated-types>=0.6.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from pydantic<3.0.0,>=2.7.4->langchain->-r requirements.txt (line 3)) (0.7.0)\n", "Requirement already satisfied: pydantic-core==2.27.2 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from pydantic<3.0.0,>=2.7.4->langchain->-r requirements.txt (line 3)) (2.27.2)\n", "Requirement already satisfied: charset-normalizer<4,>=2 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from requests>=2.32.2->datasets->-r requirements.txt (line 1)) (3.4.1)\n", "Requirement already satisfied: markdown-it-py>=2.2.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from rich>=13.9.4->smolagents->-r requirements.txt (line 5)) (3.0.0)\n", "Requirement already satisfied: pygments<3.0.0,>=2.13.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from rich>=13.9.4->smolagents->-r requirements.txt (line 5)) (2.19.1)\n", "Requirement already satisfied: greenlet!=0.4.17 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from SQLAlchemy<3,>=1.4->langchain->-r requirements.txt (line 3)) (3.1.1)\n", "Requirement already satisfied: networkx in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from torch>=1.11.0->sentence-transformers->-r requirements.txt (line 11)) (3.4.2)\n", "Requirement already satisfied: nvidia-cuda-nvrtc-cu12==12.4.127 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from torch>=1.11.0->sentence-transformers->-r requirements.txt (line 11)) (12.4.127)\n", "Requirement already satisfied: nvidia-cuda-runtime-cu12==12.4.127 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from torch>=1.11.0->sentence-transformers->-r requirements.txt (line 11)) (12.4.127)\n", "Requirement already satisfied: nvidia-cuda-cupti-cu12==12.4.127 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from torch>=1.11.0->sentence-transformers->-r requirements.txt (line 11)) (12.4.127)\n", "Requirement already satisfied: nvidia-cudnn-cu12==9.1.0.70 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from torch>=1.11.0->sentence-transformers->-r requirements.txt (line 11)) (9.1.0.70)\n", "Requirement already satisfied: nvidia-cublas-cu12==12.4.5.8 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from torch>=1.11.0->sentence-transformers->-r requirements.txt (line 11)) (12.4.5.8)\n", "Requirement already satisfied: nvidia-cufft-cu12==11.2.1.3 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from torch>=1.11.0->sentence-transformers->-r requirements.txt (line 11)) (11.2.1.3)\n", "Requirement already satisfied: nvidia-curand-cu12==10.3.5.147 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from torch>=1.11.0->sentence-transformers->-r requirements.txt (line 11)) (10.3.5.147)\n", "Requirement already satisfied: nvidia-cusolver-cu12==11.6.1.9 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from torch>=1.11.0->sentence-transformers->-r requirements.txt (line 11)) (11.6.1.9)\n", "Requirement already satisfied: nvidia-cusparse-cu12==12.3.1.170 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from torch>=1.11.0->sentence-transformers->-r requirements.txt (line 11)) (12.3.1.170)\n", "Requirement already satisfied: nvidia-cusparselt-cu12==0.6.2 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from torch>=1.11.0->sentence-transformers->-r requirements.txt (line 11)) (0.6.2)\n", "Requirement already satisfied: nvidia-nccl-cu12==2.21.5 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from torch>=1.11.0->sentence-transformers->-r requirements.txt (line 11)) (2.21.5)\n", "Requirement already satisfied: nvidia-nvtx-cu12==12.4.127 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from torch>=1.11.0->sentence-transformers->-r requirements.txt (line 11)) (12.4.127)\n", "Requirement already satisfied: nvidia-nvjitlink-cu12==12.4.127 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from torch>=1.11.0->sentence-transformers->-r requirements.txt (line 11)) (12.4.127)\n", "Requirement already satisfied: triton==3.2.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from torch>=1.11.0->sentence-transformers->-r requirements.txt (line 11)) (3.2.0)\n", "Requirement already satisfied: mpmath<1.4,>=1.1.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from sympy->onnxruntime>=1.14.1->chromadb->-r requirements.txt (line 8)) (1.3.0)\n", "Requirement already satisfied: shellingham>=1.3.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from typer>=0.9.0->chromadb->-r requirements.txt (line 8)) (1.5.4)\n", "Requirement already satisfied: httptools>=0.6.3 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from uvicorn[standard]>=0.18.3->chromadb->-r requirements.txt (line 8)) (0.6.4)\n", "Requirement already satisfied: uvloop!=0.15.0,!=0.15.1,>=0.14.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from uvicorn[standard]>=0.18.3->chromadb->-r requirements.txt (line 8)) (0.21.0)\n", "Requirement already satisfied: watchfiles>=0.13 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from uvicorn[standard]>=0.18.3->chromadb->-r requirements.txt (line 8)) (0.20.0)\n", "Requirement already satisfied: joblib>=1.2.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from scikit-learn->sentence-transformers->-r requirements.txt (line 11)) (1.4.2)\n", "Requirement already satisfied: threadpoolctl>=3.1.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from scikit-learn->sentence-transformers->-r requirements.txt (line 11)) (3.5.0)\n", "Requirement already satisfied: soupsieve>1.2 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from beautifulsoup4<5,>=4.9->markdownify>=0.14.1->smolagents->-r requirements.txt (line 5)) (2.6)\n", "Requirement already satisfied: cachetools<6.0,>=2.0.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from google-auth>=1.0.1->kubernetes>=28.1.0->chromadb->-r requirements.txt (line 8)) (5.5.1)\n", "Requirement already satisfied: pyasn1-modules>=0.2.1 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from google-auth>=1.0.1->kubernetes>=28.1.0->chromadb->-r requirements.txt (line 8)) (0.4.1)\n", "Requirement already satisfied: rsa<5,>=3.1.4 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from google-auth>=1.0.1->kubernetes>=28.1.0->chromadb->-r requirements.txt (line 8)) (4.9)\n", "Requirement already satisfied: jsonpointer>=1.9 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from jsonpatch<2.0,>=1.33->langchain-core<0.4.0,>=0.3.31->langchain->-r requirements.txt (line 3)) (3.0.0)\n", "Requirement already satisfied: mdurl~=0.1 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from markdown-it-py>=2.2.0->rich>=13.9.4->smolagents->-r requirements.txt (line 5)) (0.1.2)\n", "Requirement already satisfied: mypy-extensions>=0.3.0 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from typing-inspect<1,>=0.4.0->dataclasses-json<0.7,>=0.5.7->langchain-community->-r requirements.txt (line 4)) (1.0.0)\n", "Requirement already satisfied: humanfriendly>=9.1 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from coloredlogs->onnxruntime>=1.14.1->chromadb->-r requirements.txt (line 8)) (10.0)\n", "Requirement already satisfied: pyasn1<0.7.0,>=0.4.6 in /system/conda/miniconda3/envs/cloudspace/lib/python3.10/site-packages (from pyasn1-modules>=0.2.1->google-auth>=1.0.1->kubernetes>=28.1.0->chromadb->-r requirements.txt (line 8)) (0.6.1)\n", "Note: you may need to restart the kernel to use updated packages.\n" ] } ], "source": [ "%pip install -r requirements.txt" ] }, { "cell_type": "code", "execution_count": 18, "metadata": {}, "outputs": [], "source": [ "import os\n", "from dotenv import load_dotenv\n", "\n", "_ = load_dotenv()\n", "\n", "LLM_BASE = os.getenv(\"AZURE_OPENAI_BASE\")\n", "LLM_VERSION = os.getenv(\"AZURE_OPENAI_VERSION\")\n", "LLMI_API_KEY = os.getenv(\"AZURE_OPENAI_API_KEY\")\n", "LLM_NAME = os.getenv(\"AZURE_OPENAI_MODEL\")\n", "\n", "EMBEDDING_BASE = os.getenv(\"AZURE_OPENAI_EMBEDDING_BASE\")\n", "EMBEDDING_VERSION = os.getenv(\"AZURE_OPENAI_EMBEDDING_VERSION\")\n", "EMBEDDING_API_KEY = os.getenv(\"AZURE_OPENAI_EMBEDDING_API_KEY\")\n", "EMBEDDING_NAME = os.getenv(\"AZURE_OPENAI_EMBEDDING_MODEL\")\n", "\n", "# Path to the dataset.\n", "DATA_PATH = os.path.join(\"papers\")\n", "# Path to save Chroma database\n", "CHROMA_PATH = \"./chroma_db\"\n" ] }, { "cell_type": "code", "execution_count": 19, "metadata": {}, "outputs": [], "source": [ "from uuid import uuid4\n", "from langchain.text_splitter import RecursiveCharacterTextSplitter, MarkdownHeaderTextSplitter\n", "from langchain.document_loaders.pdf import PyPDFDirectoryLoader" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "### Read documents and preprocess" ] }, { "cell_type": "code", "execution_count": 20, "metadata": {}, "outputs": [], "source": [ "# Initialize text splitter with specified parameters\n", "# text_splitter = RecursiveCharacterTextSplitter(\n", "# chunk_size=1000, # Size of each chunk in characters\n", "# chunk_overlap=100, # Overlap between consecutive chunks\n", "# length_function=len, # Function to compute the length of the text\n", "# add_start_index=True, # Flag to add start index to each chunk\n", "# )\n", "\n", "# Load and split PDF documents\n", "# document_loader = PyPDFDirectoryLoader(DATA_PATH)\n", "# documents = document_loader.load()\n", "# chunks = text_splitter.split_documents(documents)\n", "# print(f\"Total chunks extracted from data folder: {len(chunks)}\")\n" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "### Create ChromaDB using embeddings." ] }, { "cell_type": "code", "execution_count": 21, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "[array([-0.02441968, -0.00315892, -0.00488673, ..., 0.006077 ,\n", " -0.01348737, -0.00549059], dtype=float32)]\n" ] } ], "source": [ "import chromadb.utils.embedding_functions as embedding_functions\n", "\n", "openai_embedding = embedding_functions.OpenAIEmbeddingFunction(\n", " api_key=EMBEDDING_API_KEY,\n", " api_base=EMBEDDING_BASE,\n", " api_type=\"azure\",\n", " api_version=EMBEDDING_VERSION,\n", " model_name=EMBEDDING_NAME.split(\"/\")[-1],\n", ")\n", "print(openai_embedding([\"Hola\"]))" ] }, { "cell_type": "code", "execution_count": 27, "metadata": {}, "outputs": [ { "data": { "text/plain": [ "['LLM_PAPERS', 'RENTA_2023', 'RENTA_2023_2']" ] }, "execution_count": 27, "metadata": {}, "output_type": "execute_result" } ], "source": [ "import chromadb\n", "chroma_client = chromadb.PersistentClient(path=CHROMA_PATH)\n", "# chroma_client.delete_collection('RENTA_2023')\n", "chroma_client.list_collections()" ] }, { "cell_type": "code", "execution_count": 28, "metadata": {}, "outputs": [], "source": [ "import re\n", "import random\n", "\n", "def extract_page_number(text):\n", " \"\"\"\n", " Extrae el número de página del texto si tiene el formato:\n", " \n", " \"\"\"\n", " patron = r\"\"\n", " match = re.search(patron, text)\n", " if match:\n", " return str(match.group(1))\n", " return None\n", "\n", "# Función para procesar un archivo Markdown usando MarkdownHeaderTextSplitter\n", "def process_markdown_with_header_splitter(file_path):\n", " \"\"\"\n", " Procesa un archivo Markdown usando MarkdownHeaderTextSplitter.\n", " \"\"\"\n", " # Leer el archivo Markdown\n", " with open(file_path, \"r\", encoding=\"utf-8\") as file:\n", " markdown_text = file.read()\n", "\n", " # Definir los niveles de encabezado a extraer como metadatos\n", " headers_to_split_on = [\n", " (\"#\", \"Header 1\"),\n", " (\"##\", \"Header 2\"),\n", " (\"###\", \"Header 3\"),\n", " (\"####\", \"Header 4\"),\n", " (\"#####\", \"Header 5\"),\n", " (\"######\", \"Header 6\")\n", " ]\n", "\n", " # Inicializar el splitter\n", " markdown_splitter = MarkdownHeaderTextSplitter(headers_to_split_on=headers_to_split_on)\n", "\n", " # Dividir el contenido en chunks\n", " chunks = markdown_splitter.split_text(markdown_text)\n", "\n", " chunk_size = 1000\n", " chunk_overlap = 200\n", " text_splitter = RecursiveCharacterTextSplitter(\n", " chunk_size=chunk_size, chunk_overlap=chunk_overlap\n", " )\n", "\n", " chunks = text_splitter.split_documents(chunks)\n", "\n", " # Procesar los chunks para añadir números de página y otros ajustes\n", " processed_chunks = []\n", " previous_page_number = None\n", " for chunk in chunks:\n", " # Extraer número de página\n", " page_number = extract_page_number(chunk.page_content)\n", " if page_number is None:\n", " page_number = previous_page_number\n", " else:\n", " previous_page_number = page_number\n", "\n", " # Añadir chunk procesado\n", " processed_chunks.append({\n", " \"content\": chunk.page_content,\n", " \"metadata\": {**chunk.metadata, \"page\": page_number} if chunk.metadata else {\"page\": page_number}\n", " })\n", "\n", " return processed_chunks\n", "\n", "# Función para probar una muestra aleatoria de chunks\n", "def test_random_chunks(chunks, num_samples=5):\n", " \"\"\"\n", " Imprime una muestra aleatoria de chunks para verificar el parseo.\n", " \"\"\"\n", " if not chunks:\n", " print(\"No hay chunks disponibles para probar.\")\n", " return\n", "\n", " print(f\"Mostrando {num_samples} chunks aleatorios para prueba:\")\n", " sampled_chunks = random.sample(chunks, min(num_samples, len(chunks)))\n", " for i, chunk in enumerate(sampled_chunks, start=1):\n", " print(f\"\\n--- Chunk {i} ---\")\n", " print(\"Contenido:\")\n", " print(chunk[\"content\"])\n", " print(\"Metadatos:\")\n", " print(chunk[\"metadata\"])\n", " print(\"-\" * 40)\n", "\n", "# Función para añadir chunks a ChromaDB\n", "def add_chunks_to_chroma(chunks):\n", " \"\"\"\n", " Añade los chunks y sus embeddings a la colección de ChromaDB.\n", " \"\"\"\n", " chroma_client = chromadb.PersistentClient(path=CHROMA_PATH)\n", " collection = chroma_client.get_or_create_collection(\n", " name=\"RENTA_2023_2\",\n", " embedding_function=openai_embedding,\n", " )\n", "\n", " print(f\"Chunks actuales en la colección: {collection.count()}\")\n", "\n", " if collection.count() == 0:\n", " # Generar datos en un solo bucle para mejorar la eficiencia\n", " data = [\n", " (\n", " str(uuid4()), # ID único\n", " openai_embedding([chunk[\"content\"]])[0], # Embedding\n", " chunk[\"metadata\"], # Metadatos\n", " chunk[\"content\"] # Contenido del documento\n", " )\n", " for chunk in chunks\n", " ]\n", "\n", " # Desempaquetar los datos generados\n", " docs_ids, docs_embeddings, docs_metadata, docs = zip(*data)\n", "\n", " # Crear lotes para la inserción\n", " batches = create_batches(\n", " api=chroma_client,\n", " ids=list(docs_ids),\n", " documents=list(docs),\n", " embeddings=list(docs_embeddings),\n", " metadatas=list(docs_metadata)\n", " )\n", " # Insertar los datos en Chroma DB\n", " for batch in batches:\n", " collection.add(\n", " documents=batch[3],\n", " ids=batch[0],\n", " metadatas=batch[2],\n", " embeddings=batch[1],\n", " )\n", " print(f\"Guardados {len(batch[3])}/{collection.count()} chunks en {CHROMA_PATH}.\")" ] }, { "cell_type": "code", "execution_count": 30, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "Total chunks extraídos: 1478\n", "Mostrando 5 chunks aleatorios para prueba:\n", "\n", "--- Chunk 1 ---\n", "Contenido:\n", "turísticos, que puedan llevar a cabo las entidades estatales, autonómicas o locales con\n", "competencias en materia de cultura, turismo y economía, así como por las Film Commissions\n", "o Film Offices que hayan intervenido en la realización del rodaje o producción. \n", "Atención: los requisitos establecidos en las letras b') y c') no son exigibles en el caso de\n", "producciones extranjeras de largometrajes cinematográficos y obras audiovisuales en\n", "las que el contrato por el que se encargó la ejecución de la producción se hubiese\n", "firmado con anterioridad al 11 de julio de 2021, por aplicación de lo dispuesto en la\n", "disposición transitoria cuadragésima segunda de la LIS. \n", "Límites \n", "Esta deducción queda excluida del límite conjunto (25/50 por 100) previsto el artículo\n", "39.1 LIS para las deducciones para incentivar la realización de determinadas actividades y,\n", "por tanto, esta deducción por gastos de ejecución de una producción extranjera no se\n", "computa para el cálculo de dicho límite. \n", "Finalmente, téngase en cuenta que, en caso de insuficiencia de cuota en la aplicación de\n", "esta deducción del artículo 36.2 de la LIS, la posibilidad de poder solicitar su abono a la\n", "Administración que se concede en el artículo 39.3 de la LIS, no es aplicable al IRPF, de\n", "acuerdo con lo dispuesto en el artículo 68.2 de la Ley del IRPF. \n", "\n", " \n", "\n", "Metadatos:\n", "{'Header 1': 'Capítulo 16. Deducciones generales de la cuota en el ejercicio 2023', 'Header 2': 'Deducciones por incentivos y estímulos a la inversión empresarial en actividades económicas en estimación directa', 'Header 3': 'Régimen general y regímenes especiales de deducciones por incentivos y estímulos a la inversión empresarial de la Ley del Impuesto sobre Sociedades', 'Header 4': '2. Régimen general de deducciones', 'Header 5': 'Artículo 36 LIS \"Deducción por inversiones en producciones cinematográficas, series audiovisuales y espectáculos en vivo de artes escénicas y musicales\"', 'Header 6': 'B. Deducción por producciones cinematográficas extranjeras en España (art. 36.2 LIS)', 'page': '1220'}\n", "----------------------------------------\n", "\n", "--- Chunk 2 ---\n", "Contenido:\n", "A. Delimitación positiva \n", "Normativa: Art. 21 Ley IRPF \n", "Tienen la consideración fiscal de rendimientos del capital mobiliario todas las utilidades\n", "o contraprestaciones, cualquiera que sea su denominación o naturaleza, dinerarias o en\n", "especie, que provengan, directa o indirectamente, del capital mobiliario y, en general, de\n", "bienes o derechos no clasificados como inmobiliarios, de los que sea titular el contribuyente y\n", "no se hallen afectos a actividades económicas realizadas por él mismo. \n", "Los rendimientos correspondientes a los elementos patrimoniales, bienes o derechos, que se\n", "hallen afectos de manera exclusiva a actividades económicas realizadas por el contribuyente\n", "se comprenderán entre los procedentes de las indicadas actividades. \n", "Importante: en ningún caso tienen la consideración de elementos patrimoniales afectos\n", "a actividades económicas, los activos representativos de la participación en fondos\n", "propios de una entidad y de la cesión de capitales a terceros [Art. 29.1 c) Ley IRPF]. \n", "B. Delimitación negativa \n", "\n", " \n", " \n", "No se consideran rendimientos del capital mobiliario, entre otros: \n", "· Los derivados de la entrega de acciones liberadas. \n", "Normativa: Arts. 25.1 b) y 37.1 a) y b) Ley IRPF \n", "El tratamiento fiscal de la recepción de acciones liberadas tanto en el caso de valores\n", "admitidos a negociación como de valores no admitidos a negociación se comenta en el\n", "Capítulo 11. \n", "Atención: desde el 1 de enero de 2017, el importe obtenido por la transmisión tanto\n", "de derechos de suscripción procedentes de valores admitidos a negociación como\n", "de derechos de suscripción procedentes de valores no admitidos a negociación en\n", "un mercado organizado tiene la consideración de ganancia patrimonial sujeta a\n", "retención. Véase el Capítulo 11. \n", "· Los dividendos y participaciones en beneficios distribuidos por sociedades que\n", "procedan de períodos impositivos durante los cuales dichas sociedades se hallasen\n", "en régimen de transparencia fiscal. \n", "Normativa: Art. 91.9 y disposición transitoria décima Ley IRPF; disposición\n", "transitoria cuarta Reglamento IRPF \n", "· La contraprestación obtenida por el aplazamiento o el fraccionamiento del precio de\n", "las operaciones realizadas en el desarrollo de una actividad económica habitual del\n", "contribuyente. \n", "Normativa: Art. 25.5 Ley IRPF \n", "· Los derivados de las transmisiones lucrativas, por causa de muerte del contribuyente,\n", "de los activos representativos de la captación y utilización de capitales ajenos. Tampoco\n", "se computará el rendimiento del capital mobiliario negativo derivado de la transmisión\n", "lucrativa de activos representativos de la captación y utilización de capitales ajenos\n", "por actos \"inter vivos\". \n", "Normativa: Art. 25.6 Ley IRPF \n", "· Los dividendos y participaciones en beneficios a que se refiere el artículo 25.1 a) y b)\n", "Ley IRPF que procedan de beneficios obtenidos en períodos impositivos durante los\n", "cuales la entidad que los distribuye hubiera tributado en el régimen de las\n", "sociedades patrimoniales \n", "Normativa: Disposición transitoria decima Ley IRPF \n", "· La distribución de los beneficios a que se refiere el artículo 25.1 a) y b) Ley IRPF\n", "obtenidos durante los períodos impositivos en los que hubiera sido de aplicación el\n", "régimen de atribución de rentas por sociedades civiles que hubieran llevado contabilidad\n", "ajustada al código de comercio en los ejercicios 2014 y 2015 y que pasaron a tener la \n", "\n", " \n", " \n", "consideración de contribuyentes del Impuesto sobre Sociedades a partir de 1 de enero de\n", "2016, no se integrarán en la base imponible del perceptor que sea contribuyente del\n", "IRPF, ni estarán sujetos a retención e ingreso a cuenta. \n", "Normativa: Disposición transitoria trigésima segunda.3 LIS\n", "Metadatos:\n", "{'Header 1': 'Capítulo 5. Rendimientos del capital mobiliario', 'Header 2': 'Rendimientos del capital mobiliario: cuestiones generales', 'Header 3': 'Concepto', 'Header 4': 'Delimitación positiva y negativa', 'page': '298'}\n", "----------------------------------------\n", "\n", "--- Chunk 3 ---\n", "Contenido:\n", "Provincia: Valencia \n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "
Ámbito territorialActividadÍndice rendimiento neto
Todos los términos municipales (excepto los que se detallen con un índice inferior).Cereales: Avena.0,05
Cereales: Cebada.0,05
Cereales: Centeno.0,05
Cereales: Trigo.0,05
\n", "\n", " \n", " \n", "Provincia: Valencia \n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "
Ámbito territorialActividadÍndice rendimiento neto
Forrajes.0,26
Frutos no cítricos: Albaricoque.0,19
Frutos no cítricos: Ciruela.0,19
Leguminosas.0,09
Uva para vino sin D.O.0,13
Uva para vino con D.O.0,16
Términos municipales de: Ademuz.Forrajes: Veza.0,07
Frutos no cítricos: Manzana.0,13
Frutos secos: Almendra.0,05
Términos municipales de: Ador y Xeraco.Cítricos: Naranja.0,18
Frutos no cítricos: Aguacate.0,26
Frutos no cítricos: Caqui.0,26
Otros productos agrícolas: Viveros.0,26
Términos municipales de: Atzeneta d'Albaida, Benissoda y Bufali.Frutos no cítricos: Albaricoque.0,07
Frutos no cítricos: Caqui.0,19
Términos municipales de: Agullent.Frutos no cítricos: Albaricoque.0,07
Frutos no cítricos: Caqui.0,19
\n", "\n", " \n", " \n", "Provincia: Valencia \n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "
Ámbito territorialActividadÍndice rendimiento neto
Términos municipales de: Alaquàs, Bellús, Càrcer, Navarrés, Riba-roja de Túria y Sagunto/Sagunt.Frutos no cítricos: Caqui.0,26
Términos municipales de: Aielo de Malferit, Albaida, Alborache, Manises, Marines, Pobla Llarga (La) y Riola.Frutos no cítricos: Caqui.0,19
Términos municipales de: Albal, Alberic, Almiserà, Bellreguard, Beneixida, Cotes, Guardamar de la Safor, Massanassa y Quart de les Valls.Cítricos: Naranja.0,18
Frutos no cítricos: Caqui.0,26
Términos municipales de: Albalat de la Ribera.Cítricos: Mandarina.0,13
Frutos no cítricos: Caqui.0,26
Términos municipales de: Albalat dels Sorells.Cítricos: Naranja.0,09
Frutos no cítricos: Caqui.0,07
Productos hortícolas: Alcachofa.0,18
Productos hortícolas: Cebolla.0,18
Términos municipales de: Alboraia/Alboraya.Productos hortícolas: Alcachofa.0,09
Productos hortícolas: Cebolla.0,09
Productos hortícolas: Haba verde.0,13
Términos municipales de: Albuixech.Cítricos: Naranja.0,09
Frutos no cítricos: Caqui.0,19
Productos hortícolas: Alcachofa.0,13
Términos municipales de: Alcàsser.Cítricos: Naranja.0,18
\n", "\n", " \n", " \n", "Provincia: Valencia \n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "
Ámbito territorialActividadÍndice rendimiento neto
Frutos no cítricos: Aguacate.0,26
Frutos no cítricos: Caqui.0,26
Frutos no cítricos: Granado.0,26
Frutos tropicales.0,26
Términos municipales de: Alzira.Cítricos: Mandarina.0,18
Cítricos: Naranja.0,18
Frutos no cítricos: Albaricoque.0,13
Frutos no cítricos: Caqui.0,26
Frutos no cítricos: Melocotón y nectarina.0,26
Términos municipales de: Alcublas, Domeño, Eliana (L\"), Gestalgar, Gátova y Vilamarxant.Frutos secos: Algarroba.0,18
Términos municipales de: Alcúdia (L\") y Benimodo.Cítricos: Mandarina.0,18
Cítricos: Naranja.0,18
Frutos no cítricos: Caqui.0,19
Términos municipales de: Alcúdia de Crespins (L\") y Llocnou d'En Fenollet.Cítricos: Mandarina.0,13
Cítricos: Naranja.0,09
Frutos no cítricos: Caqui.0,13
Términos municipales de: Aldaia.Frutos no cítricos: Albaricoque.0,07
\n", "\n", " \n", " \n", "Provincia: Valencia \n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "
Ámbito territorialActividadÍndice rendimiento neto
Frutos no cítricos: Granado.0,26
Términos municipales de: Alfafar, Alqueria de la Comtessa (L\"), Beniarjó, Beniflá, Benirredrà, Daimús y Guadassuar.Cítricos: Naranja.0,18
Términos municipales de: Alfauir.Cítricos: Naranja.0,18
Cítricos: Pomelo.0,18
Frutos no cítricos: Aguacate.0,26
Frutos no cítricos: Caqui.0,19
Frutos secos: Algarroba.0,13
Términos municipales de: Alfara del Patriarca.Cítricos: Naranja.0,09
Productos hortícolas: Alcachofa.0,09
Productos hortícolas: Haba verde.0,09
Términos municipales de: Alfarp.Cítricos: Mandarina.0,13
Cítricos: Naranja.0,18
Frutos no cítricos: Caqui.0,13
Términos municipales de: Aielo de Rugat, Alfarrasí y Terrateig.Frutos no cítricos: Albaricoque.0,13
Frutos no cítricos: Caqui.0,19
Frutos no cítricos: Ciruela.0,07
Frutos no cítricos: Melocotón y nectarina.0,19
\n", "\n", " \n", " \n", "Provincia: Valencia \n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "Metadatos:\n", "{'Header 1': 'Capítulo 9. Rendimientos de actividades económicas en estimación objetiva (II) (Actividades agrícolas, ganaderas y forestales)', 'Header 2': 'Apéndice: Relación de productos naturales, servicios y actividades accesorios realizados por agricultores, ganaderos y titulares de actividades forestales e índices de rendimiento aplicables en el ejercicio 2023', 'Header 3': 'Anexo: Reducción de los índices de rendimiento neto, aplicables en 2023, por las actividades agrícolas y ganaderas afectadas por circunstancias excepcionales agrupados por Comunidades Autónomas, Provincias, ámbitos territoriales y actividades', 'Header 4': 'Comunitat Valenciana', 'Header 5': 'Valencia', 'page': '814'}\n", "----------------------------------------\n", "\n", "--- Chunk 4 ---\n", "Contenido:\n", "\n", " \n", " \n", "
Ámbito territorialActividadÍndice rendimiento neto
Términos municipales de: Algemesí y Montserrat.Frutos no cítricos: Albaricoque.0,13
Términos municipales de: Almàssera.Cítricos: Naranja.0,09
\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "
MóduloDefiniciónUnidadRendimiento anual por unidad (euros)
1Personal asalariadoPersona3.476,83
2Personal no asalariadoPersona17.220,39
3Consumo de energía eléctrica100 kWh403,11
4Superficie del localm2844,02
\n", "Nota: El rendimiento neto resultante de la aplicación de los signos o módulos anteriores incluye, en su caso, el\n", "derivado de la venta de artículos de escaso valor tales como dulces, artículos de fumador, etc., los servicios de\n", "publicidad exterior y comercialización de tarjetas de transporte público, tarjetas para uso telefónico y otras\n", "similares, así como loterías, siempre que estas actividades se desarrollen con carácter accesorio a la actividad\n", "principal. \n", "Departamento de Gestión Tributaria \n", "Cuantía a efectos del índice corrector de exceso: 28.860,22 euros\n", "Metadatos:\n", "{'Header 1': 'Capítulo 8. Rendimientos de actividades económicas en estimación objetiva (I) (Actividades distintas de las agrícolas, ganaderas y forestales)', 'Header 2': 'Apéndice: Rendimientos anuales por unidad de módulo antes de amortización aplicables en el ejercicio 2023', 'Header 3': 'Módulos aplicables a cada una de las actividades, incluidos en el Anexo II de la Orden HFP/1172/2022, con indicación de su correspondiente epígrafe en el Impuesto de Actividades Económicas (IAE)', 'Header 4': 'Epígrafe IAE: 659.4 - Comercio al por menor de prensa, revistas y libros en quioscos situados en la vía pública', 'page': '554'}\n", "----------------------------------------\n", "\n", "--- Chunk 5 ---\n", "Contenido:\n", "La participación de las Comunidades Autónomas en la Agencia Estatal de Administración\n", "Tributaria, que constituye la organización administrativa responsable en nombre y por cuenta\n", "del Estado de la aplicación efectiva del sistema tributario estatal y del aduanero, se desarrolla\n", "a través de los siguientes órganos, regulados en los artículos 65 y 66 de la Ley 22/2009: \n", "Consejo Superior para la Dirección y Coordinación de la Gestión Tributaria \n", "El Consejo Superior para la Dirección y Coordinación de la Gestión Tributaria es el órgano\n", "colegiado, integrado por representantes de la Administración Tributaria del Estado y de las\n", "Comunidades Autónomas y Ciudades con Estatuto de Autonomía, encargado de coordinar la\n", "gestión de los tributos cedidos. \n", "Este órgano está presidido por el Presidente de la Agencia Estatal de Administración\n", "Tributaria e integrado por la Directora General de la Agencia Estatal de Administración\n", "Tributaria, que ostentará la Vicepresidencia primera, cinco representantes de la Agencia\n", "Estatal de Administración Tributaria, los titulares de la Secretaría General de Hacienda, de la\n", "Secretaría General de Financiación Territorial y de la Inspección General del Ministerio de\n", "Economía y Hacienda (actualmente, Ministerio de Hacienda) y por un representante de cada\n", "una de las Comunidades Autónomas de régimen común y de las Ciudades con Estatuto de\n", "Autonomía, uno de los cuales será designado por éstas cada año para ostentar la\n", "Vicepresidencia segunda. \n", "Aquellas Comunidades y Ciudades Autónomas que tengan encomendadas a dos órganos o\n", "entes distintos las funciones de aplicación de los tributos y las de diseño o interpretación de\n", "la normativa autonómica podrán designar dos representantes, si bien dispondrán de un solo\n", "voto. \n", "Precisión: téngase en cuenta que el artículo 1.8 del Real Decreto 352/2011, de 11 de marzo (BOE de 12 de\n", "marzo) suprimió la Secretaría General de Financiación Territorial y que el Real Decreto 1887/2011, de 30 de\n", "diciembre (BOE de 31 de diciembre), suprimió en su artículo 4.3, entre otros órganos directivos, la Secretaría\n", "General de Hacienda y la Inspección General del Ministerio de Economía y Hacienda. \n", "En la actualidad véase el Real Decreto 139/2020, de 28 de enero, por el que se establece la estructura\n", "orgánica básica de los departamentos ministeriales (BOE de 29 de enero), que incluye como órgano directivo\n", "de la Secretaría de Estado de Hacienda de la que depende la Secretaría General de Financiación Autonómica\n", "y Local y, dentro de la actual Subsecretaría de Hacienda y Función Pública, a la Inspección General. \n", "Consejos Territoriales para la Dirección y Coordinación de la Gestión Tributaria \n", "\n", " \n", " \n", "Los Consejos Territoriales para la Dirección y Coordinación de la Gestión Tributaria son\n", "órganos colegiados integrados por representantes de la Administración Tributaria del Estado\n", "y de la Comunidad Autónoma o de la Ciudad con Estatuto de Autonomía de que se trate a los\n", "que corresponde coordinar la gestión de los tributos cedidos en su respectivo ámbito\n", "territorial. \n", "Estos consejos están compuestos por cuatro representantes de la Agencia Estatal de\n", "Administración Tributaria y cuatro de la respectiva Comunidad Autónoma o Ciudad con\n", "Estatuto de Autonomía. Existirán tantos suplentes como titulares, que actuarán en caso de\n", "ausencia o vacante de alguno de estos últimos.\n", "Metadatos:\n", "{'Header 1': 'Capítulo 2. El Impuesto sobre la Renta de las Personas Físicas (IRPF): cuestiones generales', 'Header 2': 'Cesión parcial del IRPF a las Comunidades Autónomas', 'Header 3': 'Participación de las Comunidades Autónomas y Ciudades con Estatuto de Autonomía en la gestión del IRPF', 'page': '95'}\n", "----------------------------------------\n" ] } ], "source": [ "import chromadb\n", "from chromadb.utils.batch_utils import create_batches\n", "\n", "MARKDOWN_FILE_PATH = \"/teamspace/studios/this_studio/AgenticRAG/doc_renta/Renta_2023_doc_int_corrected_2.md\"\n", "\n", "# Procesar el archivo Markdown con MarkdownHeaderTextSplitter\n", "chunks = process_markdown_with_header_splitter(MARKDOWN_FILE_PATH)\n", "print(f\"Total chunks extraídos: {len(chunks)}\")\n", "\n", "# Realizar prueba aleatoria\n", "test_random_chunks(chunks, num_samples=5)\n", "\n", "# Añadir los chunks a ChromaDB (comentado para evitar ejecución accidental)\n", "# add_chunks_to_chroma(chunks)" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "# results = collection.query(\n", "# query_texts=[\n", "# \"What is the embedding used in llama model?\"\n", "# ], # Also available query_embeddings to use embeddings instead of text.\n", "# n_results=2, # how many results to return\n", "# )\n", "\n", "# print(results)" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "### Code to save" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "# from litellm import embedding\n", "\n", "# response = embedding(\n", "# model=EMBEDDING_NAME,\n", "# input=[\"test embedding\"],\n", "# api_key=EMBEDDING_API_KEY,\n", "# api_base=EMBEDDING_BASE,\n", "# api_version=EMBEDDING_VERSION,\n", "# )\n", "# print(response)" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [ "# chunk_step = 250\n", "# for chunk_idx in range(0, len(chunks), chunk_step):\n", "# vector_store.add_documents(\n", "# documents=chunks[chunk_idx : chunk_idx + chunk_step],\n", "# ids=uuids[chunk_idx : chunk_idx + chunk_step],\n", "# )\n", "# print(\n", "# f\"Saved {len(chunks[chunk_idx : chunk_idx + chunk_step])} chunks to {CHROMA_PATH}.\"\n", "# )\n", "# print(len(vector_store.get()[\"documents\"]))" ] } ], "metadata": { "language_info": { "name": "python" } }, "nbformat": 4, "nbformat_minor": 2 }