diff --git a/src/langchain_google_cloud_sql_pg/async_vectorstore.py b/src/langchain_google_cloud_sql_pg/async_vectorstore.py index 2e04bb93..0aa4bc07 100644 --- a/src/langchain_google_cloud_sql_pg/async_vectorstore.py +++ b/src/langchain_google_cloud_sql_pg/async_vectorstore.py @@ -327,7 +327,7 @@ async def aget_by_ids(self, ids: Sequence[str]) -> list[Document]: Document( page_content=row[self.content_column], metadata=metadata, - id=row[self.id_column], + id=str(row[self.id_column]), ) ) ) @@ -651,7 +651,7 @@ async def asimilarity_search_with_score_by_vector( Document( page_content=row[self.content_column], metadata=metadata, - id=row[self.id_column], + id=str(row[self.id_column]), ), row["distance"], ) @@ -742,7 +742,7 @@ async def amax_marginal_relevance_search_with_score_by_vector( Document( page_content=row[self.content_column], metadata=metadata, - id=row[self.id_column], + id=str(row[self.id_column]), ), row["distance"], )