Quickstart

npm install agentset

yarn add agentset

pnpm add agentset

bun add agentset

pip install agentset

import { Agentset } from "agentset";

const agentset = new Agentset({
  apiKey: process.env.AGENTSET_API_KEY,
});

const ns = agentset.namespace("YOUR_NAMESPACE_ID");

const job = await ns.ingestion.create({
  name: "Attention Is All You Need",
  payload: {
    type: "FILE",
    fileUrl: "https://arxiv.org/pdf/1706.03762.pdf",
  },
});

console.log(`Uploaded: ${job.id}`);

import os
from agentset import Agentset

client = Agentset(
    namespace_id="YOUR_NAMESPACE_ID",
    token=os.environ["AGENTSET_API_KEY"],
)

job = client.ingest_jobs.create(
    name="Attention Is All You Need",
    payload={
        "type": "FILE",
        "fileUrl": "https://arxiv.org/pdf/1706.03762.pdf",
    }
)

print(f"Uploaded: {job.data.id}")

const results = await ns.search("What is multi-head attention?");

for (const result of results) {
  console.log(result.text);
}

results = client.search.execute(query="What is multi-head attention?")

for result in results.data:
    print(result.text)

import { generateText } from "ai";
import { openai } from "@ai-sdk/openai";

const results = await ns.search("What is multi-head attention?");
const context = results.map((r) => r.text).join("\n\n");

const { text } = await generateText({
  model: openai("gpt-5.1"),
  system: `Answer questions based on the following context:\n\n${context}`,
  prompt: "What is multi-head attention?",
});

console.log(text);

from openai import OpenAI as OpenAIClient

openai = OpenAIClient()

results = client.search.execute(query="What is multi-head attention?")
context = "\n\n".join([r.text for r in results.data])

response = openai.responses.create(
    model="gpt-4.1",
    input=[
        {
            "role": "system",
            "content": f"Answer questions based on the following context:\n\n{context}",
        },
        {
            "role": "user",
            "content": "What is multi-head attention?",
        },
    ],
)

print(response.output_text)