fix circular selector dependencies (#10222)

llama_index/query_engine/router_query_engine.py

@@ -7,6 +7,7 @@
 from llama_index.callbacks.schema import CBEventType, EventPayload
 from llama_index.core.base_query_engine import BaseQueryEngine
 from llama_index.core.base_retriever import BaseRetriever
+from llama_index.core.base_selector import BaseSelector
 from llama_index.core.response.schema import (
     RESPONSE_TYPE,
     PydanticResponse,
@@ -20,7 +21,6 @@
 from llama_index.prompts.mixin import PromptMixinType
 from llama_index.response_synthesizers import TreeSummarize
 from llama_index.schema import BaseNode, QueryBundle
-from llama_index.selectors.types import BaseSelector
 from llama_index.selectors.utils import get_selector_from_context
 from llama_index.service_context import ServiceContext
 from llama_index.tools.query_engine import QueryEngineTool

llama_index/query_pipeline/components/router.py

@@ -5,6 +5,7 @@
 
 from llama_index.bridge.pydantic import Field, PrivateAttr
 from llama_index.callbacks.base import CallbackManager
+from llama_index.core.base_selector import BaseSelector
 from llama_index.core.query_pipeline.query_component import (
     QUERY_COMPONENT_TYPE,
     ChainableMixin,
@@ -13,7 +14,6 @@
     QueryComponent,
     validate_and_convert_stringable,
 )
-from llama_index.selectors.types import BaseSelector
 from llama_index.utils import print_text
 
 

llama_index/retrievers/router_retriever.py

@@ -6,9 +6,9 @@
 
 from llama_index.callbacks.schema import CBEventType, EventPayload
 from llama_index.core.base_retriever import BaseRetriever
+from llama_index.core.base_selector import BaseSelector
 from llama_index.prompts.mixin import PromptMixinType
 from llama_index.schema import IndexNode, NodeWithScore, QueryBundle
-from llama_index.selectors.types import BaseSelector
 from llama_index.selectors.utils import get_selector_from_context
 from llama_index.service_context import ServiceContext
 from llama_index.tools.retriever_tool import RetrieverTool

llama_index/selectors/__init__.py

@@ -4,11 +4,8 @@
     PydanticMultiSelector,
     PydanticSingleSelector,
 )
-from llama_index.selectors.types import BaseSelector, SelectorResult
 
 __all__ = [
-    "BaseSelector",
-    "SelectorResult",
     "LLMSingleSelector",
     "LLMMultiSelector",
     "EmbeddingSingleSelector",

llama_index/selectors/embedding_selectors.py

@@ -1,15 +1,15 @@
 from typing import Any, Dict, Optional, Sequence
 
+from llama_index.core.base_selector import (
+    BaseSelector,
+    SelectorResult,
+    SingleSelection,
+)
 from llama_index.embeddings.base import BaseEmbedding
 from llama_index.embeddings.utils import resolve_embed_model
 from llama_index.indices.query.embedding_utils import get_top_k_embeddings
 from llama_index.prompts.mixin import PromptDictType
 from llama_index.schema import QueryBundle
-from llama_index.selectors.types import (
-    BaseSelector,
-    SelectorResult,
-    SingleSelection,
-)
 from llama_index.tools.types import ToolMetadata
 
 

llama_index/selectors/llm_selectors.py

@@ -1,5 +1,10 @@
 from typing import Any, Dict, List, Optional, Sequence, cast
 
+from llama_index.core.base_selector import (
+    BaseSelector,
+    SelectorResult,
+    SingleSelection,
+)
 from llama_index.llm_predictor.base import LLMPredictorType
 from llama_index.output_parsers.base import StructuredOutput
 from llama_index.output_parsers.selection import Answer, SelectionOutputParser
@@ -12,7 +17,6 @@
     MultiSelectPrompt,
     SingleSelectPrompt,
 )
-from llama_index.selectors.types import BaseSelector, SelectorResult, SingleSelection
 from llama_index.service_context import ServiceContext
 from llama_index.tools.types import ToolMetadata
 from llama_index.types import BaseOutputParser

llama_index/selectors/pydantic_selectors.py

@@ -1,5 +1,11 @@
 from typing import Any, Dict, Optional, Sequence
 
+from llama_index.core.base_selector import (
+    BaseSelector,
+    MultiSelection,
+    SelectorResult,
+    SingleSelection,
+)
 from llama_index.llms.openai import OpenAI
 from llama_index.program.openai_program import OpenAIPydanticProgram
 from llama_index.prompts.mixin import PromptDictType
@@ -9,12 +15,6 @@
     DEFAULT_MULTI_PYD_SELECT_PROMPT_TMPL,
     DEFAULT_SINGLE_PYD_SELECT_PROMPT_TMPL,
 )
-from llama_index.selectors.types import (
-    BaseSelector,
-    MultiSelection,
-    SelectorResult,
-    SingleSelection,
-)
 from llama_index.tools.types import ToolMetadata
 from llama_index.types import BasePydanticProgram
 

llama_index/selectors/utils.py

@@ -1,11 +1,11 @@
 from typing import Optional
 
+from llama_index.core.base_selector import BaseSelector
 from llama_index.selectors.llm_selectors import LLMMultiSelector, LLMSingleSelector
 from llama_index.selectors.pydantic_selectors import (
     PydanticMultiSelector,
     PydanticSingleSelector,
 )
-from llama_index.selectors.types import BaseSelector
 from llama_index.service_context import ServiceContext
 
 

tests/query_pipeline/test_components.py

@@ -2,6 +2,12 @@
 from typing import Any, List, Sequence
 
 import pytest
+from llama_index.core.base_selector import (
+    BaseSelector,
+    MultiSelection,
+    SelectorResult,
+    SingleSelection,
+)
 from llama_index.core.query_pipeline.components import (
     ArgPackComponent,
     FnComponent,
@@ -15,12 +21,6 @@
 )
 from llama_index.query_pipeline.query import QueryPipeline
 from llama_index.schema import QueryBundle
-from llama_index.selectors.types import (
-    BaseSelector,
-    MultiSelection,
-    SelectorResult,
-    SingleSelection,
-)
 from llama_index.tools.types import ToolMetadata