adjusting imports

deepset-ai · Oct 10, 2024 · af267f5 · af267f5
1 parent c483a32
commit af267f5
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/notebooks/metadata_enrichment.ipynb b/notebooks/metadata_enrichment.ipynb
@@ -52,17 +52,16 @@
       },
       "outputs": [],
       "source": [
-        "from haystack import Document, component\n",
+        "from haystack import Document, component, Pipeline\n",
         "from haystack.components.builders import PromptBuilder\n",
         "from haystack.components.converters import HTMLToDocument\n",
         "from haystack.components.fetchers import LinkContentFetcher\n",
         "from haystack.components.generators import OpenAIGenerator\n",
         "from haystack.components.generators.openai_utils import _convert_message_to_openai_format\n",
         "from haystack.components.preprocessors import DocumentSplitter\n",
         "from haystack.dataclasses import ChatMessage, StreamingChunk\n",
-        "from haystack.document_stores.in_memory import InMemoryDocumentStore\n",
         "\n",
-        "from openai import OpenAI, Stream\n",
+        "from openai import Stream\n",
         "from openai.types.chat import ChatCompletion, ChatCompletionChunk\n",
         "from typing import List, Any, Dict, Optional, Callable, Union\n",
         "from pydantic import BaseModel"
@@ -182,9 +181,6 @@
       },
       "outputs": [],
       "source": [
-        "from haystack import component, Pipeline, Document\n",
-        "from haystack.components.builders.prompt_builder import PromptBuilder\n",
-        "\n",
         "DEFAULT_PROMPT = \"\"\"\n",
         "Given the contents of the documents, extract the requested metadata.\n",
         "The requested metadata is {{ metadata_model }}\n",