first commit

AIEC-RAG/retriver/langgraph_mermaid/Graph.txt

@@ -0,0 +1,37 @@
+---
+config:
+  flowchart:
+    curve: linear
+---
+graph TD;
+	__start__([<p>__start__</p>]):::first
+	query_complexity_check(query_complexity_check)
+	debug_mode_node(debug_mode_node)
+	simple_vector_retrieval(simple_vector_retrieval)
+	simple_answer_generation(simple_answer_generation)
+	query_decomposition(query_decomposition)
+	initial_retrieval(initial_retrieval)
+	sufficiency_check(sufficiency_check)
+	sub_query_generation(sub_query_generation)
+	parallel_retrieval(parallel_retrieval)
+	next_iteration(next_iteration)
+	final_answer(final_answer)
+	__end__([<p>__end__</p>]):::last
+	__start__ --> query_complexity_check;
+	debug_mode_node -. &nbsp;initial_retrieval&nbsp; .-> query_decomposition;
+	debug_mode_node -.-> simple_vector_retrieval;
+	initial_retrieval --> sufficiency_check;
+	next_iteration --> sufficiency_check;
+	parallel_retrieval --> next_iteration;
+	query_complexity_check --> debug_mode_node;
+	query_decomposition --> initial_retrieval;
+	simple_vector_retrieval --> simple_answer_generation;
+	sub_query_generation --> parallel_retrieval;
+	sufficiency_check -.-> final_answer;
+	sufficiency_check -.-> next_iteration;
+	sufficiency_check -. &nbsp;parallel_retrieval&nbsp; .-> sub_query_generation;
+	final_answer --> __end__;
+	simple_answer_generation --> __end__;
+	classDef default fill:#f2f0ff,line-height:1.2
+	classDef first fill-opacity:0
+	classDef last fill:#bfb6fc

AIEC-RAG/retriver/langgraph_mermaid/generate_mermaid.py

@@ -0,0 +1,56 @@
+"""
+生成LangGraph Mermaid图表的脚本
+使用LangGraph内置的draw_mermaid()方法
+"""
+
+import os
+import sys
+
+# 添加项目根目录到Python路径
+# 从当前文件位置向上找到项目根目录 (Retriver/)
+project_root = os.path.dirname(os.path.dirname(os.path.dirname(__file__)))
+sys.path.insert(0, project_root)
+
+def generate_mermaid_graph():
+    """使用LangGraph内置方法生成Mermaid图表"""
+    
+    try:
+        # 导入必要的模块
+        from retriver.langgraph.iterative_retriever import IterativeRetriever
+        
+        print("初始化检索器...")
+        # 创建一个最小化配置的迭代检索器实例
+        retriever = IterativeRetriever(
+            keyword="temp",
+            top_k=2,
+            max_iterations=3,
+            max_parallel_retrievals=2
+        )
+        
+        print("生成LangGraph Mermaid图表...")
+        # 使用LangGraph内置的draw_mermaid()方法
+        mermaid_graph = retriever.workflow.get_graph().draw_mermaid()
+        
+        # 保存到Graph.txt文件
+        output_file = os.path.join(os.path.dirname(__file__), "Graph.txt")
+        with open(output_file, 'w', encoding='utf-8') as f:
+            f.write(mermaid_graph)
+        
+        print(f"LangGraph Mermaid图表已保存到: {output_file}")
+        print("\n图表内容:")
+        print(mermaid_graph)
+        
+        return mermaid_graph
+        
+    except ImportError as e:
+        print(f"模块导入失败: {e}")
+        print("请确保在正确的Python环境中运行此脚本")
+        return None
+    except Exception as e:
+        print(f"生成Mermaid图表失败: {e}")
+        print("可能是因为缺少必要的依赖或配置")
+        return None
+
+
+if __name__ == "__main__":
+    generate_mermaid_graph()