# Copyright 2025 Redis, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""Redis tools for ADK."""

from adk_redis.tools.memory import CreateMemoryTool
from adk_redis.tools.memory import DeleteMemoryTool
from adk_redis.tools.memory import GetMemoryTool
from adk_redis.tools.memory import MemoryPromptTool
from adk_redis.tools.memory import MemoryToolConfig
from adk_redis.tools.memory import SearchMemoryTool
from adk_redis.tools.memory import UpdateMemoryTool
from adk_redis.tools.mcp_memory import ALL_MCP_TOOLS
from adk_redis.tools.mcp_memory import create_memory_mcp_toolset
from adk_redis.tools.search import BaseRedisSearchTool
from adk_redis.tools.search import RedisAggregatedHybridQueryConfig
from adk_redis.tools.search import RedisHybridQueryConfig
from adk_redis.tools.search import RedisHybridSearchTool
from adk_redis.tools.search import RedisRangeQueryConfig
from adk_redis.tools.search import RedisRangeSearchTool
from adk_redis.tools.search import RedisTextQueryConfig
from adk_redis.tools.search import RedisTextSearchTool
from adk_redis.tools.search import RedisVectorQueryConfig
from adk_redis.tools.search import RedisVectorSearchTool
from adk_redis.tools.search import VectorizedSearchTool

__all__ = [
    # Base classes
    "BaseRedisSearchTool",
    "VectorizedSearchTool",
    # Search tools
    "RedisVectorSearchTool",
    "RedisHybridSearchTool",
    "RedisRangeSearchTool",
    "RedisTextSearchTool",
    # Memory tools
    "MemoryPromptTool",
    "SearchMemoryTool",
    "CreateMemoryTool",
    "DeleteMemoryTool",
    "UpdateMemoryTool",
    "GetMemoryTool",
    # Config classes
    "RedisVectorQueryConfig",
    "RedisHybridQueryConfig",
    "RedisAggregatedHybridQueryConfig",
    "RedisRangeQueryConfig",
    "RedisTextQueryConfig",
    "MemoryToolConfig",
    # MCP tools
    "create_memory_mcp_toolset",
    "ALL_MCP_TOOLS",
]
