Update local to ensure running locally

This commit is contained in:
Lance Martin
2024-06-11 11:18:40 -07:00
parent 6e9bdcd7c7
commit ca3ed1e322
4 changed files with 12 additions and 8 deletions
@@ -45,7 +45,7 @@
"metadata": {},
"outputs": [],
"source": [
"! pip install -U langchain-nomic langchain_community tiktoken langchainhub chromadb langchain langgraph tavily-python"
"! pip install -U langchain-nomic langchain_community tiktoken langchainhub chromadb langchain langgraph tavily-python nomic[local]"
]
},
{
@@ -143,7 +143,8 @@
"vectorstore = Chroma.from_documents(\n",
" documents=doc_splits,\n",
" collection_name=\"rag-chroma\",\n",
" embedding=NomicEmbeddings(model=\"nomic-embed-text-v1.5\"),\n",
" embedding=NomicEmbeddings(model=\"nomic-embed-text-v1.5\",\n",
" inference_mode='local'),\n",
")\n",
"retriever = vectorstore.as_retriever()"
]
+2 -1
View File
@@ -202,7 +202,8 @@
"\n",
"# Embed and index\n",
"if run_local == \"Yes\":\n",
" embedding=NomicEmbeddings(model=\"nomic-embed-text-v1.5\")\n",
" embedding=NomicEmbeddings(model=\"nomic-embed-text-v1.5\",\n",
" inference_mode='local',)\n",
"else:\n",
" embedding = MistralAIEmbeddings(mistral_api_key=mistral_api_key)\n",
"\n",
@@ -51,7 +51,7 @@
"outputs": [],
"source": [
"%%capture --no-stderr\n",
"%pip install -U langchain-nomic langchain_community tiktoken langchainhub chromadb langchain langgraph tavily-python gpt4all langchain-text-splitters"
"%pip install -U langchain-nomic langchain_community tiktoken langchainhub chromadb langchain langgraph tavily-python nomic[local] langchain-text-splitters"
]
},
{
@@ -120,7 +120,8 @@
"vectorstore = Chroma.from_documents(\n",
" documents=doc_splits,\n",
" collection_name=\"rag-chroma\",\n",
" embedding=NomicEmbeddings(model=\"nomic-embed-text-v1.5\"),\n",
" embedding=NomicEmbeddings(model=\"nomic-embed-text-v1.5\",\n",
" inference_mode='local'),\n",
")\n",
"retriever = vectorstore.as_retriever()"
]
+4 -3
View File
@@ -60,7 +60,7 @@
"metadata": {},
"outputs": [],
"source": [
"! pip install -U langchain-nomic langchain_community tiktoken langchainhub chromadb langchain langgraph"
"! pip install -U langchain-nomic langchain_community tiktoken langchainhub chromadb langchain langgraph nomic[local]"
]
},
{
@@ -132,7 +132,7 @@
},
{
"cell_type": "code",
"execution_count": 4,
"execution_count": null,
"id": "c3bb9060-ad74-4470-9991-2ba167b6b8d8",
"metadata": {},
"outputs": [],
@@ -160,7 +160,8 @@
"vectorstore = Chroma.from_documents(\n",
" documents=doc_splits,\n",
" collection_name=\"rag-chroma\",\n",
" embedding=NomicEmbeddings(model=\"nomic-embed-text-v1.5\"),\n",
" embedding=NomicEmbeddings(model=\"nomic-embed-text-v1.5\",\n",
" inference_mode='local'),\n",
")\n",
"retriever = vectorstore.as_retriever()"
]