import { OpenAIProvider } from "@workstudio/ai";

const ai = new OpenAIProvider("gpt-4o");

const messages = [
  { role: "user", parts: [{ text: "Hello, how are you?" }] }
];

const struct = {
  instructions: ["You are a helpful assistant."],
  context: [],
  responseSchema: {
    type: "object",
    properties: {
      reply: { type: "string" }
    },
    required: ["reply"]
  }
};

const response = await ai.prompt(messages, struct);
console.log(response.reply);

import { GeminiProvider } from "@workstudio/ai";

const ai = new GeminiProvider(process.env.GEMINI_API_KEY);

const messages = [
  {
    role: "user",
    parts: [
      { text: "What is in this image?" },
      { 
        inlineData: { 
          data: "base64_encoded_image_data", 
          mimeType: "image/jpeg" 
        } 
      }
    ]
  }
];

const response = await ai.prompt(messages, {
  instructions: [],
  context: [],
  responseSchema: null
});

console.log(response.message);

const struct = {
  instructions: ["Help the user with weather and time."],
  context: [],
  responseSchema: null,
  tools: [
    {
      name: "get_weather",
      description: "Get current weather for a location",
      parameters: {
        type: "object",
        properties: {
          location: { type: "string" }
        },
        required: ["location"]
      }
    }
  ]
};

const response = await ai.prompt(messages, struct);
if (response.tool_calls) {
    // Handle tool calling
}