mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-09-30 19:43:15 +00:00
5a01aff1c7
* feat: only post Slack 'Working on it!' on first thread mention * feat: randomize Slack trace reply phrase Pick from a small list of friendly phrases instead of always saying 'Working on it!' so the bot feels less robotic. Explicit messages (e.g. 'Taking a look...' from PR review path) are unaffected. * adjust phrases --------- Co-authored-by: open-swe[bot] <open-swe@users.noreply.github.com> Co-authored-by: Johannes du Plessis <johannes@langchain.dev>
745 lines
26 KiB
Python
745 lines
26 KiB
Python
import asyncio
|
|
|
|
import pytest
|
|
|
|
from agent import webapp
|
|
from agent.utils import slack as slack_utils
|
|
from agent.utils.slack import (
|
|
TRACE_REPLY_PHRASES,
|
|
convert_mentions_to_slack_format,
|
|
format_slack_messages_for_prompt,
|
|
looks_like_slack_pr_review_command,
|
|
parse_github_pr_url,
|
|
parse_slack_review_command,
|
|
post_slack_trace_reply,
|
|
replace_bot_mention_with_username,
|
|
select_slack_context_messages,
|
|
strip_bot_mention,
|
|
)
|
|
from agent.webapp import generate_thread_id_from_slack_thread
|
|
|
|
|
|
class _FakeNotFoundError(Exception):
|
|
status_code = 404
|
|
|
|
|
|
class _FakeThreadsClient:
|
|
def __init__(self, thread: dict | None = None, raise_not_found: bool = False) -> None:
|
|
self.thread = thread
|
|
self.raise_not_found = raise_not_found
|
|
self.requested_thread_id: str | None = None
|
|
|
|
async def get(self, thread_id: str) -> dict:
|
|
self.requested_thread_id = thread_id
|
|
if self.raise_not_found:
|
|
raise _FakeNotFoundError("not found")
|
|
if self.thread is None:
|
|
raise AssertionError("thread must be provided when raise_not_found is False")
|
|
return self.thread
|
|
|
|
|
|
class _FakeClient:
|
|
def __init__(self, threads_client: _FakeThreadsClient) -> None:
|
|
self.threads = threads_client
|
|
|
|
|
|
def test_generate_thread_id_from_slack_thread_is_deterministic() -> None:
|
|
channel_id = "C12345"
|
|
thread_ts = "1730900000.123456"
|
|
first = generate_thread_id_from_slack_thread(channel_id, thread_ts)
|
|
second = generate_thread_id_from_slack_thread(channel_id, thread_ts)
|
|
assert first == second
|
|
assert len(first) == 36
|
|
|
|
|
|
def test_select_slack_context_messages_uses_thread_start_when_no_prior_mention() -> None:
|
|
bot_user_id = "UBOT"
|
|
messages = [
|
|
{"ts": "1.0", "text": "hello", "user": "U1"},
|
|
{"ts": "2.0", "text": "context", "user": "U2"},
|
|
{"ts": "3.0", "text": "<@UBOT> please help", "user": "U1"},
|
|
]
|
|
|
|
selected, mode = select_slack_context_messages(messages, "3.0", bot_user_id)
|
|
|
|
assert mode == "thread_start"
|
|
assert [item["ts"] for item in selected] == ["1.0", "2.0", "3.0"]
|
|
|
|
|
|
def test_select_slack_context_messages_uses_previous_mention_boundary() -> None:
|
|
bot_user_id = "UBOT"
|
|
messages = [
|
|
{"ts": "1.0", "text": "hello", "user": "U1"},
|
|
{"ts": "2.0", "text": "<@UBOT> first request", "user": "U1"},
|
|
{"ts": "3.0", "text": "extra context", "user": "U2"},
|
|
{"ts": "4.0", "text": "<@UBOT> second request", "user": "U3"},
|
|
]
|
|
|
|
selected, mode = select_slack_context_messages(messages, "4.0", bot_user_id)
|
|
|
|
assert mode == "last_mention"
|
|
assert [item["ts"] for item in selected] == ["2.0", "3.0", "4.0"]
|
|
|
|
|
|
def test_select_slack_context_messages_ignores_messages_after_current_event() -> None:
|
|
bot_user_id = "UBOT"
|
|
messages = [
|
|
{"ts": "1.0", "text": "<@UBOT> first request", "user": "U1"},
|
|
{"ts": "2.0", "text": "follow-up", "user": "U2"},
|
|
{"ts": "3.0", "text": "<@UBOT> second request", "user": "U3"},
|
|
{"ts": "4.0", "text": "after event", "user": "U4"},
|
|
]
|
|
|
|
selected, mode = select_slack_context_messages(messages, "3.0", bot_user_id)
|
|
|
|
assert mode == "last_mention"
|
|
assert [item["ts"] for item in selected] == ["1.0", "2.0", "3.0"]
|
|
|
|
|
|
def test_strip_bot_mention_removes_bot_tag() -> None:
|
|
assert strip_bot_mention("<@UBOT> please check", "UBOT") == "please check"
|
|
|
|
|
|
def test_strip_bot_mention_removes_bot_username_tag() -> None:
|
|
assert (
|
|
strip_bot_mention("@open-swe please check", "UBOT", bot_username="open-swe")
|
|
== "please check"
|
|
)
|
|
|
|
|
|
def test_replace_bot_mention_with_username() -> None:
|
|
assert (
|
|
replace_bot_mention_with_username("<@UBOT> can you help?", "UBOT", "open-swe")
|
|
== "@open-swe can you help?"
|
|
)
|
|
|
|
|
|
def test_convert_mentions_to_slack_format_basic() -> None:
|
|
assert (
|
|
convert_mentions_to_slack_format("Hey @Brace Sproul(U06KD8BFY95), check this")
|
|
== "Hey <@U06KD8BFY95>, check this"
|
|
)
|
|
|
|
|
|
def test_convert_mentions_to_slack_format_multiple() -> None:
|
|
text = "@Alice(U111) and @Bob(U222) please review"
|
|
assert convert_mentions_to_slack_format(text) == "<@U111> and <@U222> please review"
|
|
|
|
|
|
def test_convert_mentions_to_slack_format_no_match() -> None:
|
|
text = "No mentions here, just @plain text"
|
|
assert convert_mentions_to_slack_format(text) == text
|
|
|
|
|
|
def test_convert_mentions_to_slack_format_preserves_existing_slack_mentions() -> None:
|
|
text = "Already tagged <@U06KD8BFY95> correctly"
|
|
assert convert_mentions_to_slack_format(text) == text
|
|
|
|
|
|
def test_parse_github_pr_url_raw_url() -> None:
|
|
pr_ref = parse_github_pr_url("https://github.com/langchain-ai/open-swe/pull/1244")
|
|
|
|
assert pr_ref is not None
|
|
assert pr_ref.owner == "langchain-ai"
|
|
assert pr_ref.repo == "open-swe"
|
|
assert pr_ref.number == 1244
|
|
assert pr_ref.url == "https://github.com/langchain-ai/open-swe/pull/1244"
|
|
|
|
|
|
def test_parse_github_pr_url_slack_formatted_link() -> None:
|
|
pr_ref = parse_github_pr_url("<https://github.com/langchain-ai/open-swe/pull/1244|PR>")
|
|
|
|
assert pr_ref is not None
|
|
assert pr_ref.owner == "langchain-ai"
|
|
assert pr_ref.repo == "open-swe"
|
|
assert pr_ref.number == 1244
|
|
|
|
|
|
def test_parse_slack_review_command_requires_exact_review_command() -> None:
|
|
pr_ref = parse_slack_review_command("review https://github.com/langchain-ai/open-swe/pull/1244")
|
|
|
|
assert pr_ref is not None
|
|
assert pr_ref.owner == "langchain-ai"
|
|
assert pr_ref.repo == "open-swe"
|
|
assert pr_ref.number == 1244
|
|
assert (
|
|
parse_slack_review_command(
|
|
"please review https://github.com/langchain-ai/open-swe/pull/1244"
|
|
)
|
|
is None
|
|
)
|
|
assert (
|
|
parse_slack_review_command("review https://github.com/langchain-ai/open-swe/issues/1244")
|
|
is None
|
|
)
|
|
|
|
|
|
def test_parse_slack_review_command_supports_slack_link() -> None:
|
|
pr_ref = parse_slack_review_command(
|
|
"review <https://github.com/langchain-ai/open-swe/pull/1244|PR>"
|
|
)
|
|
|
|
assert pr_ref is not None
|
|
assert pr_ref.url == "https://github.com/langchain-ai/open-swe/pull/1244"
|
|
|
|
|
|
def test_parse_slack_review_command_supports_slack_wrapped_raw_link() -> None:
|
|
pr_ref = parse_slack_review_command(
|
|
"review <https://github.com/langchain-ai/open-swe/pull/1244>"
|
|
)
|
|
|
|
assert pr_ref is not None
|
|
assert pr_ref.url == "https://github.com/langchain-ai/open-swe/pull/1244"
|
|
|
|
|
|
def test_looks_like_slack_pr_review_command_validates_github_host() -> None:
|
|
assert looks_like_slack_pr_review_command(
|
|
"review https://github.com/langchain-ai/open-swe/issues/1244"
|
|
)
|
|
assert not looks_like_slack_pr_review_command(
|
|
"review https://example.com/redirect?next=https://github.com/langchain-ai/open-swe/pull/1244"
|
|
)
|
|
|
|
|
|
def test_format_slack_messages_for_prompt_uses_name_and_id() -> None:
|
|
formatted = format_slack_messages_for_prompt(
|
|
[{"ts": "1.0", "text": "hello", "user": "U123"}],
|
|
{"U123": "alice"},
|
|
)
|
|
|
|
assert formatted == "@alice(U123): hello"
|
|
|
|
|
|
def test_format_slack_messages_for_prompt_replaces_bot_id_mention_in_text() -> None:
|
|
formatted = format_slack_messages_for_prompt(
|
|
[{"ts": "1.0", "text": "<@UBOT> status update?", "user": "U123"}],
|
|
{"U123": "alice"},
|
|
bot_user_id="UBOT",
|
|
bot_username="open-swe",
|
|
)
|
|
|
|
assert formatted == "@alice(U123): @open-swe status update?"
|
|
|
|
|
|
def test_post_slack_trace_reply_picks_random_phrase_when_no_message(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
posted: list[str] = []
|
|
|
|
async def fake_post_slack_thread_reply(channel_id: str, thread_ts: str, text: str) -> None:
|
|
posted.append(text)
|
|
|
|
monkeypatch.setattr(slack_utils, "post_slack_thread_reply", fake_post_slack_thread_reply)
|
|
monkeypatch.setattr(slack_utils, "get_langsmith_trace_url", lambda thread_id: None)
|
|
|
|
asyncio.run(post_slack_trace_reply("C123", "1.0", "thread-id"))
|
|
|
|
assert len(posted) == 1
|
|
assert posted[0] in TRACE_REPLY_PHRASES
|
|
|
|
|
|
def test_post_slack_trace_reply_uses_explicit_message_when_provided(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
posted: list[str] = []
|
|
|
|
async def fake_post_slack_thread_reply(channel_id: str, thread_ts: str, text: str) -> None:
|
|
posted.append(text)
|
|
|
|
monkeypatch.setattr(slack_utils, "post_slack_thread_reply", fake_post_slack_thread_reply)
|
|
monkeypatch.setattr(slack_utils, "get_langsmith_trace_url", lambda thread_id: None)
|
|
|
|
asyncio.run(post_slack_trace_reply("C123", "1.0", "thread-id", message="Taking a look..."))
|
|
|
|
assert posted == ["Taking a look..."]
|
|
|
|
|
|
def test_select_slack_context_messages_detects_username_mention() -> None:
|
|
selected, mode = select_slack_context_messages(
|
|
[
|
|
{"ts": "1.0", "text": "@open-swe first request", "user": "U1"},
|
|
{"ts": "2.0", "text": "follow up", "user": "U2"},
|
|
{"ts": "3.0", "text": "@open-swe second request", "user": "U3"},
|
|
],
|
|
"3.0",
|
|
bot_user_id="UBOT",
|
|
bot_username="open-swe",
|
|
)
|
|
|
|
assert mode == "last_mention"
|
|
assert [item["ts"] for item in selected] == ["1.0", "2.0", "3.0"]
|
|
|
|
|
|
def test_get_slack_repo_config_message_repo_overrides_existing_thread_repo(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
threads_client = _FakeThreadsClient(
|
|
thread={"metadata": {"repo": {"owner": "saved-owner", "name": "saved-repo"}}}
|
|
)
|
|
|
|
posted = False
|
|
|
|
async def fake_post_slack_thread_reply(channel_id: str, thread_ts: str, text: str) -> bool:
|
|
nonlocal posted
|
|
posted = True
|
|
return True
|
|
|
|
monkeypatch.setattr(webapp, "get_client", lambda url: _FakeClient(threads_client))
|
|
monkeypatch.setattr(
|
|
webapp, "post_slack_thread_reply", fake_post_slack_thread_reply, raising=False
|
|
)
|
|
|
|
repo = asyncio.run(
|
|
webapp.get_slack_repo_config("please use repo:new-owner/new-repo", "C123", "1.234")
|
|
)
|
|
|
|
assert repo == {"owner": "new-owner", "name": "new-repo"}
|
|
assert threads_client.requested_thread_id is None
|
|
assert not posted
|
|
|
|
|
|
def test_get_slack_repo_config_parses_message_for_new_thread(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
threads_client = _FakeThreadsClient(raise_not_found=True)
|
|
|
|
monkeypatch.setattr(webapp, "get_client", lambda url: _FakeClient(threads_client))
|
|
|
|
repo = asyncio.run(
|
|
webapp.get_slack_repo_config("please use repo:new-owner/new-repo", "C123", "1.234")
|
|
)
|
|
|
|
assert repo == {"owner": "new-owner", "name": "new-repo"}
|
|
|
|
|
|
def test_get_slack_repo_config_existing_thread_without_repo_uses_default(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
threads_client = _FakeThreadsClient(thread={"metadata": {}})
|
|
monkeypatch.setattr(webapp, "SLACK_REPO_OWNER", "default-owner")
|
|
monkeypatch.setattr(webapp, "SLACK_REPO_NAME", "default-repo")
|
|
|
|
monkeypatch.setattr(webapp, "get_client", lambda url: _FakeClient(threads_client))
|
|
|
|
repo = asyncio.run(webapp.get_slack_repo_config("please help", "C123", "1.234"))
|
|
|
|
assert repo == {"owner": "default-owner", "name": "default-repo"}
|
|
assert threads_client.requested_thread_id == generate_thread_id_from_slack_thread(
|
|
"C123", "1.234"
|
|
)
|
|
|
|
|
|
def test_get_slack_repo_config_space_syntax_detected(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
"""repo owner/name (space instead of colon) should be detected correctly."""
|
|
threads_client = _FakeThreadsClient(raise_not_found=True)
|
|
|
|
monkeypatch.setattr(webapp, "get_client", lambda url: _FakeClient(threads_client))
|
|
|
|
repo = asyncio.run(
|
|
webapp.get_slack_repo_config(
|
|
"please fix the bug in repo langchain-ai/langchainjs", "C123", "1.234"
|
|
)
|
|
)
|
|
|
|
assert repo == {"owner": "langchain-ai", "name": "langchainjs"}
|
|
|
|
|
|
def test_get_slack_repo_config_github_url_extracted(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
"""GitHub URL in message should be used to detect the repo."""
|
|
threads_client = _FakeThreadsClient(raise_not_found=True)
|
|
|
|
monkeypatch.setattr(webapp, "get_client", lambda url: _FakeClient(threads_client))
|
|
|
|
repo = asyncio.run(
|
|
webapp.get_slack_repo_config(
|
|
"I found a bug in https://github.com/langchain-ai/langgraph-api please fix it",
|
|
"C123",
|
|
"1.234",
|
|
)
|
|
)
|
|
|
|
assert repo == {"owner": "langchain-ai", "name": "langgraph-api"}
|
|
|
|
|
|
def test_get_slack_repo_config_explicit_repo_beats_github_url(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
"""Explicit repo: syntax takes priority over a GitHub URL also present in the message."""
|
|
threads_client = _FakeThreadsClient(raise_not_found=True)
|
|
|
|
monkeypatch.setattr(webapp, "get_client", lambda url: _FakeClient(threads_client))
|
|
|
|
repo = asyncio.run(
|
|
webapp.get_slack_repo_config(
|
|
"see https://github.com/langchain-ai/langgraph-api but use repo:my-org/my-repo",
|
|
"C123",
|
|
"1.234",
|
|
)
|
|
)
|
|
|
|
assert repo == {"owner": "my-org", "name": "my-repo"}
|
|
|
|
|
|
def test_get_slack_repo_config_explicit_space_syntax_beats_thread_metadata(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
"""Explicit repo owner/name (space syntax) takes priority over saved thread metadata."""
|
|
threads_client = _FakeThreadsClient(
|
|
thread={"metadata": {"repo": {"owner": "saved-owner", "name": "saved-repo"}}}
|
|
)
|
|
|
|
monkeypatch.setattr(webapp, "get_client", lambda url: _FakeClient(threads_client))
|
|
|
|
repo = asyncio.run(
|
|
webapp.get_slack_repo_config(
|
|
"actually use repo langchain-ai/langchainjs today", "C123", "1.234"
|
|
)
|
|
)
|
|
|
|
assert repo == {"owner": "langchain-ai", "name": "langchainjs"}
|
|
|
|
|
|
def test_get_slack_repo_config_github_url_beats_thread_metadata(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
"""A GitHub URL in the message takes priority over saved thread metadata."""
|
|
threads_client = _FakeThreadsClient(
|
|
thread={"metadata": {"repo": {"owner": "saved-owner", "name": "saved-repo"}}}
|
|
)
|
|
|
|
monkeypatch.setattr(webapp, "get_client", lambda url: _FakeClient(threads_client))
|
|
|
|
repo = asyncio.run(
|
|
webapp.get_slack_repo_config(
|
|
"I found a bug in https://github.com/langchain-ai/langgraph-api",
|
|
"C123",
|
|
"1.234",
|
|
)
|
|
)
|
|
|
|
assert repo == {"owner": "langchain-ai", "name": "langgraph-api"}
|
|
|
|
|
|
def test_get_slack_repo_config_repo_name_only_defaults_org(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
"""repo:name without org should default owner to langchain-ai."""
|
|
threads_client = _FakeThreadsClient(raise_not_found=True)
|
|
|
|
monkeypatch.setattr(webapp, "get_client", lambda url: _FakeClient(threads_client))
|
|
|
|
repo = asyncio.run(
|
|
webapp.get_slack_repo_config("fix bug in repo:langchainplus", "C123", "1.234")
|
|
)
|
|
|
|
assert repo == {"owner": "langchain-ai", "name": "langchainplus"}
|
|
|
|
|
|
def test_get_slack_repo_config_repo_name_only_space_syntax(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
"""repo name (space syntax, no org) should default owner to langchain-ai."""
|
|
threads_client = _FakeThreadsClient(raise_not_found=True)
|
|
|
|
monkeypatch.setattr(webapp, "get_client", lambda url: _FakeClient(threads_client))
|
|
|
|
repo = asyncio.run(webapp.get_slack_repo_config("fix bug in repo open-swe", "C123", "1.234"))
|
|
|
|
assert repo == {"owner": "langchain-ai", "name": "open-swe"}
|
|
|
|
|
|
def _setup_slack_mention_fakes(
|
|
monkeypatch: pytest.MonkeyPatch, captured: dict[str, object]
|
|
) -> None:
|
|
async def fake_add_slack_reaction(channel_id: str, message_ts: str, emoji: str) -> bool:
|
|
captured["reaction"] = {
|
|
"channel_id": channel_id,
|
|
"message_ts": message_ts,
|
|
"emoji": emoji,
|
|
}
|
|
return True
|
|
|
|
async def fake_get_slack_user_info(user_id: str) -> dict:
|
|
return {
|
|
"profile": {
|
|
"email": "mason@example.com",
|
|
"display_name": "Mason",
|
|
}
|
|
}
|
|
|
|
async def fake_fetch_slack_thread_messages(channel_id: str, thread_ts: str) -> list[dict]:
|
|
captured["fetch_thread"] = {"channel_id": channel_id, "thread_ts": thread_ts}
|
|
return [
|
|
{"ts": "1700000000.000100", "text": "<@UBOT> first request", "user": "U123"},
|
|
{"ts": "1700000000.000150", "text": "context", "user": "U456"},
|
|
{
|
|
"ts": "1700000000.000200",
|
|
"text": "<@UBOT> continue on the branch",
|
|
"user": "U123",
|
|
},
|
|
]
|
|
|
|
async def fake_get_slack_user_names(user_ids: list[str]) -> dict[str, str]:
|
|
captured["user_ids"] = user_ids
|
|
return {"U123": "Mason", "U456": "Teammate"}
|
|
|
|
async def fake_resolve_slack_links_in_context(
|
|
context_messages: list[dict], user_names_by_id: dict[str, str]
|
|
) -> tuple[str, list[str]]:
|
|
captured["context_messages"] = context_messages
|
|
captured["user_names_by_id"] = user_names_by_id
|
|
return "", []
|
|
|
|
async def fake_is_thread_active(thread_id: str) -> bool:
|
|
captured["active_thread_id"] = thread_id
|
|
return False
|
|
|
|
async def fake_post_slack_trace_reply(
|
|
channel_id: str, thread_ts: str, thread_id: str, message: str | None = None
|
|
) -> None:
|
|
captured["trace_reply"] = {
|
|
"channel_id": channel_id,
|
|
"thread_ts": thread_ts,
|
|
"thread_id": thread_id,
|
|
"message": message,
|
|
}
|
|
|
|
class _FakeRunsClient:
|
|
async def create(self, thread_id: str, graph: str, **kwargs) -> dict[str, str]:
|
|
captured["run_create"] = {
|
|
"thread_id": thread_id,
|
|
"graph": graph,
|
|
"kwargs": kwargs,
|
|
}
|
|
return {"run_id": "run-123"}
|
|
|
|
class _FakeThreadsClientForProcess:
|
|
async def update(self, *, thread_id: str, metadata: dict) -> None:
|
|
captured["metadata_update"] = {"thread_id": thread_id, "metadata": metadata}
|
|
|
|
class _FakeLangGraphClientForProcess:
|
|
runs = _FakeRunsClient()
|
|
threads = _FakeThreadsClientForProcess()
|
|
|
|
monkeypatch.setattr(webapp, "SLACK_BOT_USERNAME", "open-swe")
|
|
monkeypatch.setattr(webapp, "add_slack_reaction", fake_add_slack_reaction)
|
|
monkeypatch.setattr(webapp, "get_slack_user_info", fake_get_slack_user_info)
|
|
monkeypatch.setattr(webapp, "fetch_slack_thread_messages", fake_fetch_slack_thread_messages)
|
|
monkeypatch.setattr(webapp, "get_slack_user_names", fake_get_slack_user_names)
|
|
monkeypatch.setattr(
|
|
webapp, "resolve_slack_links_in_context", fake_resolve_slack_links_in_context
|
|
)
|
|
monkeypatch.setattr(webapp, "is_thread_active", fake_is_thread_active)
|
|
monkeypatch.setattr(webapp, "post_slack_trace_reply", fake_post_slack_trace_reply)
|
|
monkeypatch.setattr(webapp, "get_client", lambda url: _FakeLangGraphClientForProcess())
|
|
|
|
|
|
def test_process_slack_mention_creates_thread_first_run_with_trace_reply(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
captured: dict[str, object] = {}
|
|
_setup_slack_mention_fakes(monkeypatch, captured)
|
|
|
|
async def fake_thread_exists(thread_id: str) -> bool:
|
|
captured["thread_exists_check"] = thread_id
|
|
return False
|
|
|
|
monkeypatch.setattr(webapp, "_thread_exists", fake_thread_exists)
|
|
|
|
thread_ts = "1700000000.000100"
|
|
event_ts = "1700000000.000200"
|
|
expected_thread_id = generate_thread_id_from_slack_thread("C123", thread_ts)
|
|
|
|
asyncio.run(
|
|
webapp.process_slack_mention(
|
|
{
|
|
"channel_id": "C123",
|
|
"thread_ts": thread_ts,
|
|
"event_ts": event_ts,
|
|
"user_id": "U123",
|
|
"text": "<@UBOT> continue on the branch",
|
|
"bot_user_id": "UBOT",
|
|
},
|
|
{"owner": "langchain-ai", "name": "open-swe"},
|
|
)
|
|
)
|
|
|
|
assert captured["thread_exists_check"] == expected_thread_id
|
|
assert captured["fetch_thread"] == {"channel_id": "C123", "thread_ts": thread_ts}
|
|
assert captured["active_thread_id"] == expected_thread_id
|
|
assert captured["metadata_update"] == {
|
|
"thread_id": expected_thread_id,
|
|
"metadata": {"repo": {"owner": "langchain-ai", "name": "open-swe"}},
|
|
}
|
|
assert captured["trace_reply"] == {
|
|
"channel_id": "C123",
|
|
"thread_ts": thread_ts,
|
|
"thread_id": expected_thread_id,
|
|
"message": None,
|
|
}
|
|
|
|
run_create = captured["run_create"]
|
|
assert isinstance(run_create, dict)
|
|
assert run_create["thread_id"] == expected_thread_id
|
|
assert run_create["graph"] == "agent"
|
|
kwargs = run_create["kwargs"]
|
|
assert kwargs["if_not_exists"] == "create"
|
|
assert "multitask_strategy" not in kwargs
|
|
assert kwargs["config"]["configurable"]["slack_thread"]["thread_ts"] == thread_ts
|
|
prompt_block = kwargs["input"]["messages"][0]["content"][0]
|
|
assert prompt_block["text"].count("## Slack Thread") == 1
|
|
assert f"Thread TS: {thread_ts}" in prompt_block["text"]
|
|
assert "## Latest Mention Request\ncontinue on the branch" in prompt_block["text"]
|
|
|
|
|
|
def test_process_slack_mention_skips_trace_reply_on_followup_mention(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
"""Subsequent mentions in a Slack thread should not post 'Working on it!'."""
|
|
captured: dict[str, object] = {}
|
|
_setup_slack_mention_fakes(monkeypatch, captured)
|
|
|
|
async def fake_thread_exists(thread_id: str) -> bool:
|
|
captured["thread_exists_check"] = thread_id
|
|
return True
|
|
|
|
monkeypatch.setattr(webapp, "_thread_exists", fake_thread_exists)
|
|
|
|
thread_ts = "1700000000.000100"
|
|
event_ts = "1700000000.000300"
|
|
expected_thread_id = generate_thread_id_from_slack_thread("C123", thread_ts)
|
|
|
|
asyncio.run(
|
|
webapp.process_slack_mention(
|
|
{
|
|
"channel_id": "C123",
|
|
"thread_ts": thread_ts,
|
|
"event_ts": event_ts,
|
|
"user_id": "U123",
|
|
"text": "<@UBOT> follow up question",
|
|
"bot_user_id": "UBOT",
|
|
},
|
|
{"owner": "langchain-ai", "name": "open-swe"},
|
|
)
|
|
)
|
|
|
|
assert captured["thread_exists_check"] == expected_thread_id
|
|
assert "trace_reply" not in captured
|
|
run_create = captured["run_create"]
|
|
assert isinstance(run_create, dict)
|
|
assert run_create["thread_id"] == expected_thread_id
|
|
|
|
|
|
def test_process_slack_mention_queues_active_thread_message(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
captured: dict[str, object] = {}
|
|
|
|
async def fake_add_slack_reaction(channel_id: str, message_ts: str, emoji: str) -> bool:
|
|
captured["reaction"] = {
|
|
"channel_id": channel_id,
|
|
"message_ts": message_ts,
|
|
"emoji": emoji,
|
|
}
|
|
return True
|
|
|
|
async def fake_get_slack_user_info(user_id: str) -> dict:
|
|
return {
|
|
"profile": {
|
|
"email": "mason@example.com",
|
|
"display_name": "Mason",
|
|
}
|
|
}
|
|
|
|
async def fake_fetch_slack_thread_messages(channel_id: str, thread_ts: str) -> list[dict]:
|
|
return [
|
|
{"ts": "1700000000.000100", "text": "<@UBOT> first request", "user": "U123"},
|
|
{
|
|
"ts": "1700000000.000200",
|
|
"text": "<@UBOT> include this screenshot https://example.com/image.png",
|
|
"user": "U123",
|
|
},
|
|
]
|
|
|
|
async def fake_get_slack_user_names(user_ids: list[str]) -> dict[str, str]:
|
|
captured["user_ids"] = user_ids
|
|
return {"U123": "Mason"}
|
|
|
|
async def fake_resolve_slack_links_in_context(
|
|
context_messages: list[dict], user_names_by_id: dict[str, str]
|
|
) -> tuple[str, list[str]]:
|
|
captured["context_messages"] = context_messages
|
|
return "", []
|
|
|
|
async def fake_fetch_image_block(image_url: str, http_client: object) -> None:
|
|
captured["image_url"] = image_url
|
|
return None
|
|
|
|
async def fake_is_thread_active(thread_id: str) -> bool:
|
|
captured["active_thread_id"] = thread_id
|
|
return True
|
|
|
|
async def fake_queue_message_for_thread(thread_id: str, message_content: object) -> bool:
|
|
captured["queued"] = {"thread_id": thread_id, "message_content": message_content}
|
|
return True
|
|
|
|
async def fake_post_slack_trace_reply(*args, **kwargs) -> None:
|
|
raise AssertionError("trace reply should not be posted for queued mid-run Slack messages")
|
|
|
|
async def fake_thread_exists(thread_id: str) -> bool:
|
|
return True
|
|
|
|
class _FakeRunsClient:
|
|
async def create(self, *args, **kwargs) -> None:
|
|
raise AssertionError("run should not be created for active Slack threads")
|
|
|
|
class _FakeThreadsClientForProcess:
|
|
async def update(self, *, thread_id: str, metadata: dict) -> None:
|
|
captured["metadata_update"] = {"thread_id": thread_id, "metadata": metadata}
|
|
|
|
class _FakeLangGraphClientForProcess:
|
|
runs = _FakeRunsClient()
|
|
threads = _FakeThreadsClientForProcess()
|
|
|
|
monkeypatch.setattr(webapp, "SLACK_BOT_USERNAME", "open-swe")
|
|
monkeypatch.setattr(webapp, "add_slack_reaction", fake_add_slack_reaction)
|
|
monkeypatch.setattr(webapp, "get_slack_user_info", fake_get_slack_user_info)
|
|
monkeypatch.setattr(webapp, "fetch_slack_thread_messages", fake_fetch_slack_thread_messages)
|
|
monkeypatch.setattr(webapp, "get_slack_user_names", fake_get_slack_user_names)
|
|
monkeypatch.setattr(
|
|
webapp, "resolve_slack_links_in_context", fake_resolve_slack_links_in_context
|
|
)
|
|
monkeypatch.setattr(webapp, "fetch_image_block", fake_fetch_image_block)
|
|
monkeypatch.setattr(webapp, "is_thread_active", fake_is_thread_active)
|
|
monkeypatch.setattr(webapp, "queue_message_for_thread", fake_queue_message_for_thread)
|
|
monkeypatch.setattr(webapp, "post_slack_trace_reply", fake_post_slack_trace_reply)
|
|
monkeypatch.setattr(webapp, "_thread_exists", fake_thread_exists)
|
|
monkeypatch.setattr(webapp, "get_client", lambda url: _FakeLangGraphClientForProcess())
|
|
|
|
thread_ts = "1700000000.000100"
|
|
event_ts = "1700000000.000200"
|
|
expected_thread_id = generate_thread_id_from_slack_thread("C123", thread_ts)
|
|
|
|
asyncio.run(
|
|
webapp.process_slack_mention(
|
|
{
|
|
"channel_id": "C123",
|
|
"thread_ts": thread_ts,
|
|
"event_ts": event_ts,
|
|
"user_id": "U123",
|
|
"text": "<@UBOT> include this screenshot https://example.com/image.png",
|
|
"bot_user_id": "UBOT",
|
|
},
|
|
{"owner": "langchain-ai", "name": "open-swe"},
|
|
)
|
|
)
|
|
|
|
assert captured["active_thread_id"] == expected_thread_id
|
|
assert captured["queued"]["thread_id"] == expected_thread_id
|
|
queued_payload = captured["queued"]["message_content"]
|
|
assert queued_payload["image_urls"] == ["https://example.com/image.png"]
|
|
assert "## Latest Mention Request\ninclude this screenshot" in queued_payload["text"]
|