110 lines
4.2 KiB
Python
110 lines
4.2 KiB
Python
import requests as r
|
||
import json
|
||
|
||
api_key = '9J5qFTYr6wPRxkoVoXycnoWf'
|
||
secret_key = 'Xa6eJelStx5i7Ft3qQH0NAT6AvOkqhkH'
|
||
|
||
def get_access_token():
|
||
headers = {
|
||
'Content-Type': 'application/json',
|
||
'Accept': 'application/json'
|
||
}
|
||
|
||
url = f'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id={api_key}&client_secret={secret_key}'
|
||
payload = json.dumps("")
|
||
|
||
res = r.post(
|
||
url=url,
|
||
data=payload,
|
||
headers=headers
|
||
)
|
||
|
||
resJson = res.json()
|
||
access_token = resJson.get('access_token')
|
||
assert isinstance(access_token, str), 'access_token 获取失败，详细信息' + str(resJson)
|
||
return access_token
|
||
|
||
access_token = get_access_token()
|
||
|
||
text = open('./template.txt', 'r', encoding='utf-8').read()
|
||
t2 = open('./t2.txt', 'r', encoding='utf-8').read()
|
||
|
||
payload = json.dumps({
|
||
# "messages": [
|
||
# {
|
||
# "role": "user",
|
||
# "content": text
|
||
# },
|
||
# {
|
||
# "role": "assistant",
|
||
# "content": "{ id: 3 }"
|
||
# },
|
||
# {
|
||
# "role": "user",
|
||
# "content": "Message: 大佬们，为啥我的digital ide启动之后所有功能都没启动捏？我配置了property文件，然后插件的vivado路经和modelsim路经都加上了\nIntent: "
|
||
# },
|
||
# {
|
||
# "role": "assistant",
|
||
# "content": "{ id: 0 }"
|
||
# },
|
||
# {
|
||
# "role": "user",
|
||
# "content": "话说digital-ide打开大的verilog卡死了\nIntent: "
|
||
# },
|
||
# {
|
||
# "role": "assistant",
|
||
# "content": "{ id: 1 }"
|
||
# },
|
||
# {
|
||
# 'role': 'user',
|
||
# "content": "请问一下，第一次点击对文件仿真可以出波形文件，再次点击的时候就会提示unknown module type了。是哪个配置没配置好？\nIntent: "
|
||
# },
|
||
# ]
|
||
'messages': [
|
||
# {
|
||
# 'role': 'user',
|
||
# 'content': 'Label a users message from a conversation with an intent. Reply ONLY with the name of the intent.\nThe intent should be one of the following:\n- 1\n- 2\n- 3\n- 4\nMessage: surface了解一下？\nIntent: { id: 4 }\nMessage: Metals一开直接报错\nIntent: { id: 4 }\nMessage: 大佬们，为啥我的digital ide启动之后所有功能都没启动捏？我配置了property文件，然后插件的vivado路经和modelsim路经都加上了\nIntent: { id: 1 }\nMessage: 请问 property.json 如何配置？\nIntent: { id: 1 }\nMessage: 请问一下，第一次点击对文件仿真可以出波形文件，再次点击的时候就会提示unknown module type了。是哪个配置没配置好？\nIntent: '
|
||
# },
|
||
# {
|
||
# 'role': 'assistant',
|
||
# 'content': '{id : 1}'
|
||
# },
|
||
# {
|
||
# 'role': 'user',
|
||
# 'content': 'Message: 话说digital-ide打开大的verilog卡死了\nIntent: { id: 2 }\nMessage: 帮我上传一下这份数据\nIntent: { id: 3 }\nMessage: 我的自动补全无法使用，是不是有bug？\nIntent: { id: 2 }\nMessage: 这群要被chisel夺舍了吗\nIntent: '
|
||
# },
|
||
# {
|
||
# 'role': 'assistant',
|
||
# 'content': '{id : 4}'
|
||
# },
|
||
{
|
||
"role": "user",
|
||
"content": "如何解决 digital ide 无法载入配置文件的问题？\nIntent: "
|
||
}
|
||
]
|
||
})
|
||
|
||
headers = {
|
||
'Content-Type': 'application/json'
|
||
}
|
||
|
||
url = 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie-lite-8k?access_token=' + access_token
|
||
res = r.post(url, headers=headers, data=payload)
|
||
print(res.json())
|
||
# print(res.json()['result'])
|
||
|
||
# cache = []
|
||
|
||
# for line in res.iter_lines():
|
||
# line_text: str = line.decode('UTF-8')
|
||
# if line_text.startswith('data:'):
|
||
# iter_json = json.loads(line_text.lstrip('data: '))
|
||
# result: str = iter_json['result']
|
||
# cache.append(result)
|
||
# if result.endswith('。') or result.endswith('.'):
|
||
# sentence = ''.join(cache).strip()
|
||
# print(sentence)
|
||
# cache.clear()
|
||
|
||
# if len(cache) > 0:
|
||
# print(''.join(cache).strip()) |