Skip to content

Commit 44554ee

Browse files
Restrict versions for consistency for RAG notebooks (#3101)
1 parent d420444 commit 44554ee

File tree

5 files changed

+14
-10
lines changed

5 files changed

+14
-10
lines changed

notebooks/llm-agent-rag-llamaindex/llm-agent-rag-llamaindex.ipynb

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -90,16 +90,18 @@
9090
" \"-q\",\n",
9191
" \"--extra-index-url\",\n",
9292
" \"https://download.pytorch.org/whl/cpu\",\n",
93+
" \"nncf>=2.18.0\",\n",
9394
" \"llama-index\",\n",
9495
" \"llama-index-readers-file\",\n",
9596
" \"llama-index-core\",\n",
96-
" \"transformers>=4.43.1\",\n",
97+
" \"torch==2.8\",\n",
98+
" \"transformers==4.53.3\",\n",
9799
" \"llama-index-llms-huggingface>=0.4.0,<0.5.0\",\n",
98100
" \"llama-index-embeddings-huggingface>=0.4.0,<0.5.0\",\n",
99101
" \"huggingface-hub>=0.26.5\",\n",
100102
")\n",
101-
"pip_install(\"-q\", \"git+https://github.com/huggingface/optimum-intel.git\", \"git+https://github.com/openvinotoolkit/nncf.git\", \"datasets<4.0.0\", \"accelerate\")\n",
102-
"pip_install(\"--pre\", \"-Uq\", \"openvino>=2024.2.0\", \"--extra-index-url\", \"https://storage.openvinotoolkit.org/simple/wheels/nightly\")\n",
103+
"pip_install(\"-q\", \"git+https://github.com/huggingface/optimum-intel.git\", \"datasets<4.0.0\", \"accelerate\")\n",
104+
"pip_install(\"--pre\", \"-Uq\", \"openvino>=2025.3.0\", \"--extra-index-url\", \"https://storage.openvinotoolkit.org/simple/wheels/nightly\")\n",
103105
"pip_install(\"--pre\", \"-Uq\", \"openvino-tokenizers[transformers]\", \"--extra-index-url\", \"https://storage.openvinotoolkit.org/simple/wheels/nightly\")\n",
104106
"pip_install(\"-q\", \"--no-deps\", \"llama-index-llms-openvino\", \"llama-index-embeddings-openvino\")\n",
105107
"\n",

notebooks/llm-agent-react-langchain/llm-agent-react-langchain.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@
9494
"%pip install -Uq pip\n",
9595
"%pip uninstall -q -y optimum optimum-intel\n",
9696
"%pip install --pre -Uq \"openvino>=2025.3.0\" openvino-tokenizers[transformers] --extra-index-url https://storage.openvinotoolkit.org/simple/wheels/nightly\n",
97-
"%pip install -q --extra-index-url https://download.pytorch.org/whl/cpu \"transformers==4.53.3\" \"langchain>=0.2.3\" \"langchain-huggingface>=0.1.2\" \"langchain-community>=0.2.4\" \"Wikipedia\" \\\n",
97+
"%pip install -q --extra-index-url https://download.pytorch.org/whl/cpu \"transformers==4.53.3\" \"langchain>=0.2.3,<1.0\" \"langchain-huggingface>=0.1.2\" \"langchain-community>=0.2.4\" \"Wikipedia\" \\\n",
9898
"\"nncf>=2.18.0\" \\\n",
9999
"\"torch==2.8\" \\\n",
100100
"\"datasets<4.0.0\" \\\n",

notebooks/llm-rag-langchain/llm-rag-langchain-genai.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@
130130
" \"transformers==4.53.3\",\n",
131131
" \"faiss-cpu\",\n",
132132
" \"sentence_transformers\",\n",
133-
" \"langchain>=0.2.0\",\n",
133+
" \"langchain>=0.2.0,<1.0\",\n",
134134
" \"langchain-community>=0.2.15\",\n",
135135
" \"langchainhub\",\n",
136136
" \"unstructured\",\n",

notebooks/llm-rag-langchain/llm-rag-langchain.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@
115115
" \"transformers==4.53.3\",\n",
116116
" \"faiss-cpu\",\n",
117117
" \"sentence_transformers\",\n",
118-
" \"langchain>=0.2.0\",\n",
118+
" \"langchain>=0.2.0,<1.0\",\n",
119119
" \"langchain-community>=0.2.15\",\n",
120120
" \"langchainhub\",\n",
121121
" \"unstructured\",\n",

notebooks/llm-rag-llamaindex/llm-rag-llamaindex.ipynb

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -111,11 +111,13 @@
111111
" \"-qU\",\n",
112112
" \"--extra-index-url\",\n",
113113
" \"https://download.pytorch.org/whl/cpu\",\n",
114-
" \"torch>=2.7.1\",\n",
114+
" \"nncf>=2.18.0\",\n",
115+
" \"torch==2.8\",\n",
116+
" \"transformers==4.53.3\",\n",
115117
" \"llama-index\",\n",
116118
" \"faiss-cpu\",\n",
117119
" \"pymupdf\",\n",
118-
" \"langchain\",\n",
120+
" \"langchain<1.0\",\n",
119121
" \"llama-index-readers-file\",\n",
120122
" \"llama-index-vector-stores-faiss\",\n",
121123
" \"llama-index-llms-langchain\",\n",
@@ -124,7 +126,7 @@
124126
" \"huggingface-hub>=0.26.5\",\n",
125127
")\n",
126128
"pip_install(\n",
127-
" \"-q\", \"git+https://github.com/huggingface/optimum-intel.git\", \"git+https://github.com/openvinotoolkit/nncf.git\", \"datasets<4.0.0\", \"accelerate\", \"gradio\"\n",
129+
" \"-q\", \"git+https://github.com/huggingface/optimum-intel.git\", \"datasets<4.0.0\", \"accelerate\", \"gradio\"\n",
128130
")\n",
129131
"pip_install(\"--pre\", \"-U\", \"openvino>=2025.2\", \"--extra-index-url\", \"https://storage.openvinotoolkit.org/simple/wheels/nightly\")\n",
130132
"pip_install(\n",
@@ -1422,7 +1424,7 @@
14221424
"faiss_index = faiss.IndexFlatL2(EMBEDDING_MODEL_HIDDEN_SIZE)\n",
14231425
"Settings.embed_model = embedding\n",
14241426
"\n",
1425-
"llm.max_new_tokens = 2048\n",
1427+
"llm.max_new_tokens = 1024\n",
14261428
"if stop_tokens is not None:\n",
14271429
" llm._stopping_criteria = StoppingCriteriaList(stop_tokens)\n",
14281430
"Settings.llm = llm\n",

0 commit comments

Comments
 (0)