raw func

app.py

@@ -25,6 +25,7 @@ from threading import Lock
 from langchain.llms import OpenAI
 from langchain.chains.conversation.memory import ConversationBufferMemory
 from langchain.agents import tool, load_tools, initialize_agent
+from langchain.agents import Tool
 from langchain import ConversationChain, LLMChain
 import whisper
 import warnings
@@ -89,7 +90,7 @@ def get_text(qry):
     return prediction_text
 
 
-@tool("mysearch", return_direct=True)
+@tool
 def mysearch(query: str) -> str:
     """Query our own datasets.
     """
@@ -163,7 +164,7 @@ tmdb_bearer_token = os.environ["TMDB_BEARER_TOKEN"]
 TOOLS_LIST = ['serpapi', 'wolfram-alpha', 'pal-math', 'pal-colored-objects', 'news-api', 'tmdb-api',
               'open-meteo-api']  # 'google-search'
 # TOOLS_DEFAULT_LIST = ['mysearch', 'serpapi', 'pal-math']
-TOOLS_DEFAULT_LIST = ['mysearch']
+TOOLS_DEFAULT_LIST = [mysearch]
 BUG_FOUND_MSG = "Congratulations, you've found a bug in this application!"
 AUTH_ERR_MSG = "Please paste your OpenAI key from openai.com to use this application. It is not necessary to hit a button or key after pasting it."
 MAX_TOKENS = 512