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

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