minor clean

climateqa/engine/graph.py

@@ -39,14 +39,14 @@ class GraphState(TypedDict):
     n_questions : int
     answer: str
     audience: str = "experts"
-    sources_input: List[str] = ["IPCC","IPBES"]
+    sources_input: List[str] = ["IPCC","IPBES"] # Deprecated -> used only graphs that can only be OWID
     relevant_content_sources_selection: List[str] = ["Figures (IPCC/IPBES)"]
     sources_auto: bool = True
     min_year: int = 1960
     max_year: int = None
     documents: Annotated[List[Document], operator.add]
-    related_contents : Annotated[List[Document], operator.add]
-    recommended_content : List[Document]
+    related_contents : Annotated[List[Document], operator.add] # Images
+    recommended_content : List[Document] # OWID Graphs  # TODO merge with related_contents
     search_only : bool = False
     reports : List[str] = []
 
@@ -95,20 +95,10 @@ def route_based_on_relevant_docs(state,threshold_docs=0.2):
 def route_continue_retrieve_documents(state):
     index_question_ipx = [i for i, x in enumerate(state["questions_list"]) if x["source_type"] == "IPx"]
     questions_ipx_finished = all(elem in state["handled_questions_index"] for elem in index_question_ipx)
-    # if questions_ipx_finished and state["search_only"]:
-    #     return END
     if questions_ipx_finished:
         return "end_retrieve_IPx_documents"  
     else:
         return "retrieve_documents"
-
-    
-    # if state["n_questions"]["IPx"] == len(state["handled_questions_index"]) and state["search_only"] :
-    #     return END
-    # elif state["n_questions"]["IPx"] == len(state["handled_questions_index"]):
-    #     return "answer_search"
-    # else :
-    #     return "retrieve_documents"
     
 def route_continue_retrieve_local_documents(state):
     index_question_poc = [i for i, x in enumerate(state["questions_list"]) if x["source_type"] == "POC"]
@@ -120,20 +110,6 @@ def route_continue_retrieve_local_documents(state):
     else:
         return "retrieve_local_data"
     
-    # if state["n_questions"]["POC"] == len(state["handled_questions_index"]) and state["search_only"] :
-    #     return END
-    # elif state["n_questions"]["POC"] == len(state["handled_questions_index"]):
-    #     return "answer_search"
-    # else :
-    #     return "retrieve_local_data"
-    
-    # if len(state["remaining_questions"]) == 0 and state["search_only"] :
-        # return END
-    # elif len(state["remaining_questions"]) > 0:
-    #     return "retrieve_documents"
-    # else:
-    #     return "answer_search"
-    
 def route_retrieve_documents(state):
     sources_to_retrieve = []