上传文档+ARG检索+agent

1.上传文档,并存在向量数据库中

    def post(self, request):
        file = request.FILES.get('file')
        with open('media/'+file.name,'wb') as f:
            for i in file.chunks():
                f.write(i)
        content = file.name
        ret = post_file_rag(content)
        return Response({'response': content, 'ret': ret})

def post_file_rag(x):
    file_path = os.path.join(settings.BASE_DIR, 'media', x)
    loader = TextLoader(file_path, encoding='utf-8')
    docs = loader.load()
    text_splitter = CharacterTextSplitter(chunk_size=200, chunk_overlap=50)
    chunks = text_splitter.split_documents(docs)

    embeddings = DashScopeEmbeddings()
    db = Chroma.from_documents(chunks, embeddings, persist_directory="./chroma")
    db.persist()
    return docs

2.ARG检索+agent(搜索+数学计算)

    def get(self, request):
        content = request.query_params