import erniebot
import numpy as np

erniebot.api_type = "aistudio"
erniebot.access_token = "ff1531c8c0f429f92adbc2eaed2e23bfb5349e0f"

model = 'ernie-3.5'
# messages = [{'role': 'user', 'content': "请问你能以《你好，世界》为题，写一首现代诗吗？"}]
# first_response = erniebot.ChatCompletion.create(
#     model=model,
#     messages=messages,
# )
# print(first_response.get_result())
#
# messages.append(first_response.to_message())
# messages.append({'role': 'user', 'content': "谢谢你！请问你能把这首诗改写成七言绝句吗？"})
#
# second_response = erniebot.ChatCompletion.create(
#     model=model,
#     messages=messages,
# )
# print(second_response.get_result())

response = erniebot.Embedding.create(
    model='ernie-text-embedding',
    input=[
        "我是百度公司开发的人工智能语言模型，我的中文名是文心一言，英文名是ERNIE-Bot。",
        "“喜鹊报喜、乌鸦报丧”有没有科学道理,“喜鹊报喜、乌鸦报丧”有没有科学道理喜鹊是一种较为常见的鸟类，在中国，除内蒙古和青藏高原的部分地区外，都可以见到喜鹊的踪影。“喜上眉梢”是中国国画中常见的主题，其中的“喜”便是指喜鹊，这也从侧面说明了中国人对喜鹊的喜爱之情。在中国文化中，喜鹊被认为是一种报喜鸟，是好事的征兆，而作为其近亲的乌鸦，则是厄运的象征，就如俗语所说的，“乌鸦叫，坏事到”。“喜鹊报喜、乌鸦报丧”反映了人们对这两种动物截然不同的看法和态度，那么这样的说法究竟有没有科学依据呢？喜鹊喜鹊和乌鸦都是雀形目鸦科的鸟类，属于鸣禽，鸣管较为发达。它们的叫声都谈不上动听，但对于它们自己及同类，却是一种非常重要的交流方式。在繁殖季节的清晨，喜鹊或者乌鸦们都会在自己领域范围内最高的树枝或者建筑物顶端鸣叫，宣示“主权”，同时向自己的配偶献殷勤。这是它们最常见的行为之一，它们的鸣叫，显然与人类的喜丧没什么关系。乌鸦被认为会报丧，可能与它们食腐的习性有关。在旧时的欧亚大陆，乌鸦常以死亡的动物尸体为食，甚至以战场上人的尸体为食，因为它们能够闻到腐肉的气味，并通过叫声召唤自己的同伴前来取食。其实，人类对待动物的态度，往往具有地域性，大都由本民族的文化特质所决定。如在北美土著居民文化中，渡鸦有着积极的作用，被视为英雄的化身。在农业社会，人类的死亡率很高，乌鸦可以闻到濒死之人身体发出的味道，便会飞到人类的住所附近鸣叫等待取食。所以，对于那时的人类来说，乌鸦的出现和鸣叫确实有可能是报丧。实验场：观察喜鹊筑巢喜鹊喜欢在高树上筑巢，但在一些城市周边，常缺乏可供筑巢的大树，那么，它们会在哪里筑巢呢？观察你们家和校园周围的喜鹊，看看它们都选择在什么地方筑巢，巢有多大，什么形状，筑巢的材料又是什么。"
    ])

for embedding in response.get_result():
    embedding = np.array(embedding)
    print(embedding)