@@ -136,7 +136,7 @@ async def close(self) -> None:
136
136
"""Dispose of connection pool"""
137
137
await self ._run_as_async (self ._pool .dispose ())
138
138
139
- def escape_postgres_identifier (self , name : str ) -> str :
139
+ def _escape_postgres_identifier (self , name : str ) -> str :
140
140
return name .replace ('"' , '""' )
141
141
142
142
async def _ainit_vectorstore_table (
@@ -180,19 +180,19 @@ async def _ainit_vectorstore_table(
180
180
:class:`UndefinedObjectError <asyncpg.exceptions.UndefinedObjectError>`: if the data type of the id column is not a postgreSQL data type.
181
181
"""
182
182
183
- schema_name = self .escape_postgres_identifier (schema_name )
184
- table_name = self .escape_postgres_identifier (table_name )
185
- content_column = self .escape_postgres_identifier (content_column )
186
- embedding_column = self .escape_postgres_identifier (embedding_column )
183
+ schema_name = self ._escape_postgres_identifier (schema_name )
184
+ table_name = self ._escape_postgres_identifier (table_name )
185
+ content_column = self ._escape_postgres_identifier (content_column )
186
+ embedding_column = self ._escape_postgres_identifier (embedding_column )
187
187
if metadata_columns is None :
188
188
metadata_columns = []
189
189
else :
190
190
for col in metadata_columns :
191
- col .name = self .escape_postgres_identifier (col .name )
191
+ col .name = self ._escape_postgres_identifier (col .name )
192
192
if isinstance (id_column , str ):
193
- id_column = self .escape_postgres_identifier (id_column )
193
+ id_column = self ._escape_postgres_identifier (id_column )
194
194
else :
195
- id_column .name = self .escape_postgres_identifier (id_column .name )
195
+ id_column .name = self ._escape_postgres_identifier (id_column .name )
196
196
197
197
async with self ._pool .connect () as conn :
198
198
await conn .execute (text ("CREATE EXTENSION IF NOT EXISTS vector" ))
0 commit comments