obot-platform · g-linville · May 9, 2025 · May 9, 2025 · May 9, 2025
diff --git a/search/tavily/main.py b/search/tavily/main.py
@@ -48,15 +48,23 @@ def main():
             if os.getenv("MAX_RESULTS", "").strip():
                 max_results = int(os.getenv("MAX_RESULTS"))  # override with env var
 
-            response = client.search(
-                query=query,
-                include_answer=os.getenv("INCLUDE_ANSWER", "").lower()
-                == "true",  # no LLM-generated answer needed by default - we'll do that
-                include_raw_content=os.getenv("INCLUDE_RAW_CONTENT", "").lower()
-                != "false",  # include raw content by default
-                max_results=max_results,
-                include_domains=include_domains,
-            )
+            time_range = os.getenv("TIME_RANGE", "").lower().strip()
+            if time_range not in ["", "day", "week", "month", "year"]:
+                print("Invalid time range: must be day, week, month, or year")
+                sys.exit(1)
+
+            search_params = {
+                "query": query,
+                "include_answer": os.getenv("INCLUDE_ANSWER", "").lower() == "true",  # no LLM-generated answer needed by default - we'll do that
+                "include_raw_content": os.getenv("INCLUDE_RAW_CONTENT", "").lower() != "false",  # include raw content by default
+                "max_results": max_results,
+                "include_domains": include_domains,
+            }
+
+            if time_range:
+                search_params["time_range"] = time_range
+
+            response = client.search(**search_params)
         case "extract":
             client = TavilyClient()  # env TAVILY_API_KEY required
             url = parse_url(os.getenv("URL").strip())

diff --git a/search/tavily/obot-search.gpt b/search/tavily/obot-search.gpt
@@ -1,11 +1,27 @@
 ---
 Name: Obot Search
 Description: Search the web with Obot. This returns a list of search results scored by relevance to the query and summarized.
+Share Context: Obot Search Context
 Credential: ./credential
 Param: query: The search query
+Param: time_range: (Optional) the time range to search in. Value must be exactly "day", "week", "month", or "year".
 
 #!/usr/bin/env python3 ${GPTSCRIPT_TOOL_DIR}/main.py search
 
+---
+Name: Obot Search Context
+Type: context
+
+#!sys.echo
+
+# Instructions for using Obot Search
+
+The `time_range` parameter should only be used when you know that you will get better or more accurate search results by using it.
+Examples of when using it might be appropriate include getting some statistic that fluctuates by the day, such as a stock price or currency conversion rate.
+Do not use it for information that remains relatively constant over time.
+
+# End of instructions for using Obot Search
+
 ---
 !metadata:*:icon
 /admin/assets/obot_search_icon.png
diff --git a/search/tavily/tool.gpt b/search/tavily/tool.gpt
@@ -11,6 +11,7 @@ Credential: ./credential
 Share Context: Tavily Context
 Param: query: The search query
 Param: includeDomains: A comma-separated list of domains to specifically include in the search
+Param: time_range: (Optional) the time range to search in. Value must be exactly "day", "week", "month", or "year".
 
 #!/usr/bin/env python3 ${GPTSCRIPT_TOOL_DIR}/main.py search