Function format_reference

app/utils.py:123–146

将知识库检索结果格式化为参考文档的格式

(kb_name: str, docs: List[Dict], api_base_url: str = "")

def format_reference(kb_name: str, docs: List[Dict], api_base_url: str = "") -> List[Dict]:
    '''
    将知识库检索结果格式化为参考文档的格式
    '''
    api_base_url = "http://127.0.0.1:7861"

    source_documents = []
    for inum, doc in enumerate(docs):
        filename = doc.get("metadata", {}).get("source")
        parameters = urlencode(
            {
                "knowledge_base_name": kb_name,
                "file_name": filename,
            }
        )
        api_base_url = api_base_url.strip(" /")
        url = (
                f"{api_base_url}/knowledge_base/download_doc?" + parameters
        )
        page_content = doc.get("page_content")
        ref = f"""出处 [{inum + 1}] [{filename}]({url}) \n\n{page_content}\n\n"""
        source_documents.append(ref)

    return source_documents