dnrops
/
vectordb
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: do not override dry run endpoints schemas

This commit is contained in:
Joan Fontanals Martinez 2023-06-29 15:42:35 +02:00
parent 0c0627eed9
commit 1f50dd9d6f
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vectordb/db/executors/typed_executor.py
View File

@ -1,5 +1,6 @@
from jina import Executor
from jina.serve.executors import _FunctionWithSchema
from jina.serve.executors import __dry_run_endpoint__
from typing import TypeVar, Generic, Type, Optional, TYPE_CHECKING
from vectordb.utils.create_doc_type import create_output_doc_type
@ -27,6 +28,7 @@ class TypedExecutor(Executor, Generic[InputSchema, OutputSchema]):
from docarray import DocList
self._num_replicas = getattr(self.runtime_args, 'replicas', 1)
for k, v in self._requests.items():
if k != __dry_run_endpoint__:
if k != '/search':
self._requests[k] = _FunctionWithSchema(self._requests[k].fn, DocList[self._input_schema],
DocList[self._input_schema])