from ai21 import AI21Client
from ai21.models.chat import ChatMessage

client = AI21Client(api_key="YOUR_API_KEY")

# Create and poll a Maestro run
run = client.beta.maestro.runs.create_and_poll(
    input="Summarize today's major movements in global financial markets and explain possible reasons for these changes.",
    requirements=[
        {"name": "factual_accuracy",
            "description": "Base insights on verifiable data from reliable financial sources."},
        {"name": "clarity", "description": "Keep explanations concise and free of jargon."},
        {"name": "tone", "description": "Maintain a neutral, professional tone suitable for enterprise reporting."}
    ],
    tools=[
        {
            "type": "web_search",
            "urls": ["https://www.reuters.com/markets/", "https://finance.yahoo.com/"]
        }
    ],
    include=["requirements_result", "data_sources"],
    budget="medium"
)

# Print the generated analysis
print("=== Financial Market Summary ===")
print(run.result)
print()

# Print requirement evaluation
print("=== Requirements Results ===")
for req in run.requirements_result["requirements"]:
    print(f"- {req['name']}: {req['score']} ({req['reason']})")
print()

# Print data sources
print("=== Data Sources ===")
if hasattr(run, 'data_sources') and run.data_sources:
    if run.data_sources.get('web_search'):
        print("Web Search Results:")
        for result in run.data_sources['web_search']:
            print(f"  - {result.get('url')}")
else:
    print("No data sources available")

// TypeScript example using AI21 Labs TypeScript SDK
// Install deps first: npm i ai21 ts-node typescript
// If the import path differs in your SDK version, adjust it per the SDK docs.
import { AI21} from "ai21";

const apiKey = process.env.AI21_API_KEY;
if (!apiKey) {
  throw new Error("AI21_API_KEY is not set. Please export it as an environment variable.");
}

const client = new AI21({
  apiKey,
});

const body = {
  input: [
    {
      role: "user",
      content:
        "Summarize today's major movements in global financial markets and explain possible reasons for these changes.",
    },
  ],
  requirements: [
    {
      name: "factual_accuracy",
      description:
        "Base insights on verifiable data from reliable financial sources.",
    },
    {
      name: "clarity",
      description: "Keep explanations concise and free of jargon.",
    },
    {
      name: "tone",
      description:
        "Maintain a neutral, professional tone suitable for enterprise reporting.",
    },
  ],
  tools: [
    {
      type: "web_search",
      urls: [
        "https://www.reuters.com/markets/",
        "https://finance.yahoo.com/",
      ],
    },
  ],
  budget: "low",
  include: ["requirements_result", "data_sources"],
};

async function main() {
  const result = await client.beta.maestro.runs.createAndPoll(body, {
    interval: 1000,
    timeout: 100000
  });
  console.log(JSON.stringify(result, null, 2));
}

main().catch((err) => {
  console.error(err);
  process.exit(1);
});

<?php

$curl = curl_init();

curl_setopt_array($curl, array(
  CURLOPT_URL => 'https://api.ai21.com/studio/v1/maestro/runs',
  CURLOPT_RETURNTRANSFER => true,
  CURLOPT_FOLLOWLOCATION => true,
  CURLOPT_CUSTOMREQUEST => 'POST',
  CURLOPT_HTTPHEADER => array(
    'Authorization: Bearer YOUR_API_KEY_HERE',
    'Content-Type: application/json'
  ),
  CURLOPT_POSTFIELDS => '{
    "input": [
      {
        "role": "user",
        "content": "Summarize today\'s major movements in global financial markets and explain possible reasons for these changes."
      }
    ],
    "requirements": [
      {
        "name": "factual_accuracy",
        "description": "Base insights on verifiable data from reliable financial sources."
      },
      {
        "name": "clarity",
        "description": "Keep explanations concise and free of jargon."
      },
      {
        "name": "tone",
        "description": "Maintain a neutral, professional tone suitable for enterprise reporting."
      }
    ],
    "tools": [
      {
        "type": "web_search",
        "urls": [
          "https://www.reuters.com/markets/",
          "https://finance.yahoo.com/"
        ]
      }
    ]
  }'
));

$response = curl_exec($curl);
curl_close($curl);

echo $response;

curl --location --request POST 'https://api.ai21.com/studio/v1/maestro/runs' \
  --header 'Authorization: Bearer <YOUR_API_KEY>' \
  --header 'Content-Type: application/json' \
  --data '{
    "input": [
      {
        "role": "user",
        "content": "Summarize today'\''s major movements in global financial markets and explain possible reasons for these changes."
      }
    ],
    "requirements": [
      {
        "name": "factual_accuracy",
        "description": "Base insights on verifiable data from reliable financial sources."
      },
      {
        "name": "clarity",
        "description": "Keep explanations concise and free of jargon."
      },
      {
        "name": "tone",
        "description": "Maintain a neutral, professional tone suitable for enterprise reporting."
      }
    ],
    "tools": [
      {
        "type": "web_search",
        "urls": [
          "https://www.reuters.com/markets/",
          "https://finance.yahoo.com/"
        ]
      }
    ],
    "budget": "medium",
    "include": ["requirements_result", "data_sources"]
  }'

HttpResponse<String> response = Unirest.post("https://api.ai21.com/studio/v1/maestro/runs")
  .header("Content-Type", "application/json")
  .body("{\n  \"input\": [\n    {\n      \"role\": \"<string>\",\n      \"content\": \"<string>\"\n    }\n  ],\n  \"requirements\": [\n    {\n      \"name\": \"<string>\",\n      \"description\": \"<string>\"\n    }\n  ],\n  \"tool_resources\": {\n    \"file_search\": {\n      \"labels\": [\n        \"<string>\"\n      ],\n      \"file_ids\": [\n        \"<string>\"\n      ]\n    },\n    \"web_search\": {\n      \"urls\": [\n        \"<string>\"\n      ]\n    }\n  },\n  \"models\": [\n    \"<string>\"\n  ],\n  \"budget\": \"<string>\",\n  \"include\": [\n    \"<string>\"\n  ],\n  \"response_language\": {}\n}")
  .asString();

{
  "input": [
    {
      "role": "user",
      "content": "Summarize today's major movements in global financial markets and explain possible reasons for these changes."
    }
  ],
  "requirements": [
    {
      "name": "factual_accuracy",
      "description": "Base insights on verifiable data from reliable financial sources."
    },
    {
      "name": "clarity",
      "description": "Keep explanations concise and free of jargon."
    },
    {
      "name": "tone",
      "description": "Maintain a neutral, professional tone suitable for enterprise reporting."
    }
  ],
  "tools": [
    {
      "type": "web_search",
      "urls": [
        "https://www.reuters.com/markets/",
        "https://finance.yahoo.com/"
      ]
    }
  ],
  "budget": "medium",
  "include": ["requirements_result", "data_sources"]
}