Delete ZeroShotAgent and CombineDocsChain components

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-01-25 21:19:19 -03:00
commit 53844a5963
2 changed files with 0 additions and 56 deletions

View file

@ -1,28 +0,0 @@
from typing import List, Optional
from langchain.agents.mrkl.base import ZeroShotAgent
from langchain_core.tools import BaseTool
from langflow import CustomComponent
from langflow.components.chains.LLMChain import LLMChain
class ZeroShotAgentComponent(CustomComponent):
display_name = "ZeroShotAgent"
description = "Construct an agent from an LLM and tools."
def build_config(self):
return {
"llm": {"display_name": "LLM Chain"},
"tools": {"display_name": "Tools"},
"prefix": {"display_name": "Prefix", "multiline": True},
"suffix": {"display_name": "Suffix", "multiline": True},
}
def build(
self,
llm: LLMChain,
tools: Optional[List[BaseTool]] = None,
prefix: str = "Answer the following questions as best you can. You have access to the following tools:",
suffix: str = "Begin!\n\nQuestion: {input}\nThought:{agent_scratchpad}",
) -> ZeroShotAgent:
return ZeroShotAgent(llm_chain=llm, allowed_tools=tools, prefix=prefix, suffix=suffix)

View file

@ -1,28 +0,0 @@
from langflow import CustomComponent
from langflow.field_typing import BaseLanguageModel, Chain
from typing import Union, Callable
from langchain.chains.combine_documents.base import BaseCombineDocumentsChain
class CombineDocsChainComponent(CustomComponent):
display_name = "CombineDocsChain"
description = "Load question answering chain."
def build_config(self):
return {
"llm": {"display_name": "LLM"},
"chain_type": {
"display_name": "Chain Type",
"options": ["stuff", "map_reduce", "map_rerank", "refine"],
},
}
def build(
self,
llm: BaseLanguageModel,
chain_type: str,
) -> Union[Chain, Callable]:
if chain_type not in ["stuff", "map_reduce", "map_rerank", "refine"]:
raise ValueError(f"Invalid chain_type: {chain_type}")
return BaseCombineDocumentsChain()