itam-assistant 0.1.1__py3-none-any.whl

it_assistant/ailyapp_client.py

@@ -0,0 +1,127 @@
+import json
+
+import lark_oapi as lark
+from lark_oapi.api.auth.v3 import *
+from lark_oapi.api.aily.v1 import *
+
+
+# SDK 使用说明: https://open.feishu.cn/document/uAjLw4CM/ukTMukTMukTM/server-side-sdk/python--sdk/preparations-before-development
+# 以下示例代码默认根据文档示例值填充，如果存在代码问题，请在 API 调试台填上相关必要参数后再复制代码使用
+# 复制该 Demo 后, 需要将 "YOUR_APP_ID", "YOUR_APP_SECRET" 替换为自己应用的 APP_ID, APP_SECRET.
+
+
+class AilyLarkClient():
+    def __init__(self,clientinfo):
+        """
+        初始化 Client 实例,tenant_access_token 会在 Client 初始化时自动获取
+        """
+        self.aily_app_id = clientinfo.get("aily_app_id") or "spring_f17d05d924__c"
+        self.app_id = clientinfo.get("app_id") or "cli_a6e3aea1a13c900c"
+        self.app_secret = clientinfo.get("app_secret") or "J0fAPt3BL6bv4KUJV0dJMdTUdr0pv3xx"
+        # 创建 Lark-tenant tenant客户端
+        self.tlark_client = lark.Client.builder().app_id(self.app_id).app_secret(self.app_secret).build()
+
+        # 创建 Lark-tenant user 客户端
+        self.ulark_client = lark.Client.builder().enable_set_token(True).log_level(lark.LogLevel.DEBUG).build()
+
+    def get_tenant_access_token(self):
+        # 构造请求对象
+        request: InternalTenantAccessTokenRequest = InternalTenantAccessTokenRequest.builder() \
+            .request_body(InternalTenantAccessTokenRequestBody.builder()
+                          .app_id(self.app_id)
+                          .app_secret(self.app_secret)
+                          .build()) \
+            .build()
+
+        # 发起请求
+        response: InternalTenantAccessTokenResponse = self.tlark_client.auth.v3.tenant_access_token.internal(request)
+
+        # 处理失败返回
+        if not response.success():
+            lark.logger.error(
+                f"client.auth.v3.tenant_access_token.internal failed, code: {response.code}, msg: {response.msg}, log_id: {response.get_log_id()}, resp: \n{json.dumps(json.loads(response.raw.content), indent=4, ensure_ascii=False)}")
+            return
+        # 处理业务结果
+        lark.logger.info(lark.JSON.marshal(response.raw, indent=4))
+        tenant_access_token = json.loads(response.raw.content).get("tenant_access_token")
+        if tenant_access_token:
+            return tenant_access_token
+        else:
+            lark.logger.error(
+                f"client.auth.v3.tenant_access_token.internal failed, code: {response.code}, msg: {response.msg}, log_id: {response.get_log_id()}, resp: \n{json.dumps(json.loads(response.raw.content), indent=4, ensure_ascii=False)}")
+            return
+
+    def create_ailysession(self, access_token):
+        # 创建会话
+        # 构造请求对象
+        request: CreateAilySessionRequest = CreateAilySessionRequest.builder() \
+            .request_body(CreateAilySessionRequestBody.builder()
+                          .channel_context("{}")
+                          .metadata("{}")
+                          .build()) \
+            .build()
+
+        # 发起请求
+        option = lark.RequestOption.builder().user_access_token(access_token).build()
+        response: CreateAilySessionResponse = self.ulark_client.aily.v1.aily_session.create(request, option)
+
+        # 处理失败返回
+        if not response.success():
+            lark.logger.error(
+                f"client.aily.v1.aily_session.create failed, code: {response.code}, msg: {response.msg}, log_id: {response.get_log_id()}, resp: \n{json.dumps(json.loads(response.raw.content), indent=4, ensure_ascii=False)}")
+            return
+
+        # 处理业务结果
+        lark.logger.info(lark.JSON.marshal(response.data, indent=4))
+        return response.data.session.id
+
+    def create_ailysessionaily_message(self, access_token, session_id, content):
+        # 发送智能伙伴消息
+        # 构造请求对象
+        request: CreateAilySessionAilyMessageRequest = CreateAilySessionAilyMessageRequest.builder() \
+            .aily_session_id(session_id) \
+            .request_body(CreateAilySessionAilyMessageRequestBody.builder()
+                          .content(content)
+                          .content_type("MDX")
+                          .idempotent_id("idempotent_id_1")
+                          .build()) \
+            .build()
+        # 发起请求
+        option = lark.RequestOption.builder().user_access_token(access_token).build()
+        response: CreateAilySessionAilyMessageResponse = self.ulark_client.aily.v1.aily_session_aily_message.create(
+            request,
+            option)
+        # 处理失败返回
+        if not response.success():
+            lark.logger.error(
+                f"client.aily.v1.aily_session_aily_message.create failed, code: {response.code}, msg: {response.msg}, log_id: {response.get_log_id()}, resp: \n{json.dumps(json.loads(response.raw.content), indent=4, ensure_ascii=False)}")
+            return
+        # 处理业务结果
+        lark.logger.info(lark.JSON.marshal(response.data, indent=4))
+        return response.data.message.id
+
+    def create_ailysession_run(self, access_token, aily_session_id):
+        # 创建运行
+        # 构造请求对象
+        request: CreateAilySessionRunRequest = CreateAilySessionRunRequest.builder() \
+            .aily_session_id(aily_session_id) \
+            .request_body(CreateAilySessionRunRequestBody.builder()
+                          .app_id(self.aily_app_id)
+                          .build()) \
+            .build()
+        # 发起请求
+        option = lark.RequestOption.builder().user_access_token(
+            access_token).build()
+        response: CreateAilySessionRunResponse = self.ulark_client.aily.v1.aily_session_run.create(request, option)
+        # 处理失败返回
+        if not response.success():
+            lark.logger.error(
+                f"client.aily.v1.aily_session_run.create failed, code: {response.code}, msg: {response.msg}, log_id: {response.get_log_id()}, resp: \n{json.dumps(json.loads(response.raw.content), indent=4, ensure_ascii=False)}")
+            return
+
+        # 处理业务结果
+        lark.logger.info(lark.JSON.marshal(response.data, indent=4))
+
+
+if __name__ == '__main__':
+    aily_app_id = AilyLarkClient({}).get_tenant_access_token()

