KI-Agenten in YAML.
Tools in Python.

from typing import Any, Dict
import httpx

async def before(content: Dict[str, Any], context: Dict[str, Any]) -> Dict[str, Any]:
    """Give the agent the customer's booking rules before it reads the invoice."""
    customer_id = context.get("payload", {}).get("customer_id")
    async with httpx.AsyncClient() as client:
        resp = await client.get(f"https://erp.example.com/customers/{customer_id}/rules")

    content["parts"].insert(0, {"text": f"Booking rules: {resp.text}"})
    return content

async def after(response: str, context: Dict[str, Any]) -> str:
    print(f"Run {context.get('run_id')} finished")
    return response

import re
from typing import Any
from connic import AbortTool

IBAN = re.compile(r"\b[A-Z]{2}\d{2}[A-Z0-9]{11,30}\b")

async def before(tool_name: str, params: dict[str, Any]) -> dict[str, Any]:
    if tool_name == "save_invoice" and params.get("total", 0) > 50_000:
        raise AbortTool({"error": "Totals above 50,000 need a manual booking."})
    return params

async def after(tool_name: str, params: dict[str, Any], result: Any) -> Any:
    if tool_name == "parse_pdf":
        result["text"] = IBAN.sub("[IBAN]", result["text"])
    return result

import json
from connic import GuardrailResult

def check(content: str, context: dict) -> GuardrailResult:
    """Block answers whose line items do not add up to the total."""
    invoice = json.loads(content)
    line_sum = sum(line["amount"] for line in invoice["lines"])

    if abs(line_sum - invoice["total"]) > 0.01:
        return GuardrailResult(
            passed=False,
            message="Line items do not match the total.",
        )

    return GuardrailResult(passed=True)

{
  "type": "object",
  "description": "Extracted invoice data",
  "properties": {
    "vendor": { "type": "string" },
    "invoice_date": { "type": "string", "description": "YYYY-MM-DD" },
    "currency": { "type": "string" },
    "lines": {
      "type": "array",
      "items": {
        "type": "object",
        "properties": {
          "description": { "type": "string" },
          "amount": { "type": "number" }
        },
        "required": ["description", "amount"]
      }
    },
    "total": { "type": "number" }
  },
  "required": ["vendor", "lines", "total"]
}

version: "1.0"

tests:
  - name: extracts_totals
    payload: "Extract this invoice"
    files:
      - invoice_acme.pdf
    expected_tool_calls:
      - invoices.extract_fields
    expected_result: output.total == 1249.5

  - name: reads_currency
    payload: "Extract this invoice"
    files:
      - invoice_globex_chf.pdf
    expected_result: output.currency == "CHF"