常见大模型API python调用一览

1. openai

from openai import OpenAI

def chatgpt_api(query):

    client = OpenAI(
        base_url='https://api.openai-proxy.live/v1',
        api_key='sk-xxxxxxxxxxxxxxxxxxx',
    )

    chat_completion = client.chat.completions.create(
        messages=[
            {
                "role": "user",
                "content":f"{query}",
            }
        ],
        model="gpt-3.5-turbo",
    )
    return chat_completion.choices[0].message.content
print(chatgpt_api('你是谁'))

2. chatglm

import requests

def chatglm_api(query):
    url = "https://api.siliconflow.cn/v1/chat/completions"

    payload = {
        "model": "THUDM/glm-4-9b-chat",
        "messages": [
            {
                "role": "user",
                "content": f"{query}"
            }
        ]
    }
    headers = {
        "accept": "application/json",
        "content-type": "application/json",
        "authorization": "Bearer sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
    }

    response = r