Refactor SQL agent and related nodes

- Rename `VectorStoreAgent` to `Vector store agent` in `custom.py`
- Replace "Construct a sql agent from an LLM and tools." with "Construct an SQL agent from an LLM and tools." in `custom.py`
- Update descriptions of `SQLAgentNode`, `TimeTravelGuideChainNode`, `CombineDocsChainNode`, and `ToolNode` in related files
- Increase width of `ExtraSidebarComponent` container in `index.tsx` from 52 to 56
This commit is contained in:
Rodrigo Nader 2023-06-14 23:54:39 -03:00
commit fd59bd77b3
5 changed files with 7 additions and 7 deletions

View file

@ -120,7 +120,7 @@ class CSVAgent(CustomAgentExecutor):
class VectorStoreAgent(CustomAgentExecutor):
"""Vector Store agent"""
"""Vector store agent"""
@staticmethod
def function_name():
@ -175,7 +175,7 @@ class SQLAgent(CustomAgentExecutor):
def from_toolkit_and_llm(
cls, llm: BaseLanguageModel, database_uri: str, **kwargs: Any
):
"""Construct a sql agent from an LLM and tools."""
"""Construct an SQL agent from an LLM and tools."""
db = SQLDatabase.from_uri(database_uri)
toolkit = SQLDatabaseToolkit(db=db, llm=llm)

View file

@ -37,7 +37,7 @@ class SQLAgentNode(FrontendNode):
),
],
)
description: str = """Construct a sql agent from an LLM and tools."""
description: str = """Construct an SQL agent from an LLM and tools."""
base_classes: list[str] = ["AgentExecutor"]
def to_dict(self):

View file

@ -126,7 +126,7 @@ class TimeTravelGuideChainNode(FrontendNode):
),
],
)
description: str = "Time travel guide chain to be used in the flow."
description: str = "Time travel guide chain."
base_classes: list[str] = [
"LLMChain",
"BaseCustomChain",
@ -197,7 +197,7 @@ class CombineDocsChainNode(FrontendNode):
),
],
)
description: str = """Construct a zero shot agent from an LLM and tools."""
description: str = """Construct a chain from combined documents."""
base_classes: list[str] = ["BaseCombineDocumentsChain", "function"]
def to_dict(self):

View file

@ -52,7 +52,7 @@ class ToolNode(FrontendNode):
),
],
)
description: str = "Tool to be used in the flow."
description: str = "Converts a chain, agent or function into a tool."
base_classes: list[str] = ["Tool"]
def to_dict(self):

View file

@ -72,7 +72,7 @@ export default function ExtraSidebar() {
}
return (
<div className="w-52 flex flex-col overflow-hidden scrollbar-hide h-full border-r">
<div className="w-56 flex flex-col overflow-hidden scrollbar-hide h-full border-r">
<div className="mt-2 mb-2 w-full flex gap-2 justify-between px-2 items-center">
<ShadTooltip delayDuration={1000} content="Import" side="top">
<button