Wire the s. 36 screening tools into the agentic loop

Declares canlex_us_disposition, canlex_us_equivalency and
canlex_rehabilitation to Gemini, adds the three-step US-record screening
flow to the system instruction, and gives each tool a call-trace line.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>

app.py

@@ -175,11 +175,109 @@ TOOL_DECLARATIONS = [
             "required": ["case_url"],
         },
     },
+    {
+        "name": "canlex_us_disposition",
+        "description": (
+            "Assess whether a US criminal disposition (deferred adjudication, "
+            "withheld adjudication, nolo plea, expungement, state pardon, "
+            "diversion, juvenile adjudication...) is a 'conviction' for IRPA "
+            "s. 36 inadmissibility, with a 51-jurisdiction state-by-state "
+            "breakdown. Use for ANY question involving a US criminal record. "
+            "Step 1 of the s. 36 screening flow."
+        ),
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "disposition": {
+                    "type": "string",
+                    "description": (
+                        "The disposition in the record's own words, e.g. "
+                        "'deferred adjudication', 'withheld adjudication', "
+                        "'expunged conviction' -- or 'all' for the full table."
+                    ),
+                },
+                "state": {
+                    "type": "string",
+                    "description": "Optional US state name or 2-letter code.",
+                },
+            },
+            "required": ["disposition"],
+        },
+    },
+    {
+        "name": "canlex_us_equivalency",
+        "description": (
+            "Map a common US offence (DUI, theft, assault, fraud, drug "
+            "possession, disorderly conduct...) to its Canadian equivalent "
+            "offence, verified maximum penalty, and the IRPA s. 36 branch "
+            "(serious criminality vs criminality vs none). Step 2 of the "
+            "s. 36 screening flow; its max-penalty answer feeds step 3."
+        ),
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "offense": {
+                    "type": "string",
+                    "description": "The US offence as charged, e.g. 'DUI'.",
+                },
+            },
+            "required": ["offense"],
+        },
+    },
+    {
+        "name": "canlex_rehabilitation",
+        "description": (
+            "Assess deemed rehabilitation / eligibility to apply for "
+            "individual rehabilitation under IRPA s. 36(3)(c) and IRPR "
+            "ss. 17-18.1, from offence severity, count and time elapsed. "
+            "Step 3 of the s. 36 screening flow."
+        ),
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "basis": {
+                    "type": "string",
+                    "description": (
+                        "'foreign-conviction', 'foreign-act', or "
+                        "'canada-conviction'."
+                    ),
+                },
+                "max_penalty_lt_10": {
+                    "type": "boolean",
+                    "description": (
+                        "True if the equivalent Canadian offence's maximum is "
+                        "under 10 years (see canlex_us_equivalency)."
+                    ),
+                },
+                "count": {
+                    "type": "string",
+                    "description": (
+                        "'one', 'two-plus-summary-only', or 'multiple'."
+                    ),
+                },
+                "years_elapsed": {
+                    "type": "integer",
+                    "description": (
+                        "Full years since the day after the entire sentence "
+                        "was completed (or since commission, for acts)."
+                    ),
+                },
+                "clean_record": {
+                    "type": "boolean",
+                    "description": "True if no other convictions or acts.",
+                },
+            },
+            "required": ["basis", "max_penalty_lt_10", "count",
+                         "years_elapsed", "clean_record"],
+        },
+    },
 ]
 
-# The three tools whose MCP signature wraps arguments under a single 'params'
-# object. canlex_list_acts takes none and is handled separately in _run_tool.
-_PARAMS_WRAPPED = {"canlex_search_legislation", "canlex_get_section", "canlex_case"}
+# Tools whose MCP signature wraps arguments under a single 'params' object.
+# canlex_list_acts takes none and is handled separately in _run_tool.
+_PARAMS_WRAPPED = {"canlex_search_legislation", "canlex_get_section",
+                   "canlex_case", "canlex_us_disposition",
+                   "canlex_us_equivalency", "canlex_rehabilitation"}
 
 
 # --- System prompt ------------------------------------------------------------
@@ -187,7 +285,7 @@ _PARAMS_WRAPPED = {"canlex_search_legislation", "canlex_get_section", "canlex_ca
 SYSTEM_INSTRUCTION = """\
 You are CanLex Web, a Canadian legal-research assistant. A member of the public \
 has asked the legal question shown below through a web form. Answer it by \
-agentically using the four CanLex tools to retrieve primary sources, then \
+agentically using the CanLex tools to retrieve primary sources, then \
 compose a clear, well-organised answer grounded entirely in what those tools \
 return.
 
@@ -199,6 +297,13 @@ bears on the question but is not reproduced, call canlex_get_section or \
 canlex_search_legislation again to fetch it. Do not guess its contents.
 - For a question that turns on case law, consider calling canlex_case on the \
 leading decision's neutral citation to confirm it has not been overtaken.
+- US CRIMINAL RECORDS -- the three-step screening flow: (1) \
+canlex_us_disposition: is the disposition a conviction at all (state-aware)? \
+(2) canlex_us_equivalency: what Canadian offence does it equate to, and is it \
+serious criminality or criminality? (3) canlex_rehabilitation: has time cured \
+it (deemed rehabilitation / eligibility to apply)? Chain all three when the \
+question involves a US record and enough facts are given; ask for the missing \
+facts (state, disposition wording, dates) when they are not.
 - You may call tools multiple times; iterate until you have enough material to \
 answer well. Aim for thoroughness but stop once further calls would not change \
 the answer.
@@ -361,6 +466,15 @@ def _summarize_call(name: str, args: dict) -> str:
         return f"Looking up case {args.get('case_url', '?')}"
     if name == "canlex_list_acts":
         return "Listing the CanLex corpus"
+    if name == "canlex_us_disposition":
+        state = f" ({args['state']})" if args.get("state") else ""
+        return (f"Checking US disposition "
+                f"`{args.get('disposition', '?')}`{state}")
+    if name == "canlex_us_equivalency":
+        return f"Mapping US offence `{args.get('offense', '?')}` to Canadian law"
+    if name == "canlex_rehabilitation":
+        return (f"Assessing rehabilitation "
+                f"({args.get('years_elapsed', '?')} years elapsed)")
     return f"Calling {name}"