VideoToArticles/applications/ai.py

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


def kimi_ai(prompt):
    """
    kimi extract text
    :param prompt:
    :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.replace('```json', '').replace('```', '')
    return response


def metaSo(prompt, mode="深入"):
    """
    meta
    :param prompt:
    :param mode:
    :return:
    """
    url = "http://8.217.190.241:8888/crawler/metaso/meta_ai"
    body = {
        "prompt": prompt,
        "mode": mode
    }
    header = {
        "Content-Type": "application/json"
    }
    response = requests.post(url=url, json=body, headers=header)
    return response.json()['data']