aieo 0.1.23 → 0.1.24

package/dist/index.cjs

@@ -35,13 +35,15 @@ __export(index_exports, {
   OUTRO: () => OUTRO,
   PROVIDERS: () => PROVIDERS,
   SYSTEM: () => SYSTEM,
+  callGenerateObject: () => callGenerateObject,
   callModel: () => callModel,
+  getApiKeyForProvider: () => getApiKeyForProvider,
   getModel: () => getModel,
   getProviderOptions: () => getProviderOptions
 });
 module.exports = __toCommonJS(index_exports);
 
-// ../node_modules/uuid/dist/esm/stringify.js
+// ../../node_modules/uuid/dist/esm/stringify.js
 var byteToHex = [];
 for (let i = 0; i < 256; ++i) {
   byteToHex.push((i + 256).toString(16).slice(1));
@@ -50,7 +52,7 @@ function unsafeStringify(arr, offset = 0) {
   return (byteToHex[arr[offset + 0]] + byteToHex[arr[offset + 1]] + byteToHex[arr[offset + 2]] + byteToHex[arr[offset + 3]] + "-" + byteToHex[arr[offset + 4]] + byteToHex[arr[offset + 5]] + "-" + byteToHex[arr[offset + 6]] + byteToHex[arr[offset + 7]] + "-" + byteToHex[arr[offset + 8]] + byteToHex[arr[offset + 9]] + "-" + byteToHex[arr[offset + 10]] + byteToHex[arr[offset + 11]] + byteToHex[arr[offset + 12]] + byteToHex[arr[offset + 13]] + byteToHex[arr[offset + 14]] + byteToHex[arr[offset + 15]]).toLowerCase();
 }
 
-// ../node_modules/uuid/dist/esm/rng.js
+// ../../node_modules/uuid/dist/esm/rng.js
 var import_crypto = require("crypto");
 var rnds8Pool = new Uint8Array(256);
 var poolPtr = rnds8Pool.length;
@@ -62,11 +64,11 @@ function rng() {
   return rnds8Pool.slice(poolPtr, poolPtr += 16);
 }
 
-// ../node_modules/uuid/dist/esm/native.js
+// ../../node_modules/uuid/dist/esm/native.js
 var import_crypto2 = require("crypto");
 var native_default = { randomUUID: import_crypto2.randomUUID };
 
-// ../node_modules/uuid/dist/esm/v4.js
+// ../../node_modules/uuid/dist/esm/v4.js
 function v4(options, buf, offset) {
   if (native_default.randomUUID && !buf && !options) {
     return native_default.randomUUID();
@@ -8192,9 +8194,30 @@ var PROVIDERS = [
 var SOTA = {
   anthropic: "claude-4-sonnet-20250514",
   google: "gemini-2.5-pro-preview-05-06",
-  openai: "gpt-4.1",
+  openai: "gpt-5",
   claude_code: "sonnet"
 };
+function getApiKeyForProvider(provider) {
+  let apiKey;
+  switch (provider) {
+    case "anthropic":
+      apiKey = process.env.ANTHROPIC_API_KEY;
+      break;
+    case "google":
+      apiKey = process.env.GOOGLE_API_KEY;
+      break;
+    case "openai":
+      apiKey = process.env.OPENAI_API_KEY;
+      break;
+    case "claude_code":
+      apiKey = process.env.CLAUDE_CODE_API_KEY;
+      break;
+  }
+  if (!apiKey) {
+    throw new Error(`API key not found for provider: ${provider}`);
+  }
+  return apiKey;
+}
 async function getModel(provider, apiKey, cwd, executablePath) {
   switch (provider) {
     case "anthropic":
@@ -8303,6 +8326,15 @@ async function callModel(opts) {
   }
   return fullResponse;
 }
+async function callGenerateObject(args) {
+  const model = await getModel(args.provider, args.apiKey);
+  const { object } = await (0, import_ai.generateObject)({
+    model,
+    schema: args.schema,
+    prompt: args.prompt
+  });
+  return object;
+}
 
 // src/prompt.ts
 var SYSTEM = `You are an expert dev. You are given a codebase and a task. You need to write the code for the task. You can also suggest changes to the codebase if needed.
@@ -8329,7 +8361,9 @@ Please write all the necessary code to fully implement the feature end-to-end. D
   OUTRO,
   PROVIDERS,
   SYSTEM,
+  callGenerateObject,
   callModel,
+  getApiKeyForProvider,
   getModel,
   getProviderOptions
 });