How to add an ExtractionChain?

[scorego]

I want to use extraction chain with Flowise. Extraction chain is implemented on LLMChain, with customized output_parser and llm_kwargs, and I cannot find how to configure these parameters in Flowise UI. Is there any plan to support customizing them?

SEE https://github.com/langchain-ai/langchain/blob/master/libs/langchain/langchain/chains/openai_functions/extraction.py#L46

def create_extraction_chain(
    schema: dict, llm: BaseLanguageModel, verbose: bool = False
) -> Chain:

    function = _get_extraction_function(schema)
    prompt = ChatPromptTemplate.from_template(_EXTRACTION_TEMPLATE)
    output_parser = JsonKeyOutputFunctionsParser(key_name="info")
    llm_kwargs = get_llm_kwargs(function)
    chain = LLMChain(
        llm=llm,
        prompt=prompt,
        llm_kwargs=llm_kwargs,
        output_parser=output_parser,
        verbose=verbose,
    )
    return chain

PS, I think UI should provide a way to customize all the class field of LLMChain.

[chungyau97]

Hi @scorego,

Would you like to take up this part to submit a PR?
Here's the source code location