Update main.py

main.py

@@ -1,7 +1,7 @@
 import os
 from langchain.memory import ConversationBufferMemory
 from langchain.utilities import GoogleSearchAPIWrapper
-from langchain.agents import initialize_agent, Tool
+from langchain.agents import AgentType, initialize_agent, Tool
 from lang import G4F
 from fastapi import FastAPI, Request
 from pydantic import BaseModel