refactor: Update build_config parameter type in AgentComponent
This commit is contained in:
parent
7642969f32
commit
c6ad9a56fa
2 changed files with 6 additions and 3 deletions
|
|
@ -3,11 +3,11 @@ from typing import Any
|
||||||
from .constants import (
|
from .constants import (
|
||||||
AgentExecutor,
|
AgentExecutor,
|
||||||
BaseChatMemory,
|
BaseChatMemory,
|
||||||
|
BaseChatModel,
|
||||||
BaseLanguageModel,
|
BaseLanguageModel,
|
||||||
BaseLLM,
|
BaseLLM,
|
||||||
BaseLoader,
|
BaseLoader,
|
||||||
BaseMemory,
|
BaseMemory,
|
||||||
BaseChatModel,
|
|
||||||
BaseOutputParser,
|
BaseOutputParser,
|
||||||
BasePromptTemplate,
|
BasePromptTemplate,
|
||||||
BaseRetriever,
|
BaseRetriever,
|
||||||
|
|
@ -21,10 +21,11 @@ from .constants import (
|
||||||
NestedDict,
|
NestedDict,
|
||||||
Object,
|
Object,
|
||||||
PromptTemplate,
|
PromptTemplate,
|
||||||
|
Retriever,
|
||||||
|
Text,
|
||||||
TextSplitter,
|
TextSplitter,
|
||||||
Tool,
|
Tool,
|
||||||
VectorStore,
|
VectorStore,
|
||||||
Retriever,
|
|
||||||
)
|
)
|
||||||
from .range_spec import RangeSpec
|
from .range_spec import RangeSpec
|
||||||
|
|
||||||
|
|
@ -82,4 +83,5 @@ __all__ = [
|
||||||
"VectorStore",
|
"VectorStore",
|
||||||
"BaseChatModel",
|
"BaseChatModel",
|
||||||
"Retriever",
|
"Retriever",
|
||||||
|
"Text",
|
||||||
]
|
]
|
||||||
|
|
|
||||||
|
|
@ -1,4 +1,4 @@
|
||||||
from typing import Callable, Dict, TypeAlias, TypeVar, Union
|
from typing import Callable, Dict, Text, TypeAlias, TypeVar, Union
|
||||||
|
|
||||||
from langchain.agents.agent import AgentExecutor
|
from langchain.agents.agent import AgentExecutor
|
||||||
from langchain.chains.base import Chain
|
from langchain.chains.base import Chain
|
||||||
|
|
@ -65,6 +65,7 @@ CUSTOM_COMPONENT_SUPPORTED_TYPES = {
|
||||||
"NestedDict": NestedDict,
|
"NestedDict": NestedDict,
|
||||||
"Data": Data,
|
"Data": Data,
|
||||||
"Message": Message,
|
"Message": Message,
|
||||||
|
"Text": Text,
|
||||||
"Object": Object,
|
"Object": Object,
|
||||||
"Callable": Callable,
|
"Callable": Callable,
|
||||||
"LanguageModel": LanguageModel,
|
"LanguageModel": LanguageModel,
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue