refactor: Update build_config parameter type in AgentComponent

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-23 18:14:56 -03:00
commit c6ad9a56fa
2 changed files with 6 additions and 3 deletions

View file

@ -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",
]

View file

@ -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,