Update Multi-step research agent.ipynb with local embeddings example (#…

…87)
ShoggothAI · Oct 21, 2024 · 4de890f · 4de890f
1 parent 6a4ef41
commit 4de890f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/examples/Multi-step research agent.ipynb b/examples/Multi-step research agent.ipynb
@@ -179,7 +179,12 @@
     "MAX_ITER = 3\n",
     "ANSWER_LENGTH = 200\n",
     "\n",
-    "embeddings = OpenAIEmbedding(model=\"text-embedding-ada-002\")\n",
+    "# for local embeddings use from llama_index.embeddings.ollama import OllamaEmbedding\n",
+    "#ollama_embedding = OllamaEmbedding(\n",
+    "#    model_name=\"nomic-embed-text\",\n",
+    "#    base_url=\"http://localhost:11434\",\n",
+    "#    ollama_additional_kwargs={\"mirostat\": 0},\n",
+    "#)\n",
     "llm = init_llm(LLMFramework.LANGCHAIN, llm_name=\"gpt-4o\")\n",
     "\n",
     "query_tool = SimpleRetrieverTool(DATA_DIR, PERSIST_DIR, return_strings_only=True, embeddings=embeddings)\n",