it_assistant/config.py

@@ -0,0 +1,51 @@
+# -*- coding: utf-8 -*-
+#初始化 表单数据
+
+Clientin = {
+    "aily_app_id": "spring_f17d05d924__c",
+    "app_id": "cli_a6e3aea1a13c900c",
+    "app_secret": "J0fAPt3BL6bv4KUJV0dJMdTUdr0pv3xx",
+    "cookie": "X-Kunlun-SessionId=L%3A3b34958803f34f43a52c.eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ2YWwiOnsidGVuYW50X2lkIjozOTAsInVzZXJfaWQiOjE3MjIxNjYwNzMxOTk2NDUsInRlbmFudF9kb21haW5fbmFtZSI6ImFwYWFzIiwic2Vzc2lvbl92ZXJzaW9uIjoidjIwMjAtMDUtMTkiLCJ3c190b2tlbiI6Ilc6OTk3Y2MwNTA3OTRlNGFmYWFkYzEiLCJsb2dpbl90b2tlbiI6IjE3MDE3ZmFlMWJlNjVlMzdzSzBhMzA0ZjY0N2MyZmFjY2QwSjRFYmNmNGVjNzAzZDgwOWYxNDVnNDY0MzY1ZjEyNWI0YmZlZDhhTmMiLCJzb3VyY2VfY2hhbm5lbCI6ImZlaXNodSIsInRlbmFudF9rZXkiOiI3MzY1ODhjOTI2MGYxNzVkIiwiZXh0ZXJuYWxfZG9tYWluX25hbWUiOiJieXRlZGFuY2UiLCJvcmlnaW5hbF90ZW5hbnRfaWQiOjAsIm9yaWdpbmFsX3VzZXJfaWQiOjAsImlkcF9jaGFubmVsIjoiIn0sImV4cCI6MTc1ODk0MTY3MH0.l9yn5zbWFhOEJml5iA69TpFwZ7qgLMzj7L0cj4Ryozc; passport_web_did=7487801556726579201; passport_trace_id=7487801556748156956; QXV0aHpDb250ZXh0=2f506053fdd544e7aa0df84c66a287f9; locale=zh-CN; landing_url=https://accounts.feishu.cn/accounts/page/login?app_id=107&no_trap=1&redirect_uri=https%3A%2F%2Fapaas.feishu.cn%2Fai%2Fspring_f17d05d924__c%2Fmanagement%2Fchat-log; _gcl_au=1.1.1249684330.1743389657; s_v_web_id=verify_m8wh6ssk_JRUTLUkb_AJsu_4Xjm_ANzV_gLPDip941iqw; __tea__ug__uid=7487801495396992562; _ga=GA1.2.1834362348.1743389657; _gid=GA1.2.758422620.1743389658; session=XN0YXJ0-4e7g6c2c-da65-4492-a6f6-6413002bd949-WVuZA; session_list=XN0YXJ0-4e7g6c2c-da65-4492-a6f6-6413002bd949-WVuZA; login_recently=1; _ga_VPYRHN104D=GS1.1.1743389657.1.1.1743389669.48.0.0; msToken=4W_kQaUJyB5jBl5FX8vjfY6SYAFcNAp7NiDqM3-QyBN0XIF24a5SyaOeTpfzIZAuNfH-cGjXK1u3tNXV3ETo8Z2ZTQFLGSTFF2KmMr35XQsODVrddz8FdHAfyJg4F7ayxiDsicO5ObKgK0Y_95Bq1d12vKKbJ99vm9IZWEpcRFLG; kunlun-session-v2=L%3A3b34958803f34f43a52c.eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ2YWwiOnsidGVuYW50X2lkIjozOTAsInVzZXJfaWQiOjE3MjIxNjYwNzMxOTk2NDUsInRlbmFudF9kb21haW5fbmFtZSI6ImFwYWFzIiwic2Vzc2lvbl92ZXJzaW9uIjoidjIwMjAtMDUtMTkiLCJ3c190b2tlbiI6Ilc6OTk3Y2MwNTA3OTRlNGFmYWFkYzEiLCJsb2dpbl90b2tlbiI6IjE3MDE3ZmFlMWJlNjVlMzdzSzBhMzA0ZjY0N2MyZmFjY2QwSjRFYmNmNGVjNzAzZDgwOWYxNDVnNDY0MzY1ZjEyNWI0YmZlZDhhTmMiLCJzb3VyY2VfY2hhbm5lbCI6ImZlaXNodSIsInRlbmFudF9rZXkiOiI3MzY1ODhjOTI2MGYxNzVkIiwiZXh0ZXJuYWxfZG9tYWluX25hbWUiOiJieXRlZGFuY2UiLCJvcmlnaW5hbF90ZW5hbnRfaWQiOjAsIm9yaWdpbmFsX3VzZXJfaWQiOjAsImlkcF9jaGFubmVsIjoiIn0sImV4cCI6MTc1ODk0MTY3MH0.l9yn5zbWFhOEJml5iA69TpFwZ7qgLMzj7L0cj4Ryozc; kunlun-session-token=2b32fc3c28f44fb89bab94ad072a05c9f2f844c49705c95d76bae40479a189b7; _tea_utm_cache_1229=undefined; sl_session=eyJhbGciOiJFUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOjE3NDM0MzI4NzEsInVuaXQiOiJldV9uYyIsInJhdyI6eyJtZXRhIjoiQVdIazBuRzhRUUFDQUFBQUFBQUFBQUZuNmdQWUFVNEFBV2ZxQTlnQlRnQUJaK29ENWRES3dBSUNLZ0VBUVVGQlFVRkJRVUZCUVVKdU5tZFFiRE40ZDBGQlp6MDkiLCJzdW0iOiJlMmM4YTIwMTcyMDcxNmVjYTFiOWRlOTQ5Yjc3OGJkNDczOGIzOTAwNWJiNTJhYTkyOTM2YTRhZWIzMGI2ZTY0IiwibG9jIjoiemhfY24iLCJhcGMiOiJSZWxlYXNlIiwiaWF0IjoxNzQzMzg5NjcxLCJzYWMiOnsiVXNlclN0YWZmU3RhdHVzIjoiMSIsIlVzZXJUeXBlIjoiNDIifSwibG9kIjpudWxsLCJjbmYiOnsiamt0IjoiYkx6aTdPRDBHS09mNllOQ0xGamtPZWtuQkNRSHM2ZFh5STdmcTVubE93VSJ9LCJucyI6ImxhcmsiLCJuc191aWQiOiI3MDUzOTk0MzAyMzAwNTUzMjE4IiwibnNfdGlkIjoiMSIsIm90IjozLCJjdCI6MTc0MzM4OTY3MCwicnQiOjE3NDMzODk2NzB9fQ.2pQlqU6fuqnw_iqtJe1sH1FfSSXBpFQ0RAoaRccxHEaHSBvqsdc9_7e4zjgcHOhTjISi3mGw3EC3EXftLj5Otw; passport_app_access_token=eyJhbGciOiJFUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOjE3NDM0MzI4NzIsInVuaXQiOiJldV9uYyIsInJhdyI6eyJtX2FjY2Vzc19pbmZvIjp7IjEwNyI6eyJpYXQiOjE3NDMzODk2NzIsImFjY2VzcyI6dHJ1ZX19LCJzdW0iOiJlMmM4YTIwMTcyMDcxNmVjYTFiOWRlOTQ5Yjc3OGJkNDczOGIzOTAwNWJiNTJhYTkyOTM2YTRhZWIzMGI2ZTY0In19.jtfbxALtDnZYTJx4cb6ohPy2uDVCHTuh0x-Dg7Ui1F4vMO3aka7rvOeZTIwGJ7IlAn0b-OjBOWQEVQvHthhEwQ; swp_csrf_token=a239a297-e0f7-4820-aa3a-6349c8a04977; t_beda37=10a0c227407070710f979ef9d5b530118d080fd0ec27f2c3ce04c251a5a20d70",
+    "x-kunlun-token": "17017fae1be65e37sK0a304f647c2faccd0J4Ebcf4ec703d809f145g464365f125b4bfed8aNc",
+    "x-tt-env": "ppe_cn_env_self_test_feat_cr_a",
+    "Authorization": "Basic cm40cmFpdTRwenY1cGlsYTo2bWhvOXV3ZXFrOHZpbDllcXRxMHZ1YmVnc2xjeXBucg==",
+}
+
+class SingletonMeta(type):
+    _instances = {}
+
+    def __call__(cls, *args, **kwargs):
+        if cls not in cls._instances:
+            cls._instances[cls] = super().__call__(*args, **kwargs)
+        return cls._instances[cls]
+
+class Clientassign(metaclass=SingletonMeta):
+    """
+    定义一个全局变量 Client
+    该类为单例类，确保整个应用中只有一个实例
+    """
+    def __init__(self, Clientinfox):
+
+        self.aily_app_id = Clientinfox.get("data_set_ailyappid") or Clientin["aily_app_id"]
+        self.app_id = Clientinfox.get("data_set_appid") or Clientin["app_id"]
+        self.app_secret = Clientinfox.get("data_set_appsecret") or Clientin["app_secret"]
+        self.cookie = Clientinfox.get("data_set_cookie") or Clientin["cookie"]
+        self.x_kunlun_token = Clientinfox.get("data_set_kunluntoken") or Clientin["x-kunlun-token"]
+        self.x_tt_env = Clientinfox.get("data_set_ppeenv") or Clientin["x-tt-env"]
+        self.authorization = Clientinfox.get("data_set_auth") or Clientin["Authorization"]
+
+
+if __name__ == '__main__':
+    Clientinfox = {
+        "data_set_ailyappid": "123",
+        "data_set_appid": "456",
+        "data_set_appsecret": "789",
+        "data_set_cookie": "012",
+        "data_set_kunluntoken": "345",
+        "data_set_ppeenv": "678",
+        "data_set_auth": "901",
+    }
+    b=Clientassign(Clientinfox)
+
+
+

it_assistant/do_ai.py

@@ -0,0 +1,490 @@
+# -*- coding: utf-8 -*-
+import time
+from it_assistant.ailyapp_client import AilyLarkClient
+from it_assistant.lark_client import LarkdocsClient
+from it_assistant.intent_detail import *
+from it_assistant.openapi import *
+import datetime
+import copy
+import os
+import csv
+
+
+# 定义一个全局变量Client
+# Testsuitelink = "https://bytedance.larkoffice.com/sheets/ZVzfsw4rMhkMF6tjtxmc4BdSnMb"
+
+
+def do_ai_auto(Testk_data, clientinfo):
+    """
+    自动化执行AI测试用例
+    """
+    startAt = 0
+    try:
+        # 获取租户访问令牌
+        tenant_access_token = AilyLarkClient(clientinfo).get_tenant_access_token()
+        if not tenant_access_token:
+            raise ValueError("未能获取到有效的租户访问令牌")
+        # 判断Testsuitelink中是否包含https://
+        if "https://" in Testk_data:
+
+            # 通过文档链接获取spreadsheet_token
+            spreadsheet_token = Testk_data.split("/")[-1]
+            if not spreadsheet_token:
+                raise ValueError("未能从文档链接中提取到有效的spreadsheet_token")
+            # 读取表格用户输入
+            spreadsheet = LarkdocsClient().get_the_worksheet(spreadsheet_token)
+            if not spreadsheet:
+                raise ValueError("未能获取到有效的工作表数据")
+            for i in spreadsheet.sheets:
+                column_count = i.grid_properties.column_count
+                row_count = i.grid_properties.row_count
+                sheet_id = i.sheet_id
+                title = i.title
+                if title == "测试集":
+                    # 构建JSON字符串
+                    json_str = {"ranges": [sheet_id + "!A1:A" + str(row_count)]}
+                    # 获取纯文本内容
+                    test = LarkdocsClient().get_plaintextcontent(json_str, spreadsheet_token, sheet_id)
+                    test = json.loads(test)
+                    userinput = test['data']['value_ranges'][0]['values']
+                    print(f"表头为{userinput[0]}")
+                    for i in range(1, row_count):
+                        if userinput[i][0]:
+                            if startAt == 0:
+                                startAt = int(time.time())
+                            # 创建会话
+                            seseion_id = AilyLarkClient(clientinfo).create_ailysession(tenant_access_token)
+                            if not seseion_id:
+                                raise ValueError("未能成功创建会话")
+                            # 创建消息
+                            message_id = AilyLarkClient(clientinfo).create_ailysessionaily_message(tenant_access_token,
+                                                                                                   seseion_id,
+                                                                                                   userinput[i][0])
+                            if not message_id:
+                                raise ValueError("未能成功创建消息")
+                            # 创建运行实例
+                            runs = AilyLarkClient(clientinfo).create_ailysession_run(tenant_access_token, seseion_id)
+                            # 可不需等待运行实例创建完成
+                            # if not runs:
+                            #    raise ValueError("未能成功创建运行实例")
+                            time.sleep(1)
+                        else:
+                            return startAt, i
+                            break
+                    return startAt, row_count
+                    break
+        elif Testk_data[0].get('ext'):
+            num = 0
+            for i in Testk_data:
+                aa = i['ext']['input']
+                if startAt == 0:
+                    startAt = int(time.time())
+                # 创建会话
+                seseion_id = AilyLarkClient(clientinfo).create_ailysession(tenant_access_token)
+                if not seseion_id:
+                    raise ValueError("未能成功创建会话")
+                # 创建消息
+                message_id = AilyLarkClient(clientinfo).create_ailysessionaily_message(tenant_access_token, seseion_id,
+                                                                                       aa)
+                if not message_id:
+                    raise ValueError("未能成功创建消息")
+                # 创建运行实例
+                runs = AilyLarkClient(clientinfo).create_ailysession_run(tenant_access_token, seseion_id)
+                num = num + 1
+            return startAt, num
+    except KeyError as ke:
+        print(f"KeyError 发生: 数据中缺少必要的键，错误详情: {ke}")
+        return None, None
+    except json.JSONDecodeError as jde:
+        print(f"JSON 解析错误: {jde}")
+        return None, None
+    except ValueError as ve:
+        print(f"值错误: {ve}")
+        return None, None
+    except Exception as e:
+        print(f"发生未知错误: {e}")
+        return None, None
+
+
+def get_results(num,res):
+    """
+    获取结果
+    """
+    labels = []
+    a = len(res["body"]["Results"])
+    #判断res["body"]["Results"]的个数，如果大于num，则取num个值，小于num，则取res["body"]["Results"]的个数
+    if a > num:
+        num = num
+    else:
+        num = a
+    #提取res["body"]["Results"]的前num个值
+    res["body"]["Results"] = res["body"]["Results"][:num]
+    #遍历res["body"]["Results"]提取 name_zh、brand_zh、model_zh、specification_zh 并拼接
+    for i in range(0, num):
+        res["body"]["Results"][i]['Item']['sku_zh'] = res["body"]["Results"][i]['Item']['name_zh'] + " " + res["body"]["Results"][i]['Item']['brand_zh'] + " " + res["body"]["Results"][i]['Item']['model_zh'] + " " + res["body"]["Results"][i]['Item']['specification_zh']
+        label = {"label" :res["body"]["Results"][i]['Item']['sku_zh'], "score" : res["body"]["Results"][i]['Score']}
+        labels.append(copy.deepcopy(label))
+    return labels
+
+
+
+def do_waterlevellineres_list(res, info):
+    """
+    获取结果，并组装水位线info
+    """
+    if res == '':
+        info['label'] = [{'label': '', 'score': 0}, {'label': '', 'score': 0}]
+        info['rt'] = False
+        return info
+
+
+    # 判断res["body"]["Results"]不为空，空则：label0 label1 为空，label默认2级
+    if res["body"]["Results"]:
+        # 取["Results"]下前2个结果，若只有1个结果，label1为空
+        if len(res["body"]["Results"]) > 1:
+            info['label'] = [
+                {'label': res["body"]["Results"][0]['Item']['name_zh'],
+                 'score': res["body"]["Results"][0]['Score']},
+                {'label': res["body"]["Results"][1]['Item']['name_zh'],
+                 'score': res["body"]["Results"][1]['Score']}]
+
+        else:
+            info['label'] = [
+                {'label': res["body"]["Results"][0]['Item']['name_zh'],
+                 'score': res["body"]["Results"][0]['Score']},
+                {'label': '', 'score': 0}]
+        # 判断label0和label1是否为空，为空则：label默认2级
+        # 判断exp和label是否一致，一致则：rt=True，不一致则：rt=False
+        if info['exp'][0]['label'] == info['label'][0]['label'] and info['exp'][1]['label'] == info['label'][1][
+            'label'] and info['exp'][0]['score'] <= info['label'][0]['score'] and info['exp'][1]['score'] <= \
+                info['label'][1]['score']:
+            info['rt'] = True
+        else:
+            info['rt'] = False
+
+    else:
+        info['label'] = info['label'] = [{'label': '', 'score': 0}, {'label': '', 'score': 0}]
+        info['rt'] = False
+
+    return info
+
+
+def do_metricsevaluation_list(collections,data,score_threshold):
+    """
+    指标 评测
+    """
+    info = {
+        "input": {
+            "用户输入/userInput": "我要申请软件，名字叫：ai_xzh_all_restricted_software完全受限软件"
+        },
+        "output": {
+            "用户输入/output": "我要申请软件，名字叫：ai_xzh_all_restricted_software完全受限软件"
+        },
+        "rt": True,
+        "label": [{"label": "GUI 软件申请", "score": 0.6}, {"label": "软件申请", "score": 0.5}],
+        "exp": [{"label": "GUI 软件申请", " score": 0.9}, {"label": "软件申请", "score": 0.8 }],
+        "artificial": []
+    }
+    info_list = []
+    businessscenario = []
+    for i in collections:
+        for j in data:
+            if i['content'] == j['用户输入/userInput']:
+                info['input']['用户输入/userInput'] = i['ext']['output']
+                info['output']['用户输入/output'] = j['用户输入/userInput']
+                if i['ext']['BPO标注-AP-分发技能'] != '' and i['ext']['BPO标注-AP-分发技能'] in j['分发技能/skill'][0]:
+                    info['rt'] = True
+                    info['label'] = [{'label': j['分发技能/skill'][0], 'score': score_threshold}]
+                    info['exp'] = [{'label': j['分发技能/skill'][0], 'score': score_threshold}]
+                    info['artificial'] = info['exp']
+                else:
+                    info['rt'] = False
+                    info['label'] = [{'label': j['分发技能/skill'][0], 'score': score_threshold}]
+                    info['exp'] = [{'label': i['ext']['BPO标注-AP-分发技能'], 'score': score_threshold}]
+                    info['artificial'] = info['exp']
+                if i['ext']['BPO标注-AP-分发技能'] not in businessscenario:
+                    businessscenario.append(i['ext']['BPO标注-AP-分发技能'])
+        info_list.append(copy.deepcopy(info))
+    return info_list,businessscenario
+
+
+def do_scenereview_list(collections,data,score_threshold):
+    """
+    场景 评测  提取关键词
+    """
+    score_threshold=0.8
+    info = {
+        "input": {
+            "用户输入/userInput": "我要申请软件，名字叫：ai_xzh_all_restricted_software完全受限软件"
+        },
+        "output": {
+            "用户输入/output": "我要申请软件，名字叫：ai_xzh_all_restricted_software完全受限软件"
+        },
+        "rt": True,
+        "label": [{"label": "GUI 软件申请", "score": 0.6}, {"label": "软件申请", "score": 0.5}],
+        "exp": [{"label": "GUI 软件申请", " score": 0.9}, {"label": "软件申请", "score": 0.8 }],
+        "artificial": []
+    }
+    info_list = []
+    for i in collections:
+        for j in data:
+            if i['content'] == j['用户输入/userInput']:
+                info['input']['用户输入/userInput'] = i['ext']['output']
+                info['output']['用户输入/output'] = j['用户输入/userInput']
+                if i['ext']['BPO标注-AP-资产名称'] == j['llm关键词']:
+                    info['rt'] = True
+                    info['label'] = [{'label': str(j['llm关键词']), 'score': score_threshold}]
+                else:
+                    info['rt'] = False
+                    info['label'] = [{'label': str(i['ext']['BPO标注-AP-资产名称']), 'score': score_threshold}]
+                info['exp'] = [{'label': str(i['ext']['BPO标注-AP-资产名称']), 'score': score_threshold}]
+                info['artificial'] = info['exp']
+    info_list.append(copy.deepcopy(info))
+    return info_list
+
+def do_waterlevelline_autotest(collections, clientinfo, score_threshold):
+    """
+    水位线评测- 返回 符合报告模式的结果
+    """
+    keywprd= []
+    info_list = []
+    info = {
+        "input": {
+            "用户输入/userInput": "我要申请软件，名字叫：ai_xzh_all_restricted_software完全受限软件"
+        },
+        "output": {
+            "用户输入/output": "我要申请软件，名字叫：ai_xzh_all_restricted_software完全受限软件"
+        },
+        "rt": True,
+        "label": [{"label": "GUI 软件申请", "score": 0.6}, {"label": "软件申请", "score": 0.5}],
+        "exp": [{"label": "GUI 软件申请", " score": 0.9}, {"label": "软件申请", "score": 0.8
+                                                       }],
+        "artificial": []
+    }
+    a =0
+    for i in collections:
+        info['input']['用户输入/userInput'] = i['ext']['BPO标注-AP-资产名称']
+        info['output']['用户输入/output'] = i['ext']['BPO标注-AP-资产名称']
+        info['exp'] = [
+            {'label': i['ext']['BPO标注-AP-资产型号'], 'score': score_threshold},
+            {'label': i['ext'].get('BPO标注-AP-资产型号1', ''), 'score': score_threshold}]
+        info['artificial'] = info['exp']
+        if i['ext']['BPO标注-AP-资产名称']:
+            if "软件申请" in i['ext']['BPO标注-AP-分发技能']:
+                keywprd = software_asset_sku_structure(json.loads(i['ext']['BPO标注-AP-资产名称']))
+                res = json.loads(get_query_vector(keywprd, clientinfo))
+            if "设备/配件申请" in i['ext']['BPO标注-AP-分发技能']:
+                keywprd = equipmentrequest_structure(json.loads(i['ext']['BPO标注-AP-资产名称']), i['ext']['asset_type'])
+                res0 = json.loads(get_query_vector(keywprd, clientinfo))
+                res = get_by_AssetModelBizTypes(keywprd,res0)
+            if "设备/配件退还" in i['ext']['BPO标注-AP-分发技能']:
+                keywprd = equipmentreturn_structure0(json.loads(i['ext']['BPO标注-AP-资产名称']), i['ext']['asset_type'])
+                res0 = json.loads(get_query_vector(keywprd, clientinfo))
+                res = get_by_AssetModelBizTypes(keywprd, res0)
+        else:
+            res = ""
+        infoout = do_waterlevellineres_list(res, info)
+        info_list.append(copy.deepcopy(infoout))
+        a = a+1
+        print("这是"+str(a))
+        bbb = a
+    return info_list
+
+
+def get_conversationlogs1(startAt):
+    """
+    对话ID 技能分发 用户输入
+    res_data = {
+            'intentID': 7485259579248705537,
+            'skillLabels': ["GUI 设备/配件申请"],
+            'userInput': "我要申请一个鼠标",
+
+         }
+         """
+    data = webapiClient().get_intent_detail_list(startAt)
+
+
+def get_conversationlogs(startAt, pageSize=10):
+    """
+    对话ID 技能分发 用户输入
+    res_data = {
+            'intentID': 7485259579248705537,
+            'skillLabels': ["GUI 设备/配件申请"],
+            'userInput': "我要申请一个鼠标",
+
+         }
+    """
+    try:
+        # 之前提到形参 'pageSize' 未填，这里假设默认值为 10，你可按需修改
+        data = webapiClient().get_intent_detail_list(startAt, pageSize=10)
+        return data
+    except KeyError as ke:
+        print(f"KeyError 发生: 数据中缺少必要的键，错误详情: {ke}")
+        return None
+    except IndexError as ie:
+        print(f"IndexError 发生: 索引超出范围，错误详情: {ie}")
+        return None
+    except Exception as e:
+        print(f"发生未知错误: {e}")
+        return None
+
+
+def write_reslut(data, Testsuitelink, title):
+    """
+    写入表格
+    """
+    try:
+        # 解析 spreadsheet_token
+        spreadsheet_token = Testsuitelink.split("/")[-1]
+
+        # 生成新工作表名称
+        new_sheet_title = f"{title}{datetime.datetime.now().strftime('%Y_%m_%d_%H_%M_%S')}"
+        sheetinfo = {"index": 0, "title": new_sheet_title}
+
+        # 创建新工作表
+        spreadsheet0 = LarkdocsClient().createsheets(spreadsheet_token, sheetinfo)
+        sheet_id = spreadsheet0['sheet_id']
+
+        # 准备表头数据
+        headers = list(data[0].keys())
+        header_data = [
+            {
+                "range": f"{sheet_id}!{chr(ord('A') + col)}1:{chr(ord('A') + col)}1",
+                "values": [[[{"text": {"text": header}, "type": "text"}]]]
+            }
+            for col, header in enumerate(headers)
+        ]
+
+        # 写入表头
+        LarkdocsClient().writesheets(spreadsheet_token, sheet_id, {"value_ranges": header_data})
+
+        # 写入数据
+        for row, row_data in enumerate(data, start=1):
+            row_values = [
+                {
+                    "range": f"{sheet_id}!{chr(ord('A') + col)}{row + 1}:{chr(ord('A') + col)}{row + 1}",
+                    "values": [[[{"text": {"text": str(row_data[header])}, "type": "text"}]]]
+                }
+                for col, header in enumerate(headers)
+            ]
+            LarkdocsClient().writesheets(spreadsheet_token, sheet_id, {"value_ranges": row_values})
+
+        return True
+    except KeyError as ke:
+        print(f"KeyError 发生: 数据中缺少必要的键，错误详情: {ke}")
+        return False
+    except IndexError as ie:
+        print(f"IndexError 发生: 索引超出范围，错误详情: {ie}")
+        return False
+    except Exception as e:
+        print(f"发生未知错误: {e}")
+        return False
+
+
+def write_excletolist(data_name):
+    """
+    1. 读取本地表格
+    2. 将表格内容拼接为text
+    """
+    try:
+        # 查看当前工作目录
+        print(f"当前工作目录: {os.getcwd()}")
+        # /Users/bytedance/it_assistant/it_assistant/accessory.csv
+        # 构建文件路径
+        file_path = f'data/{data_name}.csv'
+        Candidates = []
+        Candidate = {
+            "Score": 0,
+            "Text": "IOS手机",
+            "Attrs": {"id": "", "type": ""}}
+        text = ""
+        with open(file_path, 'r', encoding='utf-8', newline='') as file:
+            reader = csv.reader(file)
+            headers = next(reader)  # 读取表头
+            for header in headers:
+                text += f"{header}: "
+            text = text.rstrip(': ') + '\n'
+
+            for row in reader:
+                textout = ""
+                textout += ', '.join(row)
+                Candidate['Text'] = textout
+                Candidates.append(copy.deepcopy(Candidate))
+        return Candidates
+    except FileNotFoundError:
+        print(f"未找到文件: {file_path}")
+        return None
+    except Exception as e:
+        print(f"发生未知错误: {e}")
+        return None
+
+
+
+
+if __name__ == '__main__':
+    clientinfo={
+        "Authorization": "Basic cm40cmFpdTRwenY1cGlsYTo2bWhvOXV3ZXFrOHZpbDllcXRxMHZ1YmVnc2xjeXBucg==",
+        "x_tt_env": "ppe_cn_env_self_test_feat_cr_a"
+    }
+
+
+    # 读取 CSV 文件
+    with open('/Users/bytedance/it_assistant/test.csv', mode='r', encoding='utf-8') as file:
+        csv_reader = csv.reader(file)
+        for row in csv_reader:
+
+            print(row)
+            parm = {
+                "From": 0,
+                "Size": 10,
+                "MinScore": 0.1,
+                "AssetModelFieldsWithOr": [
+                    {
+                        "FieldName": "vec_name",
+                        "FieldType": "knn",
+                        "QueryValue": row
+                    },
+                    {
+                        "FieldName": "vec_brand",
+                        "FieldType": "knn",
+                        "QueryValue": row
+                    },
+                    {
+                        "FieldName": "vec_specification",
+                        "FieldType": "knn",
+                        "QueryValue": row
+                    },
+                    {
+                        "FieldName": "vec_model_name",
+                        "FieldType": "knn",
+                        "QueryValue": row
+                    }
+                ],
+                "AssetModelBizTypes": ["asset_sku"]
+            }
+            # 每一行是一个列表
+
+            info={
+                "score":0.8,
+                "itams": ""
+            }
+            info_list = []
+
+            cccc = get_query_vector(parm,clientinfo)
+            bbbb = json.loads(cccc)['body']['Results']
+            bbbb[0]['Score']
+            #将bbbb[0]['Score']写入test.csv的score0列
+
+
+
+            for i in bbbb:
+                info['score']=i['Score']/4
+                info['itams']=i['Item']['brand_zh']+i['Item']['model_zh']+i['Item']['specification_zh']
+
+
+
+
+
+
+
+