@arcote.tech/arc-ai-openai 0.5.0 → 0.5.2

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@arcote.tech/arc-ai-openai",
   "type": "module",
-  "version": "0.5.0",
+  "version": "0.5.2",
   "private": false,
   "description": "OpenAI adapter for Arc AI framework",
   "main": "./src/index.ts",
@@ -10,7 +10,7 @@
     "type-check": "tsc --noEmit"
   },
   "peerDependencies": {
-    "@arcote.tech/arc-ai": "^0.5.0",
+    "@arcote.tech/arc-ai": "^0.5.2",
     "typescript": "^5.0.0"
   },
   "devDependencies": {

package/src/index.ts

@@ -31,6 +31,7 @@ export function openai(config: OpenAIConfig): LLMProvider {
       name: t.name,
       description: t.description,
       parameters: t.parameters,
+      strict: false,
     }));
   }
 
@@ -146,6 +147,9 @@ export function openai(config: OpenAIConfig): LLMProvider {
     if (request.webSearch) {
       body.tools = [...(tools ?? []), { type: "web_search_preview" }];
     }
+    if (request.toolChoice) {
+      body.tool_choice = request.toolChoice;
+    }
 
     const response = await fetch(`${baseUrl}/responses`, {
       method: "POST",
@@ -202,6 +206,9 @@ export function openai(config: OpenAIConfig): LLMProvider {
     if (request.webSearch) {
       body.tools = [...(tools ?? []), { type: "web_search_preview" }];
     }
+    if (request.toolChoice) {
+      body.tool_choice = request.toolChoice;
+    }
 
     const response = await fetch(`${baseUrl}/responses`, {
       method: "POST",
@@ -281,10 +288,10 @@ export function openai(config: OpenAIConfig): LLMProvider {
 
             case "response.output_item.done":
               if (event.item?.type === "function_call") {
-                const argsStr =
-                  toolCallArgBuffers.get(event.item.call_id) ??
-                  event.item.arguments ??
-                  "{}";
+                const buffered = toolCallArgBuffers.get(event.item.call_id);
+                const argsStr = (buffered && buffered.length > 0)
+                  ? buffered
+                  : (event.item.arguments ?? "{}");
                 let args: Record<string, unknown> = {};
                 try {
                   args = JSON.parse(argsStr);
@@ -325,19 +332,22 @@ export function openai(config: OpenAIConfig): LLMProvider {
     };
   }
 
+  const pricing: Record<string, { inputPer1M: number; outputPer1M: number; cachedInputPer1M?: number; reasoningPer1M?: number }> = {
+    "gpt-4o": { inputPer1M: 2.50, outputPer1M: 10.00, cachedInputPer1M: 1.25 },
+    "gpt-4o-mini": { inputPer1M: 0.15, outputPer1M: 0.60, cachedInputPer1M: 0.075 },
+    "o3": { inputPer1M: 10.00, outputPer1M: 40.00, reasoningPer1M: 40.00 },
+    "o3-mini": { inputPer1M: 1.10, outputPer1M: 4.40, reasoningPer1M: 4.40 },
+    "gpt-4.1": { inputPer1M: 2.00, outputPer1M: 8.00, cachedInputPer1M: 0.50 },
+    "gpt-4.1-mini": { inputPer1M: 0.40, outputPer1M: 1.60, cachedInputPer1M: 0.10 },
+    "gpt-4.1-nano": { inputPer1M: 0.10, outputPer1M: 0.40, cachedInputPer1M: 0.025 },
+    "gpt-5.4-nano": { inputPer1M: 0.10, outputPer1M: 0.40, cachedInputPer1M: 0.025 },
+  };
+
   return {
     name: "openai",
-    models: [
-      "gpt-4o",
-      "gpt-4o-mini",
-      "o3",
-      "o3-mini",
-      "gpt-4.1",
-      "gpt-4.1-mini",
-      "gpt-4.1-nano",
-      "gpt-5.4-nano",
-    ],
+    models: Object.keys(pricing),
     complete,
     streamComplete,
+    getPricing: (model: string) => pricing[model],
   };
 }