Skip to content

Commit

Permalink
fix: fix incompatibility with Jina version (#55)
Browse files Browse the repository at this point in the history
  • Loading branch information
JoanFM authored Aug 23, 2023
1 parent 4c5f5eb commit a8e48b2
Show file tree
Hide file tree
Showing 2 changed files with 17 additions and 6 deletions.
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
jina>=3.19.0
jina>=3.20.0
docarray[hnswlib]>=0.34.0
21 changes: 16 additions & 5 deletions vectordb/db/executors/typed_executor.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,6 @@

methods = ['/index', '/update', '/delete', '/search']


class TypedExecutor(Executor, Generic[InputSchema, OutputSchema]):
# the BaseDoc that defines the schema of the store
# for subclasses this is filled automatically
Expand All @@ -30,11 +29,23 @@ def __init__(self, *args, **kwargs):
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])
self._requests[k] = _FunctionWithSchema(fn=self._requests[k].fn,
is_generator=self._requests[k].is_generator,
is_batch_docs=self._requests[k].is_batch_docs,
is_singleton_doc=self._requests[k].is_singleton_doc,
parameters_is_pydantic_model=self._requests[k].parameters_is_pydantic_model,
parameters_model=self._requests[k].parameters_model,
request_schema=DocList[self._input_schema],
response_schema=DocList[self._input_schema])
else:
self._requests[k] = _FunctionWithSchema(self._requests[k].fn, DocList[self._input_schema],
DocList[self._output_schema])
self._requests[k] = _FunctionWithSchema(fn=self._requests[k].fn,
is_generator=self._requests[k].is_generator,
is_batch_docs=self._requests[k].is_batch_docs,
is_singleton_doc=self._requests[k].is_singleton_doc,
parameters_is_pydantic_model=self._requests[k].parameters_is_pydantic_model,
parameters_model=self._requests[k].parameters_model,
request_schema=DocList[self._input_schema],
response_schema=DocList[self._output_schema])

@property
def handle_persistence(self):
Expand Down

0 comments on commit a8e48b2

Please sign in to comment.