Add specific reports selection

app.py

@@ -137,7 +137,7 @@ async def chat(query, history, audience, sources, reports, relevant_content_sour
     if reports is None or len(reports) == 0:
         reports = []
     
-    inputs = {"user_input": query,"audience": audience_prompt,"sources_input":sources, "relevant_content_sources" : relevant_content_sources, "search_only": search_only}
+    inputs = {"user_input": query,"audience": audience_prompt,"sources_input":sources, "relevant_content_sources" : relevant_content_sources, "search_only": search_only, "reports": reports}
     result = agent.astream_events(inputs,version = "v1") 
 
 
@@ -182,7 +182,7 @@ async def chat(query, history, audience, sources, reports, relevant_content_sour
                     if not hasattr(history[-1], 'metadata') or history[-1].metadata["title"] != event_description: # if a new step begins
                         history.append(ChatMessage(role="assistant", content = "", metadata={'title' :event_description}))
  
-                elif event["name"] != "transform_query" and event["event"] == "on_chat_model_stream" and node in ["answer_rag", "answer_search","answer_chitchat"]:# if streaming answer
+                elif event["name"] != "transform_query" and event["event"] == "on_chat_model_stream" and node in ["answer_rag", "answer_rag_no_docs","answer_search","answer_chitchat"]:# if streaming answer
                     history, start_streaming, answer_message_content = stream_answer(history, event, start_streaming, answer_message_content)
 
                 elif event["name"] in ["retrieve_graphs", "retrieve_graphs_ai"] and event["event"] == "on_chain_end":

climateqa/constants.py

@@ -1,4 +1,6 @@
 POSSIBLE_REPORTS = [
+    "IPBES IABWFH SPM",
+    "IPBES CBL SPM",
     "IPCC AR6 WGI SPM",
     "IPCC AR6 WGI FR",
     "IPCC AR6 WGI TS",

climateqa/engine/chains/retrieve_documents.py

@@ -219,6 +219,7 @@ async def retrieve_documents(state,config, vectorstore,reranker,llm,rerank_by_qu
         docs = state["documents"]
     else:
         docs = []
+        
     # Get the related_content from the state
     if "related_content" in state and state["related_content"] is not None:
         related_content = state["related_content"]
@@ -227,6 +228,8 @@ async def retrieve_documents(state,config, vectorstore,reranker,llm,rerank_by_qu
     
     search_figures = "Figures (IPCC/IPBES)" in state["relevant_content_sources"]
     search_only = state["search_only"]
+    
+    reports = state["reports"]
 
     # Get the current question
     current_question = state["remaining_questions"][0]
@@ -256,6 +259,7 @@ async def retrieve_documents(state,config, vectorstore,reranker,llm,rerank_by_qu
             k_images = k_images_by_question,
             threshold = 0.5,
             search_only = search_only,
+            reports = reports,
         )
 
     

climateqa/engine/graph.py

@@ -44,6 +44,7 @@ class GraphState(TypedDict):
     related_contents : Dict[str,Document]
     recommended_content : List[Document]
     search_only : bool = False
+    reports : List[str] = []
 
 def search(state): #TODO
     return state