task: changes rag test

src/graph.ipynb

@@ -31,19 +31,33 @@
      "text": [
       "Loaded 82 design documents\n",
       "Testing RAG retriever with requirements:\n",
-      "----------------------------------------\n"
-     ]
-    },
-    {
-     "ename": "AttributeError",
-     "evalue": "'set' object has no attribute 'items'",
-     "output_type": "error",
-     "traceback": [
-      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
-      "\u001b[0;31mAttributeError\u001b[0m                            Traceback (most recent call last)",
-      "Cell \u001b[0;32mIn[2], line 25\u001b[0m\n\u001b[1;32m     22\u001b[0m     \u001b[38;5;28mprint\u001b[39m(results)\n\u001b[1;32m     24\u001b[0m \u001b[38;5;66;03m# Run the test\u001b[39;00m\n\u001b[0;32m---> 25\u001b[0m \u001b[38;5;28;01mawait\u001b[39;00m test_rag()\n",
-      "Cell \u001b[0;32mIn[2], line 16\u001b[0m, in \u001b[0;36mtest_rag\u001b[0;34m()\u001b[0m\n\u001b[1;32m     14\u001b[0m \u001b[38;5;28mprint\u001b[39m(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mTesting RAG retriever with requirements:\u001b[39m\u001b[38;5;124m\"\u001b[39m)\n\u001b[1;32m     15\u001b[0m \u001b[38;5;28mprint\u001b[39m(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124m----------------------------------------\u001b[39m\u001b[38;5;124m\"\u001b[39m)\n\u001b[0;32m---> 16\u001b[0m \u001b[38;5;28;01mfor\u001b[39;00m key, value \u001b[38;5;129;01min\u001b[39;00m \u001b[43mtest_requirements\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mitems\u001b[49m():\n\u001b[1;32m     17\u001b[0m     \u001b[38;5;28mprint\u001b[39m(\u001b[38;5;124mf\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;132;01m{\u001b[39;00mkey\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m: \u001b[39m\u001b[38;5;132;01m{\u001b[39;00mvalue\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m\"\u001b[39m)\n\u001b[1;32m     18\u001b[0m \u001b[38;5;28mprint\u001b[39m(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;124mRetrieved Designs:\u001b[39m\u001b[38;5;124m\"\u001b[39m)\n",
-      "\u001b[0;31mAttributeError\u001b[0m: 'set' object has no attribute 'items'"
+      "\n",
+      "Retrieved Designs:\n",
+      "----------------------------------------\n",
+      "Generated query: \"vintage classic easy to use grandmother love design\"\n",
+      "Design 180:\n",
+      "Description: This design employs a vintage newspaper aesthetic with a classic serif typography that evokes an old-world charm, utilizing sepia-toned paper backgrounds to enhance its nostalgic feel. The layout is text-heavy with a deliberate obfuscation, reflecting a layered collage effect. Its balanced placement keeps the focus central, inviting closer inspection and interaction.\n",
+      "Categories: Vintage, Nostalgic, Typography, Collage, Editorial\n",
+      "Visual Characteristics: Sepia tone, Serif typography, Textured background, Layered elements, Central focus\n",
+      "URL: https://csszengarden.com/180\n",
+      "\n",
+      "Design 182:\n",
+      "Description: The design creatively utilizes a retro theme with vinyl records as the prominent visual element to evoke a sense of nostalgia and classic style, complemented by a muted green color palette that brings harmony and balance. Handwritten and vintage-style typography enhance the retro aesthetic, while background illustrations and decorative elements like stars add whimsy and depth to the composition.\n",
+      "Categories: Retro, Nostalgic, Music-themed, Decorative, Vintage\n",
+      "Visual Characteristics: Vinyl Records, Muted Green Palette, Handwritten Typography, Background Illustrations, Decorative Elements\n",
+      "URL: https://csszengarden.com/182\n",
+      "\n",
+      "Design 194:\n",
+      "Description: This design exudes a minimalist elegance with a muted, earthy color palette and a clean layout, embodying a sense of calm and sophistication. The subtle use of textures and classic serif typography enhances the refined aesthetic, while the centered alignment and generous spacing contribute to a relaxed readability. The incorporation of a delicate floral illustration adds a touch of organic charm, making the design feel both timeless and inviting.\n",
+      "Categories: Minimalism, Elegant, Organic, Sophisticated, Classic\n",
+      "Visual Characteristics: Muted Color Palette, Serif Typography, Centered Layout, Generous Spacing, Floral Illustration\n",
+      "URL: https://csszengarden.com/194\n",
+      "\n",
+      "Design 212:\n",
+      "Description: The design features a retro aesthetic using a muted color palette of browns and creams, creating a nostalgic and vintage feel. The asymmetrical layout and bold typography contribute to the visual hierarchy, guiding the viewer through the content effortlessly. Illustrations with a mid-century modern style add character, merging traditional design elements with contemporary functionality.\n",
+      "Categories: Retro, Typography, Illustration, Vintage Style, Educational\n",
+      "Visual Characteristics: Muted Color Palette, Asymmetrical Layout, Bold Typography, Retro Illustrations, Functional Design\n",
+      "URL: https://csszengarden.com/212\n"
      ]
     }
    ],
@@ -56,8 +70,8 @@
     "#design_retriever = DesignRetrieverTool(rag=design_rag)\n",
     "\n",
     "test_requirements = {\n",
-    "    \"I'm looking for something modern and space-themed, designed to appeal to adventurers, young people, and science enthusiasts\"\n",
-    "}\n",
+    "    \"I want a design that is vintage and classic, something easy to use that a grandmother would love\"\n",
+    "    }\n",
     "\n",
     "# Test the retriever\n",
     "async def test_rag():\n",
@@ -65,7 +79,7 @@
     "    print(\"\\nRetrieved Designs:\")\n",
     "    print(\"----------------------------------------\")\n",
     "    \n",
-    "    results = await design_rag.query_similar_designs(test_requirements)\n",
+    "    results = await design_rag.query_similar_designs(test_requirements, 2)\n",
     "    print(results)\n",
     "\n",
     "# Run the test\n",