
class Document:
    def __init__(self, document: str):
        self.document = document

def doc_search(query, top_k, libs, kb_ids, kb_file_ids):
    """
    实现检索逻辑，返回List[Document]
    return: List[Document]
    """
    if set(libs)==set(["es_chunk","es_summary","es_embedding","es_summary_embedding"]):
        documents=['第7航空队第607物资维修中队水港联络官达尔文·瓦伦西亚（Darwin Valencia） 美国空军三级军士长、菲律宾裔达尔文·瓦伦西亚担任韩国釜山第607物资维修中队水港联络官，重视家庭，爱好广泛，工作成就显著。 教育经历：瓦伦西亚接受过多所学校教育，拥有良好教育背景。 瓦伦西亚教育背景：关岛大学、刘易斯-麦科德联合基地皮尔斯学院和圣马丁大学。', '第7航空队第607物资维修中队水港联络官达尔文·瓦伦西亚（Darwin Valencia） 美国空军三级军士长、菲律宾裔达尔文·瓦伦西亚担任韩国釜山第607物资维修中队水港联络官，重视家庭，爱好广泛，工作成就显著。 教育经历：瓦伦西亚接受过多所学校教育，拥有良好教育背景。 瓦伦西亚宗教信仰：信仰基督教。', '第7航空队第607物资维修中队水港联络官达尔文·瓦伦西亚（Darwin Valencia） 美国空军三级军士长、菲律宾裔达尔文·瓦伦西亚担任韩国釜山第607物资维修中队水港联络官，重视家庭，爱好广泛，工作成就显著。 个人背景：达尔文·瓦伦西亚是美国空军三级军士长，菲律宾裔，重视家庭与工作。 教育经历：瓦伦西亚接受过多所学校教育，拥有良好教育背景。 社交属性：瓦伦西亚发帖频繁，内容正面，有400名左右好友。 心理属性与兴趣爱好：瓦伦西亚性格稳定，爱好广泛，注重家庭。 工作职责与目标价值综述：瓦伦西亚负责管理运输和联络服务，工作重要，有强烈家庭和国家认同感。', '第7航空队第607物资维修中队水港联络官达尔文·瓦伦西亚（Darwin Valencia） 美国空军三级军士长、菲律宾裔达尔文·瓦伦西亚担任韩国釜山第607物资维修中队水港联络官，重视家庭，爱好广泛，工作成就显著。 工作职责与目标价值综述：瓦伦西亚负责管理运输和联络服务，工作重要，有强烈家庭和国家认同感。 瓦伦西亚影响方式：妻子支持，家人对工作状态和情绪有重要影响。', '第7航空队第607物资维修中队水港联络官达尔文·瓦伦西亚（Darwin Valencia） 美国空军三级军士长、菲律宾裔达尔文·瓦伦西亚担任韩国釜山第607物资维修中队水港联络官，重视家庭，爱好广泛，工作成就显著。 工作职责与目标价值综述：瓦伦西亚负责管理运输和联络服务，工作重要，有强烈家庭和国家认同感。 瓦伦西亚目标价值综述：个体认知稳定，有强烈家庭和国家认同感，情绪敏感。', '第7航空队第607物资维修中队水港联络官达尔文·瓦伦西亚（Darwin Valencia） 美国空军三级军士长、菲律宾裔达尔文·瓦伦西亚担任韩国釜山第607物资维修中队水港联络官，重视家庭，爱好广泛，工作成就显著。 心理属性与兴趣爱好：瓦伦西亚性格稳定，爱好广泛，注重家庭。 瓦伦西亚家庭观念：非常重视家庭，组建家庭的愿望是其加入空军的动力。', '第7航空队第607物资维修中队水港联络官达尔文·瓦伦西亚（Darwin Valencia） 美国空军三级军士长、菲律宾裔达尔文·瓦伦西亚担任韩国釜山第607物资维修中队水港联络官，重视家庭，爱好广泛，工作成就显著。 社交属性：瓦伦西亚发帖频繁，内容正面，有400名左右好友。 瓦伦西亚社交关系：好友数量约400人，亲密度前十好友成分不固定。', '第7航空队第607物资维修中队水港联络官达尔文·瓦伦西亚（Darwin Valencia） 美国空军三级军士长、菲律宾裔达尔文·瓦伦西亚担任韩国釜山第607物资维修中队水港联络官，重视家庭，爱好广泛，工作成就显著。 社交属性：瓦伦西亚发帖频繁，内容正面，有400名左右好友。 瓦伦西亚目前工作地点：韩国釜山空军基地。', '第7航空队第607物资维修中队水港联络官达尔文·瓦伦西亚（Darwin Valencia） 美国空军三级军士长、菲律宾裔达尔文·瓦伦西亚担任韩国釜山第607物资维修中���水港联络官，重视家庭，爱好广泛，工作成就显著。 个人背景：达尔文·瓦伦西亚是美国空军三级军士长，菲律宾裔，重视家庭与工作。 瓦伦西亚个人资料：出生于菲律宾裔，重视家庭，育有三子一女。', '第7航空队第607物资维修中队水港联络官达尔文·瓦伦西亚（Darwin Valencia） 美国空军三级军士长、菲律宾裔达尔文·瓦伦西亚担任韩国釜山第607物资维修中队水港联络官，重视家庭，爱好广泛，工作成就显著。 心理属性与兴趣爱好：瓦伦西亚性格稳定，爱好广泛，注重家庭。 瓦伦西亚人格特质：神经质较低、外倾性一般、尽责性较高、宜人性一般、开放性一般。']
        docs = [Document(document=doc) for doc in documents]
    #     doc=Document(document="""预防糖尿病一般需要注意控制体重、注意养成良好的生活习惯、进行适度的体育锻炼、控制压力、避免过度饮酒和吸烟；饮食上要注意清淡饮食、注意选择低糖饮食、增加蔬果摄入、选择健康蛋白质、增加膳食纤维摄入。具体分析如下：

    # 一、生活上注意事项

    # 1.注意控制体重：糖尿病是一种代谢性疾病，主要表现为血糖升高。体重超重或肥胖是糖尿病发生的危险因素之一，因此需要通过适当锻炼、健康的饮食来控制体重。

    # 2.注意养成良好的生活习惯：日常生活中要保持规律的生活方式，避免过度疲劳，保持心情愉悦，有利于预防糖尿病。

    # 3.进行适度的体育锻炼：定期进行适度的有氧运动，如快走、跑步、游泳等，可以帮助提高身体的胰岛素敏感性，降低糖尿病的风险。

    # """)
    #     return [doc]*10
    elif set(libs)==set(["es_chunk","es_embedding"]):
        documents=['第7航空队第51战斗机联队牧师亚历克斯·陆（Alex Lu） 姓名 亚历克斯·陆（Alex J.Lu）\n英文名：Alex J.Lu\n常用名：Alex Lu  性别 男 族裔 美籍华人生日 1981年，现年42岁  家乡 中国  标准照 工作照 该照片拍摄于2023年1月17日，韩国乌山空军基\n地，第51战斗机联队牧师亚历克斯·陆（Alex\nLu）等人和宗教部专业人员在新教堂的施工现场\n合影。 3 \n \n生活照 \n \n2020 年3 月31 日，个人照。 1.3 家庭成员 \n1.3.1 妻子 \n姓名 \n米歇尔·张（Michelle Zhang） \n照片 \n \n简介\n米歇尔·张（Michelle Zhang），系亚历克斯·陆\n（Alex Lu）的妻子，亚裔。社交媒体 脸谱：https://www.facebook.com/jessie.zhang.3766 \n \n1.4 生活习惯 \n亚历克斯·陆（Alex Lu）网络获取习惯的时间主要集中生活照 2020 年 3 月 31 日，个人照。姓名 米歇尔·张（Michelle Zhang）照片 简介 米歇尔·张（Michelle Zhang），系亚历克斯·陆\n（Alex Lu）的妻子，亚裔。社交媒体 脸谱：https://www.facebook.com/jessie.zhang.37664 \n \n在早上6 点，中午点12 点至3 点，晚上8 点至12 点。1.5 就医情况 \n亚历克斯·陆（Alex Lu）目前身体状况良好。二、社会属性 \n2.1 工作情况 \n单位 \n美国太平洋空军第7 航空队',
 '第7航空队第51战斗机联队牧师亚历克斯·陆（Alex Lu） 历克斯·陆大五人格表现为：神经质较低、外倾性较高、\n尽责性较高、宜人性较高、开放性一般。 历克斯·陆呼吁反\n对种族歧视，停止亚洲仇恨。 2 \n \n一、自然属性 \n1.1 基本信息 \n姓名 \n亚历克斯·陆（Alex J.Lu） \n英文名：Alex J.Lu \n常用名：Alex Lu \n性别 男 \n族裔 美籍华人\n生日 1981年，现年42岁 \n家乡 中国 \n1.2 个人照片 \n标准照 \n \n工作照 \n \n该照片拍摄于2023年1月17日，韩国乌山空军基\n地，第51战斗机联队牧师亚历克斯·陆（Alex \nLu）等人和宗教部专业人员在新教堂的施工现场\n合影。姓名 亚历克斯·陆（Alex J.Lu）\n英文名：Alex J.Lu\n常用名：Alex Lu  性别 男 族裔 美籍华人生日 1981年，现年42岁  家乡 中国  标准照 工作照 该照片拍摄于2023年1月17日，韩国乌山空军基\n地，第51战斗机联队牧师亚历克斯·陆（Alex\nLu）等人和宗教部专业人员在新教堂的施工现场\n合影。3 \n \n生活照 \n \n2020 年3 月31 日，个人照。1.3 家庭成员 \n1.3.1 妻子 \n姓名 \n米歇尔·张（Michelle Zhang） \n照片 \n \n简介',
 ' 亚历克斯·陆第7航空队第51战斗机联队牧师 2023年3月16日\n目标概述 第7航空队第51战斗机联队牧师亚历克斯·陆（AlexLu），美国空军上尉，男，美籍华人，1981年出生于中国，现年 42岁，历克斯·陆妻子是米歇尔·张（Michelle Zhang）。历克斯·陆喜欢棒球运动，是波士顿红袜的球迷。联系邮箱：alex. lu@us.af.mil。历克斯·陆2013年加入陆军，后于2017年加入空军。教育经历完备，获欧洲拿撒勒学院（EuropeanNazareneCollege）宗教研究文学学士学位和戈登康威尔神学院（Gordon-Conwell Theological',
 '第7航空队第51战斗机联队牧师亚历克斯·陆（Alex Lu） （4）2017 年 2 月-2020 年 6 月，弗吉尼亚州兰利尤斯\n蒂斯联合基地，空军牧师，期间服役于第 633 空军基\n地联队。 （5）2022 年至今，韩国乌山空军基地第 51 战斗机联\n队牧师。 军衔晋升\n情况 上尉-2017 年邮箱 alex.lu@us.af.mil5 \n \n2.5 职业发展前景 \n亚历克斯·陆（Alex Lu）教育经历完备，获欧洲拿撒勒\n学院（European Nazarene College）宗教研究文学学士学位\n和戈登康威尔神学院（Gordon-Conwell Theological \nSeminary）宗教研究神学硕士学位。履历丰富，2013 年入伍\n陆军，2017 年转投空军，现任韩国乌山空军基地第51 战斗\n机联队牧师，年龄不大，在军队积累了一定人脉关系，上升\n空间较大，职业发展前景较好。三、社交属性 \n3.1 社交账号 \n平台 \n账号 \n脸谱 \n昵称：Alex J.Lu \n链接：https://www.facebook.com/luwarehouse \n领英 \n昵称：Alex Lu \n链接： \nhttps://www.linkedin.com/in/alex-lu-8b131527/ \n \n3.2 脸谱平台发帖互动 \n亚历克斯·陆（Alex Lu）共有1 个脸谱账号和1 个领',
 '第7航空队第51战斗机联队牧师亚历克斯·陆（Alex Lu） 社交媒体 脸谱：https://www.facebook.com/jessie.zhang.37664 \n \n在早上6 点，中午点12 点至3 点，晚上8 点至12 点。 1.5 就医情况 \n亚历克斯·陆（Alex Lu）目前身体状况良好。 二、社会属性 \n2.1 工作情况 \n单位 \n美国太平洋空军第7 航空队\n岗位 \n第51 战斗机联队牧师（美国空军上尉） \n入伍 \n2013 年 \n履历经历 \n（1）2010 年1 月-2011 年1 月，美国邮政总局。（2）2013 年，入伍美国陆军。（3）2013 年11 月-2017 年1 月，陆军牧师。（4）2017 年2 月-2020 年6 月，弗吉尼亚州兰利尤斯\n蒂斯联合基地，空军牧师，期间服役于第633 空军基\n地联队。（5）2022 年至今，韩国乌山空军基地第51 战斗机联\n队牧师。军衔晋升\n情况 \n上尉-2017 年 \n2.2 教育经历 \n1999 年-2003 年，欧洲拿撒勒学院（European Nazarene \nCollege）宗教研究文学学士； \n2009 年-2013 年，戈登康威尔神学院（Gordon-Conwell \nTheological Seminary）宗教研究神学硕士。2.3 宗教信仰',
 '第7航空队第51战斗机联队牧师亚历克斯·陆（Alex Lu） 5.2 个体认知稳定性 \n据调查，亚历克斯·陆（Alex Lu）的个体认知稳定。 自\n我认知清晰明确，对美国有高度的国家认同感。 亚历克\n斯·陆（Alex Lu）曾表示自己在中国出生和长大，但他的家\n人在文化大革命期间遭到迫害后计划移民并成为美国公民，\n正是在这段旅程中，他找到了自己的身份和人生使命。5.3 情绪敏感性信息 \n亚历克斯·陆（Alex Lu）是美籍华人，对种族歧视问题\n较为敏感，曾多次在发文讲述亚裔在美的压力，也曾上传带\n有“反对种族歧视，停止亚洲仇恨”字样的图片。18 \n \n \n5.4 情绪激发扳机点 \n亚历克斯·陆（Alex Lu）反对种族歧视，因此，有关种\n族歧视的言论和行为是其情绪激发扳机点。5.5 目标重要性 \n亚历克斯·陆（Alex Lu）现任韩国乌山空军基地第51\n战斗机联队牧师，为信仰者和无信仰者提供教牧关怀和咨询，\n支持空军人员的精神韧性。并就飞行员道德、伦理和生活质\n量问题与领导层进行磋商。其个人价值较大，目标较为重要。5.6 目标影响方式 \n亚历克斯·陆（Alex Lu）强烈的家庭责任感，在陆军服',
 '第7航空队第51战斗机联队牧师亚历克斯·陆（Alex Lu） （5）2022 年至今，韩国乌山空军基地第51 战斗机联\n队牧师。 军衔晋升\n情况 \n上尉-2017 年 \n2.2 教育经历 \n1999 年-2003 年，欧洲拿撒勒学院（European Nazarene \nCollege）宗教研究文学学士； \n2009 年-2013 年，戈登康威尔神学院（Gordon-Conwell \nTheological Seminary）宗教研究神学硕士。 2.3 宗教信仰\n亚历克斯·陆（Alex Lu）信仰基督教。2.4 联系方式 \n邮箱 \nalex.lu@us.af.mil单位 美国太平洋空军第 7 航空队岗位 第 51 战斗机联队牧师（美国空军上尉）入伍 2013 年履历经历 （1）2010 年 1 月-2011 年 1 月，美国邮政总局。（2）2013 年，入伍美国陆军。（3）2013 年 11 月-2017 年 1 月，陆军牧师。（4）2017 年 2 月-2020 年 6 月，弗吉尼亚州兰利尤斯\n蒂斯联合基地，空军牧师，期间服役于第 633 空军基\n地联队。（5）2022 年至今，韩国乌山空军基地第 51 战斗机联\n队牧师。军衔晋升\n情况 上尉-2017 年邮箱 alex.lu@us.af.mil5 \n \n2.5 职业发展前景 \n亚历克斯·陆（Alex Lu）教育经历完备，获欧洲拿撒勒',
 '第7航空队第51战斗机联队牧师亚历克斯·陆（Alex Lu） 4.6 资金消费情况 \n通过对亚历克斯·陆（Alex Lu）的消费情况调查发现亚\n历克斯·陆（Alex Lu）收入不菲，常驻军营，日常消费也较\n少。 4.7 生活困境 \n亚历克斯·陆（Alex Lu）目前是美国空军上尉，基本月\n薪在4500 美元至7000 美元之间，再加上驻外补贴等其他福\n利待遇，单凭薪资待遇来分析，亚历克斯·陆（Alex Lu）的\n个人收入稳定，无家庭经济压力，整体经济状况良好，目前\n不存在家庭困难。 五、目标价值综述 \n5.1 网络信息获取习惯 \n亚历克斯·陆（Alex Lu）的网络信息获取途径主要是通\n过社交网络、搜索引擎、第7 航空队脸谱主页、美国印太司\n令部官方脸谱主页、《空军月刊》（Air Forces Monthly）、\n《空军杂志》（Air Force Magazine）、《空军时报》（Air \nForce Times）、《每日报告》（Daily Report）、《AFSA 杂\n志》（AFSA Magazine）、乌山空军基地脸谱主页、美国海关\n和边境保护局领英账号、美国国务院领英账号、美国邮政服\n务领英账号等。亚历克斯·陆（Alex Lu）通过社交媒体脸谱获取网络信\n息的时间主要集中在早上6 点，中午12 点至3 点，晚上8 点17 \n \n至12 点。5.2 个体认知稳定性 \n据调查，亚历克斯·陆（Alex Lu）的个体认知稳定。自\n我认知清晰明确，对美国有高度的国家认同感。亚历克\n斯·陆（Alex Lu）曾表示自己在中国出生和长大，但他的家',
 '第7航空队第51战斗机联队牧师亚历克斯·陆（Alex Lu）:第7航空队第51战斗机联队牧师亚历克斯·陆（Alex Lu） \n美国印太司令部 \n \n \n \n \n \n \n \n \n \n亚历克斯·陆 \n第7 航空队第51 战斗机联队牧师 \n \n \n \n \n \n \n2023 年3 月16 日目 录 \n目 标 概 述 ....................................................... 1 \n一、自然属性 ...................................................... 2 \n1.1 基本信息..................................................... 2 \n1.2 个人照片..................................................... 2']
        docs=[Document(document=doc) for doc in documents]
        
    return docs


def extract_docs(docs: list[Document]):

    doc_strs = [doc.document for doc in docs]
    return doc_strs

def doc_search_main(query, top_k, libs, kb_ids, kb_file_ids):

    docs = doc_search(query=query, top_k=top_k, libs=libs, kb_ids=kb_ids, kb_file_ids=kb_file_ids)
    #doc_strs = extract_docs(docs)
    return docs
