From 53844a59637970dd40bf9d7b7a569524d68fab5f Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Thu, 25 Jan 2024 21:19:19 -0300
Subject: [PATCH] Delete ZeroShotAgent and CombineDocsChain components

---
 .../components/agents/ZeroShotAgent.py        | 28 -------------------
 .../components/chains/CombineDocsChain.py     | 28 -------------------
 2 files changed, 56 deletions(-)
 delete mode 100644 src/backend/langflow/components/agents/ZeroShotAgent.py
 delete mode 100644 src/backend/langflow/components/chains/CombineDocsChain.py

diff --git a/src/backend/langflow/components/agents/ZeroShotAgent.py b/src/backend/langflow/components/agents/ZeroShotAgent.py
deleted file mode 100644
index 538e693cd..000000000
--- a/src/backend/langflow/components/agents/ZeroShotAgent.py
+++ /dev/null
@@ -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)
diff --git a/src/backend/langflow/components/chains/CombineDocsChain.py b/src/backend/langflow/components/chains/CombineDocsChain.py
deleted file mode 100644
index bb5a32c29..000000000
--- a/src/backend/langflow/components/chains/CombineDocsChain.py
+++ /dev/null
@@ -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()