import time

# 版本 A：invoke 一次性（小林原来的写法）
t0 = time.time()
resp = model.invoke("从 1 数到 5，每行一个数字")
print(f"耗时 {time.time()-t0:.1f}s，一次性输出：{resp.content}")

# 版本 B：stream 流式（小林的新写法）
t0 = time.time()
collected = ""
for chunk in model.stream("从 1 数到 5，每行一个数字"):
    collected \+= chunk.content or ""
    print(chunk.content, end="", flush=True)   # flush=True：立即打印，不打缓冲
print(f"\n耗时 {time.time()-t0:.1f}s，流式输出完毕")

版本 A：耗时 3.2s，一次性输出：1 2 3 4 5
版本 B：1 2 3 4 5（逐个蹦出来），耗时 0.8s 就看到了第一个数字

resp = model.invoke("把这句话整理成JSON：我上周买的东西没发货！订单号20260701001，明天就要用了！")
print(resp.content)
# → "好的，我为您整理如下：\n{\n  \"intent\": \"投诉\",\n  \"order_id\": ...\n}\n希望这能帮到您！"

from pydantic import BaseModel, Field

class SupportTicket(BaseModel):
    intent: str = Field(description="用户意图：物流查询/退换货/投诉/其他")
    order_id: str | None = Field(description="订单号，用户没提则为空", default=None)
    urgency: int = Field(description="紧急程度 1~5，5 最紧急")

# 关键：把模型"变成"只输出 SupportTicket 的模型
structured_model = model.with_structured_output(SupportTicket)

ticket = structured_model.invoke(
    "我上周买的东西到现在没发货！订单号20260701001，你们到底在搞什么？？？明天就要用了！"
)
print(ticket)
# → intent='投诉' order_id='20260701001' urgency=5   ← 一个真正的 Pydantic 对象！

if ticket.intent == "投诉":
    print(f"→ 转人工处理（意图：{ticket.intent}，紧急度：{ticket.urgency}）")
print(f"→ 关联订单号：{ticket.order_id}")

model.stream(input)   # → 迭代器，每个元素是 AIMessageChunk
for chunk in model.stream("你好"):
    print(chunk.content)     # 一片文字
    print(chunk.response_metadata)  # 首块是空壳，末块带 stop_reason/usage
# 流式完成后的"收尾"信号：chunk.response_metadata["stop_reason"] == "end_turn"

StrOutputParser().parse(text: str) -> str    # 纯透传，等于 lambda x: x
# 但 invoke 输入可以是 AIMessage，自动取 .content
parser.invoke(ai_msg)   # → str

model.with_structured_output(
    schema,                # Pydantic 类，如 SupportTicket
    method="function_calling",  # 底层实现：function_calling / json_mode / json_schema
    **kwargs,
) -> Runnable   # invoke 后直接返回 Pydantic 对象（不是 AIMessage！）

from pydantic import BaseModel, Field
class SupportTicket(BaseModel):
    intent: str = Field(description="用户意图")
    order_id: str | None = Field(default=None, description="订单号，没提则为空")
    urgency: int = Field(description="紧急程度 1~5")
# with_structured_output 返回：SupportTicket(intent='投诉', order_id='...', urgency=5)