task: get better results from RAG

src/chains/design_rag.py

@@ -86,31 +86,44 @@ class DesignRAG:
             print(f"Error creating vector store: {str(e)}")
             raise
     
-    async def query_similar_designs(self, requirements: Dict, num_examples: int = 5) -> str:
-        """Find similar designs based on requirements"""
+    async def query_similar_designs(self, conversation_history: List[str], num_examples: int = 5) -> str:
+        """Find similar designs based on conversation history
+        
+        Args:
+            conversation_history: List of conversation messages
+            num_examples: Number of examples to retrieve
+        """
         # Create query generation prompt
-        query_prompt = ChatPromptTemplate.from_template("""Given these design requirements:
+        query_prompt = ChatPromptTemplate.from_template("""Based on this conversation history:
 
-        {requirements}
+        {conversation}
 
-        Create a search query that will find similar designs. Focus on:
-        1. Visual style and aesthetics
-        2. Design categories and themes
-        3. Key visual characteristics
-        4. Overall mood and impact
+        Extract the key design requirements and create a search query to find similar designs.
+        Focus on:
+        1. Visual style and aesthetics mentioned
+        2. Design categories and themes discussed
+        3. Key visual characteristics requested
+        4. Overall mood and impact desired
+        5. Any specific preferences or constraints
 
-        Return only the search query text, no additional explanation.""")
+        Return only the search query text, no additional explanation or analysis.""")
+        
+        # Format conversation history
+        conversation_text = "\n".join([
+            f"{'User' if i % 2 == 0 else 'Assistant'}: {msg}"
+            for i, msg in enumerate(conversation_history)
+        ])
         
         # Generate optimized search query
         query_response = await self.llm.ainvoke(
             query_prompt.format(
-                requirements=json.dumps(requirements, indent=2)
+                conversation=conversation_text
             )
         )
         
         print(f"Generated query: {query_response.content}")
         
-        # Get similar documents using the correct method name
+        # Get similar documents
         docs = self.retriever.get_relevant_documents(
             query_response.content, 
             k=num_examples

src/graph.ipynb

@@ -22,7 +22,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": 2,
    "metadata": {},
    "outputs": [
     {
@@ -31,59 +31,19 @@
      "text": [
       "Loaded 82 design documents\n",
       "Testing RAG retriever with requirements:\n",
-      "----------------------------------------\n",
-      "style_description: Modern minimalist design with clean lines\n",
-      "key_elements: ['clean typography', 'white space', 'grid layout']\n",
-      "color_scheme: Monochromatic with blue accents\n",
-      "layout_preferences: Grid-based with clear hierarchy\n",
-      "mood: Professional and sophisticated\n",
-      "\n",
-      "Retrieved Designs:\n",
-      "----------------------------------------\n",
-      "Generated query: Modern minimalist design with clean lines AND clean typography AND white space AND grid layout AND Monochromatic with blue accents AND Grid-based layout AND Professional AND sophisticated mood\n"
+      "----------------------------------------\n"
      ]
     },
     {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "/Users/owner/Desktop/Projects/ai-maker-space/code/ImagineUI/src/chains/design_rag.py:114: LangChainDeprecationWarning: The method `BaseRetriever.get_relevant_documents` was deprecated in langchain-core 0.1.46 and will be removed in 1.0. Use :meth:`~invoke` instead.\n",
-      "  docs = self.retriever.get_relevant_documents(\n"
-     ]
-    },
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Design 003:\n",
-      "Design 003:\n",
-      "Description: This design features a serene and calming aesthetic, employing a monochromatic blue-gray color palette that evokes a sense of tranquility. The layout is structured and organized, with a notable emphasis on clean typography and spaciousness. Each section is clearly delineated, contributing to an overall coherent and harmonious visual experience, while the soft textures in the background enhance the serene mood.\n",
-      "Categories: minimalism, typography, print design, web interface, aesthetic design\n",
-      "Visual Characteristics: monochromatic palette, serene mood, structured layout, clean typography, spaciousness\n",
-      "\n",
-      "Design 020:\n",
-      "Design 020:\n",
-      "Description: The design features a serene and professional layout with a muted color palette, emphasizing clean lines and minimalist aesthetics. It balances a centered, text-focused main content area with a sidebar offering functional navigation and supplementary information. The harmonious integration of serene imagery and subtle blue accents promotes a calm and reflective visual experience.\n",
-      "Categories: minimalist design, web aesthetics, informational layout, professional tone, navigation focus\n",
-      "Visual Characteristics: muted color palette, serene imagery, clean typography, balanced layout, subtle blue accents\n",
-      "\n",
-      "Design 217:\n",
-      "Design 217:\n",
-      "Description: The design embraces a minimalist and modern aesthetic with a refreshing color palette of soft blues, creams, and reds, highlighted by a structured layout that organizes content effectively into distinct sections. The typography is bold and clear, drawing attention to key headings while maintaining readability. The use of contrasting colors for different sections adds a dynamic feel to the design, and the sidebar provides easy navigation, enhancing user experience.\n",
-      "Categories: Modern, Minimalist, Educational, Structured, Interactive\n",
-      "Visual Characteristics: Bold typography, Contrasting colors, Organized layout, Clean lines, Effective use of white space\n",
-      "\n",
-      "Design 004:\n",
-      "Design 004:\n",
-      "Description: The design features a structured layout with a calming blue and white color scheme that conveys clarity and modernity. The large, fluid background image adds a touch of elegance, while the clear typography and organized sections contribute to the aesthetic appeal. Highlighted text links in green and orange draw attention, creating a dynamic yet harmonious look.\n",
-      "Categories: Modern, Web Design, Instructional, Clean, Professional\n",
-      "Visual Characteristics: Blue and White Palette, Fluid Background Image, Clear Typography, Highlighted Text Links, Structured Layout\n",
-      "\n",
-      "Design 203:\n",
-      "Design 203:\n",
-      "Description: This design offers a serene and minimalist aesthetic with a focus on clarity and enlightenment. Featuring a calming blue color palette highlighted by a water droplet image, it prioritizes clean typography and well-structured sections. The layout is vertical and spaced, creating a smooth flow from one section to the next, while the consistent use of blue tones reinforces a peaceful mood.\n",
-      "Categories: Minimalism, Typography, Calm Aesthetic, Water Theme, Serenity\n",
-      "Visual Characteristics: Blue Color Palette, Vertical Layout, Calming Imagery, Structured Sections, Clean Typography\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'"
      ]
     }
    ],
@@ -96,19 +56,12 @@
     "#design_retriever = DesignRetrieverTool(rag=design_rag)\n",
     "\n",
     "test_requirements = {\n",
-    "    \"style_description\": \"Modern minimalist design with clean lines\",\n",
-    "    \"key_elements\": [\"clean typography\", \"white space\", \"grid layout\"],\n",
-    "    \"color_scheme\": \"Monochromatic with blue accents\",\n",
-    "    \"layout_preferences\": \"Grid-based with clear hierarchy\",\n",
-    "    \"mood\": \"Professional and sophisticated\"\n",
+    "    \"I'm looking for something modern and space-themed, designed to appeal to adventurers, young people, and science enthusiasts\"\n",
     "}\n",
     "\n",
     "# Test the retriever\n",
     "async def test_rag():\n",
     "    print(\"Testing RAG retriever with requirements:\")\n",
-    "    print(\"----------------------------------------\")\n",
-    "    for key, value in test_requirements.items():\n",
-    "        print(f\"{key}: {value}\")\n",
     "    print(\"\\nRetrieved Designs:\")\n",
     "    print(\"----------------------------------------\")\n",
     "    \n",