Enable keying result by column name

Enable dict cursor. Allows keying the result by column name instead of
just by column number. Helps ergonomics.

Author: cevian

nbs/00_vector.ipynb

@@ -1385,6 +1385,9 @@
     "assert len(rec) == 2\n",
     "\n",
     "assert isinstance(rec[0][SEARCH_RESULT_METADATA_IDX], dict)\n",
+    "assert isinstance(rec[0][\"metadata\"], dict)\n",
+    "assert rec[0][\"contents\"] == \"the brown fox\"\n",
+    "\n",
     "\n",
     "rec = await vec.search([1.0, 2.0], limit=4, predicates=Predicates((\"key\", \"val2\")))\n",
     "assert len(rec) == 1\n",
@@ -1442,7 +1445,7 @@
     "\n",
     "rec = await vec.search([1.0, 2.0], limit=4, filter=[{\"key_1\": \"val_1\"}, {\"key2\": \"val2\"}])\n",
     "assert len(rec) == 2\n",
-    "await vec.delete_by_ids([rec[0][SEARCH_RESULT_ID_IDX]])\n",
+    "await vec.delete_by_ids([rec[0][\"id\"]])\n",
     "rec = await vec.search([1.0, 2.0], limit=4, filter=[{\"key_1\": \"val_1\"}, {\"key2\": \"val2\"}])\n",
     "assert len(rec) == 1\n",
     "await vec.delete_by_metadata([{\"key_1\": \"val_1\"}, {\"key2\": \"val2\"}])\n",
@@ -1658,7 +1661,7 @@
     "                self.max_db_connections = self.default_max_db_connections()\n",
     "\n",
     "            self.pool = psycopg2.pool.SimpleConnectionPool(\n",
-    "                1, self.max_db_connections, dsn=self.service_url)\n",
+    "                1, self.max_db_connections, dsn=self.service_url, cursor_factory=psycopg2.extras.DictCursor)\n",
     "\n",
     "        connection = self.pool.getconn()\n",
     "        pgvector.psycopg2.register_vector(connection)\n",
@@ -2181,10 +2184,14 @@
     "\n",
     "rec = vec.search([1.0, 2.0], filter={\"key_1\": \"val_1\", \"key_2\": \"val_2\"})\n",
     "assert rec[0][SEARCH_RESULT_CONTENTS_IDX] == 'the brown fox'\n",
+    "assert rec[0][\"contents\"] == 'the brown fox'\n",
     "assert rec[0][SEARCH_RESULT_METADATA_IDX] == {\n",
     "    'key_1': 'val_1', 'key_2': 'val_2'}\n",
+    "assert rec[0][\"metadata\"] == {\n",
+    "    'key_1': 'val_1', 'key_2': 'val_2'}\n",
     "assert isinstance(rec[0][SEARCH_RESULT_METADATA_IDX], dict)\n",
     "assert rec[0][SEARCH_RESULT_DISTANCE_IDX] == 0.0009438353921149556\n",
+    "assert rec[0][\"distance\"] == 0.0009438353921149556\n",
     "\n",
     "rec = vec.search([1.0, 2.0], limit=4, predicates=Predicates((\"key\", \"val2\")))\n",
     "assert len(rec) == 1\n",

timescale_vector/client.py

@@ -1049,7 +1049,7 @@ def connect(self):
                 self.max_db_connections = self.default_max_db_connections()
 
             self.pool = psycopg2.pool.SimpleConnectionPool(
-                1, self.max_db_connections, dsn=self.service_url)
+                1, self.max_db_connections, dsn=self.service_url, cursor_factory=psycopg2.extras.DictCursor)
 
         connection = self.pool.getconn()
         pgvector.psycopg2.register_vector(connection)