Update app.py

app.py

@@ -69,13 +69,32 @@ 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]
+
+tools = [
+    knowledge_base_tool,
+    google_search_tool,
+]
 
 # Create the prompt template
 prompt_message = """