[ENH] Export schema and search types from chromadb.api

Author: jairad26

chromadb/api/__init__.py

@@ -1,3 +1,40 @@
+from chromadb.api.types import *  # noqa: F401, F403
+from chromadb.execution.expression import (  # noqa: F401, F403
+    Search,
+    Key,
+    K,
+    SearchWhere,
+    And,
+    Or,
+    Eq,
+    Ne,
+    Gt,
+    Gte,
+    Lt,
+    Lte,
+    In,
+    Nin,
+    Regex,
+    NotRegex,
+    Contains,
+    NotContains,
+    Limit,
+    Select,
+    Rank,
+    Abs,
+    Div,
+    Exp,
+    Log,
+    Max,
+    Min,
+    Mul,
+    Knn,
+    Rrf,
+    Sub,
+    Sum,
+    Val,
+)
+
 from abc import ABC, abstractmethod
 from typing import Sequence, Optional, List, Dict, Any
 from uuid import UUID
@@ -30,8 +67,8 @@
     SearchResult,
     DefaultEmbeddingFunction,
 )
+
 from chromadb.auth import UserIdentity
-from chromadb.execution.expression.plan import Search
 from chromadb.config import Component, Settings
 from chromadb.types import Database, Tenant, Collection as CollectionModel
 from chromadb.api.models.Collection import Collection

chromadb/execution/expression/__init__.py

@@ -45,13 +45,16 @@
     Search,
 )
 
+SearchWhere = Where
+
 __all__ = [
     # Main search class
     "Search",
     # Field proxy
     "Key",
     "K",
     # Where expressions
+    "SearchWhere",
     "Where",
     "And",
     "Or",