Root

9269ed4f · m0hammadjaan · 06a33cc8 · 9269ed4f
Commit 9269ed4f authored Nov 03, 2023 by m0hammadjaan
Show whitespace changes
Inline Side-by-side

Showing with 1 additions and 6 deletions

2_Chatbot_Agent.py pages/2_Chatbot_Agent.py +1 -6

No files found.
--- a/pages/2_Chatbot_Agent.py
+++ b/pages/2_Chatbot_Agent.py
@@ -5,6 +5,7 @@ from langchain.chat_models import ChatOpenAI
 from langchain.tools import DuckDuckGoSearchRun
 from langchain.agents import initialize_agent, Tool
 from langchain.callbacks import StreamlitCallbackHandler
+
 from langchain.utilities import ArxivAPIWrapper


@@ -22,17 +23,11 @@ class ChatbotTools:
    def setup_agent(self):
        # Define tool
        ddg_search = DuckDuckGoSearchRun()
-        arxiv = ArxivAPIWrapper()
        tools = [
            Tool(
                name="DuckDuckGoSearch",
                func=ddg_search.run,
                description="Useful for when you need to answer questions about current events. You should ask targeted questions",
-            ),
-            Tool(
-                name="Arxiv",
-                func=arxiv.run,
-                description="Useful for when you need to answer questions about research papers, scientific articles, and preprints etc",
            )
        ]