title_with_video/applications/functions/ask_kimi.py

"""
@author: luojunhui
"""
import json
from openai import OpenAI


def ask_kimi(question):
    """
    Ask Kimi for information
    :param question: tiny text
    :return: "{}"
    """
    single_title_prompt = """
        我会给你一个视频标题，需要你帮我用你所学的知识来帮我分析出以下信息，信息我都写到 json 里面了
        {
            "key_words": [],  # 返回三个关键词
            "search_keys": [], # 标题可能的搜索关键词，返回 3 个
            "extra_keys": [], # 关心这个视频的用户还会关心哪些关键词， 返回 3 个
            "theme": 标题的主题， 用一个词概括
        }
        只需要返回一个 json，key 和上面的一样，
        我给你的标题是: 
        """
    client = OpenAI(
        api_key='sk-tz1VaKqksTzk0F8HxlU4YVGwj7oa1g0c0puGNUZrdn9MDtzm',
        base_url="https://api.moonshot.cn/v1"
    )
    chat_completion = client.chat.completions.create(
        messages=[
            {
                "role": "user",
                "content": single_title_prompt + question,
            }
        ],
        model="moonshot-v1-8k",
    )
    response = chat_completion.choices[0].message.content.replace('```json', '').replace('```', '')
    try:
        response = json.loads(response)
        return response
    except:
        return {}