helper is a tool library

helper.py

@@ -0,0 +1,94 @@
+from langchain_experimental.utilities import PythonREPL
+from langchain.tools import Tool
+from langchain_community.tools import TavilySearchResults
+
+# For newer LangChain versions, sometimes it's directly from langchain.tools.python
+# from langchain.tools.python.tool import PythonREPLTool
+
+python_repl = PythonREPL()
+
+# Create the Tool object to pass to the agent
+# Make sure to update the name if you change it from 'python_repl'
+repl_tool = Tool(
+    name="python_repl",
+    description="""
+    A Python REPL (Read-Eval-Print Loop) for executing Python code.
+    Use this tool for:
+    - Performing accurate calculations (arithmetic, complex math).
+    - Manipulating and analyzing data (e.g., lists, numbers).
+    - Executing small, self-contained Python scripts.
+
+    **Input Format (CRITICAL):**
+    The input MUST be a valid, executable Python code string.
+    If you want to see the result of an expression or variable, you MUST explicitly `print()` it.
+    Example: `print(2 + 2)`
+    Example:
+    ```python
+    x = 10
+    y = 20
+    print(x * y)
+    ```
+
+    **DO NOT:**
+    - Ask questions in natural language.
+    - Expect implicit output (always use `print()`).
+    - Attempt to run interactive commands or scripts that require user input.
+    - Write code that requires external files or non-standard libraries unless explicitly provided or permitted.
+    - Access the internet or external resources unless the environment allows it.
+
+    **Output:**
+    The tool returns the standard output (stdout) of the executed Python code.
+    If there is an error in the code, the error message will be returned instead.
+    """,
+    func=python_repl.run,
+)
+
+
+def get_travily_api_search_tool(tavily_api_key: str) -> Tool:
+    """
+    Creates and returns a LangChain Tool for Tavily Search API.
+
+    Args:
+        tavily_api_key: The API key for Tavily Search.
+
+    Returns:
+        A LangChain Tool configured for Tavily Search.
+    """
+    tavily_search = TavilySearchResults(
+        max_results=3,             # Limits the number of search results to 3
+        include_answer=True,       # Prioritizes direct answers from search results
+        include_raw_content=False, # Excludes full raw content of pages to save tokens
+        include_images=False,      # Excludes image results to save tokens
+        tavily_api_key=tavily_api_key
+    )
+
+    return Tool(
+        name="tavily_search",
+        description="""
+        A powerful search engine tool for real-time information retrieval from the web.
+        Use this tool when you need to:
+        - Find up-to-date information, facts, or statistics.
+        - Research current events, people, or concepts.
+        - Answer questions that require external knowledge not present in the model's training data.
+        - Verify information or get factual answers.
+
+        **Input Format (CRITICAL):**
+        The input MUST be a concise, clear, and specific search query string.
+        Think of what you would type into a Google search bar.
+        Example: "current weather in London"
+        Example: "population of India 2024"
+        Example: "history of the internet"
+
+        **DO NOT:**
+        - Ask natural language questions that are not search queries.
+        - Provide incomplete sentences or ambiguous terms.
+        - Include personal information or sensitive data in your queries.
+        - Expect direct execution of commands or calculations; this is a search tool.
+
+        **Output:**
+        The tool returns a JSON string containing relevant search results.
+        It prioritizes a direct 'answer' if available. Otherwise, it provides a list of
+        search results including titles, URLs, and snippets.
+        """,
+        func=tavily_search.run,
+    )