langchain-ai/anthropic-interleaved-thinking-demo
1
0
Fork 0
mirror of https://github.com/langchain-ai/anthropic-interleaved-thinking-demo.git synced 2026-07-01 21:04:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

first commit

Browse Source
This commit is contained in:
Jake Broekhuizen
2025-06-02 17:43:06 -07:00
committed by GitHub
parent 299f17bf2c
commit 707ddbed79
5 changed files with 475 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+4 -1
View File
@@ -1 +1,4 @@
# anthropic-interleaved-thinking-demo
# Anthropic extended thinking demo
Interleaved thinking demo with dummy finance data
Sample trace: https://smith.langchain.com/public/717810b1-f59f-4d79-813c-8f3adb390ef4/r
agent.py
+133
View File
@@ -0,0 +1,133 @@
from langgraph.graph import StateGraph, START, END
from langchain_core.tools import tool
from typing_extensions import TypedDict
from langchain_openai import ChatOpenAI
from typing import Annotated, List
from langgraph.graph.message import AnyMessage, add_messages
from langchain_core.messages import SystemMessage, HumanMessage
from langgraph.prebuilt import ToolNode
from langchain_anthropic import ChatAnthropic
import json
import os
class State(TypedDict):
messages: Annotated[list[AnyMessage], add_messages]
def load_finance_data():
# Get the absolute path to the finance_data.json file
current_dir = os.path.dirname(os.path.abspath(__file__))
finance_data_path = os.path.join(current_dir, "finance_data.json")
# Load the finance data from the JSON file
with open(finance_data_path, "r") as file:
finance_data = json.load(file)
return finance_data
@tool
def get_finance_data():
"""
This tool returns a dataset of company contracts containing information about:
- Company names
- Contract amounts
- Contract lengths (in months)
- Renewal dates
"""
return load_finance_data()
@tool
def multiply_by_pi(number: int):
"""
This tool multiplies a number by pi.
"""
return 3.14159 * number
llm = ChatAnthropic(
model="claude-sonnet-4-20250514",
max_tokens=16000,
thinking={"type": "enabled", "budget_tokens": 10000},
betas=["interleaved-thinking-2025-05-14"],
)
finance_tools = [get_finance_data, multiply_by_pi]
model_with_tools = llm.bind_tools(finance_tools)
finance_tool_node = ToolNode(finance_tools)
def should_continue(state: State):
messages = state["messages"]
last_message = messages[-1]
# If there is no function call, then we finish
if not last_message.tool_calls:
return "end"
# Otherwise if there is, we continue
else:
return "continue"
def finance_assistant(state: State):
# Extract the user's question from the last human message
messages = state["messages"]
# Find the last human message
user_question = ""
for message in reversed(messages):
if message.type == "human":
user_question = message.content
break
# Create a finance expert prompt
finance_expert_prompt = """You are an expert financial analyst with deep knowledge of business contracts and financial planning.
You have access to a dataset of company contracts containing information. If you need to access the data, use the get_finance_data tool, if you need to multiply a number by pi, use the multiply_by_pi tool. If not, answer the user directly.
It will provide the following information:
- Company names
- Contract amounts
- Contract lengths (in months)
- Renewal dates
Please analyze this data to provide helpful, accurate, and insightful answers to the user's questions.
The user's question is: {user_question}
"""
# Call the model with the finance expert prompt and context
response = model_with_tools.invoke(
[SystemMessage(finance_expert_prompt)] + state["messages"]
)
# Return the answer
return {"messages": [response]}
# Build the graph with explicit schemas
builder = StateGraph(State)
builder.add_node("finance_assistant", finance_assistant)
builder.add_node("finance_tool_node", finance_tool_node)
builder.add_edge(START, "finance_assistant")
builder.add_conditional_edges(
"finance_assistant",
should_continue,
{
"continue": "finance_tool_node",
"end": END,
},
)
builder.add_edge("finance_tool_node", "finance_assistant")
builder.add_edge("finance_tool_node", END)
graph = builder.compile()
if __name__ == "__main__":
graph.invoke(
{
"messages": [
HumanMessage(
content="Who are the top 3 contracts by size? Multiply the top by pi"
)
]
}
)
finance_data.json
+302
View File
@@ -0,0 +1,302 @@
[
{
"company_name": "Acme Corporation",
"contract_amount": 45000,
"contract_length": 12,
"renewal_date": "2026-07-15"
},
{
"company_name": "TechSolutions Inc.",
"contract_amount": 78500,
"contract_length": 24,
"renewal_date": "2026-03-22"
},
{
"company_name": "Global Dynamics",
"contract_amount": 125000,
"contract_length": 36,
"renewal_date": "2026-09-01"
},
{
"company_name": "Quantum Industries",
"contract_amount": 32000,
"contract_length": 12,
"renewal_date": "2026-12-10"
},
{
"company_name": "Pioneer Innovations",
"contract_amount": 67800,
"contract_length": 24,
"renewal_date": "2026-08-15"
},
{
"company_name": "Horizon Enterprises",
"contract_amount": 54200,
"contract_length": 18,
"renewal_date": "2026-01-05"
},
{
"company_name": "Summit Systems",
"contract_amount": 96400,
"contract_length": 24,
"renewal_date": "2026-05-28"
},
{
"company_name": "Crestview Analytics",
"contract_amount": 42750,
"contract_length": 12,
"renewal_date": "2026-09-30"
},
{
"company_name": "Nova Tech",
"contract_amount": 115000,
"contract_length": 36,
"renewal_date": "2026-04-12"
},
{
"company_name": "Pinnacle Software",
"contract_amount": 28500,
"contract_length": 6,
"renewal_date": "2026-06-18"
},
{
"company_name": "Atlas Consulting",
"contract_amount": 84300,
"contract_length": 24,
"renewal_date": "2026-11-23"
},
{
"company_name": "Catalyst Media",
"contract_amount": 37500,
"contract_length": 12,
"renewal_date": "2026-08-05"
},
{
"company_name": "Elevate Biotech",
"contract_amount": 156000,
"contract_length": 36,
"renewal_date": "2026-07-19"
},
{
"company_name": "Saphire Health",
"contract_amount": 92000,
"contract_length": 24,
"renewal_date": "2026-02-14"
},
{
"company_name": "Redwood Financial",
"contract_amount": 135000,
"contract_length": 36,
"renewal_date": "2026-10-08"
},
{
"company_name": "Lighthouse Education",
"contract_amount": 48500,
"contract_length": 12,
"renewal_date": "2026-11-30"
},
{
"company_name": "Vertex Manufacturing",
"contract_amount": 72000,
"contract_length": 18,
"renewal_date": "2026-04-25"
},
{
"company_name": "Emerald Retail",
"contract_amount": 35200,
"contract_length": 12,
"renewal_date": "2026-10-15"
},
{
"company_name": "Meridian Logistics",
"contract_amount": 68900,
"contract_length": 24,
"renewal_date": "2026-06-30"
},
{
"company_name": "Phoenix Energy",
"contract_amount": 184500,
"contract_length": 48,
"renewal_date": "2026-03-17"
},
{
"company_name": "Velocity Sports",
"contract_amount": 29800,
"contract_length": 12,
"renewal_date": "2026-05-22"
},
{
"company_name": "Cobalt Security",
"contract_amount": 52000,
"contract_length": 18,
"renewal_date": "2026-12-28"
},
{
"company_name": "Fusion Networks",
"contract_amount": 103000,
"contract_length": 24,
"renewal_date": "2026-09-11"
},
{
"company_name": "Echo Communications",
"contract_amount": 41500,
"contract_length": 12,
"renewal_date": "2026-08-19"
},
{
"company_name": "Orion Aerospace",
"contract_amount": 225000,
"contract_length": 48,
"renewal_date": "2026-01-31"
},
{
"company_name": "Avalon Real Estate",
"contract_amount": 59700,
"contract_length": 18,
"renewal_date": "2026-03-08"
},
{
"company_name": "Equinox Investments",
"contract_amount": 140000,
"contract_length": 36,
"renewal_date": "2026-05-24"
},
{
"company_name": "Pulse Entertainment",
"contract_amount": 32500,
"contract_length": 12,
"renewal_date": "2026-07-29"
},
{
"company_name": "Evergreen Agriculture",
"contract_amount": 78500,
"contract_length": 24,
"renewal_date": "2026-08-05"
},
{
"company_name": "Spectrum Telecom",
"contract_amount": 95400,
"contract_length": 24,
"renewal_date": "2026-02-28"
},
{
"company_name": "Silverline Automotive",
"contract_amount": 62800,
"contract_length": 18,
"renewal_date": "2026-05-12"
},
{
"company_name": "Crystal Waters",
"contract_amount": 27500,
"contract_length": 6,
"renewal_date": "2026-06-04"
},
{
"company_name": "Alpine Outdoors",
"contract_amount": 39000,
"contract_length": 12,
"renewal_date": "2026-09-17"
},
{
"company_name": "Polaris Defense",
"contract_amount": 195000,
"contract_length": 48,
"renewal_date": "2026-06-22"
},
{
"company_name": "Helix Pharmaceuticals",
"contract_amount": 142500,
"contract_length": 36,
"renewal_date": "2026-03-01"
},
{
"company_name": "Kodiak Construction",
"contract_amount": 86700,
"contract_length": 24,
"renewal_date": "2026-01-19"
},
{
"company_name": "Indigo Fashion",
"contract_amount": 34250,
"contract_length": 12,
"renewal_date": "2026-11-05"
},
{
"company_name": "Prism Architecture",
"contract_amount": 72800,
"contract_length": 18,
"renewal_date": "2026-07-14"
},
{
"company_name": "Carbon Footwear",
"contract_amount": 29500,
"contract_length": 12,
"renewal_date": "2026-10-28"
},
{
"company_name": "Zenith Insurance",
"contract_amount": 118000,
"contract_length": 36,
"renewal_date": "2026-01-15"
},
{
"company_name": "Oasis Hospitality",
"contract_amount": 47500,
"contract_length": 12,
"renewal_date": "2026-04-30"
},
{
"company_name": "Magnum Shipping",
"contract_amount": 84000,
"contract_length": 24,
"renewal_date": "2026-10-10"
},
{
"company_name": "Radiant Healthcare",
"contract_amount": 163000,
"contract_length": 36,
"renewal_date": "2026-08-25"
},
{
"company_name": "Vortex Gaming",
"contract_amount": 45800,
"contract_length": 12,
"renewal_date": "2026-12-15"
},
{
"company_name": "Titanium Metals",
"contract_amount": 112500,
"contract_length": 24,
"renewal_date": "2026-04-08"
},
{
"company_name": "Solstice Solar",
"contract_amount": 173000,
"contract_length": 36,
"renewal_date": "2026-11-12"
},
{
"company_name": "Hyperion Data",
"contract_amount": 65200,
"contract_length": 18,
"renewal_date": "2026-02-01"
},
{
"company_name": "Nebula Cloud",
"contract_amount": 58700,
"contract_length": 12,
"renewal_date": "2026-07-07"
},
{
"company_name": "Empire Foods",
"contract_amount": 42000,
"contract_length": 12,
"renewal_date": "2026-10-02"
},
{
"company_name": "Terra Organics",
"contract_amount": 37500,
"contract_length": 12,
"renewal_date": "2026-05-18"
}
]
langgraph.json
+12
View File
@@ -0,0 +1,12 @@
{
"dependencies": [
"."
],
"graphs": {
"agent": {
"path": "./agent.py:graph",
"description": "a test"
}
},
"env": ".env"
}
requirements.txt
+24
View File
@@ -0,0 +1,24 @@
python-dotenv
typing-extensions
langgraph
langchain-core
langchain-openai
langchain-anthropic
langgraph-supervisor
langchain-community
scikit-learn
openai
ipython
langgraph-sdk
langgraph-api
langchain-chroma
beautifulsoup4
ruff
langgraph-cli
langgraph-runtime-inmem
pydantic
langchain
langchain-mcp-adapters
supabase
aiohttp
langchain-anthropic