@langchain/google-common 0.1.0 → 0.1.2

package/dist/types.cjs

@@ -1,2 +1,56 @@
 "use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.GoogleAISafetyMethod = exports.GoogleAISafetyThreshold = exports.GoogleAISafetyCategory = void 0;
+__exportStar(require("./types-anthropic.cjs"), exports);
+exports.GoogleAISafetyCategory = {
+    Harassment: "HARM_CATEGORY_HARASSMENT",
+    HARASSMENT: "HARM_CATEGORY_HARASSMENT",
+    HARM_CATEGORY_HARASSMENT: "HARM_CATEGORY_HARASSMENT",
+    HateSpeech: "HARM_CATEGORY_HATE_SPEECH",
+    HATE_SPEECH: "HARM_CATEGORY_HATE_SPEECH",
+    HARM_CATEGORY_HATE_SPEECH: "HARM_CATEGORY_HATE_SPEECH",
+    SexuallyExplicit: "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+    SEXUALLY_EXPLICIT: "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+    HARM_CATEGORY_SEXUALLY_EXPLICIT: "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+    Dangerous: "HARM_CATEGORY_DANGEROUS",
+    DANGEROUS: "HARM_CATEGORY_DANGEROUS",
+    HARM_CATEGORY_DANGEROUS: "HARM_CATEGORY_DANGEROUS",
+    CivicIntegrity: "HARM_CATEGORY_CIVIC_INTEGRITY",
+    CIVIC_INTEGRITY: "HARM_CATEGORY_CIVIC_INTEGRITY",
+    HARM_CATEGORY_CIVIC_INTEGRITY: "HARM_CATEGORY_CIVIC_INTEGRITY",
+};
+exports.GoogleAISafetyThreshold = {
+    None: "BLOCK_NONE",
+    NONE: "BLOCK_NONE",
+    BLOCK_NONE: "BLOCK_NONE",
+    Few: "BLOCK_ONLY_HIGH",
+    FEW: "BLOCK_ONLY_HIGH",
+    BLOCK_ONLY_HIGH: "BLOCK_ONLY_HIGH",
+    Some: "BLOCK_MEDIUM_AND_ABOVE",
+    SOME: "BLOCK_MEDIUM_AND_ABOVE",
+    BLOCK_MEDIUM_AND_ABOVE: "BLOCK_MEDIUM_AND_ABOVE",
+    Most: "BLOCK_LOW_AND_ABOVE",
+    MOST: "BLOCK_LOW_AND_ABOVE",
+    BLOCK_LOW_AND_ABOVE: "BLOCK_LOW_AND_ABOVE",
+    Off: "OFF",
+    OFF: "OFF",
+    BLOCK_OFF: "OFF",
+};
+exports.GoogleAISafetyMethod = {
+    Severity: "SEVERITY",
+    Probability: "PROBABILITY",
+};

package/dist/types.d.ts

@@ -1,7 +1,11 @@
 import type { BaseLLMParams } from "@langchain/core/language_models/llms";
 import type { BaseChatModelCallOptions, BindToolsInput } from "@langchain/core/language_models/chat_models";
+import { BaseMessage, BaseMessageChunk, MessageContent } from "@langchain/core/messages";
+import { ChatGenerationChunk, ChatResult } from "@langchain/core/outputs";
 import type { JsonStream } from "./utils/stream.js";
 import { MediaManager } from "./experimental/utils/media_core.js";
+import { AnthropicResponseData, AnthropicAPIConfig } from "./types-anthropic.js";
+export * from "./types-anthropic.js";
 /**
  * Parameters needed to setup the client connection.
  * AuthOptions are something like GoogleAuthOptions (from google-auth-library)
@@ -33,10 +37,51 @@ export interface GoogleConnectionParams<AuthOptions> extends GoogleClientParams<
      */
     platformType?: GooglePlatformType;
 }
