ERNIE Bot SDK基础教程

安装EB SDK

使用pip可以快速安装EB SDK

!pip install erniebot=0.3.1

认证鉴权

import erniebot

erniebot.api_type = 'aistudio'
erniebot.access_token = "{YOUR-ACCESS-TOKEN}"

EB SDK Hello-World

response = erniebot.ChatCompletion.create(
    model='ernie-bot',
    messages=[{'role': 'user', 'content': "请对我说“你好，世界！”"}],
)
print(response.result)

多轮对话

文心一言大模型具备强大的上下文理解能力，在我们发送新的消息时，模型能够联系历史消息进行回复。

model = 'ernie-bot'
messages = [{'role': 'user', 'content': "请问你能以《你好，世界》为题，写一首现代诗吗？"}]
first_response = erniebot.ChatCompletion.create(
    model=model,
    messages=messages,
)
print(first_response.result)

messages.append({'role': 'assistant', 'content': first_response.result})
messages.append({'role': 'user', 'content': "谢谢你！请问你能把这首诗改写成七言绝句吗？"})

second_response = erniebot.ChatCompletion.create(
    model=model,
    messages=messages,
)
print(second_response.result)

语义向量（Embedding）

语义向量功能将文本转化为用数值表示的向量形式，从而以紧凑高效的方式编码文本。

import numpy as np

response = erniebot.Embedding.create(
    model='ernie-text-embedding',
    input=[
        "我是百度公司开发的人工智能语言模型，我的中文名是文心一言，英文名是ERNIE-Bot。",
        "2018年深圳市各区GDP"
    ])

for emb_res in response.data:
    embedding = np.array(emb_res['embedding'])
    print(embedding)