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): class VectorStoreAgent(CustomAgentExecutor):
"""Vector Store agent""" """Vector store agent"""
@staticmethod @staticmethod
def function_name(): def function_name():
@ -175,7 +175,7 @@ class SQLAgent(CustomAgentExecutor):
def from_toolkit_and_llm( def from_toolkit_and_llm(
cls, llm: BaseLanguageModel, database_uri: str, **kwargs: Any 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) db = SQLDatabase.from_uri(database_uri)
toolkit = SQLDatabaseToolkit(db=db, llm=llm) 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"] base_classes: list[str] = ["AgentExecutor"]
def to_dict(self): 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] = [ base_classes: list[str] = [
"LLMChain", "LLMChain",
"BaseCustomChain", "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"] base_classes: list[str] = ["BaseCombineDocumentsChain", "function"]
def to_dict(self): 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"] base_classes: list[str] = ["Tool"]
def to_dict(self): def to_dict(self):

View file

@ -72,7 +72,7 @@ export default function ExtraSidebar() {
} }
return ( 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"> <div className="mt-2 mb-2 w-full flex gap-2 justify-between px-2 items-center">
<ShadTooltip delayDuration={1000} content="Import" side="top"> <ShadTooltip delayDuration={1000} content="Import" side="top">
<button <button