@gammatech/aijsx 0.10.0-dev.2024-06-04.2 → 0.10.1-dev.2024-06-07

package/dist/index.d.mts

@@ -87,7 +87,6 @@ declare const tracing: AITraceAPI;
 
 declare class SpanTree {
     roots: SpanTreeNode[];
-    findSpan: any;
     addSpan(span: ReadableSpan): boolean;
     findNode(id: string): SpanTreeNode | null;
 }

package/dist/index.d.ts

@@ -87,7 +87,6 @@ declare const tracing: AITraceAPI;
 
 declare class SpanTree {
     roots: SpanTreeNode[];
-    findSpan: any;
     addSpan(span: ReadableSpan): boolean;
     findNode(id: string): SpanTreeNode | null;
 }

package/dist/index.js

@@ -1587,7 +1587,6 @@ async function* Fallback({ shouldFallback = () => true, fallback, children }, ct
 // src/tracing/SpanTree.ts
 var SpanTree = class {
   roots = [];
-  findSpan;
   addSpan(span) {
     const parentId = span.parentSpanId;
     if (!parentId) {
@@ -2574,6 +2573,33 @@ var import_sdk2 = __toESM(require("@anthropic-ai/sdk"));
 
 // src/lib/google/Google.tsx
 var import_vertexai = require("@google-cloud/vertexai");
+
+// src/lib/google/errors.ts
+var extractStatusFromMessage = (message) => {
+  const RE_STATUS = /status: (\d{3})/;
+  const matchStatus = message.match(RE_STATUS);
+  if (matchStatus !== null) {
+    return parseInt(matchStatus[1]);
+  }
+  const RE_GOOGLE_GENERATIVE_AI_ERROR = /\[VertexAI.GoogleGenerativeAIError\]/;
+  const matchErrorName = message.match(RE_GOOGLE_GENERATIVE_AI_ERROR);
+  if (matchErrorName !== null) {
+    return 500;
+  }
+  return 500;
+};
+var errorToChatCompletionError = (error, requestData) => {
+  const status = extractStatusFromMessage(error.message);
+  const shouldRetry4 = status !== 400;
+  return new ChatCompletionError(
+    error.message,
+    requestData,
+    status,
+    shouldRetry4
+  );
+};
+
+// src/lib/google/Google.tsx
 var SAFETY_SETTINGS = [
   {
     category: import_vertexai.HarmCategory.HARM_CATEGORY_HATE_SPEECH,
@@ -2734,20 +2760,7 @@ async function* GoogleChatCompletionInner(props, ctx) {
   try {
     response = await model.generateContentStream(googleCompletionRequest);
   } catch (err) {
-    if (err instanceof import_vertexai.GoogleGenerativeAIError) {
-      throw new ChatCompletionError(
-        err.message,
-        logRequestData,
-        void 0,
-        // always retry on GoogleGenerativeAIError errors, they represent non 4xx error cases
-        true
-      );
-    } else if (err instanceof Error) {
-      const status = void 0;
-      const retry = false;
-      throw new ChatCompletionError(err.message, logRequestData, status, retry);
-    }
-    throw err;
+    throw errorToChatCompletionError(err, logRequestData);
   }
   let content = "";
   let outputUsage = 0;
@@ -2793,20 +2806,7 @@ async function* GoogleChatCompletionInner(props, ctx) {
       }
     }
   } catch (err) {
-    if (err instanceof import_vertexai.GoogleGenerativeAIError) {
-      throw new ChatCompletionError(
-        err.message,
-        logRequestData,
-        void 0,
-        // always retry these errors they represent non 4xx error cases
-        true
-      );
-    } else if (err instanceof Error) {
-      const status = void 0;
-      const retry = false;
-      throw new ChatCompletionError(err.message, logRequestData, status, retry);
-    }
-    throw err;
+    throw errorToChatCompletionError(err, logRequestData);
   }
   const outputMessage = {
     role: "assistant",

package/dist/index.mjs

@@ -1487,7 +1487,6 @@ async function* Fallback({ shouldFallback = () => true, fallback, children }, ct
 // src/tracing/SpanTree.ts
 var SpanTree = class {
   roots = [];
-  findSpan;
   addSpan(span) {
     const parentId = span.parentSpanId;
     if (!parentId) {
@@ -2475,10 +2474,36 @@ import AnthropicClient2 from "@anthropic-ai/sdk";
 // src/lib/google/Google.tsx
 import {
   VertexAI,
-  GoogleGenerativeAIError,
   HarmCategory,
   HarmBlockThreshold
 } from "@google-cloud/vertexai";
+
+// src/lib/google/errors.ts
+var extractStatusFromMessage = (message) => {
+  const RE_STATUS = /status: (\d{3})/;
+  const matchStatus = message.match(RE_STATUS);
+  if (matchStatus !== null) {
+    return parseInt(matchStatus[1]);
+  }
+  const RE_GOOGLE_GENERATIVE_AI_ERROR = /\[VertexAI.GoogleGenerativeAIError\]/;
+  const matchErrorName = message.match(RE_GOOGLE_GENERATIVE_AI_ERROR);
+  if (matchErrorName !== null) {
+    return 500;
+  }
+  return 500;
+};
+var errorToChatCompletionError = (error, requestData) => {
+  const status = extractStatusFromMessage(error.message);
+  const shouldRetry4 = status !== 400;
+  return new ChatCompletionError(
+    error.message,
+    requestData,
+    status,
+    shouldRetry4
+  );
+};
+
+// src/lib/google/Google.tsx
 var SAFETY_SETTINGS = [
   {
     category: HarmCategory.HARM_CATEGORY_HATE_SPEECH,
@@ -2639,20 +2664,7 @@ async function* GoogleChatCompletionInner(props, ctx) {
   try {
     response = await model.generateContentStream(googleCompletionRequest);
   } catch (err) {
-    if (err instanceof GoogleGenerativeAIError) {
-      throw new ChatCompletionError(
-        err.message,
-        logRequestData,
-        void 0,
-        // always retry on GoogleGenerativeAIError errors, they represent non 4xx error cases
-        true
-      );
-    } else if (err instanceof Error) {
-      const status = void 0;
-      const retry = false;
-      throw new ChatCompletionError(err.message, logRequestData, status, retry);
-    }
-    throw err;
+    throw errorToChatCompletionError(err, logRequestData);
   }
   let content = "";
   let outputUsage = 0;
@@ -2698,20 +2710,7 @@ async function* GoogleChatCompletionInner(props, ctx) {
       }
     }
   } catch (err) {
-    if (err instanceof GoogleGenerativeAIError) {
-      throw new ChatCompletionError(
-        err.message,
-        logRequestData,
-        void 0,
-        // always retry these errors they represent non 4xx error cases
-        true
-      );
-    } else if (err instanceof Error) {
-      const status = void 0;
-      const retry = false;
-      throw new ChatCompletionError(err.message, logRequestData, status, retry);
-    }
-    throw err;
+    throw errorToChatCompletionError(err, logRequestData);
   }
   const outputMessage = {
     role: "assistant",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gammatech/aijsx",
-  "version": "0.10.0-dev.2024-06-04.2",
+  "version": "0.10.1-dev.2024-06-07",
   "description": "Rewrite of aijsx",
   "author": "Jordan Garcia",
   "license": "MIT",