From 4741977504bf17679cf9faf9ef0d6e326e459d14 Mon Sep 17 00:00:00 2001
From: venti <1308199824@qq.com>
Date: Sat, 30 May 2026 20:58:49 +0800
Subject: [PATCH] fix: update example notebook imports to match v3 module
 structure (fixes #2356)

---
 docs/examples_notebooks/local_search.ipynb | 68 ++++++++++------------
 1 file changed, 32 insertions(+), 36 deletions(-)

diff --git a/docs/examples_notebooks/local_search.ipynb b/docs/examples_notebooks/local_search.ipynb
index f7f0c5a54b..14cc1213d8 100644
--- a/docs/examples_notebooks/local_search.ipynb
+++ b/docs/examples_notebooks/local_search.ipynb
@@ -189,42 +189,38 @@
    "execution_count": null,
    "metadata": {},
    "outputs": [],
-   "source": [
-    "from graphrag.config.enums import ModelType\n",
-    "from graphrag.config.models.language_model_config import LanguageModelConfig\n",
-    "from graphrag.language_model.manager import ModelManager\n",
-    "from graphrag.tokenizer.get_tokenizer import get_tokenizer\n",
-    "\n",
-    "api_key = os.environ[\"GRAPHRAG_API_KEY\"]\n",
-    "\n",
-    "chat_config = LanguageModelConfig(\n",
-    "    api_key=api_key,\n",
-    "    type=ModelType.Chat,\n",
-    "    model_provider=\"openai\",\n",
-    "    model=\"gpt-4.1\",\n",
-    "    max_retries=20,\n",
-    ")\n",
-    "chat_model = ModelManager().get_or_create_chat_model(\n",
-    "    name=\"local_search\",\n",
-    "    model_type=ModelType.Chat,\n",
-    "    config=chat_config,\n",
-    ")\n",
-    "\n",
-    "embedding_config = LanguageModelConfig(\n",
-    "    api_key=api_key,\n",
-    "    type=ModelType.Embedding,\n",
-    "    model_provider=\"openai\",\n",
-    "    model=\"text-embedding-3-small\",\n",
-    "    max_retries=20,\n",
-    ")\n",
-    "\n",
-    "text_embedder = ModelManager().get_or_create_embedding_model(\n",
-    "    name=\"local_search_embedding\",\n",
-    "    model_type=ModelType.Embedding,\n",
-    "    config=embedding_config,\n",
-    ")\n",
-    "\n",
-    "tokenizer = get_tokenizer(chat_config)"
+    "source": [
+     "from graphrag_llm.config import ModelConfig\n",
+     "from graphrag_llm.completion.completion_factory import completion_factory\n",
+     "from graphrag_llm.embedding.embedding_factory import embedding_factory\n",
+     "from graphrag.tokenizer.get_tokenizer import get_tokenizer\n",
+     "\n",
+     "api_key = os.environ[\"GRAPHRAG_API_KEY\"]\n",
+     "\n",
+     "chat_config = ModelConfig(\n",
+     "    api_key=api_key,\n",
+     "    model_provider=\"openai\",\n",
+     "    model=\"gpt-4.1\",\n",
+     "    max_retries=20,\n",
+     ")\n",
+     "chat_model = completion_factory.create(\n",
+     "    name=\"local_search\",\n",
+     "    config=chat_config,\n",
+     ")\n",
+     "\n",
+     "embedding_config = ModelConfig(\n",
+     "    api_key=api_key,\n",
+     "    model_provider=\"openai\",\n",
+     "    model=\"text-embedding-3-small\",\n",
+     "    max_retries=20,\n",
+     ")\n",
+     "\n",
+     "text_embedder = embedding_factory.create(\n",
+     "    name=\"local_search_embedding\",\n",
+     "    config=embedding_config,\n",
+     ")\n",
+     "\n",
+     "tokenizer = get_tokenizer(chat_config)"
    ]
   },
   {