Decide if a draft is ready to send

from sqwishai import OneHot

with OneHot() as client:  # reads SQWISH_API_KEY
    result = client.decide(
        model="kettei-core",
        context={
            "conversation": [
                {
                    "role": "user",
                    "content": "Write a Python function that removes duplicates from a list but keeps the original order.",
                },
            ],
            "draft": "Use a set:\n\ndef dedupe(items):\n    return list(set(items))\n\nA set can't hold the same value twice, so every item appears once.",
        },
        decisions=[
            {
                "id": "quality",
                "kind": "ordinal",
                "question": "Is this draft reply ready to send to the user?",
                "outcomes": [
                    "0",
                    "1",
                    "2",
                ],
                "rubric": [
                    "No. It is wrong, unsafe or unhelpful, or it ignores part of what the user asked.",
                    "Nearly. It is on the right track, but has an error, a gap or a missing step to fix first.",
                    "Yes. It is correct and complete, and does what the user asked.",
                ],
            },
        ],
    )

import { OneHot } from "sqwishai";

const client = new OneHot(); // reads SQWISH_API_KEY
const result = await client.decide({
  "model": "kettei-core",
  "context": {
    "conversation": [
      {
        "role": "user",
        "content": "Write a Python function that removes duplicates from a list but keeps the original order."
      }
    ],
    "draft": "Use a set:\n\ndef dedupe(items):\n    return list(set(items))\n\nA set can't hold the same value twice, so every item appears once."
  },
  "decisions": [
    {
      "id": "quality",
      "kind": "ordinal",
      "question": "Is this draft reply ready to send to the user?",
      "outcomes": [
        "0",
        "1",
        "2"
      ],
      "rubric": [
        "No. It is wrong, unsafe or unhelpful, or it ignores part of what the user asked.",
        "Nearly. It is on the right track, but has an error, a gap or a missing step to fix first.",
        "Yes. It is correct and complete, and does what the user asked."
      ]
    }
  ]
});

curl --fail-with-body --silent --show-error https://api.sqwish.ai/v1/decide \
  -H "Authorization: Bearer $SQWISH_API_KEY" \
  -H "Content-Type: application/json" \
  --data-binary @- <<'JSON'
{
  "model": "kettei-core",
  "context": {
    "conversation": [
      {
        "role": "user",
        "content": "Write a Python function that removes duplicates from a list but keeps the original order."
      }
    ],
    "draft": "Use a set:\n\ndef dedupe(items):\n    return list(set(items))\n\nA set can't hold the same value twice, so every item appears once."
  },
  "decisions": [
    {
      "id": "quality",
      "kind": "ordinal",
      "question": "Is this draft reply ready to send to the user?",
      "outcomes": [
        "0",
        "1",
        "2"
      ],
      "rubric": [
        "No. It is wrong, unsafe or unhelpful, or it ignores part of what the user asked.",
        "Nearly. It is on the right track, but has an error, a gap or a missing step to fix first.",
        "Yes. It is correct and complete, and does what the user asked."
      ]
    }
  ]
}
JSON

p_ready = result["decisions"]["quality"]["probabilities"]["2"]  # the keys are strings

if p_ready > 0.5:  # start here, and raise it until you would send what it lets through
    ...  # send the draft
else:
    ...  # write it again, or pass it to a person

const pReady = result.decisions.quality.probabilities["2"]; // the keys are strings

if (pReady > 0.5) { // start here, and raise it until you would send what it lets through
  // send the draft
} else {
  // write it again, or pass it to a person
}