Update src/agents/tools.py

src/agents/tools.py

@@ -1,19 +1,73 @@
 from langchain.agents import Tool
 from langchain_community.utilities import GoogleSerperAPIWrapper
 from src.config.env import SERPER_API_KEY
+from langchain_google_community import GmailToolkit
+from langchain_google_community.gmail.utils import (
+    build_resource_service,
+    get_gmail_credentials,
+)
+import os
 
+# Gmail Send Helper
 
+def send_email_function(input_data):
+    if isinstance(input_data, str):
+        input_data = {"message": input_data}
+
+    input_data.setdefault("to", "default_advisor@gmail.com")
+    input_data.setdefault("subject", "Meeting Request")
+    input_data.setdefault("message", "Dear Advisor, the student would like to book a meeting with you.")
+
+    return gmail_send_tool(input_data)
+    
 def get_tools(rag_chain):
-    search = GoogleSerperAPIWrapper(serper_api_key=SERPER_API_KEY)
-    return [
+    tools = []
+
+    # === RAG Tool ===
+    tools.append(
         Tool(
             name="RAG",
             func=lambda x: rag_chain.invoke({"input": x}),
-            description="Useful when you're asked questions based on the UWF handbook using Retrieval Augmented Generation."
-        ),
+            description="Use this for questions about UWF, using the Student Handbook as context."
+        )
+    )
+
+    # === Google Search Tool ===
+    search = GoogleSerperAPIWrapper(serper_api_key=SERPER_API_KEY)
+    tools.append(
         Tool(
             name="Google Search",
             func=search.run,
-            description="Useful when information is not found in the internal knowledge base."
+            description="Fallback when RAG does not contain the information."
         )
-    ]
+    )
+
+    # === Gmail Tool ===
+    try:
+        credentials = get_gmail_credentials(
+            token_file="token.json",
+            scopes=["https://mail.google.com/"],
+            client_secrets_file="credentials.json",
+        )
+        api_resource = build_resource_service(credentials=credentials)
+        toolkit = GmailToolkit(api_resource=api_resource)
+        gmail_send_tool = next(
+            (tool for tool in toolkit.get_tools() if tool.__class__.__name__ == "GmailSendMessage"),
+            None
+        )
+
+        if gmail_send_tool:
+            tools.append(
+                Tool(
+                    name="Gmail",
+                    func=send_email_function,
+                    description="Send email to the advisor if the student requests it."
+                )
+            )
+        else:
+            print("[Warning] GmailSendMessage tool not found.")
+
+    except Exception as e:
+        print(f"[Warning] Gmail tool could not be initialized: {e}")
+
+    return tools