ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter5/dynamic-form/test_round_trip_destination.py

@@ -0,0 +1,57 @@
+import subprocess
+import sys
+from pathlib import Path
+
+import pytest
+
+
+@pytest.mark.parametrize(
+    ("user_request", "expected_destination"),
+    [
+        ("我想订一张去上海的机票", "上海"),
+        ("我想订一张去上海的往返机票", "上海"),
+        ("我想订一张去上海往返的机票", "上海"),
+        ("我想订一张去上海往返机票", "上海"),
+        ("我想订一张去广州的单程机票", "广州"),
+        ("我想订一张去广州单程的票", "广州"),
+        ("我想订一张去广州单程机票", "广州"),
+        ("我想订一张去深圳的航班", "深圳"),
+        ("我想订一张去深圳往返的航班", "深圳"),
+        ("我想订一张去杭州单程的票", "杭州"),
+    ],
+    ids=[
+        "simple-jipiao",
+        "round-trip-with-de-before",
+        "round-trip-with-de-after",
+        "round-trip-without-de",
+        "one-way-with-de-before",
+        "one-way-with-de-after",
+        "one-way-without-de",
+        "hangban-simple",
+        "hangban-round-trip",
+        "piao-one-way",
+    ],
+)
+def test_offline_cli_extracts_only_the_destination(user_request, expected_destination, tmp_path):
+    """Trip-type modifiers must not become part of the submitted destination."""
+    demo = Path(__file__).with_name("demo.py")
+    output_html = tmp_path / "form.html"
+    result = subprocess.run(
+        [
+            sys.executable,
+            str(demo),
+            "--offline",
+            "--request",
+            user_request,
+            "--output",
+            str(output_html),
+        ],
+        capture_output=True,
+        text=True,
+    )
+
+    assert result.returncode == 0, result.stderr
+    assert f'"destination_city": "{expected_destination}"' in result.stdout
+    assert f"已收到您的订票信息：上海 → {expected_destination}，出发日期" in result.stdout
+    form_html_content = output_html.read_text(encoding="utf-8")
+    assert f'"destination_city": "{expected_destination}"' in form_html_content