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 (
|
||||
AgentExecutor,
|
||||
BaseChatMemory,
|
||||
BaseChatModel,
|
||||
BaseLanguageModel,
|
||||
BaseLLM,
|
||||
BaseLoader,
|
||||
BaseMemory,
|
||||
BaseChatModel,
|
||||
BaseOutputParser,
|
||||
BasePromptTemplate,
|
||||
BaseRetriever,
|
||||
|
|
@ -21,10 +21,11 @@ from .constants import (
|
|||
NestedDict,
|
||||
Object,
|
||||
PromptTemplate,
|
||||
Retriever,
|
||||
Text,
|
||||
TextSplitter,
|
||||
Tool,
|
||||
VectorStore,
|
||||
Retriever,
|
||||
)
|
||||
from .range_spec import RangeSpec
|
||||
|
||||
|
|
@ -82,4 +83,5 @@ __all__ = [
|
|||
"VectorStore",
|
||||
"BaseChatModel",
|
||||
"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.chains.base import Chain
|
||||
|
|
@ -65,6 +65,7 @@ CUSTOM_COMPONENT_SUPPORTED_TYPES = {
|
|||
"NestedDict": NestedDict,
|
||||
"Data": Data,
|
||||
"Message": Message,
|
||||
"Text": Text,
|
||||
"Object": Object,
|
||||
"Callable": Callable,
|
||||
"LanguageModel": LanguageModel,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue