LongArticlesJob/applications/kimi_api.py

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


def kimi_process(prompt):
    """
    Ask Kimi for information
    :param prompt: tiny text
    :return: "{}"
    """
    client = OpenAI(
        api_key='sk-5DqYCa88kche6nwIWjLE1p4oMm8nXrR9kQMKbBolNAWERu7q',
        base_url="https://api.moonshot.cn/v1"
    )
    chat_completion = client.chat.completions.create(
        messages=[
            {
                "role": "user",
                "content": prompt,
            }
        ],
        model="moonshot-v1-8k",
    )
    response = chat_completion.choices[0].message.content
    return response