Merge branch 'main' into redirect_readme

Author: amychen1776

.changes/unreleased/Under the Hood-20250811-163302.yaml

@@ -0,0 +1,3 @@
+kind: Under the Hood
+body: LangGraph create_react_agent example
+time: 2025-08-11T16:33:02.306539-05:00

.changes/v0.4.2.md

@@ -0,0 +1,7 @@
+## v0.4.2 - 2025-08-13
+### Enhancement or New Feature
+* Add default --limit to show tool
+### Under the Hood
+* Define toolsets
+### Bug Fix
+* Fix the prompt to ensure grain is passed even for non-time group by"

CHANGELOG.md

@@ -7,6 +7,14 @@ adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html),
 and is generated by [Changie](https://github.com/miniscruff/changie).
 
 
+## v0.4.2 - 2025-08-13
+### Enhancement or New Feature
+* Add default --limit to show tool
+### Under the Hood
+* Define toolsets
+### Bug Fix
+* Fix the prompt to ensure grain is passed even for non-time group by"
+
 ## v0.4.1 - 2025-08-08
 ### Under the Hood
 * Upgrade dbt-sl-sdk

examples/langgraph_agent/.python-version

@@ -0,0 +1 @@
+3.13

examples/langgraph_agent/README.md

@@ -0,0 +1,8 @@
+# LangGraph Agent Example
+
+This is a simple example of how to create conversational agent with remote dbt MCP & LangGraph.
+
+## Usage
+
+1. Set an `ANTHROPIC_API_KEY` environment variable with your Anthropic API key.
+2. Run `uv run main.py`.

examples/langgraph_agent/__init__.py

@@ -0,0 +1,67 @@
+# mypy: ignore-errors
+
+import asyncio
+import os
+
+from langchain_mcp_adapters.client import MultiServerMCPClient
+from langgraph.checkpoint.memory import InMemorySaver
+from langgraph.prebuilt import create_react_agent
+
+
+def print_stream_item(item):
+    if "agent" in item:
+        content = [
+            part
+            for message in item["agent"]["messages"]
+            for part in (
+                message.content
+                if isinstance(message.content, list)
+                else [message.content]
+            )
+        ]
+        for c in content:
+            if isinstance(c, str):
+                print(f"Agent > {c}")
+            elif "text" in c:
+                print(f"Agent > {c['text']}")
+            elif c["type"] == "tool_use":
+                print(f"    using tool: {c['name']}")
+
+
+async def main():
+    url = f"https://{os.environ.get('DBT_HOST')}/api/ai/v1/mcp/"
+    headers = {
+        "x-dbt-user-id": os.environ.get("DBT_USER_ID"),
+        "x-dbt-prod-environment-id": os.environ.get("DBT_PROD_ENV_ID"),
+        "x-dbt-dev-environment-id": os.environ.get("DBT_DEV_ENV_ID"),
+        "Authorization": f"token {os.environ.get('DBT_TOKEN')}",
+    }
+    client = MultiServerMCPClient(
+        {
+            "dbt": {
+                "url": url,
+                "headers": headers,
+                "transport": "streamable_http",
+            }
+        }
+    )
+    tools = await client.get_tools()
+    agent = create_react_agent(
+        model="anthropic:claude-3-7-sonnet-latest",
+        tools=tools,
+        # This allows the agent to have conversational memory.
+        checkpointer=InMemorySaver(),
+    )
+    # This config maintains the conversation thread.
+    config = {"configurable": {"thread_id": "1"}}
+    while True:
+        user_input = input("User > ")
+        async for item in agent.astream(
+            {"messages": {"role": "user", "content": user_input}},
+            config,
+        ):
+            print_stream_item(item)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

examples/langgraph_agent/main.py

@@ -0,0 +1,11 @@
+[project]
+name = "langgraph-agent"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "langchain-mcp-adapters>=0.1.9",
+    "langchain[anthropic]>=0.3.27",
+    "langgraph>=0.6.4",
+]

examples/langgraph_agent/pyproject.toml

@@ -117,9 +117,7 @@ def test(
 
     def show(
         sql_query: str = Field(description=get_prompt("dbt_cli/args/sql_query")),
-        limit: int | None = Field(
-            default=None, description=get_prompt("dbt_cli/args/limit")
-        ),
+        limit: int = Field(default=5, description=get_prompt("dbt_cli/args/limit")),
     ) -> str:
         args = ["show", "--inline", sql_query, "--favor-state"]
         # This is quite crude, but it should be okay for now