remove unused code

climateqa/engine/chains/retrieve_documents.py

@@ -258,7 +258,6 @@ async def retrieve_documents(state,config, vectorstore,reranker,llm,rerank_by_qu
     # Add to the list of docs
     docs.extend(docs_question)
     related_content.extend(images_question)
-    # related_content=[]
     new_state = {"documents":docs, "related_contents": related_content,"remaining_questions":remaining_questions}
     return new_state
     

climateqa/engine/graph.py

@@ -43,7 +43,6 @@ class GraphState(TypedDict):
     documents: List[Document]
     related_contents : Dict[str,Document]
     recommended_content : List[Document]
-    # graphs_returned: Dict[str,str]
 
 def search(state): #TODO
     return state
@@ -108,16 +107,10 @@ def make_graph_agent(llm, vectorstore_ipcc, vectorstore_graphs, reranker, thresh
     workflow.add_node("answer_search", answer_search)
     workflow.add_node("transform_query", transform_query)
     workflow.add_node("translate_query", translate_query)
-    # workflow.add_node("transform_query_ai", transform_query)
-    # workflow.add_node("translate_query_ai", translate_query)
     workflow.add_node("answer_chitchat", answer_chitchat)
     workflow.add_node("chitchat_categorize_intent", chitchat_categorize_intent)
-    # workflow.add_node("answer_ai_impact", answer_ai_impact)
     workflow.add_node("retrieve_graphs", retrieve_graphs)
     workflow.add_node("retrieve_graphs_chitchat", retrieve_graphs)
-    # workflow.add_node("retrieve_graphs_ai", retrieve_graphs)
-    # workflow.add_node("answer_rag_graph", answer_rag_graph)
-    # workflow.add_node("answer_rag_graph_ai", answer_rag_graph)
     workflow.add_node("retrieve_documents", retrieve_documents)
     workflow.add_node("answer_rag", answer_rag)
     workflow.add_node("answer_rag_no_docs", answer_rag_no_docs)
@@ -161,27 +154,14 @@ def make_graph_agent(llm, vectorstore_ipcc, vectorstore_graphs, reranker, thresh
     )
 
     # Define the edges
-    # workflow.add_edge("set_defaults", "categorize_intent")
     workflow.add_edge("translate_query", "transform_query")
-    # workflow.add_edge("transform_query", "retrieve_graphs")
     workflow.add_edge("transform_query", "retrieve_documents")
 
-    # workflow.add_edge("retrieve_graphs", "answer_rag_graph")
     workflow.add_edge("retrieve_graphs", END)
-    # workflow.add_edge("retrieve_graphs", "retrieve_documents")
-    # workflow.add_edge("answer_rag_graph", "retrieve_documents")
     workflow.add_edge("answer_rag", END)
     workflow.add_edge("answer_rag_no_docs", END)
     workflow.add_edge("answer_chitchat", "chitchat_categorize_intent")
-    # workflow.add_edge("answer_chitchat", END)
-    # workflow.add_edge("answer_ai_impact", END)
-    # workflow.add_edge("retrieve_graphs_chitchat", END)
-    # workflow.add_edge("answer_ai_impact", "translate_query_ai")
-    # workflow.add_edge("translate_query_ai", "transform_query_ai")
-    # workflow.add_edge("transform_query_ai", "retrieve_graphs_ai")
-    # workflow.add_edge("retrieve_graphs_ai", "answer_rag_graph_ai")
-    # workflow.add_edge("answer_rag_graph_ai", END)
-    # workflow.add_edge("retrieve_graphs_ai", END)
+
 
     # Compile
     app = workflow.compile()