fix: Error processing query: 'str' object does not support item assignment

Author: jasperan

agentic_rag/gradio_app.py

@@ -198,10 +198,31 @@ def chat(message: str, history: List[List[str]], agent_type: str, use_cot: bool,
             # Add final formatted response to history
             history.append([message, formatted_response])
         else:
+            # For standard response (no CoT)
             formatted_response = response["answer"]
             print("\nStandard Response:")
             print("-" * 50)
             print(formatted_response)
+            
+            # Add sources if available
+            if response.get("context"):
+                print("\nSources Used:")
+                print("-" * 50)
+                sources_text = "\n\n📚 Sources used:\n"
+                formatted_response += sources_text
+                print(sources_text)
+                
+                for ctx in response["context"]:
+                    source = ctx["metadata"].get("source", "Unknown")
+                    if "page_numbers" in ctx["metadata"]:
+                        pages = ctx["metadata"].get("page_numbers", [])
+                        source_line = f"- {source} (pages: {pages})\n"
+                    else:
+                        file_path = ctx["metadata"].get("file_path", "Unknown")
+                        source_line = f"- {source} (file: {file_path})\n"
+                    formatted_response += source_line
+                    print(source_line)
+            
             history.append([message, formatted_response])
 
         print("\n" + "="*50)

agentic_rag/local_rag_agent.py

@@ -431,12 +431,12 @@ def _generate_response(self, query: str, context: List[Dict[str, Any]]) -> Dict[
 Answer:"""
 
         prompt = template.format(context=context_str, query=query)
-        response = self._generate_text(prompt)
+        response_text = self._generate_text(prompt)
 
         # Add sources to response if available
+        sources = {}
         if context:
             # Group sources by document
-            sources = {}
             for item in context:
                 source = item['metadata'].get('source', 'Unknown')
                 if source not in sources:
@@ -457,12 +457,11 @@ def _generate_response(self, query: str, context: List[Dict[str, Any]]) -> Dict[
                     print(f"Document: {source} (pages: {pages})")
                 else:  # Code with file path
                     print(f"Code file: {source}")
-            
-            response['sources'] = sources
 
         return {
-            "answer": response,
-            "context": context
+            "answer": response_text,
+            "context": context,
+            "sources": sources
         }
 
     def _generate_general_response(self, query: str) -> Dict[str, Any]: