Skip to content

Commit

Permalink
v45 vector search
Browse files Browse the repository at this point in the history
Signed-off-by: Praneeth Bedapudi <[email protected]>
  • Loading branch information
bedapudi6788 committed Jan 31, 2024
1 parent 604db6f commit 8961b0a
Show file tree
Hide file tree
Showing 3 changed files with 12 additions and 5 deletions.
13 changes: 10 additions & 3 deletions liteindex/defined_index.py
Original file line number Diff line number Diff line change
Expand Up @@ -712,7 +712,14 @@ def math(self, key, op, query={}):
return self.__connection.execute(sql_query, sql_params).fetchone()[0]

def create_trigger(
self, trigger_name, for_key, function_to_trigger, before_delete=False, after_set=False, before_set=False, for_keys=None
self,
trigger_name,
for_key,
function_to_trigger,
before_delete=False,
after_set=False,
before_set=False,
for_keys=None,
):
if after_set:
operation = "INSERT"
Expand All @@ -728,9 +735,9 @@ def create_trigger(

if for_keys is None:
for_keys = [for_key]

for_keys = [self.__co]

function_name = f"{trigger_name}_function"

with self.__connection as conn:
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
EMAIL = "[email protected]"
AUTHOR = "BEDAPUDI PRANEETH"
REQUIRES_PYTHON = ">=3.6.0"
VERSION = "0.0.2.dev44"
VERSION = "0.0.2.dev45"

# What packages are required for this module to be executed?
REQUIRED = []
Expand Down
2 changes: 1 addition & 1 deletion tests/test_vector_search.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@
{},
sort_by="embedding",
reversed_sort=True,
query_vector=query_vector,
sort_by_embedding=query_vector,
select_keys=[],
return_metadata=True,
n=1
Expand Down

0 comments on commit 8961b0a

Please sign in to comment.