@langchain/google-genai 0.2.18 → 1.0.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +17 -0
- package/LICENSE +6 -6
- package/README.md +8 -8
- package/dist/_virtual/rolldown_runtime.cjs +25 -0
- package/dist/chat_models.cjs +688 -847
- package/dist/chat_models.cjs.map +1 -0
- package/dist/chat_models.d.cts +575 -0
- package/dist/chat_models.d.cts.map +1 -0
- package/dist/chat_models.d.ts +190 -157
- package/dist/chat_models.d.ts.map +1 -0
- package/dist/chat_models.js +686 -842
- package/dist/chat_models.js.map +1 -0
- package/dist/embeddings.cjs +97 -151
- package/dist/embeddings.cjs.map +1 -0
- package/dist/embeddings.d.cts +104 -0
- package/dist/embeddings.d.cts.map +1 -0
- package/dist/embeddings.d.ts +76 -70
- package/dist/embeddings.d.ts.map +1 -0
- package/dist/embeddings.js +93 -144
- package/dist/embeddings.js.map +1 -0
- package/dist/index.cjs +5 -18
- package/dist/index.d.cts +3 -0
- package/dist/index.d.ts +3 -2
- package/dist/index.js +4 -2
- package/dist/output_parsers.cjs +47 -75
- package/dist/output_parsers.cjs.map +1 -0
- package/dist/output_parsers.js +47 -72
- package/dist/output_parsers.js.map +1 -0
- package/dist/profiles.cjs +345 -0
- package/dist/profiles.cjs.map +1 -0
- package/dist/profiles.js +344 -0
- package/dist/profiles.js.map +1 -0
- package/dist/types.d.cts +8 -0
- package/dist/types.d.cts.map +1 -0
- package/dist/types.d.ts +7 -2
- package/dist/types.d.ts.map +1 -0
- package/dist/utils/common.cjs +356 -549
- package/dist/utils/common.cjs.map +1 -0
- package/dist/utils/common.js +357 -545
- package/dist/utils/common.js.map +1 -0
- package/dist/utils/tools.cjs +65 -102
- package/dist/utils/tools.cjs.map +1 -0
- package/dist/utils/tools.js +64 -99
- package/dist/utils/tools.js.map +1 -0
- package/dist/utils/zod_to_genai_parameters.cjs +31 -49
- package/dist/utils/zod_to_genai_parameters.cjs.map +1 -0
- package/dist/utils/zod_to_genai_parameters.js +29 -45
- package/dist/utils/zod_to_genai_parameters.js.map +1 -0
- package/package.json +45 -51
- package/dist/output_parsers.d.ts +0 -20
- package/dist/types.cjs +0 -2
- package/dist/types.js +0 -1
- package/dist/utils/common.d.ts +0 -22
- package/dist/utils/tools.d.ts +0 -10
- package/dist/utils/zod_to_genai_parameters.d.ts +0 -14
- package/index.cjs +0 -1
- package/index.d.cts +0 -1
- package/index.d.ts +0 -1
- package/index.js +0 -1
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"common.js","names":["message: BaseMessage","author: string","content: MessageContentComplex","message: ToolMessage | ToolMessageChunk","previousMessages: BaseMessage[]","isMultimodalModel: boolean","standardContentBlockConverter: StandardContentBlockConverter<{\n text: TextPart;\n image: FileDataPart | InlineDataPart;\n audio: FileDataPart | InlineDataPart;\n file: FileDataPart | InlineDataPart | TextPart;\n }>","functionCalls: FunctionCallPart[]","messageParts: Part[]","messages: BaseMessage[]","convertSystemMessageToHumanContent: boolean","prevContent","content: Content","response: EnhancedGenerateContentResponse","extra?: {\n usageMetadata: UsageMetadata | undefined;\n }","content: MessageContent | undefined","generation: ChatGeneration","uuidv4","extra: {\n usageMetadata?: UsageMetadata | undefined;\n index: number;\n }","toolCallChunks: ToolCallChunk[]","tools: GoogleGenerativeAIToolType[]"],"sources":["../../src/utils/common.ts"],"sourcesContent":["import {\n EnhancedGenerateContentResponse,\n Content,\n Part,\n type FunctionDeclarationsTool as GoogleGenerativeAIFunctionDeclarationsTool,\n type FunctionDeclaration as GenerativeAIFunctionDeclaration,\n POSSIBLE_ROLES,\n FunctionCallPart,\n TextPart,\n FileDataPart,\n InlineDataPart,\n} from \"@google/generative-ai\";\nimport {\n AIMessage,\n AIMessageChunk,\n BaseMessage,\n ChatMessage,\n ToolMessage,\n ToolMessageChunk,\n MessageContent,\n MessageContentComplex,\n UsageMetadata,\n isAIMessage,\n isBaseMessage,\n isToolMessage,\n StandardContentBlockConverter,\n parseBase64DataUrl,\n convertToProviderContentBlock,\n isDataContentBlock,\n} from \"@langchain/core/messages\";\nimport {\n ChatGeneration,\n ChatGenerationChunk,\n ChatResult,\n} from \"@langchain/core/outputs\";\nimport { isLangChainTool } from \"@langchain/core/utils/function_calling\";\nimport { isOpenAITool } from \"@langchain/core/language_models/base\";\nimport { ToolCallChunk } from \"@langchain/core/messages/tool\";\nimport { v4 as uuidv4 } from \"uuid\";\nimport {\n jsonSchemaToGeminiParameters,\n schemaToGenerativeAIParameters,\n} from \"./zod_to_genai_parameters.js\";\nimport { GoogleGenerativeAIToolType } from \"../types.js\";\n\nexport function getMessageAuthor(message: BaseMessage) {\n const type = message._getType();\n if (ChatMessage.isInstance(message)) {\n return message.role;\n }\n if (type === \"tool\") {\n return type;\n }\n return message.name ?? type;\n}\n\n/**\n * Maps a message type to a Google Generative AI chat author.\n * @param message The message to map.\n * @param model The model to use for mapping.\n * @returns The message type mapped to a Google Generative AI chat author.\n */\nexport function convertAuthorToRole(\n author: string\n): (typeof POSSIBLE_ROLES)[number] {\n switch (author) {\n /**\n * Note: Gemini currently is not supporting system messages\n * we will convert them to human messages and merge with following\n * */\n case \"supervisor\":\n case \"ai\":\n case \"model\": // getMessageAuthor returns message.name. code ex.: return message.name ?? type;\n return \"model\";\n case \"system\":\n return \"system\";\n case \"human\":\n return \"user\";\n case \"tool\":\n case \"function\":\n return \"function\";\n default:\n throw new Error(`Unknown / unsupported author: ${author}`);\n }\n}\n\nfunction messageContentMedia(content: MessageContentComplex): Part {\n if (\"mimeType\" in content && \"data\" in content) {\n return {\n inlineData: {\n mimeType: content.mimeType,\n data: content.data,\n },\n };\n }\n if (\"mimeType\" in content && \"fileUri\" in content) {\n return {\n fileData: {\n mimeType: content.mimeType,\n fileUri: content.fileUri,\n },\n };\n }\n\n throw new Error(\"Invalid media content\");\n}\n\nfunction inferToolNameFromPreviousMessages(\n message: ToolMessage | ToolMessageChunk,\n previousMessages: BaseMessage[]\n): string | undefined {\n return previousMessages\n .map((msg) => {\n if (isAIMessage(msg)) {\n return msg.tool_calls ?? [];\n }\n return [];\n })\n .flat()\n .find((toolCall) => {\n return toolCall.id === message.tool_call_id;\n })?.name;\n}\n\nfunction _getStandardContentBlockConverter(isMultimodalModel: boolean) {\n const standardContentBlockConverter: StandardContentBlockConverter<{\n text: TextPart;\n image: FileDataPart | InlineDataPart;\n audio: FileDataPart | InlineDataPart;\n file: FileDataPart | InlineDataPart | TextPart;\n }> = {\n providerName: \"Google Gemini\",\n\n fromStandardTextBlock(block) {\n return {\n text: block.text,\n };\n },\n\n fromStandardImageBlock(block): FileDataPart | InlineDataPart {\n if (!isMultimodalModel) {\n throw new Error(\"This model does not support images\");\n }\n if (block.source_type === \"url\") {\n const data = parseBase64DataUrl({ dataUrl: block.url });\n if (data) {\n return {\n inlineData: {\n mimeType: data.mime_type,\n data: data.data,\n },\n };\n } else {\n return {\n fileData: {\n mimeType: block.mime_type ?? \"\",\n fileUri: block.url,\n },\n };\n }\n }\n\n if (block.source_type === \"base64\") {\n return {\n inlineData: {\n mimeType: block.mime_type ?? \"\",\n data: block.data,\n },\n };\n }\n\n throw new Error(`Unsupported source type: ${block.source_type}`);\n },\n\n fromStandardAudioBlock(block): FileDataPart | InlineDataPart {\n if (!isMultimodalModel) {\n throw new Error(\"This model does not support audio\");\n }\n if (block.source_type === \"url\") {\n const data = parseBase64DataUrl({ dataUrl: block.url });\n if (data) {\n return {\n inlineData: {\n mimeType: data.mime_type,\n data: data.data,\n },\n };\n } else {\n return {\n fileData: {\n mimeType: block.mime_type ?? \"\",\n fileUri: block.url,\n },\n };\n }\n }\n\n if (block.source_type === \"base64\") {\n return {\n inlineData: {\n mimeType: block.mime_type ?? \"\",\n data: block.data,\n },\n };\n }\n\n throw new Error(`Unsupported source type: ${block.source_type}`);\n },\n\n fromStandardFileBlock(block): FileDataPart | InlineDataPart | TextPart {\n if (!isMultimodalModel) {\n throw new Error(\"This model does not support files\");\n }\n if (block.source_type === \"text\") {\n return {\n text: block.text,\n };\n }\n if (block.source_type === \"url\") {\n const data = parseBase64DataUrl({ dataUrl: block.url });\n if (data) {\n return {\n inlineData: {\n mimeType: data.mime_type,\n data: data.data,\n },\n };\n } else {\n return {\n fileData: {\n mimeType: block.mime_type ?? \"\",\n fileUri: block.url,\n },\n };\n }\n }\n\n if (block.source_type === \"base64\") {\n return {\n inlineData: {\n mimeType: block.mime_type ?? \"\",\n data: block.data,\n },\n };\n }\n throw new Error(`Unsupported source type: ${block.source_type}`);\n },\n };\n return standardContentBlockConverter;\n}\n\nfunction _convertLangChainContentToPart(\n content: MessageContentComplex,\n isMultimodalModel: boolean\n): Part | undefined {\n if (isDataContentBlock(content)) {\n return convertToProviderContentBlock(\n content,\n _getStandardContentBlockConverter(isMultimodalModel)\n );\n }\n\n if (content.type === \"text\") {\n return { text: content.text };\n } else if (content.type === \"executableCode\") {\n return { executableCode: content.executableCode };\n } else if (content.type === \"codeExecutionResult\") {\n return { codeExecutionResult: content.codeExecutionResult };\n } else if (content.type === \"image_url\") {\n if (!isMultimodalModel) {\n throw new Error(`This model does not support images`);\n }\n let source;\n if (typeof content.image_url === \"string\") {\n source = content.image_url;\n } else if (\n typeof content.image_url === \"object\" &&\n \"url\" in content.image_url\n ) {\n source = content.image_url.url;\n } else {\n throw new Error(\"Please provide image as base64 encoded data URL\");\n }\n const [dm, data] = source.split(\",\");\n if (!dm.startsWith(\"data:\")) {\n throw new Error(\"Please provide image as base64 encoded data URL\");\n }\n\n const [mimeType, encoding] = dm.replace(/^data:/, \"\").split(\";\");\n if (encoding !== \"base64\") {\n throw new Error(\"Please provide image as base64 encoded data URL\");\n }\n\n return {\n inlineData: {\n data,\n mimeType,\n },\n };\n } else if (content.type === \"media\") {\n return messageContentMedia(content);\n } else if (content.type === \"tool_use\") {\n return {\n functionCall: {\n name: content.name,\n args: content.input,\n },\n };\n } else if (\n content.type?.includes(\"/\") &&\n // Ensure it's a single slash.\n content.type.split(\"/\").length === 2 &&\n \"data\" in content &&\n typeof content.data === \"string\"\n ) {\n return {\n inlineData: {\n mimeType: content.type,\n data: content.data,\n },\n };\n } else if (\"functionCall\" in content) {\n // No action needed here — function calls will be added later from message.tool_calls\n return undefined;\n } else {\n if (\"type\" in content) {\n throw new Error(`Unknown content type ${content.type}`);\n } else {\n throw new Error(`Unknown content ${JSON.stringify(content)}`);\n }\n }\n}\n\nexport function convertMessageContentToParts(\n message: BaseMessage,\n isMultimodalModel: boolean,\n previousMessages: BaseMessage[]\n): Part[] {\n if (isToolMessage(message)) {\n const messageName =\n message.name ??\n inferToolNameFromPreviousMessages(message, previousMessages);\n if (messageName === undefined) {\n throw new Error(\n `Google requires a tool name for each tool call response, and we could not infer a called tool name for ToolMessage \"${message.id}\" from your passed messages. Please populate a \"name\" field on that ToolMessage explicitly.`\n );\n }\n\n const result = Array.isArray(message.content)\n ? (message.content\n .map((c) => _convertLangChainContentToPart(c, isMultimodalModel))\n .filter((p) => p !== undefined) as Part[])\n : message.content;\n\n if (message.status === \"error\") {\n return [\n {\n functionResponse: {\n name: messageName,\n // The API expects an object with an `error` field if the function call fails.\n // `error` must be a valid object (not a string or array), so we wrap `message.content` here\n response: { error: { details: result } },\n },\n },\n ];\n }\n\n return [\n {\n functionResponse: {\n name: messageName,\n // again, can't have a string or array value for `response`, so we wrap it as an object here\n response: { result },\n },\n },\n ];\n }\n\n let functionCalls: FunctionCallPart[] = [];\n const messageParts: Part[] = [];\n\n if (typeof message.content === \"string\" && message.content) {\n messageParts.push({ text: message.content });\n }\n\n if (Array.isArray(message.content)) {\n messageParts.push(\n ...(message.content\n .map((c) => _convertLangChainContentToPart(c, isMultimodalModel))\n .filter((p) => p !== undefined) as Part[])\n );\n }\n\n if (isAIMessage(message) && message.tool_calls?.length) {\n functionCalls = message.tool_calls.map((tc) => {\n return {\n functionCall: {\n name: tc.name,\n args: tc.args,\n },\n };\n });\n }\n\n return [...messageParts, ...functionCalls];\n}\n\nexport function convertBaseMessagesToContent(\n messages: BaseMessage[],\n isMultimodalModel: boolean,\n convertSystemMessageToHumanContent: boolean = false\n) {\n return messages.reduce<{\n content: Content[];\n mergeWithPreviousContent: boolean;\n }>(\n (acc, message, index) => {\n if (!isBaseMessage(message)) {\n throw new Error(\"Unsupported message input\");\n }\n const author = getMessageAuthor(message);\n if (author === \"system\" && index !== 0) {\n throw new Error(\"System message should be the first one\");\n }\n const role = convertAuthorToRole(author);\n\n const prevContent = acc.content[acc.content.length];\n if (\n !acc.mergeWithPreviousContent &&\n prevContent &&\n prevContent.role === role\n ) {\n throw new Error(\n \"Google Generative AI requires alternate messages between authors\"\n );\n }\n\n const parts = convertMessageContentToParts(\n message,\n isMultimodalModel,\n messages.slice(0, index)\n );\n\n if (acc.mergeWithPreviousContent) {\n const prevContent = acc.content[acc.content.length - 1];\n if (!prevContent) {\n throw new Error(\n \"There was a problem parsing your system message. Please try a prompt without one.\"\n );\n }\n prevContent.parts.push(...parts);\n\n return {\n mergeWithPreviousContent: false,\n content: acc.content,\n };\n }\n let actualRole = role;\n if (\n actualRole === \"function\" ||\n (actualRole === \"system\" && !convertSystemMessageToHumanContent)\n ) {\n // GenerativeAI API will throw an error if the role is not \"user\" or \"model.\"\n actualRole = \"user\";\n }\n const content: Content = {\n role: actualRole,\n parts,\n };\n return {\n mergeWithPreviousContent:\n author === \"system\" && !convertSystemMessageToHumanContent,\n content: [...acc.content, content],\n };\n },\n { content: [], mergeWithPreviousContent: false }\n ).content;\n}\n\nexport function mapGenerateContentResultToChatResult(\n response: EnhancedGenerateContentResponse,\n extra?: {\n usageMetadata: UsageMetadata | undefined;\n }\n): ChatResult {\n // if rejected or error, return empty generations with reason in filters\n if (\n !response.candidates ||\n response.candidates.length === 0 ||\n !response.candidates[0]\n ) {\n return {\n generations: [],\n llmOutput: {\n filters: response.promptFeedback,\n },\n };\n }\n\n const functionCalls = response.functionCalls();\n const [candidate] = response.candidates;\n const { content: candidateContent, ...generationInfo } = candidate;\n let content: MessageContent | undefined;\n\n if (\n Array.isArray(candidateContent?.parts) &&\n candidateContent.parts.length === 1 &&\n candidateContent.parts[0].text\n ) {\n content = candidateContent.parts[0].text;\n } else if (\n Array.isArray(candidateContent?.parts) &&\n candidateContent.parts.length > 0\n ) {\n content = candidateContent.parts.map((p) => {\n if (\"text\" in p) {\n return {\n type: \"text\",\n text: p.text,\n };\n } else if (\"inlineData\" in p) {\n return {\n type: \"inlineData\",\n inlineData: p.inlineData,\n };\n } else if (\"functionCall\" in p) {\n return {\n type: \"functionCall\",\n functionCall: p.functionCall,\n };\n } else if (\"functionResponse\" in p) {\n return {\n type: \"functionResponse\",\n functionResponse: p.functionResponse,\n };\n } else if (\"fileData\" in p) {\n return {\n type: \"fileData\",\n fileData: p.fileData,\n };\n } else if (\"executableCode\" in p) {\n return {\n type: \"executableCode\",\n executableCode: p.executableCode,\n };\n } else if (\"codeExecutionResult\" in p) {\n return {\n type: \"codeExecutionResult\",\n codeExecutionResult: p.codeExecutionResult,\n };\n }\n return p;\n });\n } else {\n // no content returned - likely due to abnormal stop reason, e.g. malformed function call\n content = [];\n }\n\n let text = \"\";\n if (typeof content === \"string\") {\n text = content;\n } else if (Array.isArray(content) && content.length > 0) {\n const block = content.find((b) => \"text\" in b) as\n | { text: string }\n | undefined;\n text = block?.text ?? text;\n }\n\n const generation: ChatGeneration = {\n text,\n message: new AIMessage({\n content: content ?? \"\",\n tool_calls: functionCalls?.map((fc) => {\n return {\n ...fc,\n type: \"tool_call\",\n id: \"id\" in fc && typeof fc.id === \"string\" ? fc.id : uuidv4(),\n };\n }),\n additional_kwargs: {\n ...generationInfo,\n },\n usage_metadata: extra?.usageMetadata,\n }),\n generationInfo,\n };\n\n return {\n generations: [generation],\n llmOutput: {\n tokenUsage: {\n promptTokens: extra?.usageMetadata?.input_tokens,\n completionTokens: extra?.usageMetadata?.output_tokens,\n totalTokens: extra?.usageMetadata?.total_tokens,\n },\n },\n };\n}\n\nexport function convertResponseContentToChatGenerationChunk(\n response: EnhancedGenerateContentResponse,\n extra: {\n usageMetadata?: UsageMetadata | undefined;\n index: number;\n }\n): ChatGenerationChunk | null {\n if (!response.candidates || response.candidates.length === 0) {\n return null;\n }\n const functionCalls = response.functionCalls();\n const [candidate] = response.candidates;\n const { content: candidateContent, ...generationInfo } = candidate;\n let content: MessageContent | undefined;\n // Checks if some parts do not have text. If false, it means that the content is a string.\n if (\n Array.isArray(candidateContent?.parts) &&\n candidateContent.parts.every((p) => \"text\" in p)\n ) {\n content = candidateContent.parts.map((p) => p.text).join(\"\");\n } else if (Array.isArray(candidateContent?.parts)) {\n content = candidateContent.parts.map((p) => {\n if (\"text\" in p) {\n return {\n type: \"text\",\n text: p.text,\n };\n } else if (\"inlineData\" in p) {\n return {\n type: \"inlineData\",\n inlineData: p.inlineData,\n };\n } else if (\"functionCall\" in p) {\n return {\n type: \"functionCall\",\n functionCall: p.functionCall,\n };\n } else if (\"functionResponse\" in p) {\n return {\n type: \"functionResponse\",\n functionResponse: p.functionResponse,\n };\n } else if (\"fileData\" in p) {\n return {\n type: \"fileData\",\n fileData: p.fileData,\n };\n } else if (\"executableCode\" in p) {\n return {\n type: \"executableCode\",\n executableCode: p.executableCode,\n };\n } else if (\"codeExecutionResult\" in p) {\n return {\n type: \"codeExecutionResult\",\n codeExecutionResult: p.codeExecutionResult,\n };\n }\n return p;\n });\n } else {\n // no content returned - likely due to abnormal stop reason, e.g. malformed function call\n content = [];\n }\n\n let text = \"\";\n if (content && typeof content === \"string\") {\n text = content;\n } else if (Array.isArray(content)) {\n const block = content.find((b) => \"text\" in b) as\n | { text: string }\n | undefined;\n text = block?.text ?? \"\";\n }\n\n const toolCallChunks: ToolCallChunk[] = [];\n if (functionCalls) {\n toolCallChunks.push(\n ...functionCalls.map((fc) => ({\n ...fc,\n args: JSON.stringify(fc.args),\n index: extra.index,\n type: \"tool_call_chunk\" as const,\n id: \"id\" in fc && typeof fc.id === \"string\" ? fc.id : uuidv4(),\n }))\n );\n }\n\n return new ChatGenerationChunk({\n text,\n message: new AIMessageChunk({\n content: content || \"\",\n name: !candidateContent ? undefined : candidateContent.role,\n tool_call_chunks: toolCallChunks,\n // Each chunk can have unique \"generationInfo\", and merging strategy is unclear,\n // so leave blank for now.\n additional_kwargs: {},\n response_metadata: {\n model_provider: \"google-genai\",\n },\n usage_metadata: extra.usageMetadata,\n }),\n generationInfo,\n });\n}\n\nexport function convertToGenerativeAITools(\n tools: GoogleGenerativeAIToolType[]\n): GoogleGenerativeAIFunctionDeclarationsTool[] {\n if (\n tools.every(\n (tool) =>\n \"functionDeclarations\" in tool &&\n Array.isArray(tool.functionDeclarations)\n )\n ) {\n return tools as GoogleGenerativeAIFunctionDeclarationsTool[];\n }\n return [\n {\n functionDeclarations: tools.map(\n (tool): GenerativeAIFunctionDeclaration => {\n if (isLangChainTool(tool)) {\n const jsonSchema = schemaToGenerativeAIParameters(tool.schema);\n if (\n jsonSchema.type === \"object\" &&\n \"properties\" in jsonSchema &&\n Object.keys(jsonSchema.properties).length === 0\n ) {\n return {\n name: tool.name,\n description: tool.description,\n };\n }\n return {\n name: tool.name,\n description: tool.description,\n parameters: jsonSchema,\n };\n }\n if (isOpenAITool(tool)) {\n return {\n name: tool.function.name,\n description:\n tool.function.description ?? `A function available to call.`,\n parameters: jsonSchemaToGeminiParameters(\n tool.function.parameters\n ),\n };\n }\n return tool as unknown as GenerativeAIFunctionDeclaration;\n }\n ),\n },\n ];\n}\n"],"mappings":";;;;;;;;AA6CA,SAAgB,iBAAiBA,SAAsB;CACrD,MAAM,OAAO,QAAQ,UAAU;AAC/B,KAAI,YAAY,WAAW,QAAQ,CACjC,QAAO,QAAQ;AAEjB,KAAI,SAAS,OACX,QAAO;AAET,QAAO,QAAQ,QAAQ;AACxB;;;;;;;AAQD,SAAgB,oBACdC,QACiC;AACjC,SAAQ,QAAR;EAKE,KAAK;EACL,KAAK;EACL,KAAK,QACH,QAAO;EACT,KAAK,SACH,QAAO;EACT,KAAK,QACH,QAAO;EACT,KAAK;EACL,KAAK,WACH,QAAO;EACT,QACE,OAAM,IAAI,MAAM,CAAC,8BAA8B,EAAE,QAAQ;CAC5D;AACF;AAED,SAAS,oBAAoBC,SAAsC;AACjE,KAAI,cAAc,WAAW,UAAU,QACrC,QAAO,EACL,YAAY;EACV,UAAU,QAAQ;EAClB,MAAM,QAAQ;CACf,EACF;AAEH,KAAI,cAAc,WAAW,aAAa,QACxC,QAAO,EACL,UAAU;EACR,UAAU,QAAQ;EAClB,SAAS,QAAQ;CAClB,EACF;AAGH,OAAM,IAAI,MAAM;AACjB;AAED,SAAS,kCACPC,SACAC,kBACoB;AACpB,QAAO,iBACJ,IAAI,CAAC,QAAQ;AACZ,MAAI,YAAY,IAAI,CAClB,QAAO,IAAI,cAAc,CAAE;AAE7B,SAAO,CAAE;CACV,EAAC,CACD,MAAM,CACN,KAAK,CAAC,aAAa;AAClB,SAAO,SAAS,OAAO,QAAQ;CAChC,EAAC,EAAE;AACP;AAED,SAAS,kCAAkCC,mBAA4B;CACrE,MAAMC,gCAKD;EACH,cAAc;EAEd,sBAAsB,OAAO;AAC3B,UAAO,EACL,MAAM,MAAM,KACb;EACF;EAED,uBAAuB,OAAsC;AAC3D,OAAI,CAAC,kBACH,OAAM,IAAI,MAAM;AAElB,OAAI,MAAM,gBAAgB,OAAO;IAC/B,MAAM,OAAO,mBAAmB,EAAE,SAAS,MAAM,IAAK,EAAC;AACvD,QAAI,KACF,QAAO,EACL,YAAY;KACV,UAAU,KAAK;KACf,MAAM,KAAK;IACZ,EACF;QAED,QAAO,EACL,UAAU;KACR,UAAU,MAAM,aAAa;KAC7B,SAAS,MAAM;IAChB,EACF;GAEJ;AAED,OAAI,MAAM,gBAAgB,SACxB,QAAO,EACL,YAAY;IACV,UAAU,MAAM,aAAa;IAC7B,MAAM,MAAM;GACb,EACF;AAGH,SAAM,IAAI,MAAM,CAAC,yBAAyB,EAAE,MAAM,aAAa;EAChE;EAED,uBAAuB,OAAsC;AAC3D,OAAI,CAAC,kBACH,OAAM,IAAI,MAAM;AAElB,OAAI,MAAM,gBAAgB,OAAO;IAC/B,MAAM,OAAO,mBAAmB,EAAE,SAAS,MAAM,IAAK,EAAC;AACvD,QAAI,KACF,QAAO,EACL,YAAY;KACV,UAAU,KAAK;KACf,MAAM,KAAK;IACZ,EACF;QAED,QAAO,EACL,UAAU;KACR,UAAU,MAAM,aAAa;KAC7B,SAAS,MAAM;IAChB,EACF;GAEJ;AAED,OAAI,MAAM,gBAAgB,SACxB,QAAO,EACL,YAAY;IACV,UAAU,MAAM,aAAa;IAC7B,MAAM,MAAM;GACb,EACF;AAGH,SAAM,IAAI,MAAM,CAAC,yBAAyB,EAAE,MAAM,aAAa;EAChE;EAED,sBAAsB,OAAiD;AACrE,OAAI,CAAC,kBACH,OAAM,IAAI,MAAM;AAElB,OAAI,MAAM,gBAAgB,OACxB,QAAO,EACL,MAAM,MAAM,KACb;AAEH,OAAI,MAAM,gBAAgB,OAAO;IAC/B,MAAM,OAAO,mBAAmB,EAAE,SAAS,MAAM,IAAK,EAAC;AACvD,QAAI,KACF,QAAO,EACL,YAAY;KACV,UAAU,KAAK;KACf,MAAM,KAAK;IACZ,EACF;QAED,QAAO,EACL,UAAU;KACR,UAAU,MAAM,aAAa;KAC7B,SAAS,MAAM;IAChB,EACF;GAEJ;AAED,OAAI,MAAM,gBAAgB,SACxB,QAAO,EACL,YAAY;IACV,UAAU,MAAM,aAAa;IAC7B,MAAM,MAAM;GACb,EACF;AAEH,SAAM,IAAI,MAAM,CAAC,yBAAyB,EAAE,MAAM,aAAa;EAChE;CACF;AACD,QAAO;AACR;AAED,SAAS,+BACPJ,SACAG,mBACkB;AAClB,KAAI,mBAAmB,QAAQ,CAC7B,QAAO,8BACL,SACA,kCAAkC,kBAAkB,CACrD;AAGH,KAAI,QAAQ,SAAS,OACnB,QAAO,EAAE,MAAM,QAAQ,KAAM;UACpB,QAAQ,SAAS,iBAC1B,QAAO,EAAE,gBAAgB,QAAQ,eAAgB;UACxC,QAAQ,SAAS,sBAC1B,QAAO,EAAE,qBAAqB,QAAQ,oBAAqB;UAClD,QAAQ,SAAS,aAAa;AACvC,MAAI,CAAC,kBACH,OAAM,IAAI,MAAM,CAAC,kCAAkC,CAAC;EAEtD,IAAI;AACJ,MAAI,OAAO,QAAQ,cAAc,UAC/B,SAAS,QAAQ;WAEjB,OAAO,QAAQ,cAAc,YAC7B,SAAS,QAAQ,WAEjB,SAAS,QAAQ,UAAU;MAE3B,OAAM,IAAI,MAAM;EAElB,MAAM,CAAC,IAAI,KAAK,GAAG,OAAO,MAAM,IAAI;AACpC,MAAI,CAAC,GAAG,WAAW,QAAQ,CACzB,OAAM,IAAI,MAAM;EAGlB,MAAM,CAAC,UAAU,SAAS,GAAG,GAAG,QAAQ,UAAU,GAAG,CAAC,MAAM,IAAI;AAChE,MAAI,aAAa,SACf,OAAM,IAAI,MAAM;AAGlB,SAAO,EACL,YAAY;GACV;GACA;EACD,EACF;CACF,WAAU,QAAQ,SAAS,QAC1B,QAAO,oBAAoB,QAAQ;UAC1B,QAAQ,SAAS,WAC1B,QAAO,EACL,cAAc;EACZ,MAAM,QAAQ;EACd,MAAM,QAAQ;CACf,EACF;UAED,QAAQ,MAAM,SAAS,IAAI,IAE3B,QAAQ,KAAK,MAAM,IAAI,CAAC,WAAW,KACnC,UAAU,WACV,OAAO,QAAQ,SAAS,SAExB,QAAO,EACL,YAAY;EACV,UAAU,QAAQ;EAClB,MAAM,QAAQ;CACf,EACF;UACQ,kBAAkB,QAE3B,QAAO;UAEH,UAAU,QACZ,OAAM,IAAI,MAAM,CAAC,qBAAqB,EAAE,QAAQ,MAAM;KAEtD,OAAM,IAAI,MAAM,CAAC,gBAAgB,EAAE,KAAK,UAAU,QAAQ,EAAE;AAGjE;AAED,SAAgB,6BACdL,SACAK,mBACAD,kBACQ;AACR,KAAI,cAAc,QAAQ,EAAE;EAC1B,MAAM,cACJ,QAAQ,QACR,kCAAkC,SAAS,iBAAiB;AAC9D,MAAI,gBAAgB,OAClB,OAAM,IAAI,MACR,CAAC,oHAAoH,EAAE,QAAQ,GAAG,2FAA2F,CAAC;EAIlO,MAAM,SAAS,MAAM,QAAQ,QAAQ,QAAQ,GACxC,QAAQ,QACN,IAAI,CAAC,MAAM,+BAA+B,GAAG,kBAAkB,CAAC,CAChE,OAAO,CAAC,MAAM,MAAM,OAAU,GACjC,QAAQ;AAEZ,MAAI,QAAQ,WAAW,QACrB,QAAO,CACL,EACE,kBAAkB;GAChB,MAAM;GAGN,UAAU,EAAE,OAAO,EAAE,SAAS,OAAQ,EAAE;EACzC,EACF,CACF;AAGH,SAAO,CACL,EACE,kBAAkB;GAChB,MAAM;GAEN,UAAU,EAAE,OAAQ;EACrB,EACF,CACF;CACF;CAED,IAAIG,gBAAoC,CAAE;CAC1C,MAAMC,eAAuB,CAAE;AAE/B,KAAI,OAAO,QAAQ,YAAY,YAAY,QAAQ,SACjD,aAAa,KAAK,EAAE,MAAM,QAAQ,QAAS,EAAC;AAG9C,KAAI,MAAM,QAAQ,QAAQ,QAAQ,EAChC,aAAa,KACX,GAAI,QAAQ,QACT,IAAI,CAAC,MAAM,+BAA+B,GAAG,kBAAkB,CAAC,CAChE,OAAO,CAAC,MAAM,MAAM,OAAU,CAClC;AAGH,KAAI,YAAY,QAAQ,IAAI,QAAQ,YAAY,QAC9C,gBAAgB,QAAQ,WAAW,IAAI,CAAC,OAAO;AAC7C,SAAO,EACL,cAAc;GACZ,MAAM,GAAG;GACT,MAAM,GAAG;EACV,EACF;CACF,EAAC;AAGJ,QAAO,CAAC,GAAG,cAAc,GAAG,aAAc;AAC3C;AAED,SAAgB,6BACdC,UACAJ,mBACAK,qCAA8C,OAC9C;AACA,QAAO,SAAS,OAId,CAAC,KAAK,SAAS,UAAU;AACvB,MAAI,CAAC,cAAc,QAAQ,CACzB,OAAM,IAAI,MAAM;EAElB,MAAM,SAAS,iBAAiB,QAAQ;AACxC,MAAI,WAAW,YAAY,UAAU,EACnC,OAAM,IAAI,MAAM;EAElB,MAAM,OAAO,oBAAoB,OAAO;EAExC,MAAM,cAAc,IAAI,QAAQ,IAAI,QAAQ;AAC5C,MACE,CAAC,IAAI,4BACL,eACA,YAAY,SAAS,KAErB,OAAM,IAAI,MACR;EAIJ,MAAM,QAAQ,6BACZ,SACA,mBACA,SAAS,MAAM,GAAG,MAAM,CACzB;AAED,MAAI,IAAI,0BAA0B;GAChC,MAAMC,gBAAc,IAAI,QAAQ,IAAI,QAAQ,SAAS;AACrD,OAAI,CAACA,cACH,OAAM,IAAI,MACR;GAGJA,cAAY,MAAM,KAAK,GAAG,MAAM;AAEhC,UAAO;IACL,0BAA0B;IAC1B,SAAS,IAAI;GACd;EACF;EACD,IAAI,aAAa;AACjB,MACE,eAAe,cACd,eAAe,YAAY,CAAC,oCAG7B,aAAa;EAEf,MAAMC,UAAmB;GACvB,MAAM;GACN;EACD;AACD,SAAO;GACL,0BACE,WAAW,YAAY,CAAC;GAC1B,SAAS,CAAC,GAAG,IAAI,SAAS,OAAQ;EACnC;CACF,GACD;EAAE,SAAS,CAAE;EAAE,0BAA0B;CAAO,EACjD,CAAC;AACH;AAED,SAAgB,qCACdC,UACAC,OAGY;AAEZ,KACE,CAAC,SAAS,cACV,SAAS,WAAW,WAAW,KAC/B,CAAC,SAAS,WAAW,GAErB,QAAO;EACL,aAAa,CAAE;EACf,WAAW,EACT,SAAS,SAAS,eACnB;CACF;CAGH,MAAM,gBAAgB,SAAS,eAAe;CAC9C,MAAM,CAAC,UAAU,GAAG,SAAS;CAC7B,MAAM,EAAE,SAAS,iBAAkB,GAAG,gBAAgB,GAAG;CACzD,IAAIC;AAEJ,KACE,MAAM,QAAQ,kBAAkB,MAAM,IACtC,iBAAiB,MAAM,WAAW,KAClC,iBAAiB,MAAM,GAAG,MAE1B,UAAU,iBAAiB,MAAM,GAAG;UAEpC,MAAM,QAAQ,kBAAkB,MAAM,IACtC,iBAAiB,MAAM,SAAS,GAEhC,UAAU,iBAAiB,MAAM,IAAI,CAAC,MAAM;AAC1C,MAAI,UAAU,EACZ,QAAO;GACL,MAAM;GACN,MAAM,EAAE;EACT;WACQ,gBAAgB,EACzB,QAAO;GACL,MAAM;GACN,YAAY,EAAE;EACf;WACQ,kBAAkB,EAC3B,QAAO;GACL,MAAM;GACN,cAAc,EAAE;EACjB;WACQ,sBAAsB,EAC/B,QAAO;GACL,MAAM;GACN,kBAAkB,EAAE;EACrB;WACQ,cAAc,EACvB,QAAO;GACL,MAAM;GACN,UAAU,EAAE;EACb;WACQ,oBAAoB,EAC7B,QAAO;GACL,MAAM;GACN,gBAAgB,EAAE;EACnB;WACQ,yBAAyB,EAClC,QAAO;GACL,MAAM;GACN,qBAAqB,EAAE;EACxB;AAEH,SAAO;CACR,EAAC;MAGF,UAAU,CAAE;CAGd,IAAI,OAAO;AACX,KAAI,OAAO,YAAY,UACrB,OAAO;UACE,MAAM,QAAQ,QAAQ,IAAI,QAAQ,SAAS,GAAG;EACvD,MAAM,QAAQ,QAAQ,KAAK,CAAC,MAAM,UAAU,EAAE;EAG9C,OAAO,OAAO,QAAQ;CACvB;CAED,MAAMC,aAA6B;EACjC;EACA,SAAS,IAAI,UAAU;GACrB,SAAS,WAAW;GACpB,YAAY,eAAe,IAAI,CAAC,OAAO;AACrC,WAAO;KACL,GAAG;KACH,MAAM;KACN,IAAI,QAAQ,MAAM,OAAO,GAAG,OAAO,WAAW,GAAG,KAAKC,IAAQ;IAC/D;GACF,EAAC;GACF,mBAAmB,EACjB,GAAG,eACJ;GACD,gBAAgB,OAAO;EACxB;EACD;CACD;AAED,QAAO;EACL,aAAa,CAAC,UAAW;EACzB,WAAW,EACT,YAAY;GACV,cAAc,OAAO,eAAe;GACpC,kBAAkB,OAAO,eAAe;GACxC,aAAa,OAAO,eAAe;EACpC,EACF;CACF;AACF;AAED,SAAgB,4CACdJ,UACAK,OAI4B;AAC5B,KAAI,CAAC,SAAS,cAAc,SAAS,WAAW,WAAW,EACzD,QAAO;CAET,MAAM,gBAAgB,SAAS,eAAe;CAC9C,MAAM,CAAC,UAAU,GAAG,SAAS;CAC7B,MAAM,EAAE,SAAS,iBAAkB,GAAG,gBAAgB,GAAG;CACzD,IAAIH;AAEJ,KACE,MAAM,QAAQ,kBAAkB,MAAM,IACtC,iBAAiB,MAAM,MAAM,CAAC,MAAM,UAAU,EAAE,EAEhD,UAAU,iBAAiB,MAAM,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,KAAK,GAAG;UACnD,MAAM,QAAQ,kBAAkB,MAAM,EAC/C,UAAU,iBAAiB,MAAM,IAAI,CAAC,MAAM;AAC1C,MAAI,UAAU,EACZ,QAAO;GACL,MAAM;GACN,MAAM,EAAE;EACT;WACQ,gBAAgB,EACzB,QAAO;GACL,MAAM;GACN,YAAY,EAAE;EACf;WACQ,kBAAkB,EAC3B,QAAO;GACL,MAAM;GACN,cAAc,EAAE;EACjB;WACQ,sBAAsB,EAC/B,QAAO;GACL,MAAM;GACN,kBAAkB,EAAE;EACrB;WACQ,cAAc,EACvB,QAAO;GACL,MAAM;GACN,UAAU,EAAE;EACb;WACQ,oBAAoB,EAC7B,QAAO;GACL,MAAM;GACN,gBAAgB,EAAE;EACnB;WACQ,yBAAyB,EAClC,QAAO;GACL,MAAM;GACN,qBAAqB,EAAE;EACxB;AAEH,SAAO;CACR,EAAC;MAGF,UAAU,CAAE;CAGd,IAAI,OAAO;AACX,KAAI,WAAW,OAAO,YAAY,UAChC,OAAO;UACE,MAAM,QAAQ,QAAQ,EAAE;EACjC,MAAM,QAAQ,QAAQ,KAAK,CAAC,MAAM,UAAU,EAAE;EAG9C,OAAO,OAAO,QAAQ;CACvB;CAED,MAAMI,iBAAkC,CAAE;AAC1C,KAAI,eACF,eAAe,KACb,GAAG,cAAc,IAAI,CAAC,QAAQ;EAC5B,GAAG;EACH,MAAM,KAAK,UAAU,GAAG,KAAK;EAC7B,OAAO,MAAM;EACb,MAAM;EACN,IAAI,QAAQ,MAAM,OAAO,GAAG,OAAO,WAAW,GAAG,KAAKF,IAAQ;CAC/D,GAAE,CACJ;AAGH,QAAO,IAAI,oBAAoB;EAC7B;EACA,SAAS,IAAI,eAAe;GAC1B,SAAS,WAAW;GACpB,MAAM,CAAC,mBAAmB,SAAY,iBAAiB;GACvD,kBAAkB;GAGlB,mBAAmB,CAAE;GACrB,mBAAmB,EACjB,gBAAgB,eACjB;GACD,gBAAgB,MAAM;EACvB;EACD;CACD;AACF;AAED,SAAgB,2BACdG,OAC8C;AAC9C,KACE,MAAM,MACJ,CAAC,SACC,0BAA0B,QAC1B,MAAM,QAAQ,KAAK,qBAAqB,CAC3C,CAED,QAAO;AAET,QAAO,CACL,EACE,sBAAsB,MAAM,IAC1B,CAAC,SAA0C;AACzC,MAAI,gBAAgB,KAAK,EAAE;GACzB,MAAM,aAAa,+BAA+B,KAAK,OAAO;AAC9D,OACE,WAAW,SAAS,YACpB,gBAAgB,cAChB,OAAO,KAAK,WAAW,WAAW,CAAC,WAAW,EAE9C,QAAO;IACL,MAAM,KAAK;IACX,aAAa,KAAK;GACnB;AAEH,UAAO;IACL,MAAM,KAAK;IACX,aAAa,KAAK;IAClB,YAAY;GACb;EACF;AACD,MAAI,aAAa,KAAK,CACpB,QAAO;GACL,MAAM,KAAK,SAAS;GACpB,aACE,KAAK,SAAS,eAAe,CAAC,6BAA6B,CAAC;GAC9D,YAAY,6BACV,KAAK,SAAS,WACf;EACF;AAEH,SAAO;CACR,EACF,CACF,CACF;AACF"}
|
package/dist/utils/tools.cjs
CHANGED
|
@@ -1,110 +1,73 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
const
|
|
5
|
-
const
|
|
6
|
-
const
|
|
7
|
-
|
|
8
|
-
|
|
1
|
+
const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
|
|
2
|
+
const require_zod_to_genai_parameters = require('./zod_to_genai_parameters.cjs');
|
|
3
|
+
const require_common = require('./common.cjs');
|
|
4
|
+
const __google_generative_ai = require_rolldown_runtime.__toESM(require("@google/generative-ai"));
|
|
5
|
+
const __langchain_core_utils_function_calling = require_rolldown_runtime.__toESM(require("@langchain/core/utils/function_calling"));
|
|
6
|
+
const __langchain_core_language_models_base = require_rolldown_runtime.__toESM(require("@langchain/core/language_models/base"));
|
|
7
|
+
|
|
8
|
+
//#region src/utils/tools.ts
|
|
9
9
|
function convertToolsToGenAI(tools, extra) {
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
10
|
+
const genAITools = processTools(tools);
|
|
11
|
+
const toolConfig = createToolConfig(genAITools, extra);
|
|
12
|
+
return {
|
|
13
|
+
tools: genAITools,
|
|
14
|
+
toolConfig
|
|
15
|
+
};
|
|
15
16
|
}
|
|
16
17
|
function processTools(tools) {
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
}
|
|
40
|
-
});
|
|
41
|
-
const genAIFunctionDeclaration = genAITools.find((t) => "functionDeclarations" in t);
|
|
42
|
-
if (genAIFunctionDeclaration) {
|
|
43
|
-
return genAITools.map((tool) => {
|
|
44
|
-
if (functionDeclarationTools?.length > 0 &&
|
|
45
|
-
"functionDeclarations" in tool) {
|
|
46
|
-
const newTool = {
|
|
47
|
-
functionDeclarations: [
|
|
48
|
-
...(tool.functionDeclarations || []),
|
|
49
|
-
...functionDeclarationTools,
|
|
50
|
-
],
|
|
51
|
-
};
|
|
52
|
-
// Clear the functionDeclarationTools array so it is not passed again
|
|
53
|
-
functionDeclarationTools = [];
|
|
54
|
-
return newTool;
|
|
55
|
-
}
|
|
56
|
-
return tool;
|
|
57
|
-
});
|
|
58
|
-
}
|
|
59
|
-
return [
|
|
60
|
-
...genAITools,
|
|
61
|
-
...(functionDeclarationTools.length > 0
|
|
62
|
-
? [
|
|
63
|
-
{
|
|
64
|
-
functionDeclarations: functionDeclarationTools,
|
|
65
|
-
},
|
|
66
|
-
]
|
|
67
|
-
: []),
|
|
68
|
-
];
|
|
18
|
+
let functionDeclarationTools = [];
|
|
19
|
+
const genAITools = [];
|
|
20
|
+
tools.forEach((tool) => {
|
|
21
|
+
if ((0, __langchain_core_utils_function_calling.isLangChainTool)(tool)) {
|
|
22
|
+
const [convertedTool] = require_common.convertToGenerativeAITools([tool]);
|
|
23
|
+
if (convertedTool.functionDeclarations) functionDeclarationTools.push(...convertedTool.functionDeclarations);
|
|
24
|
+
} else if ((0, __langchain_core_language_models_base.isOpenAITool)(tool)) {
|
|
25
|
+
const { functionDeclarations } = convertOpenAIToolToGenAI(tool);
|
|
26
|
+
if (functionDeclarations) functionDeclarationTools.push(...functionDeclarations);
|
|
27
|
+
else throw new Error("Failed to convert OpenAI structured tool to GenerativeAI tool");
|
|
28
|
+
} else genAITools.push(tool);
|
|
29
|
+
});
|
|
30
|
+
const genAIFunctionDeclaration = genAITools.find((t) => "functionDeclarations" in t);
|
|
31
|
+
if (genAIFunctionDeclaration) return genAITools.map((tool) => {
|
|
32
|
+
if (functionDeclarationTools?.length > 0 && "functionDeclarations" in tool) {
|
|
33
|
+
const newTool = { functionDeclarations: [...tool.functionDeclarations || [], ...functionDeclarationTools] };
|
|
34
|
+
functionDeclarationTools = [];
|
|
35
|
+
return newTool;
|
|
36
|
+
}
|
|
37
|
+
return tool;
|
|
38
|
+
});
|
|
39
|
+
return [...genAITools, ...functionDeclarationTools.length > 0 ? [{ functionDeclarations: functionDeclarationTools }] : []];
|
|
69
40
|
}
|
|
70
41
|
function convertOpenAIToolToGenAI(tool) {
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
parameters: (0, zod_to_genai_parameters_js_1.removeAdditionalProperties)(tool.function.parameters),
|
|
77
|
-
},
|
|
78
|
-
],
|
|
79
|
-
};
|
|
42
|
+
return { functionDeclarations: [{
|
|
43
|
+
name: tool.function.name,
|
|
44
|
+
description: tool.function.description,
|
|
45
|
+
parameters: require_zod_to_genai_parameters.removeAdditionalProperties(tool.function.parameters)
|
|
46
|
+
}] };
|
|
80
47
|
}
|
|
81
48
|
function createToolConfig(genAITools, extra) {
|
|
82
|
-
|
|
83
|
-
|
|
84
|
-
|
|
85
|
-
|
|
86
|
-
|
|
87
|
-
|
|
88
|
-
|
|
89
|
-
|
|
90
|
-
|
|
91
|
-
|
|
92
|
-
|
|
93
|
-
|
|
94
|
-
|
|
95
|
-
|
|
96
|
-
|
|
97
|
-
|
|
98
|
-
|
|
99
|
-
|
|
100
|
-
|
|
101
|
-
|
|
102
|
-
allowedFunctionNames: [
|
|
103
|
-
...(allowedFunctionNames ?? []),
|
|
104
|
-
...(toolChoice && typeof toolChoice === "string" ? [toolChoice] : []),
|
|
105
|
-
],
|
|
106
|
-
},
|
|
107
|
-
};
|
|
108
|
-
}
|
|
109
|
-
return undefined;
|
|
49
|
+
if (!genAITools.length || !extra) return void 0;
|
|
50
|
+
const { toolChoice, allowedFunctionNames } = extra;
|
|
51
|
+
const modeMap = {
|
|
52
|
+
any: __google_generative_ai.FunctionCallingMode.ANY,
|
|
53
|
+
auto: __google_generative_ai.FunctionCallingMode.AUTO,
|
|
54
|
+
none: __google_generative_ai.FunctionCallingMode.NONE
|
|
55
|
+
};
|
|
56
|
+
if (toolChoice && [
|
|
57
|
+
"any",
|
|
58
|
+
"auto",
|
|
59
|
+
"none"
|
|
60
|
+
].includes(toolChoice)) return { functionCallingConfig: {
|
|
61
|
+
mode: modeMap[toolChoice] ?? "MODE_UNSPECIFIED",
|
|
62
|
+
allowedFunctionNames
|
|
63
|
+
} };
|
|
64
|
+
if (typeof toolChoice === "string" || allowedFunctionNames) return { functionCallingConfig: {
|
|
65
|
+
mode: __google_generative_ai.FunctionCallingMode.ANY,
|
|
66
|
+
allowedFunctionNames: [...allowedFunctionNames ?? [], ...toolChoice && typeof toolChoice === "string" ? [toolChoice] : []]
|
|
67
|
+
} };
|
|
68
|
+
return void 0;
|
|
110
69
|
}
|
|
70
|
+
|
|
71
|
+
//#endregion
|
|
72
|
+
exports.convertToolsToGenAI = convertToolsToGenAI;
|
|
73
|
+
//# sourceMappingURL=tools.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tools.cjs","names":["tools: GoogleGenerativeAIToolType[]","extra?: {\n toolChoice?: ToolChoice;\n allowedFunctionNames?: string[];\n }","functionDeclarationTools: FunctionDeclaration[]","genAITools: GenerativeAITool[]","convertToGenerativeAITools","tool: ToolDefinition","removeAdditionalProperties","modeMap: Record<string, FunctionCallingMode>","FunctionCallingMode"],"sources":["../../src/utils/tools.ts"],"sourcesContent":["import {\n Tool as GenerativeAITool,\n ToolConfig,\n FunctionCallingMode,\n FunctionDeclaration,\n FunctionDeclarationsTool,\n FunctionDeclarationSchema,\n} from \"@google/generative-ai\";\nimport { ToolChoice } from \"@langchain/core/language_models/chat_models\";\nimport { StructuredToolInterface } from \"@langchain/core/tools\";\nimport { isLangChainTool } from \"@langchain/core/utils/function_calling\";\nimport {\n isOpenAITool,\n ToolDefinition,\n} from \"@langchain/core/language_models/base\";\nimport { convertToGenerativeAITools } from \"./common.js\";\nimport { GoogleGenerativeAIToolType } from \"../types.js\";\nimport { removeAdditionalProperties } from \"./zod_to_genai_parameters.js\";\n\nexport function convertToolsToGenAI(\n tools: GoogleGenerativeAIToolType[],\n extra?: {\n toolChoice?: ToolChoice;\n allowedFunctionNames?: string[];\n }\n): {\n tools: GenerativeAITool[];\n toolConfig?: ToolConfig;\n} {\n // Extract function declaration processing to a separate function\n const genAITools = processTools(tools);\n\n // Simplify tool config creation\n const toolConfig = createToolConfig(genAITools, extra);\n\n return { tools: genAITools, toolConfig };\n}\n\nfunction processTools(tools: GoogleGenerativeAIToolType[]): GenerativeAITool[] {\n let functionDeclarationTools: FunctionDeclaration[] = [];\n const genAITools: GenerativeAITool[] = [];\n\n tools.forEach((tool) => {\n if (isLangChainTool(tool)) {\n const [convertedTool] = convertToGenerativeAITools([\n tool as StructuredToolInterface,\n ]);\n if (convertedTool.functionDeclarations) {\n functionDeclarationTools.push(...convertedTool.functionDeclarations);\n }\n } else if (isOpenAITool(tool)) {\n const { functionDeclarations } = convertOpenAIToolToGenAI(tool);\n if (functionDeclarations) {\n functionDeclarationTools.push(...functionDeclarations);\n } else {\n throw new Error(\n \"Failed to convert OpenAI structured tool to GenerativeAI tool\"\n );\n }\n } else {\n genAITools.push(tool as GenerativeAITool);\n }\n });\n\n const genAIFunctionDeclaration = genAITools.find(\n (t) => \"functionDeclarations\" in t\n );\n if (genAIFunctionDeclaration) {\n return genAITools.map((tool) => {\n if (\n functionDeclarationTools?.length > 0 &&\n \"functionDeclarations\" in tool\n ) {\n const newTool = {\n functionDeclarations: [\n ...(tool.functionDeclarations || []),\n ...functionDeclarationTools,\n ],\n };\n // Clear the functionDeclarationTools array so it is not passed again\n functionDeclarationTools = [];\n return newTool;\n }\n return tool;\n });\n }\n\n return [\n ...genAITools,\n ...(functionDeclarationTools.length > 0\n ? [\n {\n functionDeclarations: functionDeclarationTools,\n },\n ]\n : []),\n ];\n}\n\nfunction convertOpenAIToolToGenAI(\n tool: ToolDefinition\n): FunctionDeclarationsTool {\n return {\n functionDeclarations: [\n {\n name: tool.function.name,\n description: tool.function.description,\n parameters: removeAdditionalProperties(\n tool.function.parameters\n ) as FunctionDeclarationSchema,\n },\n ],\n };\n}\n\nfunction createToolConfig(\n genAITools: GenerativeAITool[],\n extra?: {\n toolChoice?: ToolChoice;\n allowedFunctionNames?: string[];\n }\n): ToolConfig | undefined {\n if (!genAITools.length || !extra) return undefined;\n\n const { toolChoice, allowedFunctionNames } = extra;\n\n const modeMap: Record<string, FunctionCallingMode> = {\n any: FunctionCallingMode.ANY,\n auto: FunctionCallingMode.AUTO,\n none: FunctionCallingMode.NONE,\n };\n\n if (toolChoice && [\"any\", \"auto\", \"none\"].includes(toolChoice as string)) {\n return {\n functionCallingConfig: {\n mode: modeMap[toolChoice as keyof typeof modeMap] ?? \"MODE_UNSPECIFIED\",\n allowedFunctionNames,\n },\n };\n }\n\n if (typeof toolChoice === \"string\" || allowedFunctionNames) {\n return {\n functionCallingConfig: {\n mode: FunctionCallingMode.ANY,\n allowedFunctionNames: [\n ...(allowedFunctionNames ?? []),\n ...(toolChoice && typeof toolChoice === \"string\" ? [toolChoice] : []),\n ],\n },\n };\n }\n\n return undefined;\n}\n"],"mappings":";;;;;;;;AAmBA,SAAgB,oBACdA,OACAC,OAOA;CAEA,MAAM,aAAa,aAAa,MAAM;CAGtC,MAAM,aAAa,iBAAiB,YAAY,MAAM;AAEtD,QAAO;EAAE,OAAO;EAAY;CAAY;AACzC;AAED,SAAS,aAAaD,OAAyD;CAC7E,IAAIE,2BAAkD,CAAE;CACxD,MAAMC,aAAiC,CAAE;CAEzC,MAAM,QAAQ,CAAC,SAAS;AACtB,mEAAoB,KAAK,EAAE;GACzB,MAAM,CAAC,cAAc,GAAGC,0CAA2B,CACjD,IACD,EAAC;AACF,OAAI,cAAc,sBAChB,yBAAyB,KAAK,GAAG,cAAc,qBAAqB;EAEvE,mEAAuB,KAAK,EAAE;GAC7B,MAAM,EAAE,sBAAsB,GAAG,yBAAyB,KAAK;AAC/D,OAAI,sBACF,yBAAyB,KAAK,GAAG,qBAAqB;OAEtD,OAAM,IAAI,MACR;EAGL,OACC,WAAW,KAAK,KAAyB;CAE5C,EAAC;CAEF,MAAM,2BAA2B,WAAW,KAC1C,CAAC,MAAM,0BAA0B,EAClC;AACD,KAAI,yBACF,QAAO,WAAW,IAAI,CAAC,SAAS;AAC9B,MACE,0BAA0B,SAAS,KACnC,0BAA0B,MAC1B;GACA,MAAM,UAAU,EACd,sBAAsB,CACpB,GAAI,KAAK,wBAAwB,CAAE,GACnC,GAAG,wBACJ,EACF;GAED,2BAA2B,CAAE;AAC7B,UAAO;EACR;AACD,SAAO;CACR,EAAC;AAGJ,QAAO,CACL,GAAG,YACH,GAAI,yBAAyB,SAAS,IAClC,CACE,EACE,sBAAsB,yBACvB,CACF,IACD,CAAE,CACP;AACF;AAED,SAAS,yBACPC,MAC0B;AAC1B,QAAO,EACL,sBAAsB,CACpB;EACE,MAAM,KAAK,SAAS;EACpB,aAAa,KAAK,SAAS;EAC3B,YAAYC,2DACV,KAAK,SAAS,WACf;CACF,CACF,EACF;AACF;AAED,SAAS,iBACPH,YACAF,OAIwB;AACxB,KAAI,CAAC,WAAW,UAAU,CAAC,MAAO,QAAO;CAEzC,MAAM,EAAE,YAAY,sBAAsB,GAAG;CAE7C,MAAMM,UAA+C;EACnD,KAAKC,2CAAoB;EACzB,MAAMA,2CAAoB;EAC1B,MAAMA,2CAAoB;CAC3B;AAED,KAAI,cAAc;EAAC;EAAO;EAAQ;CAAO,EAAC,SAAS,WAAqB,CACtE,QAAO,EACL,uBAAuB;EACrB,MAAM,QAAQ,eAAuC;EACrD;CACD,EACF;AAGH,KAAI,OAAO,eAAe,YAAY,qBACpC,QAAO,EACL,uBAAuB;EACrB,MAAMA,2CAAoB;EAC1B,sBAAsB,CACpB,GAAI,wBAAwB,CAAE,GAC9B,GAAI,cAAc,OAAO,eAAe,WAAW,CAAC,UAAW,IAAG,CAAE,CACrE;CACF,EACF;AAGH,QAAO;AACR"}
|
package/dist/utils/tools.js
CHANGED
|
@@ -1,107 +1,72 @@
|
|
|
1
|
-
import { FunctionCallingMode, } from "@google/generative-ai";
|
|
2
|
-
import { isLangChainTool } from "@langchain/core/utils/function_calling";
|
|
3
|
-
import { isOpenAITool, } from "@langchain/core/language_models/base";
|
|
4
|
-
import { convertToGenerativeAITools } from "./common.js";
|
|
5
1
|
import { removeAdditionalProperties } from "./zod_to_genai_parameters.js";
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
2
|
+
import { convertToGenerativeAITools } from "./common.js";
|
|
3
|
+
import { FunctionCallingMode } from "@google/generative-ai";
|
|
4
|
+
import { isLangChainTool } from "@langchain/core/utils/function_calling";
|
|
5
|
+
import { isOpenAITool } from "@langchain/core/language_models/base";
|
|
6
|
+
|
|
7
|
+
//#region src/utils/tools.ts
|
|
8
|
+
function convertToolsToGenAI(tools, extra) {
|
|
9
|
+
const genAITools = processTools(tools);
|
|
10
|
+
const toolConfig = createToolConfig(genAITools, extra);
|
|
11
|
+
return {
|
|
12
|
+
tools: genAITools,
|
|
13
|
+
toolConfig
|
|
14
|
+
};
|
|
12
15
|
}
|
|
13
16
|
function processTools(tools) {
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
}
|
|
37
|
-
});
|
|
38
|
-
const genAIFunctionDeclaration = genAITools.find((t) => "functionDeclarations" in t);
|
|
39
|
-
if (genAIFunctionDeclaration) {
|
|
40
|
-
return genAITools.map((tool) => {
|
|
41
|
-
if (functionDeclarationTools?.length > 0 &&
|
|
42
|
-
"functionDeclarations" in tool) {
|
|
43
|
-
const newTool = {
|
|
44
|
-
functionDeclarations: [
|
|
45
|
-
...(tool.functionDeclarations || []),
|
|
46
|
-
...functionDeclarationTools,
|
|
47
|
-
],
|
|
48
|
-
};
|
|
49
|
-
// Clear the functionDeclarationTools array so it is not passed again
|
|
50
|
-
functionDeclarationTools = [];
|
|
51
|
-
return newTool;
|
|
52
|
-
}
|
|
53
|
-
return tool;
|
|
54
|
-
});
|
|
55
|
-
}
|
|
56
|
-
return [
|
|
57
|
-
...genAITools,
|
|
58
|
-
...(functionDeclarationTools.length > 0
|
|
59
|
-
? [
|
|
60
|
-
{
|
|
61
|
-
functionDeclarations: functionDeclarationTools,
|
|
62
|
-
},
|
|
63
|
-
]
|
|
64
|
-
: []),
|
|
65
|
-
];
|
|
17
|
+
let functionDeclarationTools = [];
|
|
18
|
+
const genAITools = [];
|
|
19
|
+
tools.forEach((tool) => {
|
|
20
|
+
if (isLangChainTool(tool)) {
|
|
21
|
+
const [convertedTool] = convertToGenerativeAITools([tool]);
|
|
22
|
+
if (convertedTool.functionDeclarations) functionDeclarationTools.push(...convertedTool.functionDeclarations);
|
|
23
|
+
} else if (isOpenAITool(tool)) {
|
|
24
|
+
const { functionDeclarations } = convertOpenAIToolToGenAI(tool);
|
|
25
|
+
if (functionDeclarations) functionDeclarationTools.push(...functionDeclarations);
|
|
26
|
+
else throw new Error("Failed to convert OpenAI structured tool to GenerativeAI tool");
|
|
27
|
+
} else genAITools.push(tool);
|
|
28
|
+
});
|
|
29
|
+
const genAIFunctionDeclaration = genAITools.find((t) => "functionDeclarations" in t);
|
|
30
|
+
if (genAIFunctionDeclaration) return genAITools.map((tool) => {
|
|
31
|
+
if (functionDeclarationTools?.length > 0 && "functionDeclarations" in tool) {
|
|
32
|
+
const newTool = { functionDeclarations: [...tool.functionDeclarations || [], ...functionDeclarationTools] };
|
|
33
|
+
functionDeclarationTools = [];
|
|
34
|
+
return newTool;
|
|
35
|
+
}
|
|
36
|
+
return tool;
|
|
37
|
+
});
|
|
38
|
+
return [...genAITools, ...functionDeclarationTools.length > 0 ? [{ functionDeclarations: functionDeclarationTools }] : []];
|
|
66
39
|
}
|
|
67
40
|
function convertOpenAIToolToGenAI(tool) {
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
parameters: removeAdditionalProperties(tool.function.parameters),
|
|
74
|
-
},
|
|
75
|
-
],
|
|
76
|
-
};
|
|
41
|
+
return { functionDeclarations: [{
|
|
42
|
+
name: tool.function.name,
|
|
43
|
+
description: tool.function.description,
|
|
44
|
+
parameters: removeAdditionalProperties(tool.function.parameters)
|
|
45
|
+
}] };
|
|
77
46
|
}
|
|
78
47
|
function createToolConfig(genAITools, extra) {
|
|
79
|
-
|
|
80
|
-
|
|
81
|
-
|
|
82
|
-
|
|
83
|
-
|
|
84
|
-
|
|
85
|
-
|
|
86
|
-
|
|
87
|
-
|
|
88
|
-
|
|
89
|
-
|
|
90
|
-
|
|
91
|
-
|
|
92
|
-
|
|
93
|
-
|
|
94
|
-
|
|
95
|
-
|
|
96
|
-
|
|
97
|
-
|
|
98
|
-
|
|
99
|
-
allowedFunctionNames: [
|
|
100
|
-
...(allowedFunctionNames ?? []),
|
|
101
|
-
...(toolChoice && typeof toolChoice === "string" ? [toolChoice] : []),
|
|
102
|
-
],
|
|
103
|
-
},
|
|
104
|
-
};
|
|
105
|
-
}
|
|
106
|
-
return undefined;
|
|
48
|
+
if (!genAITools.length || !extra) return void 0;
|
|
49
|
+
const { toolChoice, allowedFunctionNames } = extra;
|
|
50
|
+
const modeMap = {
|
|
51
|
+
any: FunctionCallingMode.ANY,
|
|
52
|
+
auto: FunctionCallingMode.AUTO,
|
|
53
|
+
none: FunctionCallingMode.NONE
|
|
54
|
+
};
|
|
55
|
+
if (toolChoice && [
|
|
56
|
+
"any",
|
|
57
|
+
"auto",
|
|
58
|
+
"none"
|
|
59
|
+
].includes(toolChoice)) return { functionCallingConfig: {
|
|
60
|
+
mode: modeMap[toolChoice] ?? "MODE_UNSPECIFIED",
|
|
61
|
+
allowedFunctionNames
|
|
62
|
+
} };
|
|
63
|
+
if (typeof toolChoice === "string" || allowedFunctionNames) return { functionCallingConfig: {
|
|
64
|
+
mode: FunctionCallingMode.ANY,
|
|
65
|
+
allowedFunctionNames: [...allowedFunctionNames ?? [], ...toolChoice && typeof toolChoice === "string" ? [toolChoice] : []]
|
|
66
|
+
} };
|
|
67
|
+
return void 0;
|
|
107
68
|
}
|
|
69
|
+
|
|
70
|
+
//#endregion
|
|
71
|
+
export { convertToolsToGenAI };
|
|
72
|
+
//# sourceMappingURL=tools.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tools.js","names":["tools: GoogleGenerativeAIToolType[]","extra?: {\n toolChoice?: ToolChoice;\n allowedFunctionNames?: string[];\n }","functionDeclarationTools: FunctionDeclaration[]","genAITools: GenerativeAITool[]","tool: ToolDefinition","modeMap: Record<string, FunctionCallingMode>"],"sources":["../../src/utils/tools.ts"],"sourcesContent":["import {\n Tool as GenerativeAITool,\n ToolConfig,\n FunctionCallingMode,\n FunctionDeclaration,\n FunctionDeclarationsTool,\n FunctionDeclarationSchema,\n} from \"@google/generative-ai\";\nimport { ToolChoice } from \"@langchain/core/language_models/chat_models\";\nimport { StructuredToolInterface } from \"@langchain/core/tools\";\nimport { isLangChainTool } from \"@langchain/core/utils/function_calling\";\nimport {\n isOpenAITool,\n ToolDefinition,\n} from \"@langchain/core/language_models/base\";\nimport { convertToGenerativeAITools } from \"./common.js\";\nimport { GoogleGenerativeAIToolType } from \"../types.js\";\nimport { removeAdditionalProperties } from \"./zod_to_genai_parameters.js\";\n\nexport function convertToolsToGenAI(\n tools: GoogleGenerativeAIToolType[],\n extra?: {\n toolChoice?: ToolChoice;\n allowedFunctionNames?: string[];\n }\n): {\n tools: GenerativeAITool[];\n toolConfig?: ToolConfig;\n} {\n // Extract function declaration processing to a separate function\n const genAITools = processTools(tools);\n\n // Simplify tool config creation\n const toolConfig = createToolConfig(genAITools, extra);\n\n return { tools: genAITools, toolConfig };\n}\n\nfunction processTools(tools: GoogleGenerativeAIToolType[]): GenerativeAITool[] {\n let functionDeclarationTools: FunctionDeclaration[] = [];\n const genAITools: GenerativeAITool[] = [];\n\n tools.forEach((tool) => {\n if (isLangChainTool(tool)) {\n const [convertedTool] = convertToGenerativeAITools([\n tool as StructuredToolInterface,\n ]);\n if (convertedTool.functionDeclarations) {\n functionDeclarationTools.push(...convertedTool.functionDeclarations);\n }\n } else if (isOpenAITool(tool)) {\n const { functionDeclarations } = convertOpenAIToolToGenAI(tool);\n if (functionDeclarations) {\n functionDeclarationTools.push(...functionDeclarations);\n } else {\n throw new Error(\n \"Failed to convert OpenAI structured tool to GenerativeAI tool\"\n );\n }\n } else {\n genAITools.push(tool as GenerativeAITool);\n }\n });\n\n const genAIFunctionDeclaration = genAITools.find(\n (t) => \"functionDeclarations\" in t\n );\n if (genAIFunctionDeclaration) {\n return genAITools.map((tool) => {\n if (\n functionDeclarationTools?.length > 0 &&\n \"functionDeclarations\" in tool\n ) {\n const newTool = {\n functionDeclarations: [\n ...(tool.functionDeclarations || []),\n ...functionDeclarationTools,\n ],\n };\n // Clear the functionDeclarationTools array so it is not passed again\n functionDeclarationTools = [];\n return newTool;\n }\n return tool;\n });\n }\n\n return [\n ...genAITools,\n ...(functionDeclarationTools.length > 0\n ? [\n {\n functionDeclarations: functionDeclarationTools,\n },\n ]\n : []),\n ];\n}\n\nfunction convertOpenAIToolToGenAI(\n tool: ToolDefinition\n): FunctionDeclarationsTool {\n return {\n functionDeclarations: [\n {\n name: tool.function.name,\n description: tool.function.description,\n parameters: removeAdditionalProperties(\n tool.function.parameters\n ) as FunctionDeclarationSchema,\n },\n ],\n };\n}\n\nfunction createToolConfig(\n genAITools: GenerativeAITool[],\n extra?: {\n toolChoice?: ToolChoice;\n allowedFunctionNames?: string[];\n }\n): ToolConfig | undefined {\n if (!genAITools.length || !extra) return undefined;\n\n const { toolChoice, allowedFunctionNames } = extra;\n\n const modeMap: Record<string, FunctionCallingMode> = {\n any: FunctionCallingMode.ANY,\n auto: FunctionCallingMode.AUTO,\n none: FunctionCallingMode.NONE,\n };\n\n if (toolChoice && [\"any\", \"auto\", \"none\"].includes(toolChoice as string)) {\n return {\n functionCallingConfig: {\n mode: modeMap[toolChoice as keyof typeof modeMap] ?? \"MODE_UNSPECIFIED\",\n allowedFunctionNames,\n },\n };\n }\n\n if (typeof toolChoice === \"string\" || allowedFunctionNames) {\n return {\n functionCallingConfig: {\n mode: FunctionCallingMode.ANY,\n allowedFunctionNames: [\n ...(allowedFunctionNames ?? []),\n ...(toolChoice && typeof toolChoice === \"string\" ? [toolChoice] : []),\n ],\n },\n };\n }\n\n return undefined;\n}\n"],"mappings":";;;;;;;AAmBA,SAAgB,oBACdA,OACAC,OAOA;CAEA,MAAM,aAAa,aAAa,MAAM;CAGtC,MAAM,aAAa,iBAAiB,YAAY,MAAM;AAEtD,QAAO;EAAE,OAAO;EAAY;CAAY;AACzC;AAED,SAAS,aAAaD,OAAyD;CAC7E,IAAIE,2BAAkD,CAAE;CACxD,MAAMC,aAAiC,CAAE;CAEzC,MAAM,QAAQ,CAAC,SAAS;AACtB,MAAI,gBAAgB,KAAK,EAAE;GACzB,MAAM,CAAC,cAAc,GAAG,2BAA2B,CACjD,IACD,EAAC;AACF,OAAI,cAAc,sBAChB,yBAAyB,KAAK,GAAG,cAAc,qBAAqB;EAEvE,WAAU,aAAa,KAAK,EAAE;GAC7B,MAAM,EAAE,sBAAsB,GAAG,yBAAyB,KAAK;AAC/D,OAAI,sBACF,yBAAyB,KAAK,GAAG,qBAAqB;OAEtD,OAAM,IAAI,MACR;EAGL,OACC,WAAW,KAAK,KAAyB;CAE5C,EAAC;CAEF,MAAM,2BAA2B,WAAW,KAC1C,CAAC,MAAM,0BAA0B,EAClC;AACD,KAAI,yBACF,QAAO,WAAW,IAAI,CAAC,SAAS;AAC9B,MACE,0BAA0B,SAAS,KACnC,0BAA0B,MAC1B;GACA,MAAM,UAAU,EACd,sBAAsB,CACpB,GAAI,KAAK,wBAAwB,CAAE,GACnC,GAAG,wBACJ,EACF;GAED,2BAA2B,CAAE;AAC7B,UAAO;EACR;AACD,SAAO;CACR,EAAC;AAGJ,QAAO,CACL,GAAG,YACH,GAAI,yBAAyB,SAAS,IAClC,CACE,EACE,sBAAsB,yBACvB,CACF,IACD,CAAE,CACP;AACF;AAED,SAAS,yBACPC,MAC0B;AAC1B,QAAO,EACL,sBAAsB,CACpB;EACE,MAAM,KAAK,SAAS;EACpB,aAAa,KAAK,SAAS;EAC3B,YAAY,2BACV,KAAK,SAAS,WACf;CACF,CACF,EACF;AACF;AAED,SAAS,iBACPD,YACAF,OAIwB;AACxB,KAAI,CAAC,WAAW,UAAU,CAAC,MAAO,QAAO;CAEzC,MAAM,EAAE,YAAY,sBAAsB,GAAG;CAE7C,MAAMI,UAA+C;EACnD,KAAK,oBAAoB;EACzB,MAAM,oBAAoB;EAC1B,MAAM,oBAAoB;CAC3B;AAED,KAAI,cAAc;EAAC;EAAO;EAAQ;CAAO,EAAC,SAAS,WAAqB,CACtE,QAAO,EACL,uBAAuB;EACrB,MAAM,QAAQ,eAAuC;EACrD;CACD,EACF;AAGH,KAAI,OAAO,eAAe,YAAY,qBACpC,QAAO,EACL,uBAAuB;EACrB,MAAM,oBAAoB;EAC1B,sBAAsB,CACpB,GAAI,wBAAwB,CAAE,GAC9B,GAAI,cAAc,OAAO,eAAe,WAAW,CAAC,UAAW,IAAG,CAAE,CACrE;CACF,EACF;AAGH,QAAO;AACR"}
|
|
@@ -1,53 +1,35 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
const
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
}
|
|
20
|
-
if ("strict" in newObj) {
|
|
21
|
-
delete newObj.strict;
|
|
22
|
-
}
|
|
23
|
-
for (const key in newObj) {
|
|
24
|
-
if (key in newObj) {
|
|
25
|
-
if (Array.isArray(newObj[key])) {
|
|
26
|
-
newObj[key] = newObj[key].map(removeAdditionalProperties);
|
|
27
|
-
}
|
|
28
|
-
else if (typeof newObj[key] === "object" && newObj[key] !== null) {
|
|
29
|
-
newObj[key] = removeAdditionalProperties(newObj[key]);
|
|
30
|
-
}
|
|
31
|
-
}
|
|
32
|
-
}
|
|
33
|
-
return newObj;
|
|
34
|
-
}
|
|
35
|
-
return obj;
|
|
1
|
+
const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
|
|
2
|
+
const __langchain_core_utils_types = require_rolldown_runtime.__toESM(require("@langchain/core/utils/types"));
|
|
3
|
+
const __langchain_core_utils_json_schema = require_rolldown_runtime.__toESM(require("@langchain/core/utils/json_schema"));
|
|
4
|
+
|
|
5
|
+
//#region src/utils/zod_to_genai_parameters.ts
|
|
6
|
+
function removeAdditionalProperties(obj) {
|
|
7
|
+
if (typeof obj === "object" && obj !== null) {
|
|
8
|
+
const newObj = { ...obj };
|
|
9
|
+
if ("additionalProperties" in newObj) delete newObj.additionalProperties;
|
|
10
|
+
if ("$schema" in newObj) delete newObj.$schema;
|
|
11
|
+
if ("strict" in newObj) delete newObj.strict;
|
|
12
|
+
for (const key in newObj) if (key in newObj) {
|
|
13
|
+
if (Array.isArray(newObj[key])) newObj[key] = newObj[key].map(removeAdditionalProperties);
|
|
14
|
+
else if (typeof newObj[key] === "object" && newObj[key] !== null) newObj[key] = removeAdditionalProperties(newObj[key]);
|
|
15
|
+
}
|
|
16
|
+
return newObj;
|
|
17
|
+
}
|
|
18
|
+
return obj;
|
|
36
19
|
}
|
|
37
20
|
function schemaToGenerativeAIParameters(schema) {
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
const { $schema, ...rest } = jsonSchema;
|
|
42
|
-
return rest;
|
|
21
|
+
const jsonSchema = removeAdditionalProperties((0, __langchain_core_utils_types.isInteropZodSchema)(schema) ? (0, __langchain_core_utils_json_schema.toJsonSchema)(schema) : schema);
|
|
22
|
+
const { $schema,...rest } = jsonSchema;
|
|
23
|
+
return rest;
|
|
43
24
|
}
|
|
44
|
-
function jsonSchemaToGeminiParameters(
|
|
45
|
-
|
|
46
|
-
schema
|
|
47
|
-
|
|
48
|
-
// attributes, so we need to explicitly remove them.
|
|
49
|
-
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
50
|
-
const jsonSchema = removeAdditionalProperties(schema);
|
|
51
|
-
const { $schema, ...rest } = jsonSchema;
|
|
52
|
-
return rest;
|
|
25
|
+
function jsonSchemaToGeminiParameters(schema) {
|
|
26
|
+
const jsonSchema = removeAdditionalProperties(schema);
|
|
27
|
+
const { $schema,...rest } = jsonSchema;
|
|
28
|
+
return rest;
|
|
53
29
|
}
|
|
30
|
+
|
|
31
|
+
//#endregion
|
|
32
|
+
exports.jsonSchemaToGeminiParameters = jsonSchemaToGeminiParameters;
|
|
33
|
+
exports.removeAdditionalProperties = removeAdditionalProperties;
|
|
34
|
+
exports.schemaToGenerativeAIParameters = schemaToGenerativeAIParameters;
|
|
35
|
+
//# sourceMappingURL=zod_to_genai_parameters.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"zod_to_genai_parameters.cjs","names":["obj: Record<string, any>","schema: InteropZodType<RunOutput> | JsonSchema7Type","schema: Record<string, any>"],"sources":["../../src/utils/zod_to_genai_parameters.ts"],"sourcesContent":["import {\n type FunctionDeclarationSchema as GenerativeAIFunctionDeclarationSchema,\n type SchemaType as FunctionDeclarationSchemaType,\n} from \"@google/generative-ai\";\nimport {\n InteropZodType,\n isInteropZodSchema,\n} from \"@langchain/core/utils/types\";\nimport {\n type JsonSchema7Type,\n toJsonSchema,\n} from \"@langchain/core/utils/json_schema\";\n\nexport interface GenerativeAIJsonSchema extends Record<string, unknown> {\n properties?: Record<string, GenerativeAIJsonSchema>;\n type: FunctionDeclarationSchemaType;\n}\n\nexport interface GenerativeAIJsonSchemaDirty extends GenerativeAIJsonSchema {\n properties?: Record<string, GenerativeAIJsonSchemaDirty>;\n additionalProperties?: boolean;\n}\n\nexport function removeAdditionalProperties(\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n obj: Record<string, any>\n): GenerativeAIJsonSchema {\n if (typeof obj === \"object\" && obj !== null) {\n const newObj = { ...obj };\n\n if (\"additionalProperties\" in newObj) {\n delete newObj.additionalProperties;\n }\n if (\"$schema\" in newObj) {\n delete newObj.$schema;\n }\n if (\"strict\" in newObj) {\n delete newObj.strict;\n }\n\n for (const key in newObj) {\n if (key in newObj) {\n if (Array.isArray(newObj[key])) {\n newObj[key] = newObj[key].map(removeAdditionalProperties);\n } else if (typeof newObj[key] === \"object\" && newObj[key] !== null) {\n newObj[key] = removeAdditionalProperties(newObj[key]);\n }\n }\n }\n\n return newObj as GenerativeAIJsonSchema;\n }\n\n return obj as GenerativeAIJsonSchema;\n}\n\nexport function schemaToGenerativeAIParameters<\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n RunOutput extends Record<string, any> = Record<string, any>\n>(\n schema: InteropZodType<RunOutput> | JsonSchema7Type\n): GenerativeAIFunctionDeclarationSchema {\n // GenerativeAI doesn't accept either the $schema or additionalProperties\n // attributes, so we need to explicitly remove them.\n const jsonSchema = removeAdditionalProperties(\n isInteropZodSchema(schema) ? toJsonSchema(schema) : schema\n );\n const { $schema, ...rest } = jsonSchema;\n\n return rest as GenerativeAIFunctionDeclarationSchema;\n}\n\nexport function jsonSchemaToGeminiParameters(\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n schema: Record<string, any>\n): GenerativeAIFunctionDeclarationSchema {\n // Gemini doesn't accept either the $schema or additionalProperties\n // attributes, so we need to explicitly remove them.\n const jsonSchema = removeAdditionalProperties(\n schema as GenerativeAIJsonSchemaDirty\n );\n const { $schema, ...rest } = jsonSchema;\n\n return rest as GenerativeAIFunctionDeclarationSchema;\n}\n"],"mappings":";;;;;AAuBA,SAAgB,2BAEdA,KACwB;AACxB,KAAI,OAAO,QAAQ,YAAY,QAAQ,MAAM;EAC3C,MAAM,SAAS,EAAE,GAAG,IAAK;AAEzB,MAAI,0BAA0B,QAC5B,OAAO,OAAO;AAEhB,MAAI,aAAa,QACf,OAAO,OAAO;AAEhB,MAAI,YAAY,QACd,OAAO,OAAO;AAGhB,OAAK,MAAM,OAAO,OAChB,KAAI,OAAO,QACT;OAAI,MAAM,QAAQ,OAAO,KAAK,EAC5B,OAAO,OAAO,OAAO,KAAK,IAAI,2BAA2B;YAChD,OAAO,OAAO,SAAS,YAAY,OAAO,SAAS,MAC5D,OAAO,OAAO,2BAA2B,OAAO,KAAK;EACtD;AAIL,SAAO;CACR;AAED,QAAO;AACR;AAED,SAAgB,+BAIdC,QACuC;CAGvC,MAAM,aAAa,gFACE,OAAO,wDAAgB,OAAO,GAAG,OACrD;CACD,MAAM,EAAE,QAAS,GAAG,MAAM,GAAG;AAE7B,QAAO;AACR;AAED,SAAgB,6BAEdC,QACuC;CAGvC,MAAM,aAAa,2BACjB,OACD;CACD,MAAM,EAAE,QAAS,GAAG,MAAM,GAAG;AAE7B,QAAO;AACR"}
|