xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-17 03:04:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: 4355f0fa78 ruff lint

Browse Source
This commit is contained in:
youhuanghe
2026-03-16 06:39:29 +00:00
parent 4355f0fa78
commit 254a13bba3
6 changed files with 45 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/langbot/pkg/vector/vdbs/pgvector_db.py
+12 -12
View File
@@ -338,9 +338,7 @@ class PgVectorDatabase(VectorDatabase):
)
count_stmt = (
select(func.count())
.select_from(PgVectorEntry)
.filter(PgVectorEntry.collection == collection)
select(func.count()).select_from(PgVectorEntry).filter(PgVectorEntry.collection == collection)
)
if filter:
@@ -356,15 +354,17 @@ class PgVectorDatabase(VectorDatabase):
items = []
for row in rows:
items.append({
'id': row.id,
'document': row.text or '',
'metadata': {
'text': row.text or '',
'file_id': row.file_id or '',
'uuid': row.chunk_uuid or '',
},
})
items.append(
{
'id': row.id,
'document': row.text or '',
'metadata': {
'text': row.text or '',
'file_id': row.file_id or '',
'uuid': row.chunk_uuid or '',
},
}
)
return items, total
except Exception as e: