Skip to content

Commit b9526c6

Browse files
committed
Make escape sql string function private
1 parent cf58c2a commit b9526c6

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

langchain_postgres/engine.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ async def close(self) -> None:
136136
"""Dispose of connection pool"""
137137
await self._run_as_async(self._pool.dispose())
138138

139-
def escape_postgres_identifier(self, name: str) -> str:
139+
def _escape_postgres_identifier(self, name: str) -> str:
140140
return name.replace('"', '""')
141141

142142
async def _ainit_vectorstore_table(
@@ -180,19 +180,19 @@ async def _ainit_vectorstore_table(
180180
:class:`UndefinedObjectError <asyncpg.exceptions.UndefinedObjectError>`: if the data type of the id column is not a postgreSQL data type.
181181
"""
182182

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)
187187
if metadata_columns is None:
188188
metadata_columns = []
189189
else:
190190
for col in metadata_columns:
191-
col.name = self.escape_postgres_identifier(col.name)
191+
col.name = self._escape_postgres_identifier(col.name)
192192
if isinstance(id_column, str):
193-
id_column = self.escape_postgres_identifier(id_column)
193+
id_column = self._escape_postgres_identifier(id_column)
194194
else:
195-
id_column.name = self.escape_postgres_identifier(id_column.name)
195+
id_column.name = self._escape_postgres_identifier(id_column.name)
196196

197197
async with self._pool.connect() as conn:
198198
await conn.execute(text("CREATE EXTENSION IF NOT EXISTS vector"))

0 commit comments

Comments
 (0)