+export declare const GoogleAISafetyCategory: {
+    readonly Harassment: "HARM_CATEGORY_HARASSMENT";
+    readonly HARASSMENT: "HARM_CATEGORY_HARASSMENT";
+    readonly HARM_CATEGORY_HARASSMENT: "HARM_CATEGORY_HARASSMENT";
+    readonly HateSpeech: "HARM_CATEGORY_HATE_SPEECH";
+    readonly HATE_SPEECH: "HARM_CATEGORY_HATE_SPEECH";
+    readonly HARM_CATEGORY_HATE_SPEECH: "HARM_CATEGORY_HATE_SPEECH";
+    readonly SexuallyExplicit: "HARM_CATEGORY_SEXUALLY_EXPLICIT";
+    readonly SEXUALLY_EXPLICIT: "HARM_CATEGORY_SEXUALLY_EXPLICIT";
+    readonly HARM_CATEGORY_SEXUALLY_EXPLICIT: "HARM_CATEGORY_SEXUALLY_EXPLICIT";
+    readonly Dangerous: "HARM_CATEGORY_DANGEROUS";
+    readonly DANGEROUS: "HARM_CATEGORY_DANGEROUS";
+    readonly HARM_CATEGORY_DANGEROUS: "HARM_CATEGORY_DANGEROUS";
+    readonly CivicIntegrity: "HARM_CATEGORY_CIVIC_INTEGRITY";
+    readonly CIVIC_INTEGRITY: "HARM_CATEGORY_CIVIC_INTEGRITY";
+    readonly HARM_CATEGORY_CIVIC_INTEGRITY: "HARM_CATEGORY_CIVIC_INTEGRITY";
+};
+export type GoogleAISafetyCategory = (typeof GoogleAISafetyCategory)[keyof typeof GoogleAISafetyCategory];
+export declare const GoogleAISafetyThreshold: {
+    readonly None: "BLOCK_NONE";
+    readonly NONE: "BLOCK_NONE";
+    readonly BLOCK_NONE: "BLOCK_NONE";
+    readonly Few: "BLOCK_ONLY_HIGH";
+    readonly FEW: "BLOCK_ONLY_HIGH";
+    readonly BLOCK_ONLY_HIGH: "BLOCK_ONLY_HIGH";
+    readonly Some: "BLOCK_MEDIUM_AND_ABOVE";
+    readonly SOME: "BLOCK_MEDIUM_AND_ABOVE";
+    readonly BLOCK_MEDIUM_AND_ABOVE: "BLOCK_MEDIUM_AND_ABOVE";
+    readonly Most: "BLOCK_LOW_AND_ABOVE";
+    readonly MOST: "BLOCK_LOW_AND_ABOVE";
+    readonly BLOCK_LOW_AND_ABOVE: "BLOCK_LOW_AND_ABOVE";
+    readonly Off: "OFF";
+    readonly OFF: "OFF";
+    readonly BLOCK_OFF: "OFF";
+};
+export type GoogleAISafetyThreshold = (typeof GoogleAISafetyThreshold)[keyof typeof GoogleAISafetyThreshold];
+export declare const GoogleAISafetyMethod: {
+    readonly Severity: "SEVERITY";
+    readonly Probability: "PROBABILITY";
+};
+export type GoogleAISafetyMethod = (typeof GoogleAISafetyMethod)[keyof typeof GoogleAISafetyMethod];
 export interface GoogleAISafetySetting {
-    category: string;
-    threshold: string;
-    method?: string;
+    category: GoogleAISafetyCategory | string;
+    threshold: GoogleAISafetyThreshold | string;
+    method?: GoogleAISafetyMethod | string;
 }
 export type GoogleAIResponseMimeType = "text/plain" | "application/json";
 export interface GoogleAIModelParams {
@@ -117,7 +162,7 @@ export interface GoogleAIModelRequestParams extends GoogleAIModelParams {
      */
     allowed_function_names?: string[];
 }
-export interface GoogleAIBaseLLMInput<AuthOptions> extends BaseLLMParams, GoogleConnectionParams<AuthOptions>, GoogleAIModelParams, GoogleAISafetyParams, GeminiAPIConfig {
+export interface GoogleAIBaseLLMInput<AuthOptions> extends BaseLLMParams, GoogleConnectionParams<AuthOptions>, GoogleAIModelParams, GoogleAISafetyParams, GoogleAIAPIParams {
 }
 export interface GoogleAIBaseLanguageModelCallOptions extends BaseChatModelCallOptions, GoogleAIModelRequestParams, GoogleAISafetyParams {
     /**
@@ -240,9 +285,10 @@ export interface GenerateContentResponseData {
     usageMetadata: Record<string, unknown>;
 }
 export type GoogleLLMModelFamily = null | "palm" | "gemini";
+export type VertexModelFamily = GoogleLLMModelFamily | "claude";
 export type GoogleLLMResponseData = JsonStream | GenerateContentResponseData | GenerateContentResponseData[];
 export interface GoogleLLMResponse extends GoogleResponse {
-    data: GoogleLLMResponseData;
+    data: GoogleLLMResponseData | AnthropicResponseData;
 }
 export interface GoogleAISafetyHandler {
     /**
@@ -265,7 +311,23 @@ export interface GeminiJsonSchemaDirty extends GeminiJsonSchema {
     properties?: Record<string, GeminiJsonSchemaDirty>;
     additionalProperties?: boolean;
 }
+export type GoogleAIAPI = {
+    messageContentToParts?: (content: MessageContent) => Promise<GeminiPart[]>;
+    baseMessageToContent?: (message: BaseMessage, prevMessage: BaseMessage | undefined, useSystemInstruction: boolean) => Promise<GeminiContent[]>;
+    responseToString: (response: GoogleLLMResponse) => string;
+    responseToChatGeneration: (response: GoogleLLMResponse) => ChatGenerationChunk | null;
+    chunkToString: (chunk: BaseMessageChunk) => string;
+    responseToBaseMessage: (response: GoogleLLMResponse) => BaseMessage;
+    responseToChatResult: (response: GoogleLLMResponse) => ChatResult;
+    formatData: (input: unknown, parameters: GoogleAIModelRequestParams) => Promise<unknown>;
+};
 export interface GeminiAPIConfig {
+    safetyHandler?: GoogleAISafetyHandler;
     mediaManager?: MediaManager;
+    useSystemInstruction?: boolean;
+}
+export type GoogleAIAPIConfig = GeminiAPIConfig | AnthropicAPIConfig;
+export interface GoogleAIAPIParams {
+    apiName?: string;
+    apiConfig?: GoogleAIAPIConfig;
 }
-export {};

package/dist/types.js

@@ -1 +1,39 @@
-export {};
+export * from "./types-anthropic.js";
+export const GoogleAISafetyCategory = {
+    Harassment: "HARM_CATEGORY_HARASSMENT",
+    HARASSMENT: "HARM_CATEGORY_HARASSMENT",
+    HARM_CATEGORY_HARASSMENT: "HARM_CATEGORY_HARASSMENT",
+    HateSpeech: "HARM_CATEGORY_HATE_SPEECH",
+    HATE_SPEECH: "HARM_CATEGORY_HATE_SPEECH",
+    HARM_CATEGORY_HATE_SPEECH: "HARM_CATEGORY_HATE_SPEECH",
+    SexuallyExplicit: "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+    SEXUALLY_EXPLICIT: "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+    HARM_CATEGORY_SEXUALLY_EXPLICIT: "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+    Dangerous: "HARM_CATEGORY_DANGEROUS",
+    DANGEROUS: "HARM_CATEGORY_DANGEROUS",
+    HARM_CATEGORY_DANGEROUS: "HARM_CATEGORY_DANGEROUS",
+    CivicIntegrity: "HARM_CATEGORY_CIVIC_INTEGRITY",
+    CIVIC_INTEGRITY: "HARM_CATEGORY_CIVIC_INTEGRITY",
+    HARM_CATEGORY_CIVIC_INTEGRITY: "HARM_CATEGORY_CIVIC_INTEGRITY",
+};
+export const GoogleAISafetyThreshold = {
+    None: "BLOCK_NONE",
+    NONE: "BLOCK_NONE",
+    BLOCK_NONE: "BLOCK_NONE",
+    Few: "BLOCK_ONLY_HIGH",
+    FEW: "BLOCK_ONLY_HIGH",
+    BLOCK_ONLY_HIGH: "BLOCK_ONLY_HIGH",
+    Some: "BLOCK_MEDIUM_AND_ABOVE",
+    SOME: "BLOCK_MEDIUM_AND_ABOVE",
+    BLOCK_MEDIUM_AND_ABOVE: "BLOCK_MEDIUM_AND_ABOVE",
+    Most: "BLOCK_LOW_AND_ABOVE",
+    MOST: "BLOCK_LOW_AND_ABOVE",
+    BLOCK_LOW_AND_ABOVE: "BLOCK_LOW_AND_ABOVE",
+    Off: "OFF",
+    OFF: "OFF",
+    BLOCK_OFF: "OFF",
+};
+export const GoogleAISafetyMethod = {
+    Severity: "SEVERITY",
+    Probability: "PROBABILITY",
+};