diff --git a/src/crewai_tools/aws/__init__.py b/src/crewai_tools/aws/__init__.py index ea4626a32..dd01fd8fe 100644 --- a/src/crewai_tools/aws/__init__.py +++ b/src/crewai_tools/aws/__init__.py @@ -1,3 +1,9 @@ from .s3 import S3ReaderTool, S3WriterTool +from .bedrock import BedrockKBRetrieverTool, BedrockInvokeAgentTool -__all__ = ['S3ReaderTool', 'S3WriterTool'] \ No newline at end of file +__all__ = [ + 'S3ReaderTool', + 'S3WriterTool', + 'BedrockKBRetrieverTool', + 'BedrockInvokeAgentTool' +] \ No newline at end of file diff --git a/src/crewai_tools/aws/bedrock/__init__.py b/src/crewai_tools/aws/bedrock/__init__.py new file mode 100644 index 000000000..ded472062 --- /dev/null +++ b/src/crewai_tools/aws/bedrock/__init__.py @@ -0,0 +1,4 @@ +from .knowledge_base.retriever_tool import BedrockKBRetrieverTool +from .agents.invoke_agent_tool import BedrockInvokeAgentTool + +__all__ = ["BedrockKBRetrieverTool", "BedrockInvokeAgentTool"] diff --git a/src/crewai_tools/aws/bedrock/agents/__init__.py b/src/crewai_tools/aws/bedrock/agents/__init__.py new file mode 100644 index 000000000..b1f799872 --- /dev/null +++ b/src/crewai_tools/aws/bedrock/agents/__init__.py @@ -0,0 +1,3 @@ +from .invoke_agent_tool import BedrockInvokeAgentTool + +__all__ = ["BedrockInvokeAgentTool"] diff --git a/src/crewai_tools/aws/bedrock/knowledge_base/__init__.py b/src/crewai_tools/aws/bedrock/knowledge_base/__init__.py new file mode 100644 index 000000000..013d94cf3 --- /dev/null +++ b/src/crewai_tools/aws/bedrock/knowledge_base/__init__.py @@ -0,0 +1,3 @@ +from .retriever_tool import BedrockKBRetrieverTool + +__all__ = ["BedrockKBRetrieverTool"]