Extractor

use alith::{Extractor, LLM};
use schemars::JsonSchema;
use serde::{Deserialize, Serialize};
 
#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
struct Person {
    name: String,
    age: usize,
}
 
#[tokio::main]
async fn main() -> Result<(), anyhow::Error> {
    let model = LLM::from_model_name("gpt-4")?;
    let extractor = Extractor::new::<Person>(model);
    let response: Person = extractor.extract("Alice is 18 years old").await?;
    println!("{:?}", response);
    Ok(())
}

from alith import Agent, Extractor
from pydantic import BaseModel
 
 
class Person(BaseModel):
    name: str
    age: int
 
 
print(
    Extractor(
        Agent(
            model="gpt-4",
        ),
        Person,
    ).extract("Alice is 18 years old!")
)

import { Agent, Extractor } from "alith";
import { z } from "zod";
 
export const personSchema = z
  .object({
    name: z.string(),
    age: z.number(),
  })
  .strip();
 
const agent = new Agent({ model: "gpt-4" });
const extractor = new Extractor(agent, personSchema);
console.log(extractor.extract("Alice is 18 years old!"));