First slack tool pydantic ai

app/models/tools.py

@@ -2,5 +2,9 @@
 
 
 class SlackChatPostMessageParams(BaseModel):
+    """
+    Parameters for the slack.chat.postMessage tool
+    """
+
     channel: str
     text: str

app/tools/registry.py

@@ -1,13 +1,5 @@
-from collections.abc import Callable
+from pydantic_ai.toolsets import FunctionToolset
 
-from slack_sdk.web.slack_response import SlackResponse
+from app.tools.slack import slack_toolset
 
-from app.models.tools import SlackChatPostMessageParams
-from app.tools.slack import post_message
-
-TOOLS: dict[str, dict[str, type | Callable[..., SlackResponse]]] = {
-    "slack.chat.postMessage": {
-        "schema": SlackChatPostMessageParams,
-        "fn": post_message,
-    },
-}
+TOOLS: dict[str, FunctionToolset] = {"slack.tools": slack_toolset}

app/tools/slack.py

@@ -1,9 +1,26 @@
+from dataclasses import dataclass
+
+from pydantic_ai import RunContext
+from pydantic_ai.toolsets import FunctionToolset
 from slack_sdk import WebClient
 from slack_sdk.web.slack_response import SlackResponse
 
 from app.models.tools import SlackChatPostMessageParams
 
 
-def post_message(token: str, params: SlackChatPostMessageParams) -> SlackResponse:
-    client = WebClient(token=token)
-    return client.chat_postMessage(channel=params.channel, text=params.text)
+@dataclass
+class Deps:
+    client: WebClient
+
+
+slack_toolset = FunctionToolset()
+
+
+@slack_toolset.tool(name="slack.chat.postMessage")
+def post_message(
+    ctx: RunContext[Deps], params: SlackChatPostMessageParams
+) -> SlackResponse:
+    """
+    Use this function to post a message in the specified channel
+    """
+    return ctx.deps.client.chat_postMessage(channel=params.channel, text=params.text)

pyproject.toml

@@ -20,4 +20,5 @@ dev = [
     "pre-commit",
     "coverage",
     "ruff",
+    "ipython>=8.37.0",
 ]

tests/test_ava_slack_post_message.py

@@ -0,0 +1,77 @@
+import os
+
+import pytest
+from dotenv import load_dotenv
+from pydantic_ai import Agent, RunContext, RunUsage
+from pydantic_ai.models.test import TestModel
+from slack_sdk import WebClient
+from slack_sdk.errors import SlackApiError
+
+from app.models import tools
+from app.tools import slack
+
+_ = load_dotenv()
+
+
+def test_health_ava_can_post_message():
+    token = os.environ.get("SLACK_BOT_TOKEN")
+    if not token:
+        pytest.skip("SLACK_BOT_TOKEN not set; live Slack test skipped")
+
+    client = WebClient(token=token)
+    message_text = "AVA backend live smoke :rocket:"
+
+    try:
+        response = client.chat_postMessage(channel="sandbox", text=message_text)
+    except SlackApiError as exc:
+        pytest.fail(f"Slack API error: {exc.response['error']}")
+
+    assert response["ok"] is True
+    assert response["message"]["text"] == message_text  # type: ignore
+    assert "ts" in response
+
+
+@pytest.mark.parametrize(
+    "channel,text,token",
+    [
+        (
+            "sandbox",
+            "Hello, AVA backend live smoke :rocket:",
+            os.environ.get("SLACK_BOT_TOKEN"),
+        ),
+        (
+            "general",
+            "Hello, User is live smoke :rocket:",
+            os.environ.get("SLACK_USER_TOKEN"),
+        ),
+    ],
+)
+def test_post_message_with_ctx(channel: str, text: str, token: str | None):
+    if token is not None:
+        client = WebClient(token=token)
+        ctx = RunContext(
+            deps=slack.Deps(client=client),
+            model=TestModel(),
+            usage=RunUsage(),
+        )
+        tool_params = tools.SlackChatPostMessageParams(channel=channel, text=text)
+        try:
+            response = slack.post_message(ctx=ctx, params=tool_params)
+        except SlackApiError as exc:
+            pytest.fail(f"Slack API error: {exc.response['error']}")
+
+        assert response["ok"] is True
+        assert response["message"]["text"] == text
+        assert "ts" in response
+
+
+def test_slack_tools_are_synced():
+    token = os.environ.get("SLACK_BOT_TOKEN")
+    client = WebClient(token=token)
+    test_model = TestModel(call_tools=[])
+    agent = Agent(test_model, toolsets=[slack.slack_toolset], deps_type=slack.Deps)
+    _ = agent.run_sync("What tools are available?", deps=slack.Deps(client=client))
+
+    assert [
+        t.name for t in test_model.last_model_request_parameters.function_tools
+    ] == ["slack.chat.postMessage"]