@gammatech/aijsx 0.10.1-dev.2024-06-07 → 0.10.2-dev.2024-06-11

package/dist/index.d.mts

@@ -6,8 +6,8 @@ export { OpenAI as OpenAIClient } from 'openai';
 import { ChatCompletionSystemMessageParam, ChatCompletionUserMessageParam, ChatCompletionAssistantMessageParam, ChatCompletionCreateParams } from 'openai/resources';
 import AnthropicClient from '@anthropic-ai/sdk';
 export { default as AnthropicClient } from '@anthropic-ai/sdk';
-import { PromptFeedback, GenerateContentRequest, VertexAI } from '@google-cloud/vertexai';
-export { VertexAI } from '@google-cloud/vertexai';
+import { PromptFeedback, GenerateContentRequest, VertexAI, HarmCategory, HarmBlockThreshold } from '@google-cloud/vertexai';
+export { HarmBlockThreshold as GoogleHarmBlockThreshold, HarmCategory as GoogleHarmCategory, VertexAI } from '@google-cloud/vertexai';
 
 declare class ChatCompletionError extends Error {
     readonly chatCompletionRequest: LogChatCompletionRequest;
@@ -299,6 +299,10 @@ type GoogleChatCompletionProps = {
     stop?: string | string[];
     maxRetries?: number;
     children: AINode;
+    safetySettings?: {
+        category: HarmCategory;
+        threshold: HarmBlockThreshold;
+    }[];
 };
 declare function GoogleChatCompletion(props: GoogleChatCompletionProps, ctx: RenderContext): JSX.Element;
 

package/dist/index.d.ts

@@ -6,8 +6,8 @@ export { OpenAI as OpenAIClient } from 'openai';
 import { ChatCompletionSystemMessageParam, ChatCompletionUserMessageParam, ChatCompletionAssistantMessageParam, ChatCompletionCreateParams } from 'openai/resources';
 import AnthropicClient from '@anthropic-ai/sdk';
 export { default as AnthropicClient } from '@anthropic-ai/sdk';
-import { PromptFeedback, GenerateContentRequest, VertexAI } from '@google-cloud/vertexai';
-export { VertexAI } from '@google-cloud/vertexai';
+import { PromptFeedback, GenerateContentRequest, VertexAI, HarmCategory, HarmBlockThreshold } from '@google-cloud/vertexai';
+export { HarmBlockThreshold as GoogleHarmBlockThreshold, HarmCategory as GoogleHarmCategory, VertexAI } from '@google-cloud/vertexai';
 
 declare class ChatCompletionError extends Error {
     readonly chatCompletionRequest: LogChatCompletionRequest;
@@ -299,6 +299,10 @@ type GoogleChatCompletionProps = {
     stop?: string | string[];
     maxRetries?: number;
     children: AINode;
+    safetySettings?: {
+        category: HarmCategory;
+        threshold: HarmBlockThreshold;
+    }[];
 };
 declare function GoogleChatCompletion(props: GoogleChatCompletionProps, ctx: RenderContext): JSX.Element;
 

package/dist/index.js

@@ -44,6 +44,8 @@ __export(src_exports, {
   Fallback: () => Fallback,
   GoogleChatCompletion: () => GoogleChatCompletion,
   GoogleClientContext: () => GoogleClientContext,
+  GoogleHarmBlockThreshold: () => import_vertexai2.HarmBlockThreshold,
+  GoogleHarmCategory: () => import_vertexai2.HarmCategory,
   ImagePart: () => ImagePart,
   LogImplementation: () => LogImplementation,
   NoopLogImplementation: () => NoopLogImplementation,
@@ -2600,7 +2602,7 @@ var errorToChatCompletionError = (error, requestData) => {
 };
 
 // src/lib/google/Google.tsx
-var SAFETY_SETTINGS = [
+var DEFAULT_SAFETY_SETTINGS = [
   {
     category: import_vertexai.HarmCategory.HARM_CATEGORY_HATE_SPEECH,
     threshold: import_vertexai.HarmBlockThreshold.BLOCK_ONLY_HIGH
@@ -2754,7 +2756,7 @@ async function* GoogleChatCompletionInner(props, ctx) {
   });
   const model = client.getGenerativeModel({
     model: props.model,
-    safetySettings: SAFETY_SETTINGS
+    safetySettings: props.safetySettings || DEFAULT_SAFETY_SETTINGS
   });
   let response;
   try {
@@ -2882,6 +2884,8 @@ var import_vertexai2 = require("@google-cloud/vertexai");
   Fallback,
   GoogleChatCompletion,
   GoogleClientContext,
+  GoogleHarmBlockThreshold,
+  GoogleHarmCategory,
   ImagePart,
   LogImplementation,
   NoopLogImplementation,

package/dist/index.mjs

@@ -2473,9 +2473,9 @@ import AnthropicClient2 from "@anthropic-ai/sdk";
 
 // src/lib/google/Google.tsx
 import {
-  VertexAI,
+  HarmBlockThreshold,
   HarmCategory,
-  HarmBlockThreshold
+  VertexAI
 } from "@google-cloud/vertexai";
 
 // src/lib/google/errors.ts
@@ -2504,7 +2504,7 @@ var errorToChatCompletionError = (error, requestData) => {
 };
 
 // src/lib/google/Google.tsx
-var SAFETY_SETTINGS = [
+var DEFAULT_SAFETY_SETTINGS = [
   {
     category: HarmCategory.HARM_CATEGORY_HATE_SPEECH,
     threshold: HarmBlockThreshold.BLOCK_ONLY_HIGH
@@ -2658,7 +2658,7 @@ async function* GoogleChatCompletionInner(props, ctx) {
   });
   const model = client.getGenerativeModel({
     model: props.model,
-    safetySettings: SAFETY_SETTINGS
+    safetySettings: props.safetySettings || DEFAULT_SAFETY_SETTINGS
   });
   let response;
   try {
@@ -2768,7 +2768,11 @@ function cleanChatCompletionRequest3(chatCompletionRequest) {
 }
 
 // src/lib/google/index.ts
-import { VertexAI as VertexAI2 } from "@google-cloud/vertexai";
+import {
+  VertexAI as VertexAI2,
+  HarmBlockThreshold as HarmBlockThreshold2,
+  HarmCategory as HarmCategory2
+} from "@google-cloud/vertexai";
 export {
   AIFragment,
   AISpanProcessor,
@@ -2785,6 +2789,8 @@ export {
   Fallback,
   GoogleChatCompletion,
   GoogleClientContext,
+  HarmBlockThreshold2 as GoogleHarmBlockThreshold,
+  HarmCategory2 as GoogleHarmCategory,
   ImagePart,
   LogImplementation,
   NoopLogImplementation,

package/package.json

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