fix(update_requirements): Update requirements

src/graph/state_vector_nodes.py

@@ -15,7 +15,7 @@ from langchain_community.utilities.semanticscholar import SemanticScholarAPIWrap
 from langchain_community.tools.tavily_search import TavilySearchResults
 import pandas as pd
 import torch.nn.functional as F
-
+import os
 torch.classes.__path__ = []
 class question_model:
     def __init__(self,loaded_tokenizer,loaded_model, llm, df_keys):
@@ -113,7 +113,7 @@ class question_model:
         return state
 
 class research_model:
-    def __init__(self,llm):
+    def __init__(self,llm,tavily_api_key):
         self.llm=llm
         self.local_analysis_file='src/graph/data_analyst_prompts.csv'
         self.tool_names=["direct_semantic_scholar_query", "direct_tavily_search" ]
@@ -123,6 +123,7 @@ class research_model:
         self.tools=[semantic_scholar_tool,self.direct_tavily_search]
         # Bind the tool to the LLM
         self.llm_with_tools = self.llm.bind_tools(self.tools)
+        os.environ['TAVILY_API_KEY']=tavily_api_key
         #self.tavily_api_key=tavily_api_key
 
     def direct_semantic_scholar_query(self,query: str):

src/streamlit_app.py

@@ -77,6 +77,8 @@ class LoadStreamlitUI:
                 st.warning("⚠️ Please enter a Gemini or Open AI ChatGPT API key to proceed. Don't have? refer : https://platform.openai.com/api-keys or https://aistudio.google.com/welcome?gclsrc=aw.ds&gad_source=1&gad_campaignid=21521909442 ")
             self.user_controls["selected_usecase"]=st.selectbox("Select Usecases",usecase_options)
             self.user_controls['UN SDG Country']= st.selectbox("Choose a country (start typing to search):",options=[""] + self.unsdg_countries)
+            if self.user_controls['selected_usecase']=='DeepRishSearch':
+                self.user_controls["TAVILY_API_KEY"] = st.session_state["TAVILY_API_KEY"]=st.text_input("Tavily API Key",type="password") 
         return self.user_controls
 def display_result_on_ui( state_graph,mode="Question Refining Mode:"):
 	if mode =="Question Refining Mode:":
@@ -133,7 +135,7 @@ if __name__=='__main__':
                                     loaded_tokenizer=loaded_tokenizer,
                                     loaded_model=loaded_model, llm=llm, df_keys=df_keys)
             builder=BuildGraphOptions(SmartQuestions)
-            ResearchModel=research_model(llm=SmartQuestions.genai_model)
+            ResearchModel=research_model(llm=SmartQuestions.genai_model, tavily_api_key=user_input['TAVILY_API_KEY'])
             graph=builder.build_research_graph(ResearchModel)
             with st.chat_message("assistant"):
                 intro="Hello, I am an assistant designed to help you learn about the 17 UN SDG goals listed here: https://sdgs.un.org/goals.\