更新kimi_task

applications/functions/kimi.py

@@ -30,9 +30,23 @@ class KimiServer(object):
         try:
             kimi_title = await cls.kimi_title(title, ai_model=deep_seek)
             # 判断kimi 标题是否安全
-            title_score = await cls.get_kimi_title_safe_score(kimi_title)
+            title_score = await cls.get_kimi_title_safe_score(kimi_title, ai_model=moon_shot)
+            kimi_green_title = None
             if int(title_score) > safe_score:
-                kimi_title = await cls.make_kimi_title_safer(title)
+                kimi_green_title = await cls.make_kimi_title_safer(title, ai_model=moon_shot)
+                
+            logging(
+                code="1005",
+                info="generate_title",
+                data={
+                    "ori_title": title,
+                    "kimi_title": kimi_title,
+                    "title_score": title_score,
+                    "new_kimi_title": kimi_green_title
+                },
+                trace_id=trace_id
+            )
+            kimi_title = kimi_green_title if kimi_green_title else kimi_title
 
         except Exception as e:
             logging(
@@ -220,7 +234,7 @@ class KimiServer(object):
         return content
 
     @classmethod
-    async def get_kimi_title_safe_score(cls, kimi_title):
+    async def get_kimi_title_safe_score(cls, kimi_title, ai_model):
         """
         获取kimi_title的安全分数
         """
@@ -260,8 +274,8 @@ class KimiServer(object):
         输入的标题是： {kimi_title}
         """
         client = OpenAI(
-            api_key='sk-5DqYCa88kche6nwIWjLE1p4oMm8nXrR9kQMKbBolNAWERu7q',
-            base_url="https://api.moonshot.cn/v1"
+            api_key=ai_model['api_key'],
+            base_url=ai_model['base_url']
         )
         chat_completion = client.chat.completions.create(
             messages=[
@@ -270,13 +284,13 @@ class KimiServer(object):
                     "content": prompt
                 }
             ],
-            model="moonshot-v1-32k"
+            model=ai_model['model']
         )
         response = chat_completion.choices[0].message.content
         return response
 
     @classmethod
-    async def make_kimi_title_safer(cls, ori_title):
+    async def make_kimi_title_safer(cls, ori_title, ai_model):
         """
         安全化 kimi title
         """
@@ -295,8 +309,8 @@ class KimiServer(object):
             输入的标题是: 
         """
         client = OpenAI(
-            api_key='sk-5DqYCa88kche6nwIWjLE1p4oMm8nXrR9kQMKbBolNAWERu7q',
-            base_url="https://api.moonshot.cn/v1"
+            api_key=ai_model['api_key'],
+            base_url=ai_model['base_url']
         )
         chat_completion = client.chat.completions.create(
             messages=[
@@ -305,7 +319,7 @@ class KimiServer(object):
                     "content": prompt + ori_title,
                 }
             ],
-            model="moonshot-v1-32k",
+            model=ai_model['model'],
             response_format={"type": "json_object"}
         )
         response = chat_completion.choices[0].message.content