bugfixs

- dummy search and answer_search
- langchain_community packages
- typing class
- fix classes import

app.py

@@ -1,7 +1,7 @@
 from climateqa.engine.embeddings import get_embeddings_function
 embeddings_function = get_embeddings_function()
 
-from climateqa.papers.openalex import OpenAlex
+from climateqa.knowledge.openalex import OpenAlex
 from sentence_transformers import CrossEncoder
 
 # reranker = CrossEncoder("mixedbread-ai/mxbai-rerank-xsmall-v1")
@@ -30,7 +30,7 @@ from utils import create_user_id
 # ClimateQ&A imports
 from climateqa.engine.llm import get_llm
 from climateqa.engine.vectorstore import get_pinecone_vectorstore
-from climateqa.engine.retriever import ClimateQARetriever
+from climateqa.knowledge.retriever import ClimateQARetriever
 from climateqa.engine.reranker import get_reranker
 from climateqa.engine.embeddings import get_embeddings_function
 from climateqa.engine.chains.prompts import audience_prompts
@@ -113,8 +113,8 @@ async def chat(query,history,audience,sources,reports):
     if len(sources) == 0:
         sources = ["IPCC"]
 
-    if len(reports) == 0:
-        reports = []
+    # if len(reports) == 0: # TODO 
+    reports = []
     
     inputs = {"user_input": query,"audience": audience_prompt,"sources":sources}
     result = agent.astream_events(inputs,version = "v1") #{"callbacks":[MyCustomAsyncHandler()]})
@@ -141,7 +141,6 @@ async def chat(query,history,audience,sources,reports):
 
     try:
         async for event in result:
-
             if event["event"] == "on_chat_model_stream":
                 if start_streaming == False:
                     start_streaming = True

climateqa/engine/graph.py

@@ -4,7 +4,7 @@ from contextlib import contextmanager
 
 from langchain.schema import Document
 from langgraph.graph import END, StateGraph
-from langchain_core.runnables.graph import CurveStyle, NodeColors, MermaidDrawMethod
+from langchain_core.runnables.graph import CurveStyle, MermaidDrawMethod
 
 from typing_extensions import TypedDict
 from typing import List
@@ -37,11 +37,11 @@ class GraphState(TypedDict):
     max_year: int = None
     documents: List[Document]
 
-def search(state):
-    return {}
+def search(state): #TODO
+    return state
 
-def answer_search(state):
-    return {}
+def answer_search(state):#TODO
+    return state
 
 def route_intent(state):
     intent = state["intent"]

climateqa/knowledge/openalex.py

@@ -163,7 +163,7 @@ class OpenAlex():
 class OpenAlexRetriever(BaseRetriever):
     min_year:int = 1960
     max_year:int = None
-    k = 100
+    k:int = 100
 
     def _get_relevant_documents(
         self, query: str, *, run_manager: CallbackManagerForRetrieverRun

requirements.txt

@@ -15,3 +15,4 @@ flashrank==0.2.5
 rerankers==0.3.0
 torch==2.3.0
 nvidia-cudnn-cu12==8.9.2.26
+langchain-community==0.2