From 2c92cd00ae689c4a25b528f85dba8cdb311c7b5f Mon Sep 17 00:00:00 2001 From: lijia <469037951@qq.com> Date: Tue, 16 Jan 2024 16:58:19 +0800 Subject: [PATCH] =?UTF-8?q?langchain=E6=96=B0=E7=89=88=E6=9C=ACPGVector?= =?UTF-8?q?=E4=BC=A0=E5=85=A5connection=EF=BC=8C=E6=8F=92=E5=85=A5?= =?UTF-8?q?=E3=80=81=E5=88=A0=E9=99=A4=E6=93=8D=E4=BD=9C=E4=BC=9A=E5=9B=9E?= =?UTF-8?q?=E6=BB=9A?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- server/knowledge_base/kb_service/pg_kb_service.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/server/knowledge_base/kb_service/pg_kb_service.py b/server/knowledge_base/kb_service/pg_kb_service.py index 92d74cf3..ec0e147b 100644 --- a/server/knowledge_base/kb_service/pg_kb_service.py +++ b/server/knowledge_base/kb_service/pg_kb_service.py @@ -23,11 +23,11 @@ class PGKBService(KBService): self.pg_vector = PGVector(embedding_function=EmbeddingsFunAdapter(self.embed_model), collection_name=self.kb_name, distance_strategy=DistanceStrategy.EUCLIDEAN, - connection=self.engine, + connection=PGKBService.engine, connection_string=kbs_config.get("pg").get("connection_uri")) def get_doc_by_ids(self, ids: List[str]) -> List[Document]: - with Session(self.engine) as session: + with Session(PGKBService.engine) as session: stmt = text("SELECT document, cmetadata FROM langchain_pg_embedding WHERE collection_id in :ids") results = [Document(page_content=row[0], metadata=row[1]) for row in session.execute(stmt, {'ids': ids}).fetchall()] @@ -47,7 +47,7 @@ class PGKBService(KBService): return SupportedVSType.PG def do_drop_kb(self): - with Session(self.engine) as session: + with Session(PGKBService.engine) as session: session.execute(text(f''' -- 删除 langchain_pg_embedding 表中关联到 langchain_pg_collection 表中 的记录 DELETE FROM langchain_pg_embedding @@ -72,7 +72,7 @@ class PGKBService(KBService): return doc_infos def do_delete_doc(self, kb_file: KnowledgeFile, **kwargs): - with Session(self.engine) as session: + with Session(PGKBService.engine) as session: filepath = kb_file.filepath.replace('\\', '\\\\') session.execute( text(