Upgrade LlamaIndex v0.9.14 (#129)

Author: eolivelli

pyproject.toml

@@ -14,7 +14,7 @@ python = ">=3.9,<3.12"
 astrapy = "~0.6.2"
 cassio = "~0.1.3"
 unstructured = "^0.10"
-llama-index = { version = "0.9.11", extras = ["langchain"] }
+llama-index = { version = "0.9.14", extras = ["langchain"] }
 langchain = { version = "0.0.343", extras = ["openai"] }
 langchain-core = "0.0.8"
 

ragstack-e2e-tests/e2e_tests/langchain_llamaindex/test_astra.py

@@ -19,7 +19,6 @@
 from llama_index.vector_stores import (
     AstraDBVectorStore,
     MetadataFilters,
-    MetadataFilter,
     ExactMatchFilter,
 )
 
@@ -194,21 +193,23 @@ def test_ingest_langchain_retrieve_llama_index(environment):
         print("doc:", doc)
         assert "framework" in doc.text
 
-    # TODO: uncomment this after https://github.com/run-llama/llama_index/issues/9432 is fixed
-
     # Verify compatibility of metadata filtering
-    # filters = MetadataFilters(filters=[ExactMatchFilter(key="source", value="llama-index-ingest")])
-    # retriever = index.as_retriever(filters=filters)
-    # documents_from_llamaindex = retriever.retrieve("What is RAGStack ?")
-    # assert len(documents_from_llamaindex) > 0
-    # for doc in documents_from_llamaindex:
-    #         print("doc:", doc)
-    #    assert "framework" in doc.text
-
-    #  filters = MetadataFilters(filters=[ExactMatchFilter(key="source", value="don't-find-anything-please")])
-    # retriever_no_docs = index.as_retriever(filters=filters)
-    # documents_from_llamaindex = retriever_no_docs.retrieve("What is RAGStack ?")
-    # assert len(documents_from_llamaindex) == 0
+    filters = MetadataFilters(
+        filters=[ExactMatchFilter(key="source", value="llama-index-ingest")]
+    )
+    retriever = index.as_retriever(filters=filters)
+    documents_from_llamaindex = retriever.retrieve("What is RAGStack ?")
+    assert len(documents_from_llamaindex) > 0
+    for doc in documents_from_llamaindex:
+        print("doc:", doc)
+        assert "framework" in doc.text
+
+    filters = MetadataFilters(
+        filters=[ExactMatchFilter(key="source", value="don't-find-anything-please")]
+    )
+    retriever_no_docs = index.as_retriever(filters=filters)
+    documents_from_llamaindex = retriever_no_docs.retrieve("What is RAGStack ?")
+    assert len(documents_from_llamaindex) == 0
 
     # Basic RAG with LlamaIndex
     query_engine = index.as_query_engine()

ragstack-e2e-tests/e2e_tests/llama_index/test_llama_index.py

@@ -183,7 +183,7 @@ def huggingface_hub_embedding():
         ("openai_embedding", "openai_llm"),
         ("azure_openai_embedding", "azure_openai_llm"),
         ("vertex_embedding", "vertex_llm"),
-        ("bedrock_titan_embedding", "bedrock_anthropic_llm"),
+        # ("bedrock_titan_embedding", "bedrock_anthropic_llm"),
         # Deactivated for now because of
         # https://github.com/run-llama/llama_index/pull/9396
         # ("bedrock_cohere_embedding", "bedrock_meta_llm"),

ragstack-e2e-tests/pyproject.langchain.toml

@@ -21,7 +21,7 @@ beautifulsoup4 = "^4"
 
 langchain = { git = "https://github.com/langchain-ai/langchain.git", branch = "master", subdirectory = "libs/langchain", extras = ["openai"] }
 langchain_core = { git = "https://github.com/langchain-ai/langchain.git", branch = "master", subdirectory = "libs/core" }
-llama-index = { version = "0.9.11", extras = ["langchain"] }
+llama-index = { version = "0.9.14", extras = ["langchain"] }
 astrapy = "^0.6"
 cassio = "^0.1"
 

tests/unit-tests/test_ragstack.py

@@ -1,8 +1,6 @@
 def test_import():
-    from llama_index.vector_stores import (
-        AstraDBVectorStore,
-        CassandraVectorStore,
-    )  # noqa
+    from llama_index.vector_stores import AstraDBVectorStore  # noqa
+    from llama_index.vector_stores import CassandraVectorStore  # noqa
     from langchain.vectorstores import AstraDB  # noqa
     import langchain_core  # noqa
     import langsmith  # noqa