make jsonb decode to a dict in the async client

Author: cevian

nbs/00_vector.ipynb

@@ -354,6 +354,13 @@
     "        if self.pool == None:\n",
     "            async def init(conn):\n",
     "                await register_vector(conn)\n",
+    "                #decode to a dict, but accept a string as input in upsert\n",
+    "                await conn.set_type_codec(\n",
+    "                    'jsonb',\n",
+    "                    encoder=str,\n",
+    "                    decoder=json.loads,\n",
+    "                    schema='pg_catalog')\n",
+    "\n",
     "            self.pool = await asyncpg.create_pool(dsn=self.service_url, init=init)\n",
     "        return self.pool.acquire()\n",
     "\n",
@@ -694,6 +701,8 @@
     "rec = await vec.search([1.0, 2.0], k=4, filter=[{\"key_1\":\"val_1\"}, {\"key2\":\"val2\"}, {\"no such key\": \"no such val\"}])\n",
     "assert len(rec) == 2\n",
     "\n",
+    "assert isinstance(rec[0][SEARCH_RESULT_METADATA_IDX], dict)\n",
+    "\n",
     "try:\n",
     "    # can't upsert using both keys and dictionaries\n",
     "    await vec.upsert([ \\\n",
@@ -1212,6 +1221,7 @@
     "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][SEARCH_RESULT_METADATA_IDX] == {'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",
     "\n",
     "vec.delete_all()\n",

timescale_vector/client.py

@@ -219,6 +219,13 @@ async def connect(self):
         if self.pool == None:
             async def init(conn):
                 await register_vector(conn)
+                #decode to a dict, but accept a string as input in upsert
+                await conn.set_type_codec(
+                    'jsonb',
+                    encoder=str,
+                    decoder=json.loads,
+                    schema='pg_catalog')
+
             self.pool = await asyncpg.create_pool(dsn=self.service_url, init=init)
         return self.pool.acquire()