DrishtiSharma commited on
Commit
c0f8ffc
·
verified ·
1 Parent(s): 1b2c3c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -16,9 +16,9 @@ from langchain.output_parsers.openai_functions import JsonOutputFunctionsParser
16
  from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
17
  from langgraph.graph import StateGraph, END
18
  from typing import Annotated, Sequence, TypedDict
19
- from langchain_core.tools import tool
20
  import functools
21
  import operator
 
22
 
23
  # Load environment variables
24
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
@@ -47,6 +47,7 @@ def agent_node(state, agent, name):
47
 
48
  @tool
49
  def RAG(state):
 
50
  st.session_state.outputs.append('-> Calling RAG ->')
51
  question = state
52
  template = """Answer the question based only on the following context:\n{context}\nQuestion: {question}"""
 
16
  from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
17
  from langgraph.graph import StateGraph, END
18
  from typing import Annotated, Sequence, TypedDict
 
19
  import functools
20
  import operator
21
+ from langchain_core.tools import tool
22
 
23
  # Load environment variables
24
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 
47
 
48
  @tool
49
  def RAG(state):
50
+ """Use this tool to execute RAG. If the question is related to Japan or Sports, this tool retrieves the results."""
51
  st.session_state.outputs.append('-> Calling RAG ->')
52
  question = state
53
  template = """Answer the question based only on the following context:\n{context}\nQuestion: {question}"""