improve queries/data

Author: davidhou17

ai-integrations/langchain-self-query-retrieval.ipynb

@@ -116,6 +116,10 @@
     "        metadata={\"year\": 1993, \"rating\": 7.7, \"genre\": \"action\"},\n",
     "    ),\n",
     "    Document(\n",
+    "        page_content=\"A fight club that is not a fight club, but is a fight club\",\n",
+    "        metadata={\"year\": 1994, \"rating\": 8.7, \"genre\": \"action\"},\n",
+    "    ),\n",
+    "    Document(\n",
     "        page_content=\"Leo DiCaprio gets lost in a dream within a dream within a dream within a ...\",\n",
     "        metadata={\"year\": 2010, \"genre\": \"thriller\", \"rating\": 8.2},\n",
     "    ),\n",
@@ -135,6 +139,10 @@
     "        page_content=\"Toys come alive and have a blast doing so\",\n",
     "        metadata={\"year\": 1995, \"genre\": \"animated\", \"rating\": 9.3},\n",
     "    ),\n",
+    "    Document(\n",
+    "        page_content=\"The toys come together to save their friend from a kid who doesn't know how to play with them\",\n",
+    "        metadata={\"year\": 1997, \"genre\": \"animated\", \"rating\": 9.1},\n",
+    "    ),\n",
     "]\n",
     "\n",
     "# Add data to the vector store, which automaticaly embeds the documents\n",
@@ -298,7 +306,7 @@
     "# This example specifies a query and composite filter (year > 1990 and year < 2005 and genre = action)\n",
     "retriever.invoke(\n",
     "    \"What's a movie after 1990 but before 2005 that's all about dinosaurs, \" +\n",
-    "    \"and preferably has a lot of action\"\n",
+    "    \"and preferably has the action genre\"\n",
     ")"
    ]
   },
@@ -367,7 +375,7 @@
     ")\n",
     "\n",
     "# Prompt the chain\n",
-    "question = \"What are two movies about dinosaurs after 1990?\" # year > 1990 and document limit of 2\n",
+    "question = \"What are two movies about toys after 1990?\" # year > 1990 and document limit of 2\n",
     "answer = chain.invoke(question)\n",
     "\n",
     "print(\"Question: \" + question)\n",