From 851327e5726c46000ac3f1464f249cc0155a3ab6 Mon Sep 17 00:00:00 2001 From: wwwttt001 Date: Mon, 11 Aug 2025 08:42:20 +0000 Subject: [PATCH 1/2] update RAGSDK/MainRepo/Samples/RagDemo/dify/dify_demo.py. Signed-off-by: wwwttt001 --- RAGSDK/MainRepo/Samples/RagDemo/dify/dify_demo.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/RAGSDK/MainRepo/Samples/RagDemo/dify/dify_demo.py b/RAGSDK/MainRepo/Samples/RagDemo/dify/dify_demo.py index b2edaf3fc..9e082fd3e 100644 --- a/RAGSDK/MainRepo/Samples/RagDemo/dify/dify_demo.py +++ b/RAGSDK/MainRepo/Samples/RagDemo/dify/dify_demo.py @@ -23,11 +23,12 @@ from pydantic import BaseModel from pymilvus import MilvusClient from starlette.responses import JSONResponse +from paddle.base import libpaddle +from mx_rag.embedding.service import TEIEmbedding from langchain_community.document_loaders import TextLoader from langchain_text_splitters import RecursiveCharacterTextSplitter from mx_rag.document import LoaderMng from mx_rag.document.loader import DocxLoader, PdfLoader -from mx_rag.embedding.service import TEIEmbedding from mx_rag.knowledge import KnowledgeStore, KnowledgeDB from mx_rag.reranker.service import TEIReranker from mx_rag.retrievers import Retriever, FullTextRetriever -- Gitee From 9e21a81c47f196ad69d2fad6cdc1609cee424ce8 Mon Sep 17 00:00:00 2001 From: wwwttt001 Date: Mon, 11 Aug 2025 08:51:08 +0000 Subject: [PATCH 2/2] update RAGSDK/MainRepo/Samples/RagDemo/embedding_finetune/finetune.py. Signed-off-by: wwwttt001 --- RAGSDK/MainRepo/Samples/RagDemo/embedding_finetune/finetune.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/RAGSDK/MainRepo/Samples/RagDemo/embedding_finetune/finetune.py b/RAGSDK/MainRepo/Samples/RagDemo/embedding_finetune/finetune.py index 7dab74837..5b3952bce 100644 --- a/RAGSDK/MainRepo/Samples/RagDemo/embedding_finetune/finetune.py +++ b/RAGSDK/MainRepo/Samples/RagDemo/embedding_finetune/finetune.py @@ -3,6 +3,7 @@ import argparse import os +from paddle.base import libpaddle import torch import torch_npu from datasets import load_dataset @@ -16,7 +17,6 @@ from mx_rag.reranker.local import LocalReranker from mx_rag.tools.finetune.generator import TrainDataGenerator, DataProcessConfig from mx_rag.utils import ClientParam from mx_rag.utils.file_check import FileCheck -from paddle.base import libpaddle from sentence_transformers import SentenceTransformer from sentence_transformers import SentenceTransformerTrainer from sentence_transformers import SentenceTransformerTrainingArguments -- Gitee