Update agent.py

agent.py

@@ -43,9 +43,12 @@ from io import StringIO
 from transformers import BertTokenizer, BertModel
 import torch
 import torch.nn.functional as F
-from langchain.tools import Tool
+#from langchain.tools import Tool
 from langchain.agents import initialize_agent, AgentType
 from langchain_community.chat_models import ChatOpenAI
+#from langchain_openai import ChatOpenAI
+from langchain_community.tools import Tool
+
 import time
 
 load_dotenv()
@@ -444,10 +447,10 @@ retriever = vector_store.as_retriever()
 # -------------------------------
 # Step 6: Create LangChain Tools
 # -------------------------------
-wiki_tool = WikipediaAPIWrapper()  # If it's a proper LangChain tool
 calc_tool = calculator
 file_tool = analyze_attachment
 web_tool = web_search
+wiki_tool = wiki_search
 arvix_tool = arvix_search
 youtube_tool = get_youtube_transcript
 video_tool = extract_video_id