Remove unecessary tools (#233)

Author: vhaldemar

README.md

@@ -36,7 +36,7 @@ Also there is some cross-domain functionality for features above:
 - [**Batch processing**](https://aistudio.yandex.ru/docs/ai-studio/concepts/generation/batch-processing)
   - Run long‑running batch tasks with automatic polling.
 - **Tools**
-  - Built‑in tools such as Generative Search Tool, [Function Tool](https://aistudio.yandex.ru/docs/ai-studio/concepts/generation/function-call) and Search Index Tool that can be used in Completions and Chat, providing a unified way to extend functionality across the SDK.
+  - [Function Tool](https://aistudio.yandex.ru/docs/ai-studio/concepts/generation/function-call) for extending models with custom callable functions in Completions and Chat.
 - [**Tuning**](https://aistudio.yandex.ru/docs/ai-studio/concepts/tuning/)
   - Fine‑tune models with configurable optimizers, schedulers and other hyper‑parameters.
 - [**Datasets**](https://aistudio.yandex.ru/docs/ai-studio/concepts/resources/dataset)

docs/types/tools.rst

@@ -1,29 +1,8 @@
 Tools
 =====
 
-Search index tool
------------------
-
-.. autoclass:: yandex_ai_studio_sdk._tools.search_index.tool.SearchIndexTool
-   :undoc-members:
-
-.. autoclass:: yandex_ai_studio_sdk._tools.search_index.rephraser.function.RephraserFunction
-   :undoc-members:
-
-.. autoclass:: yandex_ai_studio_sdk._tools.search_index.rephraser.model.Rephraser
-   :undoc-members:
-
-.. autoclass:: yandex_ai_studio_sdk._tools.search_index.call_strategy.CallStrategy
-   :undoc-members:
-
-
 Function tool
 -------------
 
 .. autoclass:: yandex_ai_studio_sdk._tools.tool.FunctionTool
    :undoc-members:
-
-Generative search tool
-----------------------
-
-.. autoclass:: yandex_ai_studio_sdk._tools.generative_search.GenerativeSearchTool

src/yandex_ai_studio_sdk/_search_api/generative/function.py

@@ -30,10 +30,6 @@ def __call__(
         """
         Creates generative search object which provides methods for invoking generative search.
 
-        Not to be confused with ``sdk.tools.generative_search``.
-        Tools domain is for creating tools for using in LLMs/Assistants and search_api domain
-        is for using Search API directly.
-
         To learn more about parameters and their formats and possible values,
         refer to
         `generative search documentation <https://aistudio.yandex.ru/docs/search-api/concepts/generative-response#body>`_

src/yandex_ai_studio_sdk/_search_api/generative/generative.py

@@ -8,7 +8,6 @@
 from yandex.cloud.searchapi.v2.gen_search_service_pb2_grpc import GenSearchServiceStub
 
 from yandex_ai_studio_sdk._logging import get_logger
-from yandex_ai_studio_sdk._tools.generative_search import GenerativeSearchTool
 from yandex_ai_studio_sdk._types.misc import UNDEFINED, UndefinedOr
 from yandex_ai_studio_sdk._types.model import ModelSyncMixin
 from yandex_ai_studio_sdk._types.string import SmartStringSequence
@@ -152,28 +151,6 @@ async def _run(self, request: MessageInputType, *, timeout: float = 60) -> Gener
 
         raise RuntimeError("call returned less then one result")
 
-    def as_tool(self, description: str) -> GenerativeSearchTool:
-        """
-        Converts generative search instance to :py:class:`~.GenerativeSearchTool` object
-        which is eligible to use as tools in LLMs/Assistants.
-
-        :param description: description of tool instance which also instructs model when to call it.
-        """
-        c = self.config
-        if c.fix_misspell is not None:
-            raise ValueError("Option fix_misspell doest not supported in GenSearchTool")
-
-        return GenerativeSearchTool(
-            description=description,
-            enable_nrfm_docs=c.enable_nrfm_docs,
-            search_filters=c.search_filters,
-            fix_misspell=None,
-            host=c.host,
-            site=c.site,
-            url=c.url,
-        )
-
-
 @doc_from(BaseGenerativeSearch)
 class AsyncGenerativeSearch(BaseGenerativeSearch):
     @doc_from(BaseGenerativeSearch._run)

src/yandex_ai_studio_sdk/_tools/domain.py

@@ -4,19 +4,10 @@
 from functools import cached_property
 from typing import Generic
 
-from yandex_ai_studio_sdk._search_api.generative.config import SmartFilterSequence
-from yandex_ai_studio_sdk._search_indexes.search_index import BaseSearchIndex
 from yandex_ai_studio_sdk._types.domain import BaseDomain
-from yandex_ai_studio_sdk._types.misc import UNDEFINED, UndefinedOr, get_defined_value, is_defined
-from yandex_ai_studio_sdk._types.string import SmartStringSequence
-from yandex_ai_studio_sdk._utils.coerce import ResourceType, coerce_resource_ids
 from yandex_ai_studio_sdk._utils.doc import doc_from
 
 from .function import AsyncFunctionTools, FunctionTools, FunctionToolsTypeT
-from .generative_search import GenerativeSearchTool
-from .search_index.call_strategy import CallStrategy, CallStrategyInputType
-from .search_index.rephraser.function import RephraserFunction, RephraserInputType
-from .search_index.tool import SearchIndexTool
 
 
 class BaseTools(BaseDomain, Generic[FunctionToolsTypeT]):
@@ -34,8 +25,6 @@ class BaseTools(BaseDomain, Generic[FunctionToolsTypeT]):
 
     Tools are particularly useful in:
 
-    - **AI Assistants**: Extending conversational agents with external capabilities like web search,
-      database queries, or API calls
     - **Completions**: Enabling language models to invoke functions during text generation for
       dynamic content creation and problem-solving
 
@@ -54,121 +43,6 @@ def function(self) -> FunctionToolsTypeT:
             sdk=self._sdk
         )
 
-    @cached_property
-    def rephraser(self) -> RephraserFunction:
-        """
-        Get the rephraser for creating query transformation models.
-
-        The rephraser provides access to specialized language models designed to intelligently
-        rewrite and enhance user search queries by incorporating conversational context. This is
-        particularly useful in multi-turn conversations where the latest user message may lack context
-        from previous exchanges.
-
-        The rephraser works by:
-        - Analyzing the conversation history and current user query
-        - Reformulating the query to be more specific and contextually complete
-        - Improving search relevance by expanding abbreviated or ambiguous terms
-        - Maintaining semantic intent while adding necessary context
-
-        The rephraser returns a factory that can create Rephraser model instances with different
-        configurations, supporting various model types including the default 'rephraser' model or
-        custom rephrasing models.
-        """
-        return RephraserFunction(
-            name='tools.rehraser',
-            sdk=self._sdk,
-            parent_resource=self
-        )
-
-    def search_index(
-        self,
-        indexes: ResourceType[BaseSearchIndex],
-        *,
-        max_num_results: UndefinedOr[int] = UNDEFINED,
-        rephraser: UndefinedOr[RephraserInputType] = UNDEFINED,
-        call_strategy: UndefinedOr[CallStrategyInputType] = UNDEFINED,
-    ) -> SearchIndexTool:
-        """
-        Creates SearchIndexTool (not to be confused with :py:class:`~.SearchIndex`/:py:class:`~.AsyncSearchIndex`).
-
-        :param indexes: parameter takes :py:class:`~.BaseSearchIndex`, string with search index id,
-            or a list of this values in any combination.
-        :param max_num_results: the maximum number of results to return from the search.
-            Fewer results may be returned if necessary to fit within the prompt's token limit.
-            This ensures that the combined prompt and search results do not exceed the token constraints.
-        :param rephraser: setting for rephrasing user queries; refer to :py:class:`~.Rephraser` documentation
-            for details.
-        """
-
-        index_ids = coerce_resource_ids(indexes, BaseSearchIndex)
-        max_num_results_ = get_defined_value(max_num_results, None)
-
-        rephraser_ = None
-        if is_defined(rephraser):
-            # this is coercing any RephraserInputType to Rephraser
-            rephraser_ = self.rephraser(rephraser)  # type: ignore[arg-type]
-
-        call_strategy_ = None
-        if is_defined(call_strategy):
-            call_strategy_ = CallStrategy._coerce(call_strategy)
-
-        return SearchIndexTool(
-            search_index_ids=tuple(index_ids),
-            max_num_results=max_num_results_,
-            rephraser=rephraser_,
-            call_strategy=call_strategy_,
-        )
-
-    def generative_search(
-        self,
-        *,
-        description: str,
-        site: UndefinedOr[SmartStringSequence] = UNDEFINED,
-        host: UndefinedOr[SmartStringSequence] = UNDEFINED,
-        url: UndefinedOr[SmartStringSequence] = UNDEFINED,
-        enable_nrfm_docs: UndefinedOr[bool] = UNDEFINED,
-        search_filters: UndefinedOr[SmartFilterSequence] = UNDEFINED,
-    ) -> GenerativeSearchTool:
-        """Creates GenerativeSearch tool which provide access to
-        generative search by `Yandex Search API <https://aistudio.yandex.ru/docs/search-api/concepts/index>`_ for LLMs.
-
-        Not to be confused with ``sdk.search_api.generative``.
-        Tools domain is for creating tools for using in LLMs/Assistants and search_api domain
-        is for using Yandex Search API directly.
-
-        To learn more about parameters, their formats, and possible values,
-        refer to
-        `generative search documentation <https://aistudio.yandex.ru/docs/search-api/concepts/generative-response#body>`_
-
-        NB: All of the ``site``, ``host``, ``url`` parameters are mutually exclusive.
-
-        :param site: parameter for limiting search to specific location or list of sites.
-        :param host: parameter for limiting search to specific location or list of hosts.
-        :param url: parameter for limiting search to specific location or list of URLs.
-        :param enable_nrfm_docs: tells to backend to include or not to include pages,
-            which are not available via direct clicks from given sites/hosts/urls
-            to search result.
-        :param search_filters: allows to limit search results with additional filters.
-
-            >>> date_filter = {'date': '<20250101'}
-            >>> format_filter = {'format': 'doc'}
-            >>> lang_filter = {'lang': 'ru'}
-            >>> tool = sdk.tools.generative_search(
-            ...     search_filters=[date_filter, format_filter, lang_filter],
-            ...     description="description when model should call a tool"
-            ... )
-
-        """
-
-        search_api = self._sdk.search_api.generative(
-            site=site,
-            host=host,
-            url=url,
-            enable_nrfm_docs=enable_nrfm_docs,
-            search_filters=search_filters
-        )
-        return search_api.as_tool(description=description)
-
 
 @doc_from(BaseTools)
 class AsyncTools(BaseTools[AsyncFunctionTools]):