minor slackbot improvements

Author: zzstoatzz

examples/slackbot/src/slackbot/api.py

@@ -1,6 +1,7 @@
 import asyncio
 import re
 import time
+from collections import defaultdict
 from contextlib import asynccontextmanager
 from typing import Any
 
@@ -31,7 +32,7 @@
     post_slack_message,
 )
 from slackbot.strings import count_tokens, slice_tokens
-from slackbot.wrap import WatchToolCalls, _progress_message
+from slackbot.wrap import WatchToolCalls, _progress_message, _tool_usage_counts
 
 BOT_MENTION = r"<@(\w+)>"
 
@@ -62,6 +63,8 @@ async def run_agent(
 
     try:
         token = _progress_message.set(progress)
+        # Initialize tool usage counts for this agent run
+        counts_token = _tool_usage_counts.set(defaultdict(int))
 
         try:
             with WatchToolCalls(settings=decorator_settings):
@@ -72,6 +75,7 @@ async def run_agent(
                 )
         finally:
             _progress_message.reset(token)
+            _tool_usage_counts.reset(counts_token)
 
         await progress.update(
             f"✅ thought for {time.monotonic() - start_time:.1f} seconds"

examples/slackbot/src/slackbot/research_agent.py

@@ -54,10 +54,11 @@ def create_research_agent(
 2. Use multiple search queries with different keywords - don't stop at first result
 3. Use explore_module_offerings to understand what's available in relevant modules (i.e. valid imports, types and functions available)
 4. Use display_callable_signature to get detailed signatures of functions, classes, and methods when needed
-5. Focus on Prefect 3.x documentation unless explicitly asked about 2.x or older versions
+5. **IMPORTANT**: ONLY use search_prefect_3x_docs unless the user explicitly mentions "2.x", "Prefect 2", or version compatibility
 6. Review gotchas and release notes for recent changes
 
 CRITICAL VERSION-SPECIFIC RULES:
+- **DEFAULT TO PREFECT 3.x**: Do NOT use search_prefect_2x_docs unless user explicitly mentions "2.x", "Prefect 2", or asks about version differences
 - **NEVER** suggest `Deployment.build_from_flow()` for Prefect 3.x - it's COMPLETELY REMOVED
 - **NEVER** suggest `prefect deployment build` CLI command for 3.x - use `prefect deploy` instead
 - The correct deployment pattern in 3.x is: `flow.from_source(...).deploy(...)`

examples/slackbot/src/slackbot/wrap.py

@@ -1,4 +1,5 @@
 import inspect
+from collections import defaultdict
 from contextlib import ContextDecorator
 from contextvars import ContextVar
 from functools import wraps
@@ -11,6 +12,10 @@
 T = TypeVar("T")
 
 _progress_message: ContextVar[Any] = ContextVar("progress_message", default=None)
+_tool_usage_counts: ContextVar[dict[str, int] | None] = ContextVar(
+    "tool_usage_counts", default=None
+)
+_current_tool: ContextVar[str | None] = ContextVar("current_tool", default=None)
 
 
 class DecorateMethodContext(ContextDecorator):
@@ -77,18 +82,44 @@ async def wrapper(*args, **kwargs) -> T:
             ):
                 tool_name = args[0].function.__name__
 
+            # Update tool usage counts
+            counts = _tool_usage_counts.get()
+            if counts is None:
+                counts = defaultdict(int)
+                _tool_usage_counts.set(counts)
+            counts[tool_name] += 1
+
+            # Set current tool
+            _current_tool_token = _current_tool.set(tool_name)
+
             try:
-                await _progress.append(f"🔧 {tool_name}")
+                # Build the progress message with better formatting
+                lines = []
+                lines.append(f"🔧 Using: `{tool_name}`")
+                lines.append("")  # Empty line for spacing
+
+                # Build summary of all tools used
+                if counts:
+                    lines.append("📊 Tools used:")
+                    for tool, count in sorted(counts.items()):
+                        lines.append(f"  • `{tool}` ({count}x)")
+
+                full_message = "\n".join(lines)
+                await _progress.update(full_message)
             except Exception:
                 pass
 
-        wrapped_callable = decorator(**settings or {})(func)
-        with prefect_tags(*tags):
-            result = wrapped_callable(*args, **kwargs)  # type: ignore
-            if inspect.isawaitable(result):
-                result = await result
-
-            return result
+        try:
+            wrapped_callable = decorator(**settings or {})(func)
+            with prefect_tags(*tags):
+                result = wrapped_callable(*args, **kwargs)  # type: ignore
+                if inspect.isawaitable(result):
+                    result = await result
+
+                return result
+        finally:
+            if _progress:
+                _current_tool.reset(_current_tool_token)
 
     return wrapper  # type: ignore