Quickstart

pip install weave

pnpm install weave

import weave
from openai import OpenAI

client = OpenAI()

# Weave tracks the inputs, outputs and code of this function
@weave.op()
def extract_dinos(sentence):
    response = client.chat.completions.create(
        model="gpt-4o",
        messages=[
            {
                "role": "system",
                "content": """Extract dinosaurs from the text. Return JSON with:
- dinosaurs: list of dinosaurs
- name: scientific name  
- common_name: nickname
- diet: herbivore or carnivore."""
            },
            {
                "role": "user",
                "content": sentence
            }
        ],
        response_format={ "type": "json_object" }
    )
    return response.choices[0].message.content


# Start the weave project
weave.init('jurassic-park')

sentence = """I saw a T. rex chase a Triceratops. 
The T. rex eats meat. The Triceratops eats plants.
A Brachiosaurus ate leaves from tall trees nearby."""

result = extract_dinos(sentence)
print(result)

import OpenAI from 'openai';
import * as weave from 'weave';

const openai = new OpenAI();

async function extractDinos(input) {
  const response = await openai.chat.completions.create({
    model: 'gpt-4o',
    messages: [
      {
        role: 'user',
        content: `Extract dinosaurs from this text. Return JSON with name, common_name, and diet: ${input}`,
      },
    ],
  });
  return response.choices[0].message.content;
}

const extractDinosOp = weave.op(extractDinos);

async function main() {
  await weave.init('examples');
  const result = await extractDinosOp(
    'I saw a T. rex chase a Triceratops. The T. rex eats meat. The Triceratops eats plants. A Brachiosaurus ate leaves from tall trees nearby.'
  );
  console.log(result);
}

main();