Update langchain versions in llama-index e2e tests

datastax · Jun 14, 2024 · 3bee8d3 · 3bee8d3
1 parent afd3f4e
commit 3bee8d3
Showing 1 changed file with 14 additions and 11 deletions.
diff --git a/libs/e2e-tests/pyproject.llamaindex.toml b/libs/e2e-tests/pyproject.llamaindex.toml
@@ -16,23 +16,26 @@ azure-storage-blob = "^12.19.0"
 pillow = "^10.2.0"
 python-dotenv = "^1.0.1"
 trulens-eval = "^0.31.0"
-nemoguardrails = "^0.8.0"
 langchainhub = "^0.1.15"
 
+# Temporarily use nemoguardrails HEAD to get https://github.com/NVIDIA/NeMo-Guardrails/pull/551
+# nemoguardrails = "^0.8.0"
+nemoguardrails = { git = "https://github.com/NVIDIA/NeMo-Guardrails.git", branch = "develop" }
+
+
 # From LangChain optional deps, needed by WebBaseLoader
 beautifulsoup4 = "^4"
 
 llama-index = { git = "https://github.com/run-llama/llama_index.git", branch = "main" }
+llama-index-core = { git = "https://github.com/run-llama/llama_index.git", branch = "main" }
 llama-index-readers-llama-parse = { git = "https://github.com/run-llama/llama_index.git", branch = "main", subdirectory = "llama-index-integrations/readers/llama-index-readers-llama-parse" }
 llama-index-embeddings-langchain = { git = "https://github.com/run-llama/llama_index.git", branch = "main", subdirectory = "llama-index-integrations/embeddings/llama-index-embeddings-langchain" }
 llama-index-vector-stores-astra-db = { git = "https://github.com/run-llama/llama_index.git", branch = "main", subdirectory = "llama-index-integrations/vector_stores/llama-index-vector-stores-astra-db" }
 llama-index-vector-stores-cassandra = { git = "https://github.com/run-llama/llama_index.git", branch = "main", subdirectory = "llama-index-integrations/vector_stores/llama-index-vector-stores-cassandra" }
+llama-index-tools-cassandra = { git = "https://github.com/run-llama/llama_index.git", branch = "main", subdirectory = "llama-index-integrations/tools/llama-index-tools-cassandra" }
 llama-index-llms-bedrock = { git = "https://github.com/run-llama/llama_index.git", branch = "main", subdirectory = "llama-index-integrations/llms/llama-index-llms-bedrock" }
 llama-index-llms-azure-openai = { git = "https://github.com/run-llama/llama_index.git", branch = "main", subdirectory = "llama-index-integrations/llms/llama-index-llms-azure-openai" }
 llama-index-llms-gemini = { git = "https://github.com/run-llama/llama_index.git", branch = "main", subdirectory = "llama-index-integrations/llms/llama-index-llms-gemini" }
-# huggingface-hub is not usable because nemo-guardrails needs >=0.20,<0.21 (via fastembed)
-# and llama-index-llms-huggingface needs >=0.23.0
-#llama-index-llms-huggingface = { git = "https://github.com/run-llama/llama_index.git", branch = "main", subdirectory = "llama-index-integrations/llms/llama-index-llms-huggingface" }
 llama-index-llms-vertex = { git = "https://github.com/run-llama/llama_index.git", branch = "main", subdirectory = "llama-index-integrations/llms/llama-index-llms-vertex" }
 llama-index-embeddings-bedrock = { git = "https://github.com/run-llama/llama_index.git", branch = "main", subdirectory = "llama-index-integrations/embeddings/llama-index-embeddings-bedrock" }
 llama-index-embeddings-azure-openai = { git = "https://github.com/run-llama/llama_index.git", branch = "main", subdirectory = "llama-index-integrations/embeddings/llama-index-embeddings-azure-openai" }
@@ -41,14 +44,14 @@ llama-index-multi-modal-llms-gemini = { git = "https://github.com/run-llama/llam
 
 llama-parse = { git = "https://github.com/run-llama/llama_parse.git", branch = "main" }
 
-langchain = "0.1.17"
-langchain-core = "0.1.48"
-langchain-community = "0.0.36"
+langchain = "0.2.3"
+langchain-core = "0.2.5"
+langchain-community = "0.2.4"
 langchain-astradb = "0.3.3"
-langchain-openai = "0.1.3"
-langchain-google-genai = "0.0.11"
-langchain-google-vertexai = "1.0.1"
-langchain-nvidia-ai-endpoints = "0.0.9"
+langchain-openai = "0.1.8"
+langchain-google-genai = { version = "1.0.6" }
+langchain-google-vertexai = { version = "1.0.5" }
+langchain-nvidia-ai-endpoints = { version = "0.1.1" }
 
 unstructured = "0.14.5"