LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2026-05-06 16:25:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: pgvec

Browse Source
This commit is contained in:
Alex
2026-05-05 01:55:23 +01:00
parent b4c4ab68f0
commit e692c645b9
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
application/vectorstore/pgvector.py
View File

@@ -1,5 +1,8 @@
import logging
from typing import List, Optional, Any, Dict
from psycopg.types.json import Jsonb
from application.core.settings import settings
from application.vectorstore.base import BaseVectorStore
from application.vectorstore.document_class import Document
@@ -175,7 +178,7 @@ class PGVectorStore(BaseVectorStore):
for text, embedding, metadata in zip(texts, embeddings, metadatas):
cursor.execute(
insert_query,
(text, embedding, metadata, self._source_id)
(text, embedding, Jsonb(metadata), self._source_id)
)
inserted_id = cursor.fetchone()[0]
inserted_ids.append(str(inserted_id))
@@ -266,7 +269,7 @@ class PGVectorStore(BaseVectorStore):
cursor.execute(
insert_query,
(text, embeddings[0], final_metadata, self._source_id)
(text, embeddings[0], Jsonb(final_metadata), self._source_id)
)
inserted_id = cursor.fetchone()[0]
conn.commit()