[CLN] Cleanup frontend after query pushdown

chroma-core · Dec 17, 2024 · c6258fd · c6258fd
1 parent beca297
commit c6258fd
Show file tree

Hide file tree

Showing 5 changed files with 109 additions and 894 deletions.
diff --git a/chromadb/execution/executor/distributed.py b/chromadb/execution/executor/distributed.py
@@ -57,11 +57,6 @@ def count(self, plan: CountPlan) -> int:
         try:
             count_result = executor.Count(convert.to_proto_count_plan(plan))
         except grpc.RpcError as rpc_error:
-            if (
-                rpc_error.code() == grpc.StatusCode.INTERNAL
-                and "version mismatch" in rpc_error.details()
-            ):
-                raise VersionMismatchError()
             raise rpc_error
         return convert.from_proto_count_result(count_result)
 
@@ -71,11 +66,6 @@ def get(self, plan: GetPlan) -> GetResult:
         try:
             get_result = executor.Get(convert.to_proto_get_plan(plan))
         except grpc.RpcError as rpc_error:
-            if (
-                rpc_error.code() == grpc.StatusCode.INTERNAL
-                and "version mismatch" in rpc_error.details()
-            ):
-                raise VersionMismatchError()
             raise rpc_error
         records = convert.from_proto_get_result(get_result)
 
@@ -118,11 +108,6 @@ def knn(self, plan: KNNPlan) -> QueryResult:
         try:
             knn_result = executor.KNN(convert.to_proto_knn_plan(plan))
         except grpc.RpcError as rpc_error:
-            if (
-                rpc_error.code() == grpc.StatusCode.INTERNAL
-                and "version mismatch" in rpc_error.details()
-            ):
-                raise VersionMismatchError()
             raise rpc_error
         results = convert.from_proto_knn_batch_result(knn_result)