[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

comps/dataprep/deployment/docker_compose/compose.yaml

@@ -213,7 +213,7 @@ services:
       timeout: 5s
       retries: 10
     restart: unless-stopped
-    
+
   dataprep-pgvector:
     image: ${REGISTRY:-opea}/dataprep:${TAG:-latest}
     container_name: dataprep-pgvector-server

comps/dataprep/src/integrations/opengauss.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2025 Huawei Technologies Co., Ltd. 
+# Copyright (C) 2025 Huawei Technologies Co., Ltd.
 # SPDX-License-Identifier: Apache-2.0
 
 
@@ -12,8 +12,8 @@
 from fastapi import Body, File, Form, HTTPException, UploadFile
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain_community.embeddings import HuggingFaceBgeEmbeddings, HuggingFaceInferenceAPIEmbeddings
-from langchain_opengauss import OpenGauss, OpenGaussSettings
 from langchain_huggingface import HuggingFaceEmbeddings
+from langchain_opengauss import OpenGauss, OpenGaussSettings
 
 from comps import CustomLogger, DocPath, OpeaComponent, OpeaComponentRegistry, ServiceType
 from comps.cores.proto.api_protocol import DataprepRequest
@@ -94,12 +94,7 @@ def check_health(self) -> bool:
             port = result.port
             psycopg2.connect(database=database, user=username, password=password, host=hostname, port=port)
             self.opengauss_config = OpenGaussSettings(
-                host=hostname,
-                port=port,
-                user=username,
-                password=password,
-                database=database,
-                embedding_dimension=768
+                host=hostname, port=port, user=username, password=password, database=database, embedding_dimension=768
             )
             return True
         except psycopg2.Error as e:
@@ -200,10 +195,7 @@ async def ingest_doc_to_opengauss(self, doc_path: DocPath):
             batch_metadatas = metadatas[i : i + batch_size]
             batch_texts = batch_chunks
             _ = OpenGauss.from_texts(
-                texts=batch_texts,
-                embedding=self.embedder,
-                metadatas=batch_metadatas,
-                config=self.opengauss_config
+                texts=batch_texts, embedding=self.embedder, metadatas=batch_metadatas, config=self.opengauss_config
             )
             if logflag:
                 logger.info(f"Processed batch {i//batch_size + 1}/{(num_chunks-1)//batch_size + 1}")
@@ -237,10 +229,7 @@ async def ingest_link_to_opengauss(self, link_list: List[str]):
                 batch_texts = batch_chunks
 
                 _ = OpenGauss.from_texts(
-                    texts=batch_texts,
-                    embedding=self.embedder,
-                    metadatas=batch_metadatas,
-                    config=self.opengauss_config
+                    texts=batch_texts, embedding=self.embedder, metadatas=batch_metadatas, config=self.opengauss_config
                 )
                 if logflag:
                     logger.info(f"Processed batch {i//batch_size + 1}/{(num_chunks-1)//batch_size + 1}")
@@ -375,4 +364,4 @@ async def delete_files(self, file_path: str = Body(..., embed=True)):
                 logger.info({"status": True})
             return {"status": True}
         else:
-            raise HTTPException(status_code=404, detail="File/folder not found. Please check del_path.")
+            raise HTTPException(status_code=404, detail="File/folder not found. Please check del_path.")

comps/dataprep/src/opea_dataprep_microservice.py

@@ -12,14 +12,14 @@
 from integrations.mariadb import OpeaMariaDBDataprep
 from integrations.milvus import OpeaMilvusDataprep
 from integrations.neo4j_llamaindex import OpeaNeo4jLlamaIndexDataprep
+from integrations.opengauss import OpeaOpenGaussDataprep
 from integrations.opensearch import OpeaOpenSearchDataprep
 from integrations.pgvect import OpeaPgvectorDataprep
 from integrations.pipecone import OpeaPineConeDataprep
 from integrations.qdrant import OpeaQdrantDataprep
 from integrations.redis import OpeaRedisDataprep
 from integrations.redis_finance import OpeaRedisDataprepFinance
 from integrations.vdms import OpeaVdmsDataprep
-from integrations.opengauss import OpeaOpenGaussDataprep
 from opea_dataprep_loader import OpeaDataprepLoader
 
 from comps import (
@@ -253,4 +253,4 @@ async def get_list_of_indices():
 if __name__ == "__main__":
     logger.info("OPEA Dataprep Microservice is starting...")
     create_upload_folder(upload_folder)
-    opea_microservices["opea_service@dataprep"].start()
+    opea_microservices["opea_service@dataprep"].start()

comps/dataprep/src/requirements-cpu.txt

@@ -401,9 +401,9 @@ langchain-mariadb==0.0.17
     # via -r ./comps/dataprep/src/requirements.in
 langchain-milvus==0.1.10
     # via -r ./comps/dataprep/src/requirements.in
-langchain-opengauss==0.1.4
-    # via -r ./comps/dataprep/src/requirements.in
 langchain-openai==0.3.23
+    # via -r ./comps/dataprep/src/requirements.in
+langchain-opengauss==0.1.4
     # via
     #   -r ./comps/dataprep/src/requirements.in
     #   langchain-pinecone

comps/third_parties/opengauss/deployment/docker_compose/compose.yaml

@@ -1,4 +1,4 @@
-# Copyright (C) 2025 Huawei Technologies Co., Ltd. 
+# Copyright (C) 2025 Huawei Technologies Co., Ltd.
 # SPDX-License-Identifier: Apache-2.0
 
 services:
@@ -16,4 +16,4 @@ services:
         interval: 60s
         retries: 3
         start_period: 10s
-        timeout: 10s
+        timeout: 10s

tests/dataprep/test_dataprep_opengauss.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-# Copyright (C) 2025 Huawei Technologies Co., Ltd. 
+# Copyright (C) 2025 Huawei Technologies Co., Ltd.
 # SPDX-License-Identifier: Apache-2.0
 
 set -x
@@ -101,4 +101,4 @@ function main() {
 
 }
 
-main
+main