add folder list tool

memory_chatbot.py

@@ -2,7 +2,7 @@ import json
 from typing import TypedDict, Annotated
 
 from dotenv import load_dotenv
-from langchain_core.messages import BaseMessage, HumanMessage,ToolMessage,AIMessage
+from langchain_core.messages import BaseMessage, HumanMessage,ToolMessage,AIMessage,SystemMessage
 from langchain_groq import ChatGroq
 
 from langgraph.graph import START, END, StateGraph
@@ -16,12 +16,12 @@ from langgraph.store.memory import InMemoryStore
 
 # Agent
 from langchain.agents import create_agent
-from tool import tool_tavily,time_date,calculator,python_exec,get_weather,wikipedia_search,scrape_website,read_file,format_json,generate_sql,system_info,save_user_preference
+from tool import tool_tavily,time_date,calculator,python_exec,get_weather,wikipedia_search,scrape_website,read_file,format_json,generate_sql,system_info,save_user_preference,list_folders
 
 
 load_dotenv()
 
-tools = [tool_tavily,time_date,calculator,python_exec,get_weather,wikipedia_search,scrape_website,read_file,format_json,generate_sql,system_info,save_user_preference]
+tools = [tool_tavily,time_date,calculator,python_exec,get_weather,wikipedia_search,scrape_website,read_file,format_json,generate_sql,system_info,save_user_preference,list_folders]
 
 # -----------------------------
 # Memory Systems

tool.py

@@ -7,6 +7,7 @@ import json
 import requests
 import wikipedia
 from bs4 import BeautifulSoup
+import os
 
 from dotenv import load_dotenv
 
@@ -105,4 +106,20 @@ def system_info() -> str:
 def save_user_preference(text: str) -> str:
     """Save user preference or important info"""
     # later connect to store.put()
-    return f"Saved memory: {text}"
+    return f"Saved memory: {text}"
+
+
+@tool
+def list_folders(path: str = ".") -> str:
+    """List all folders in the given directory path"""
+    try:
+        items = os.listdir(path)
+        folders = [f for f in items if os.path.isdir(os.path.join(path, f))]
+
+        if not folders:
+            return "No folders found."
+
+        return "\n".join(folders)
+
+    except Exception as e:
+        return f"Error: {str(e)}"