Something went wrong. Try again.
lol
Something went wrong. Try again.
4.0 kB · 169 lines
Nix
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170{ lib, aiohttp, aiosqlite, banks, buildPythonPackage, dataclasses-json, deprecated, dirtyjson, fetchFromGitHub, filetype, fsspec, hatchling, jsonpath-ng, llama-index-workflows, nest-asyncio, networkx, nltk-data, nltk, numpy, openai, pandas, pillow, pytest-asyncio, pytest-mock, pytestCheckHook, pyvis, pyyaml, requests, spacy, sqlalchemy, tenacity, tinytag, tiktoken, tree-sitter, typing-inspect,}:
buildPythonPackage (finalAttrs: { pname = "llama-index-core"; version = "0.14.19"; pyproject = true;
src = fetchFromGitHub { owner = "run-llama"; repo = "llama_index"; tag = "v${finalAttrs.version}"; hash = "sha256-xcssJPBXq3bjSD13nsR6jRTmTWPVks8aKHZCZ3lSKY4="; };
sourceRoot = "${finalAttrs.src.name}/${finalAttrs.pname}";
# When `llama-index` is imported, it uses `nltk` to look for the following files and tries to # download them if they aren't present. # https://github.com/run-llama/llama_index/blob/6efa53cebd5c8ccf363582c932fffde44d61332e/llama-index-core/llama_index/core/utils.py#L59-L67 # Setting `NLTK_DATA` to a writable path can also solve this problem, but it needs to be done in # every package that depends on `llama-index-core` for `pythonImportsCheck` not to fail, so this # solution seems more elegant. postPatch = '' mkdir -p llama_index/core/_static/nltk_cache/corpora/stopwords/ cp -r ${nltk-data.stopwords}/corpora/stopwords/* llama_index/core/_static/nltk_cache/corpora/stopwords/
mkdir -p llama_index/core/_static/nltk_cache/tokenizers/punkt/ cp -r ${nltk-data.punkt}/tokenizers/punkt/* llama_index/core/_static/nltk_cache/tokenizers/punkt/ '';
pythonRelaxDeps = [ "setuptools" "tenacity" ];
build-system = [ hatchling ];
dependencies = [ aiohttp aiosqlite banks dataclasses-json deprecated dirtyjson filetype fsspec jsonpath-ng llama-index-workflows nest-asyncio networkx nltk numpy openai pandas pillow pyvis pyyaml requests spacy sqlalchemy tenacity tinytag tiktoken typing-inspect ];
nativeCheckInputs = [ tree-sitter pytest-asyncio pytest-mock pytestCheckHook ];
pythonImportsCheck = [ "llama_index" ];
disabledTestPaths = [ # Tests require network access "tests/agent/" "tests/callbacks/" "tests/chat_engine/" "tests/evaluation/" "tests/indices/" "tests/ingestion/" "tests/memory/" "tests/node_parser/" "tests/objects/" "tests/playground/" "tests/postprocessor/" "tests/query_engine/" "tests/question_gen/" "tests/response_synthesizers/" "tests/retrievers/" "tests/selectors/" "tests/test_utils.py" "tests/text_splitter/" "tests/token_predictor/" "tests/tools/" "tests/schema/" "tests/multi_modal_llms/" "tests/prompts/" "tests/base/llms/" ];
disabledTests = [ # Tests require network access "test_context_extraction_basic" "test_context_extraction_custom_prompt" "test_context_extraction_oversized_document" "test_document_block_from_b64" "test_document_block_from_bytes" "test_document_block_from_path" "test_document_block_from_url" "test_from_namespaced_persist_dir" "test_from_persist_dir" "test_mimetype_raw_data" "test_multiple_documents_context" "test_predict_and_call_via_react_agent" "test_resource" # asyncio.exceptions.InvalidStateError: invalid state "test_workflow_context_to_dict_mid_run" "test_SimpleDirectoryReader" # RuntimeError "test_str" ];
meta = { description = "Data framework for your LLM applications"; homepage = "https://github.com/run-llama/llama_index/"; changelog = "https://github.com/run-llama/llama_index/blob/${finalAttrs.src.tag}/CHANGELOG.md"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ fab ]; };})