赞
踩
近日,百度宣布其两款主力模型 ENIRE Speed、ENIRE Lite 可以免费使用。试了一下怎么程序调用。
按照百度的说明文档,可以直接HTTPS的方式调用API,也可以使用百度提供的SDK。SDK提供了多个语言版本,但是python版SDK支持的功能最全。本次测试使用python版SDK。
pip install qianfan
写入你的Access Key和Secret Key。如果不使用.env文件,也可以在环境变量中设置好下面的两个变量。或者在程序中直接用参数方式调用。
- QIANFAN_ACCESS_KEY=abcd
- QIANFAN_SECRET_KEY=efgh
- import qianfan
-
- chat_comp = qianfan.ChatCompletion(model='ERNIE-Speed-128K')
- messages = []
- user_message = '你好,你是谁?'
- print('你说:', user_message)
- while user_message != '结束':
- messages.append({'role': 'user','content': user_message})
- resp = chat_comp.do(messages, top_p=0.8, temperature=0.9, penalty_score=1.0, system='你是abcd公司制作的AI助手')
- assistant_message = resp.body['result']
- messages.append({'role': 'assistant','content': assistant_message})
- print('机器人说:', assistant_message)
-
- print('你说:', end='')
- user_message = input()
-
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。