23 lines
1.6 KiB
TypeScript
23 lines
1.6 KiB
TypeScript
|
import { z } from "zod";
|
||
|
|
import { BaseChatModel } from "@langchain/core/language_models/chat_models";
|
||
|
|
import { BaseFunctionCallOptions } from "@langchain/core/language_models/base";
|
||
|
|
import { FunctionParameters } from "../../output_parsers/openai_functions.js";
|
||
|
|
import { LLMChain } from "../llm_chain.js";
|
||
|
|
/**
|
||
|
|
* Function that creates an extraction chain using the provided JSON schema.
|
||
|
|
* It sets up the necessary components, such as the prompt, output parser, and tags.
|
||
|
|
* @param schema JSON schema of the function parameters.
|
||
|
|
* @param llm Must be a ChatOpenAI or AnthropicFunctions model that supports function calling.
|
||
|
|
* @returns A LLMChain instance configured to return data matching the schema.
|
||
|
|
*/
|
||
|
|
export declare function createExtractionChain(schema: FunctionParameters, llm: BaseChatModel<BaseFunctionCallOptions>): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, import("@langchain/core/messages").BaseMessageChunk>>;
|
||
|
|
/**
|
||
|
|
* Function that creates an extraction chain from a Zod schema. It
|
||
|
|
* converts the Zod schema to a JSON schema using zod-to-json-schema
|
||
|
|
* before creating the extraction chain.
|
||
|
|
* @param schema The Zod schema which extracted data should match
|
||
|
|
* @param llm Must be a ChatOpenAI or AnthropicFunctions model that supports function calling.
|
||
|
|
* @returns A LLMChain instance configured to return data matching the schema.
|
||
|
|
*/
|
||
|
|
export declare function createExtractionChainFromZod(schema: z.ZodObject<any, any, any, any>, llm: BaseChatModel<BaseFunctionCallOptions>): LLMChain<object, BaseChatModel<BaseFunctionCallOptions, import("@langchain/core/messages").BaseMessageChunk>>;
|