Use JSONB instead of JSON for thread metadata.

langchain-ai · May 8, 2024 · 712ccc6 · 712ccc6
1 parent bf2264f
commit 712ccc6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/backend/app/lifespan.py b/backend/app/lifespan.py
@@ -19,6 +19,12 @@ async def _init_connection(conn) -> None:
         decoder=orjson.loads,
         schema="pg_catalog",
     )
+    await conn.set_type_codec(
+        "jsonb",
+        encoder=lambda v: orjson.dumps(v).decode(),
+        decoder=orjson.loads,
+        schema="pg_catalog",
+    )
     await conn.set_type_codec(
         "uuid", encoder=lambda v: str(v), decoder=lambda v: v, schema="pg_catalog"
     )

diff --git a/backend/migrations/000004_add_metadata_to_thread.up.sql b/backend/migrations/000004_add_metadata_to_thread.up.sql
@@ -1,5 +1,5 @@
 ALTER TABLE thread
-ADD COLUMN metadata JSON;
+ADD COLUMN metadata JSONB;
 
 UPDATE thread
 SET metadata = json_build_object(