Mykola Chaban commited on
Commit
4dad7b8
·
1 Parent(s): 6d4939f

updated field type

Browse files
Files changed (1) hide show
  1. lightrag/kg/postgres_impl.py +3 -3
lightrag/kg/postgres_impl.py CHANGED
@@ -2131,7 +2131,7 @@ SQL_TEMPLATES = {
2131
  WITH relevant_chunks AS (
2132
  SELECT id as chunk_id
2133
  FROM LIGHTRAG_DOC_CHUNKS
2134
- WHERE $2::text[] IS NULL OR full_doc_id = ANY($2::text[])
2135
  )
2136
  SELECT source_id as src_id, target_id as tgt_id
2137
  FROM (
@@ -2148,7 +2148,7 @@ SQL_TEMPLATES = {
2148
  WITH relevant_chunks AS (
2149
  SELECT id as chunk_id
2150
  FROM LIGHTRAG_DOC_CHUNKS
2151
- WHERE $2::text[] IS NULL OR full_doc_id = ANY($2::text[])
2152
  )
2153
  SELECT entity_name FROM
2154
  (
@@ -2165,7 +2165,7 @@ SQL_TEMPLATES = {
2165
  WITH relevant_chunks AS (
2166
  SELECT id as chunk_id
2167
  FROM LIGHTRAG_DOC_CHUNKS
2168
- WHERE $2::text[] IS NULL OR full_doc_id = ANY($2::text[])
2169
  )
2170
  SELECT id, content, file_path FROM
2171
  (
 
2131
  WITH relevant_chunks AS (
2132
  SELECT id as chunk_id
2133
  FROM LIGHTRAG_DOC_CHUNKS
2134
+ WHERE $2::varchar[] IS NULL OR full_doc_id = ANY($2::varchar[])
2135
  )
2136
  SELECT source_id as src_id, target_id as tgt_id
2137
  FROM (
 
2148
  WITH relevant_chunks AS (
2149
  SELECT id as chunk_id
2150
  FROM LIGHTRAG_DOC_CHUNKS
2151
+ WHERE $2::varchar[] IS NULL OR full_doc_id = ANY($2::varchar[])
2152
  )
2153
  SELECT entity_name FROM
2154
  (
 
2165
  WITH relevant_chunks AS (
2166
  SELECT id as chunk_id
2167
  FROM LIGHTRAG_DOC_CHUNKS
2168
+ WHERE $2::varchar[] IS NULL OR full_doc_id = ANY($2::varchar[])
2169
  )
2170
  SELECT id, content, file_path FROM
2171
  (