Skip to content

支持设置多个关键字提交去chatgpt #1

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
widyhu2020 wants to merge 2 commits into
base: main
Choose a base branch
from
Open

支持设置多个关键字提交去chatgpt #1

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
7052b5f
支持设置多个关键字提交去chatgpt
widyhu2020 Mar 25, 2023
d74a0c4
说明不能使用的参数
widyhu2020 Mar 25, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ Play ChatGPT with Xiaomi AI Speaker
7. 因为现在必须指定 conversation_id 和 parent_id 来持续对话，会自动建一个新的 conversation
8. 如果上面不可用，可以尝试用手机抓包，https://userprofile.mina.mi.com/device_profile/v2/conversation 找到 cookie 利用 --cookie '${cookie}' cookie 别忘了用单引号包裹
9. 默认用目前 ubus, 如果你的设备不支持 ubus 可以使用 --use_command 来使用 command 来 tts
10. 使用 --mute_xiaoai 选项，可以让小爱不回答，但会频繁请求，玩一下可以使用，不建议一直用
10. 使用 --mute_xiaoai 选项，可以让小爱不回答，但会频繁请求，玩一下可以使用，不建议一直用（该参数有bug，无法使用）
11. 使用 --account ‘${account}’ --password ‘${password}’ 可以不进行步骤 2
12. 如果有能力可以自行替换唤醒词，也可以去掉唤醒词，源码在 https://github.com/yihong0618/xiaogpt/blob/main/xiaogpt.py#L32
13. 可以使用 gpt-3 的 api 那样可以更流畅的对话，速度快, 请 google 如何用 openai api, 命令 --use_gpt3
Expand All @@ -46,9 +46,9 @@ python3 xiaogpt.py --hardware LX06 --conversation_id="xxxxxxxx";
python3 xiaogpt.py --hardware LX06 --cookie ${cookie};
# 如果你想直接输入账号密码
python3 xiaogpt.py --hardware LX06 --account ${your_xiaomi_account} --password ${your_password};
# 如果你想 mute 小米的回答
# 如果你想 mute 小米的回答（--mute_xiaoai该参数不能使用）
python3 xiaogpt.py --hardware LX06 --mute_xiaoai
# 如果你想使用 gpt3 ai
# 如果你想使用 gpt3 ai （--mute_xiaoai该参数不能使用）
export OPENAI_API_KEY=${your_api_key}
python3 xiaogpt.py --hardware LX06 --mute_xiaoai --use_gpt3
# 如果你想用 chatgpt api
Expand Down
77 changes: 43 additions & 34 deletions xiaogpt.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -34,8 +34,8 @@
MI_USER = ""
MI_PASS = ""
OPENAI_API_KEY = ""
KEY_WORD = "帮我"
PROMPT = "请用100字以内回答"
KEY_WORD = ["请问", "帮我", "给我", "你觉得", "告诉我"]

# simulate the response from xiaoai server by type the input.
CLI_INTERACTIVE_MODE = False
Expand Down Expand Up @@ -304,7 +304,9 @@ async def ask_chatgpt(self, query):
return ""

async def get_if_xiaoai_is_playing(self):
# 这里有bug，后面的库并没有这个方法
playing_info = await self.mina_service.player_get_status(self.device_id)
print(playing_info)
# WTF xiaomi api
is_playing = (
json.loads(playing_info.get("data", {}).get("info", "{}")).get("status", -1)
Expand All @@ -313,13 +315,50 @@ async def get_if_xiaoai_is_playing(self):
return is_playing

async def stop_if_xiaoai_is_playing(self):
# 这里有bug，后面的库并没有这个方法
is_playing = await self.get_if_xiaoai_is_playing()
if is_playing:
# stop it
await self.mina_service.player_pause(self.device_id)

async def do_chat_with_xiaoai(self, last_record, query, key):
if query.find(key) == 0:
# only mute when your clause start's with the keyword
if self.this_mute_xiaoai:
await self.stop_if_xiaoai_is_playing()
self.this_mute_xiaoai = False
# drop 帮我回答
# query = query.replace(KEY_WORD, "")
query = f"{query}，{PROMPT}"
# waiting for xiaoai speaker done
if not self.mute_xiaoai:
await asyncio.sleep(4)
await self.do_tts("正在问GPT请耐心等待")
try:
print(
"以下是小爱的回答: ",
last_record.get("answers")[0]
.get("tts", {})
.get("text"),
)
except Exception as e:
print(e)
print("小爱没回")
message = await self.ask_gpt(query)
# tts to xiaoai with ChatGPT answer
print("以下是GPT的回答: " + message)
await self.do_tts(message)
if self.mute_xiaoai:
while 1:
is_playing = await self.get_if_xiaoai_is_playing()
time.sleep(2)
if not is_playing:
break
self.this_mute_xiaoai = True


async def run_forever(self):
print(f"Running xiaogpt now, 用`{KEY_WORD}`开头来提问")
print(f"Running xiaogpt now, 用`请问`开头来提问")
async with ClientSession() as session:
await self.init_all_data(session)
while 1:
Expand All @@ -343,38 +382,8 @@ async def run_forever(self):
if new_timestamp > self.last_timestamp:
self.last_timestamp = new_timestamp
query = last_record.get("query", "")
if query.find(KEY_WORD) != -1:
# only mute when your clause start's with the keyword
if self.this_mute_xiaoai:
await self.stop_if_xiaoai_is_playing()
self.this_mute_xiaoai = False
# drop 帮我回答
query = query.replace(KEY_WORD, "")
query = f"{query}，{PROMPT}"
# waiting for xiaoai speaker done
if not self.mute_xiaoai:
await asyncio.sleep(4)
await self.do_tts("正在问GPT请耐心等待")
try:
print(
"以下是小爱的回答: ",
last_record.get("answers")[0]
.get("tts", {})
.get("text"),
)
except:
print("小爱没回")
message = await self.ask_gpt(query)
# tts to xiaoai with ChatGPT answer
print("以下是GPT的回答: " + message)
await self.do_tts(message)
if self.mute_xiaoai:
while 1:
is_playing = await self.get_if_xiaoai_is_playing()
time.sleep(2)
if not is_playing:
break
self.this_mute_xiaoai = True
for key in KEY_WORD:
await self.do_chat_with_xiaoai(last_record, query, key)
else:
if self.verbose:
print("No new xiao ai record")
Expand Down