#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2025/3/16 20:09
# @Author  : old-tom
# @File    : vector_agent
# @Project : llmFunctionCallDemo
# @Desc    : 向量数据库,解决相似度查询,例如 相机名称
import marqo

# 索引名称
INDEX_NAME = 'test_index'
# 初始化marqo
mq = marqo.Client(url='http://localhost:8882')

# mq.delete_index(INDEX_NAME)

settings = {
    "treatUrlsAndPointersAsImages": False,
    "model": "hf/bge-large-zh-v1.5",
    "normalizeEmbeddings": True,
}


def create_and_set_index():
    """
    全局只能调用一次
    :return:
    """
    mq.create_index(INDEX_NAME, settings_dict=settings)
    # 添加文档(测试用)
    mq.index(INDEX_NAME).add_documents([
        {
            "carme_name": "中方国门AI算法",
            "ip": "192.168.10.80",
            "location": "国门顶部"
        },
        {
            "carme_name": "中方国门AI算法0102",
            "ip": "192.168.10.80",
            "location": "国门顶部"
        },
        {
            "carme_name": "中方国门入境1",
            "ip": "192.168.10.77",
            "location": "国门通道"
        },
        {
            "carme_name": "中方国门入境2",
            "ip": "192.168.10.79",
            "location": "国门通道"
        },
        {
            "carme_name": "中方国门入境摄像头中",
            "ip": "192.168.10.72",
            "location": "登临检验"
        },
        {
            "carme_name": "中方国门入境摄像头右",
            "ip": "192.168.10.74",
            "location": "登临检验"
        },
        {
            "carme_name": "中方国门入境摄像头左",
            "ip": "192.168.10.75",
            "location": "登临检验"
        },
        {
            "carme_name": "中方国门出境1",
            "ip": "192.168.10.76",
            "location": "国门通道"
        },
        {
            "carme_name": "中方国门出境2",
            "ip": "192.168.10.78",
            "location": "国门通道"
        },
        {
            "carme_name": "中方国门出境摄像头右",
            "ip": "192.168.10.71",
            "location": "登临检验"
        },
        {
            "carme_name": "中方国门出境摄像头左",
            "ip": "192.168.10.73",
            "location": "登临检验"
        },
        {
            "carme_name": "中方国门-面向国门球机",
            "ip": "192.168.10.215",
            "location": "国门通道拐角"
        },
        {
            "carme_name": "中方国门高点1",
            "ip": "192.168.10.70",
            "location": "国门顶部"
        },
        {
            "carme_name": "北卡口AI摄像头全景",
            "ip": "192.168.10.30",
            "location": "北卡口对面"
        },
        {
            "carme_name": "北卡口AI摄像头细节",
            "ip": "192.168.10.30",
            "location": "北卡口对面"
        },
        {
            "carme_name": "北卡口入境摄像头入场4号通道",
            "ip": "192.168.10.112",
            "location": "入场通道"
        },
        {
            "carme_name": "北卡口入境摄像头入场5号通道",
            "ip": "192.168.10.114",
            "location": "入场通道"
        },
        {
            "carme_name": "北卡口入境摄像头入场6号通道",
            "ip": "192.168.10.117",
            "location": "入场通道"
        },
        {
            "carme_name": "北卡口入境摄像头出场1号通道",
            "ip": "192.168.10.115",
            "location": "出场通道"
        },
        {
            "carme_name": "北卡口入境摄像头出场2号通道",
            "ip": "192.168.10.116",
            "location": "出场通道"
        },
        {
            "carme_name": "北卡口入境摄像头出场3号通道",
            "ip": "192.168.10.113",
            "location": "出场通道"
        },
        {
            "carme_name": "北卡口出口道路监控",
            "ip": "192.168.10.153",
            "location": "路口"
        },
        {
            "carme_name": "能投大厦高点1",
            "ip": "192.168.10.89",
            "location": "能投楼顶"
        },
        {
            "carme_name": "南卡口AI算法2",
            "ip": "192.168.10.210",
            "location": "南卡口顶部"
        },
        {
            "carme_name": "南卡口AI算法20102",
            "ip": "192.168.10.210",
            "location": "南卡口顶部"
        },
        {
            "carme_name": "南卡口AI算法识别摄像机",
            "ip": "192.168.10.91",
            "location": "南卡口顶部"
        },
        {
            "carme_name": "南卡口AI算法识别摄像机0102",
            "ip": "192.168.10.91",
            "location": "南卡口顶部"
        },
        {
            "carme_name": "南卡口出境摄像头1号通道",
            "ip": "192.168.10.100",
            "location": "出境通道"
        },
        {
            "carme_name": "南卡口出境摄像头2号通道",
            "ip": "192.168.10.103",
            "location": "出境通道"
        },
        {
            "carme_name": "南卡口出境摄像头3号通道",
            "ip": "192.168.10.104",
            "location": "出境通道"
        },
        {
            "carme_name": "南卡口出境摄像头4号通道",
            "ip": "192.168.10.102",
            "location": "出境通道"
        },
        {
            "carme_name": "南卡口出境摄像头5号通道",
            "ip": "192.168.10.99",
            "location": "出境通道"
        },
        {
            "carme_name": "南卡口出境摄像头6号通道",
            "ip": "192.168.10.101",
            "location": "出境通道"
        },
        {
            "carme_name": "南卡口入境摄像头7号通道",
            "ip": "192.168.10.98",
            "location": "入境通道"
        },
        {
            "carme_name": "南卡口入境摄像头8号通道",
            "ip": "192.168.10.93",
            "location": "入境通道"
        },
        {
            "carme_name": "南卡口入境摄像头9号通道",
            "ip": "192.168.10.97",
            "location": "入境通道"
        },
        {
            "carme_name": "南卡口入境摄像头10号通道",
            "ip": "192.168.10.96",
            "location": "入境通道"
        },
        {
            "carme_name": "南卡口入境摄像头11号通道",
            "ip": "192.168.10.94",
            "location": "入境通道"
        },
        {
            "carme_name": "南卡口入境摄像头12号通道",
            "ip": "192.168.10.95",
            "location": "入境通道"
        },
        {
            "carme_name": "南卡口高点-1",
            "ip": "192.168.10.90",
            "location": "南卡口顶部"
        },
        {
            "carme_name": "南卡口高点-2",
            "ip": "192.168.10.92",
            "location": "南卡口顶部"
        },
        {
            "carme_name": "南卡口高点-3",
            "ip": "192.168.3.12",
            "location": "南卡口顶部"
        },
        {
            "carme_name": "1.3公里封闭道路入境出场1",
            "ip": "192.168.10.82",
            "location": "南卡口顶部"
        },
        {
            "carme_name": "1.3公里封闭道路入境出场2",
            "ip": "192.168.10.88",
            "location": "南卡口顶部"
        },
        {
            "carme_name": "1.3公里封闭道路入境出场20102",
            "ip": "192.168.10.88",
            "location": "南卡口顶部"
        },
        {
            "carme_name": "1.3公里封闭道路入境摄像头入场2",
            "ip": "192.168.10.87",
            "location": "南卡口顶部"
        },
        {
            "carme_name": "1.3公里封闭道路入境摄像头入场20102",
            "ip": "192.168.10.87",
            "location": "南卡口顶部"
        },
        {
            "carme_name": "1.3公里封闭道路入境入场1",
            "ip": "192.168.10.81",
            "location": "南卡口顶部"
        },
        {
            "carme_name": "H9861号口摄像头入场",
            "ip": "192.168.10.107",
            "location": "H986停车场出入口"
        },
        {
            "carme_name": "H9862号口摄像头出场",
            "ip": "192.168.10.109",
            "location": "H987停车场出入口"
        },
        {
            "carme_name": "H9862号口摄像头入场",
            "ip": "192.168.10.190",
            "location": "H988停车场出入口"
        },
        {
            "carme_name": "车辆定位AI摄像头",
            "ip": "192.168.10.29",
            "location": "H988停车场山上"
        },
        {
            "carme_name": "车辆定位AI摄像头0102",
            "ip": "192.168.10.29",
            "location": "H989停车场山上"
        },
        {
            "carme_name": "北山货场森林公园制高点",
            "ip": "192.168.10.191",
            "location": "H990停车场山上"
        },
        {
            "carme_name": "边民互市入境摄像头出场",
            "ip": "192.168.10.119",
            "location": "边民互市"
        },
        {
            "carme_name": "边民互市入境摄像头入场",
            "ip": "192.168.10.118",
            "location": "出入口"
        },
        {
            "carme_name": "车辆缓冲区1号门入口",
            "ip": "192.168.10.193",
            "location": "出入口"
        },
        {
            "carme_name": "车辆缓冲区2号门入口",
            "ip": "192.168.10.195",
            "location": "出入口"
        },
        {
            "carme_name": "车辆缓冲区2号门出口",
            "ip": "192.168.10.194",
            "location": "出入口"
        },
        {
            "carme_name": "车辆缓冲区1号门出口",
            "ip": "192.168.10.192",
            "location": "出入口"
        },
        {
            "carme_name": "大贸查验场进口入境摄像头入场东口",
            "ip": "192.168.10.181",
            "location": "出入口"
        },
        {
            "carme_name": "大贸查验场进口入境摄像头出场东口",
            "ip": "192.168.10.180",
            "location": "出入口"
        },
        {
            "carme_name": "大贸查验场（进口）入境摄像头入场（西口）",
            "ip": "192.168.10.106",
            "location": "出入口"
        },
        {
            "carme_name": "大贸查验场（进口）高点1",
            "ip": "192.168.10.156",
            "location": "高点"
        },
        {
            "carme_name": "大贸查验场出口出境摄像头出场2",
            "ip": "192.168.10.110",
            "location": "出入口"
        },
        {
            "carme_name": "大贸查验场出口出境摄像头入场左",
            "ip": "192.168.10.184",
            "location": "出入口"
        },
        {
            "carme_name": "大贸查验场出口出境摄像头出场1",
            "ip": "192.168.10.111",
            "location": "出入口"
        },
        {
            "carme_name": "北山高速收费站入境摄像头入场出方向",
            "ip": "192.168.10.170",
            "location": "道路"
        },
        {
            "carme_name": "北山高速收费站入境摄像头入场入方向",
            "ip": "192.168.10.169",
            "location": "道路"
        },
        {
            "carme_name": "坝洒高点",
            "ip": "192.168.10.164",
            "location": ""
        },
        {
            "carme_name": "停车场卡口（坝洒）入境摄像头入场",
            "ip": "192.168.10.165",
            "location": "出入口"
        },
        {
            "carme_name": "停车场卡口（坝洒）入境摄像头出场",
            "ip": "192.168.10.166",
            "location": "出入口"
        },
        {
            "carme_name": "东西干道卡口入境摄像头入场",
            "ip": "192.168.10.136",
            "location": "出入口"
        },
        {
            "carme_name": "东西干道卡口入境摄像头出场",
            "ip": "192.168.10.135",
            "location": "出入口"
        },
        {
            "carme_name": "主卡口（临时）货场大门-高点",
            "ip": "192.168.10.146",
            "location": ""
        },
        {
            "carme_name": "利丰酒店往边民互市高点",
            "ip": "192.168.10.211",
            "location": "利丰酒店楼顶"
        },
        {
            "carme_name": "利丰货场近御峰货场入境摄像头出场2",
            "ip": "192.168.10.132",
            "location": "出入口"
        },
        {
            "carme_name": "利丰酒店往边民互市高点2",
            "ip": "192.168.10.212",
            "location": "利丰酒店楼顶"
        },
        {
            "carme_name": "利丰货场近御峰货场入境摄像头入场2",
            "ip": "192.168.10.139",
            "location": "出入口"
        },
        {
            "carme_name": "利丰货场入境摄像头出场左",
            "ip": "192.168.10.127",
            "location": "出入口"
        },
        {
            "carme_name": "利丰货场入境摄像头入场左",
            "ip": "192.168.10.128",
            "location": "出入口"
        },
        {
            "carme_name": "利丰货场入境摄像头出场右",
            "ip": "192.168.10.126",
            "location": "出入口"
        },
        {
            "carme_name": "利丰货场-高点",
            "ip": "192.168.10.125",
            "location": ""
        },
        {
            "carme_name": "利丰货场近御峰货场入境摄像头出场1",
            "ip": "192.168.10.131",
            "location": "出入口"
        },
        {
            "carme_name": "利丰货场入境摄像头入场右",
            "ip": "192.168.10.129",
            "location": "出入口"
        },
        {
            "carme_name": "利丰货场近御峰货场入境摄像头入场1",
            "ip": "192.168.10.130",
            "location": "出入口"
        },
        {
            "carme_name": "南屏高速收费站入境摄像头出场",
            "ip": "192.168.10.168",
            "location": "道路"
        },
        {
            "carme_name": "南屏高速收费站入境摄像头入场",
            "ip": "192.168.10.167",
            "location": "道路"
        },
        {
            "carme_name": "停车场卡口老表入境摄像头入场",
            "ip": "192.168.10.160",
            "location": "出入口"
        },
        {
            "carme_name": "停车场卡口老表右入境摄像头入场",
            "ip": "192.168.10.162",
            "location": "出入口"
        },
        {
            "carme_name": "停车场卡口老表入境摄像头出场",
            "ip": "192.168.10.161",
            "location": "出入口"
        },
        {
            "carme_name": "停车场卡口老表右入境摄像头出场",
            "ip": "192.168.10.163",
            "location": "出入口"
        },
        {
            "carme_name": "槟榔寨站入境摄像头入场",
            "ip": "192.168.10.173",
            "location": "道路"
        },
        {
            "carme_name": "槟榔寨站入境摄像头出场",
            "ip": "192.168.10.174",
            "location": "道路"
        },
        {
            "carme_name": "外围冷链停车区入境摄像头出场",
            "ip": "192.168.10.158",
            "location": "出入口"
        },
        {
            "carme_name": "外围冷链停车区入境摄像头入场",
            "ip": "192.168.10.157",
            "location": "出入口"
        },
        {
            "carme_name": "清水河隧道出",
            "ip": "192.168.10.175",
            "location": "道路"
        },
        {
            "carme_name": "清水河隧道入",
            "ip": "192.168.10.186",
            "location": "道路"
        },
        {
            "carme_name": "商阜路闭环区道路监控-高点",
            "ip": "192.168.10.154",
            "location": "道路"
        },
        {
            "carme_name": "商阜路闭环区道路监控道路监控高点2",
            "ip": "192.168.10.155",
            "location": "道路"
        },
        {
            "carme_name": "商阜路闭环区道路监控道路监控-后",
            "ip": "192.168.10.152",
            "location": "道路"
        },
        {
            "carme_name": "越南城十字路口-东西干道入境摄像头出场",
            "ip": "192.168.10.172",
            "location": "道路"
        },
        {
            "carme_name": "越南城十字路口-东西干道入境摄像头入场",
            "ip": "192.168.10.171",
            "location": "道路"
        },
        {
            "carme_name": "停车场卡口能投入境摄像头出场",
            "ip": "192.168.10.124",
            "location": "出入口"
        },
        {
            "carme_name": "停车场卡口能投入境摄像头入场2",
            "ip": "192.168.10.123",
            "location": "出入口"
        },
        {
            "carme_name": "停车场卡口能投入境摄像头入场1",
            "ip": "192.168.10.122",
            "location": "出入口"
        },
        {
            "carme_name": "御峰货场冷链入境摄像头出场出口",
            "ip": "192.168.10.145",
            "location": "出入口"
        },
        {
            "carme_name": "御峰货场（冷链）-高点",
            "ip": "192.168.10.137",
            "location": ""
        },
        {
            "carme_name": "御峰货场冷链入境摄像头入场进口",
            "ip": "192.168.10.133",
            "location": "出入口"
        },
        {
            "carme_name": "御峰货场冷链入境摄像头入场出口",
            "ip": "192.168.10.138",
            "location": "出入口"
        },
        {
            "carme_name": "御峰货场冷链入境摄像头出场进口",
            "ip": "192.168.10.141",
            "location": "出入口"
        },
        {
            "carme_name": "御峰货场干货入境摄像头入场",
            "ip": "192.168.10.140",
            "location": "出入口"
        },
        {
            "carme_name": "御峰货场干货入境摄像头入场2",
            "ip": "192.168.10.148",
            "location": "出入口"
        },
        {
            "carme_name": "御峰货场干货入境摄像头出场",
            "ip": "192.168.10.144",
            "location": "出入口"
        },
        {
            "carme_name": "御峰货场（干货）-高点1",
            "ip": "192.168.10.142",
            "location": ""
        },
        {
            "carme_name": "御峰货场（干货）-高点2",
            "ip": "192.168.10.143",
            "location": ""
        },
        {
            "carme_name": "御峰货场干货入境摄像头出场2",
            "ip": "192.168.10.147",
            "location": "出入口"
        }],
        tensor_fields=["carme_name"]
    )


def query_vector_db(query):
    return mq.index(INDEX_NAME).search(q=query)


if __name__ == '__main__':
    # create_and_set_index()
    rt = query_vector_db('利丰酒店高点')
    # TODO 根据 _score字段 取出相似度最高的结果
    if rt:
        for ele in rt['hits']:
            print(ele)