Update app.py

app.py

@@ -104,10 +104,25 @@ def rag_response(query):
 # Define tools using LangChain's `tool` decorator
 @tool
 def knowledge_base_tool(query: str):
+    """
+    Tool function to query the knowledge base and retrieve a response.
+    Args:
+        query (str): The query to search the knowledge base.
+    Returns:
+        str: The response retrieved from the knowledge base.
+    """
     return rag_response(query)
 
 @tool
 def google_search_tool(query: str):
+    """
+    Tool function to perform a Google search using the SERPER API.
+    Args:
+        query (str): The query to search on Google.
+    Returns:
+        list: List of snippets extracted from search results.
+    """
+    
     return google_search(query)
 
 tools = [knowledge_base_tool, google_search_tool]