Merge pull request #140 from deepset-ai/fix-decomposition-2

moving import up
deepset-ai · Sep 27, 2024 · ba1e9bf · ba1e9bf
2 parents 045357b + a273b93
commit ba1e9bf
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/notebooks/query_decomposition.ipynb b/notebooks/query_decomposition.ipynb
@@ -53,7 +53,8 @@
         "from datasets import load_dataset\n",
         "from openai import OpenAI, Stream\n",
         "from openai.types.chat import ChatCompletion, ChatCompletionChunk\n",
-        "from typing import List, Any, Dict, Optional, Callable, Union"
+        "from typing import List, Any, Dict, Optional, Callable, Union\n",
+        "from pydantic import BaseModel"
       ]
     },
     {
@@ -223,8 +224,6 @@
       },
       "outputs": [],
       "source": [
-        "from pydantic import BaseModel\n",
-        "\n",
         "class Question(BaseModel):\n",
         "    question: str\n",
         "    answer: Optional[str] = None\n",