local-deep-research 0.4.4__py3-none-any.whl → 0.5.2__py3-none-any.whl

local_deep_research/advanced_search_system/strategies/iterdrag_strategy.py

@@ -71,7 +71,7 @@ class IterDRAGStrategy(BaseSearchStrategy):
         try:
             # Format context for question generation
             context = f"""Current Query: {query}
-Current Date: {datetime.now().strftime('%Y-%m-%d')}
+Current Date: {datetime.now().strftime("%Y-%m-%d")}
 Past Questions: {self.questions_by_iteration}
 Current Knowledge: {current_knowledge}
 
@@ -80,7 +80,9 @@ Initial Search Results:
 
             # Generate sub-queries using the question generator
             return self.question_generator.generate_questions(
-                query, context, int(get_db_setting("search.questions_per_iteration"))
+                query,
+                context,
+                int(get_db_setting("search.questions_per_iteration")),
             )
         except Exception:
             logger.exception("Error generating sub-queries")
@@ -100,9 +102,7 @@ Initial Search Results:
 
         # Check if search engine is available
         if self.search is None:
-            error_msg = (
-                "Error: No search engine available. Please check your configuration."
-            )
+            error_msg = "Error: No search engine available. Please check your configuration."
             self._update_progress(
                 error_msg,
                 100,
@@ -140,7 +140,10 @@ Initial Search Results:
             self._update_progress(
                 f"Found {len(initial_results)} initial results",
                 15,
-                {"phase": "search_complete", "result_count": len(initial_results)},
+                {
+                    "phase": "search_complete",
+                    "result_count": len(initial_results),
+                },
             )
 
         # Extract and save links
@@ -161,7 +164,9 @@ Initial Search Results:
 
         # Store questions in repository and in self.questions_by_iteration
         self.questions_by_iteration = {0: sub_queries}
-        self.findings_repository.set_questions_by_iteration(self.questions_by_iteration)
+        self.findings_repository.set_questions_by_iteration(
+            self.questions_by_iteration
+        )
 
         if not sub_queries:
             logger.error("No sub-queries were generated to analyze.")
@@ -231,11 +236,15 @@ Initial Search Results:
                         }
                         findings.append(finding)
                         self.findings_repository.add_finding(sub_query, finding)
-                        self.findings_repository.add_documents(result["documents"])
+                        self.findings_repository.add_documents(
+                            result["documents"]
+                        )
 
                         # Add to current knowledge with space around +
                         current_knowledge = (
-                            current_knowledge + "\n\n\n New: \n" + result["content"]
+                            current_knowledge
+                            + "\n\n\n New: \n"
+                            + result["content"]
                         )
                 except Exception:
                     logger.exception("Error analyzing sub-query results:")
@@ -259,7 +268,9 @@ Initial Search Results:
 
             try:
                 # Extract finding contents for synthesis
-                finding_contents = [f["content"] for f in findings if "content" in f]
+                finding_contents = [
+                    f["content"] for f in findings if "content" in f
+                ]
 
                 # Synthesize findings into a final answer
                 final_answer = self.findings_repository.synthesize_findings(
@@ -270,7 +281,9 @@ Initial Search Results:
                 )
 
                 # Check if the synthesis failed with an error
-                if isinstance(final_answer, str) and final_answer.startswith("Error:"):
+                if isinstance(final_answer, str) and final_answer.startswith(
+                    "Error:"
+                ):
                     logger.error(f"Synthesis returned an error: {final_answer}")
 
                     # Extract error type for better handling
@@ -287,7 +300,9 @@ Initial Search Results:
                         error_type = "connection"
 
                     # Log with specific error type
-                    logger.error(f"Synthesis failed with error type: {error_type}")
+                    logger.error(
+                        f"Synthesis failed with error type: {error_type}"
+                    )
 
                     # Add error information to progress update
                     self._update_progress(
@@ -304,7 +319,9 @@ Initial Search Results:
                                 longest_finding = f["content"]
 
                     if longest_finding:
-                        logger.info("Using longest finding as fallback synthesis")
+                        logger.info(
+                            "Using longest finding as fallback synthesis"
+                        )
                         final_answer = f"""
 # Research Results (Fallback Mode)
 
@@ -316,7 +333,9 @@ This is a fallback response using the most detailed individual finding.
                         """
                     else:
                         # If we don't have any findings with content, use current_knowledge
-                        logger.info("Using current knowledge as fallback synthesis")
+                        logger.info(
+                            "Using current knowledge as fallback synthesis"
+                        )
                         final_answer = f"""
 # Research Results (Fallback Mode)
 
@@ -338,7 +357,9 @@ This is a fallback response using the accumulated knowledge.
                 findings.append(finding)
 
                 # Store the synthesized content
-                self.findings_repository.add_finding(query + "_synthesis", final_answer)
+                self.findings_repository.add_finding(
+                    query + "_synthesis", final_answer
+                )
 
                 # Update current knowledge with the synthesized version
                 current_knowledge = final_answer
@@ -366,7 +387,11 @@ This is a fallback response using the accumulated knowledge.
                     # Extract best content from findings
                     key_findings = []
                     for i, f in enumerate(findings):
-                        if isinstance(f, dict) and "content" in f and f.get("content"):
+                        if (
+                            isinstance(f, dict)
+                            and "content" in f
+                            and f.get("content")
+                        ):
                             # Only take the first 500 chars of each finding for the fallback
                             content_preview = f.get("content", "")[:500]
                             if content_preview:
@@ -405,10 +430,15 @@ Please try again with a different query or contact support.
                     """
 
         # Compress knowledge if needed
-        if get_db_setting("general.knowledge_accumulation", "ITERATION") == "ITERATION":
+        if (
+            get_db_setting("general.knowledge_accumulation", "ITERATION")
+            == "ITERATION"
+        ):
             try:
                 self._update_progress(
-                    "Compressing knowledge", 90, {"phase": "knowledge_compression"}
+                    "Compressing knowledge",
+                    90,
+                    {"phase": "knowledge_compression"},
                 )
                 current_knowledge = self.knowledge_generator.compress_knowledge(
                     current_knowledge, query, section_links
@@ -435,12 +465,16 @@ Please try again with a different query or contact support.
             )
 
             # Now format the findings with search questions and sources
-            formatted_findings = self.findings_repository.format_findings_to_text(
-                findings, final_answer
+            formatted_findings = (
+                self.findings_repository.format_findings_to_text(
+                    findings, final_answer
+                )
             )
         except Exception:
             logger.exception("Error formatting final findings")
-            formatted_findings = "Error: Could not format findings due to an error."
+            formatted_findings = (
+                "Error: Could not format findings due to an error."
+            )
 
         self._update_progress("Research complete", 100, {"phase": "complete"})