@langchain/google-common 0.0.26 → 0.1.0

package/dist/utils/gemini.d.ts

@@ -1,67 +1,33 @@
-import { AIMessageChunkFields, BaseMessage, BaseMessageChunk, BaseMessageFields, MessageContent } from "@langchain/core/messages";
-import { ChatGeneration, ChatGenerationChunk, ChatResult, Generation } from "@langchain/core/outputs";
-import type { GoogleLLMResponse, GoogleAIModelParams, GeminiPart, GeminiContent, GenerateContentResponseData, GoogleAISafetyHandler } from "../types.js";
-export declare function messageContentToParts(content: MessageContent): GeminiPart[];
-export declare function baseMessageToContent(message: BaseMessage, prevMessage: BaseMessage | undefined, useSystemInstruction: boolean): GeminiContent[];
-export declare function partsToMessageContent(parts: GeminiPart[]): MessageContent;
-interface FunctionCall {
+import { BaseMessage, BaseMessageChunk, MessageContent } from "@langchain/core/messages";
+import { ChatGenerationChunk, ChatResult } from "@langchain/core/outputs";
+import type { GoogleLLMResponse, GoogleAIModelParams, GeminiPart, GeminiContent, GenerateContentResponseData, GoogleAISafetyHandler, GeminiAPIConfig } from "../types.js";
+export interface FunctionCall {
     name: string;
     arguments: string;
 }
-interface ToolCall {
+export interface ToolCall {
     id: string;
     type: "function";
     function: FunctionCall;
 }
-interface FunctionCallRaw {
+export interface FunctionCallRaw {
     name: string;
     arguments: object;
 }
-interface ToolCallRaw {
+export interface ToolCallRaw {
     id: string;
     type: "function";
     function: FunctionCallRaw;
 }
-export declare function partsToToolsRaw(parts: GeminiPart[]): ToolCallRaw[];
-export declare function toolsRawToTools(raws: ToolCallRaw[]): ToolCall[];
-export declare function responseToGenerateContentResponseData(response: GoogleLLMResponse): GenerateContentResponseData;
-export declare function responseToParts(response: GoogleLLMResponse): GeminiPart[];
-export declare function partToText(part: GeminiPart): string;
-export declare function responseToString(response: GoogleLLMResponse): string;
-export declare function safeResponseToString(response: GoogleLLMResponse, safetyHandler: GoogleAISafetyHandler): string;
-export declare function responseToGenerationInfo(response: GoogleLLMResponse): {
-    usage_metadata?: undefined;
-    safety_ratings?: undefined;
-    finish_reason?: undefined;
-} | {
-    usage_metadata: {
-        prompt_token_count: unknown;
-        candidates_token_count: unknown;
-        total_token_count: unknown;
-    };
-    safety_ratings: {
-        category: string;
-        probability: string;
-        probability_score: unknown;
-        severity: unknown;
-        severity_score: unknown;
-    }[];
-    finish_reason: string;
+export declare function getGeminiAPI(config?: GeminiAPIConfig): {
+    messageContentToParts: (content: MessageContent) => Promise<GeminiPart[]>;
+    baseMessageToContent: (message: BaseMessage, prevMessage: BaseMessage | undefined, useSystemInstruction: boolean) => Promise<GeminiContent[]>;
+    safeResponseToString: (response: GoogleLLMResponse, safetyHandler: GoogleAISafetyHandler) => string;
+    safeResponseToChatGeneration: (response: GoogleLLMResponse, safetyHandler: GoogleAISafetyHandler) => ChatGenerationChunk;
+    chunkToString: (chunk: BaseMessageChunk) => string;
+    safeResponseToBaseMessage: (response: GoogleLLMResponse, safetyHandler: GoogleAISafetyHandler) => BaseMessage;
+    safeResponseToChatResult: (response: GoogleLLMResponse, safetyHandler: GoogleAISafetyHandler) => ChatResult;
 };
-export declare function responseToGeneration(response: GoogleLLMResponse): Generation;
-export declare function safeResponseToGeneration(response: GoogleLLMResponse, safetyHandler: GoogleAISafetyHandler): Generation;
-export declare function responseToChatGeneration(response: GoogleLLMResponse): ChatGenerationChunk;
-export declare function safeResponseToChatGeneration(response: GoogleLLMResponse, safetyHandler: GoogleAISafetyHandler): ChatGenerationChunk;
-export declare function chunkToString(chunk: BaseMessageChunk): string;
-export declare function partToMessageChunk(part: GeminiPart): BaseMessageChunk;
-export declare function partToChatGeneration(part: GeminiPart): ChatGeneration;
-export declare function responseToChatGenerations(response: GoogleLLMResponse): ChatGeneration[];
-export declare function responseToBaseMessageFields(response: GoogleLLMResponse): BaseMessageFields;
-export declare function partsToBaseMessageChunkFields(parts: GeminiPart[]): AIMessageChunkFields;
-export declare function responseToBaseMessage(response: GoogleLLMResponse): BaseMessage;
-export declare function safeResponseToBaseMessage(response: GoogleLLMResponse, safetyHandler: GoogleAISafetyHandler): BaseMessage;
-export declare function responseToChatResult(response: GoogleLLMResponse): ChatResult;
-export declare function safeResponseToChatResult(response: GoogleLLMResponse, safetyHandler: GoogleAISafetyHandler): ChatResult;
 export declare function validateGeminiParams(params: GoogleAIModelParams): void;
 export declare function isModelGemini(modelName: string): boolean;
 export interface DefaultGeminiSafetySettings {
@@ -86,4 +52,3 @@ export declare class MessageGeminiSafetyHandler extends DefaultGeminiSafetyHandl
     setMessage(data: GenerateContentResponseData): GenerateContentResponseData;
     handleData(response: GoogleLLMResponse, data: GenerateContentResponseData): GenerateContentResponseData;
 }
-export {};