Add the STACKONE_API_KEY too in example

shashi-stackone · shashi-stackone · commit 04b8bd0e9726 · 2026-04-07T14:43:05.000+01:00
diff --git a/examples/search_tool_example.py b/examples/search_tool_example.py
@@ -3,6 +3,10 @@
 For semantic search basics, see semantic_search_example.py.
 For full agent execution, see agent_tool_search.py.
 
+Prerequisites:
+    - STACKONE_API_KEY environment variable
+    - STACKONE_ACCOUNT_ID environment variable
+
 Run with:
     uv run python examples/search_tool_example.py
 """
@@ -22,37 +26,39 @@
 
 
 def main() -> None:
-    account_id = os.getenv("STACKONE_ACCOUNT_ID", "")
-    _account_ids = [a.strip() for a in account_id.split(",") if a.strip()] if account_id else []
+    api_key = os.getenv("STACKONE_API_KEY")
+    account_id = os.getenv("STACKONE_ACCOUNT_ID")
+
+    if not api_key:
+        print("Set STACKONE_API_KEY to run this example.")
+        return
+    if not account_id:
+        print("Set STACKONE_ACCOUNT_ID to run this example.")
+        return
 
     # --- Example 1: get_search_tool() callable ---
     print("=== get_search_tool() callable ===\n")
 
-    toolset = StackOneToolSet(search={})
+    toolset = StackOneToolSet(api_key=api_key, account_id=account_id, search={})
     search_tool = toolset.get_search_tool()
 
     queries = ["cancel an event", "list employees", "send a message"]
     for query in queries:
-        tools = search_tool(query, top_k=3, account_ids=_account_ids)
+        tools = search_tool(query, top_k=3)
         names = [t.name for t in tools]
         print(f'  "{query}" -> {", ".join(names) or "(none)"}')
 
     # --- Example 2: Constructor top_k vs per-call override ---
     print("\n=== Constructor top_k vs per-call override ===\n")
 
-    toolset_3 = StackOneToolSet(search={"top_k": 3})
-    toolset_10 = StackOneToolSet(search={"top_k": 10})
+    toolset_3 = StackOneToolSet(api_key=api_key, account_id=account_id, search={"top_k": 3})
 
     query = "manage employee records"
 
-    tools_3 = toolset_3.search_tools(query, account_ids=_account_ids)
+    tools_3 = toolset_3.search_tools(query)
     print(f"Constructor top_k=3: got {len(tools_3)} tools")
 
-    tools_10 = toolset_10.search_tools(query, account_ids=_account_ids)
-    print(f"Constructor top_k=10: got {len(tools_10)} tools")
-
-    # Per-call override: constructor says 3 but this call says 10
-    tools_override = toolset_3.search_tools(query, top_k=10, account_ids=_account_ids)
+    tools_override = toolset_3.search_tools(query, top_k=10)
     print(f"Per-call top_k=10 (overrides constructor 3): got {len(tools_override)} tools")
 
 
diff --git a/examples/workday_integration.py b/examples/workday_integration.py
@@ -3,6 +3,11 @@
 Workday can take 10-15s to respond. This example shows how to configure
 timeout and account_ids through the execute config.
 
+Prerequisites:
+    - STACKONE_API_KEY environment variable
+    - STACKONE_ACCOUNT_ID environment variable (a Workday-connected account)
+    - OPENAI_API_KEY environment variable
+
 Run with:
     uv run python examples/workday_integration.py
 """
@@ -25,12 +30,22 @@
 
 
 def main() -> None:
-    account_id = os.getenv("STACKONE_ACCOUNT_ID", "")
+    api_key = os.getenv("STACKONE_API_KEY")
+    account_id = os.getenv("STACKONE_ACCOUNT_ID")
+
+    if not api_key:
+        print("Set STACKONE_API_KEY to run this example.")
+        return
+    if not account_id:
+        print("Set STACKONE_ACCOUNT_ID to run this example.")
+        return
 
-    # Timeout and account_ids both live in the execute config
+    # Timeout for slow providers, account_id for scoping
     toolset = StackOneToolSet(
+        api_key=api_key,
+        account_id=account_id,
         search={"method": "auto", "top_k": 5},
-        execute={"account_ids": [account_id], "timeout": 120},
+        timeout=120,
     )
     client = OpenAI()
 
