AI入门示例

发布于:2025-05-31 ⋅ 阅读:(19) ⋅ 点赞:(0)

市面上有很多AI大模型,这里以 智谱的大模型 为示例

1.先要注册智谱AI开放平台

2.注册成功后,会赠送3个月的免费额度,如下

3.然后去控制台,创建一个API KEY

4.接着就可以开始写代码了

提前导入包: openai

示例1:

from openai import OpenAI

client = OpenAI(
    api_key="***", #这里写你自己的KEY
    base_url="https://open.bigmodel.cn/api/paas/v4/"
)

completion = client.chat.completions.create(
    model="glm-4-air-0111",
    messages=[
        {"role": "system", "content": "你是一个幽默的小说作家"},
        {"role": "user",
         "content": "请你作为故事大王,写一篇短篇童话故事,故事的主题是要永远保持一颗善良的心,要能够激发儿童的学习兴趣和想象力,同时也能够帮助儿童更好地理解和接受故事中所蕴含的道理和价值观。"}
    ],
    top_p=0.7,
    temperature=0.9
)

print(completion.choices[0].message.content)

注意:这里 top_p,temperature 参数的值是在0-1之间

第1次运行有点慢,运行结果:

示例2:

from zhipuai import ZhipuAI

client = ZhipuAI(api_key="***")  # 请填写您自己的APIKey

response = client.chat.completions.create(
    model="glm-4-air-0111",  # 请填写您要调用的模型名称
    messages=[
        {"role": "user", "content": "作为一名营销专家,请为我的产品创作一个吸引人的口号"},
        {"role": "assistant", "content": "当然,要创作一个吸引人的口号,请告诉我一些关于您产品的信息"},
        {"role": "user", "content": "智谱AI开放平台"},
        {"role": "assistant", "content": "点燃未来,智谱AI绘制无限,让创新触手可及!"},
        {"role": "user", "content": "创作一个更精准且吸引人的口号"}
    ],
    temperature=0.9,
    top_p=0.9
)
print(response.choices[0].message.content)

运行结果:

示例3:

如果你有deepseek的API KEY,也可以这样。

这里使用 阿里云百炼的API,(注册https://bailian.console.aliyun.com/会送deepseek-r1的免费token)

import os
from openai import OpenAI

client = OpenAI(
    # 若没有配置环境变量,请用百炼API Key将下行替换为:api_key="sk-xxx",
    api_key=os.getenv("DASHSCOPE_API_KEY"),
    # 如何获取API Key:https://help.aliyun.com/zh/model-studio/developer-reference/get-api-key
    base_url="https://dashscope.aliyuncs.com/compatible-mode/v1"
)

completion = client.chat.completions.create(
    model="deepseek-r1",  # 此处以 deepseek-r1 为例,可按需更换模型名称。
    messages=[
        {'role': 'user', 'content': '10.9和11.11谁大'}
    ]
)

# 通过reasoning_content字段打印思考过程
print("下面是思考过程:")
print(completion.choices[0].message.reasoning_content)

# 通过content字段打印最终答案
print("问题的最终答案:")
print(completion.choices[0].message.content)

需要提前在环境中配置DASHSCOPE_API_KEY,这个KEY就是阿里云百炼中申请的API key


网站公告

今日签到

点亮在社区的每一天
去签到