cdryzun/tg_bot_collections
1
0
Fork 0
mirror of https://github.com/cdryzun/tg_bot_collections.git synced 2025-11-04 16:56:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: make gemini safe to None

Browse Source 
Signed-off-by: yihong0618 <zouzou0208@gmail.com>
This commit is contained in:
yihong0618
2024-03-17 18:30:55 +08:00
parent 9e5d5db767
commit 2952b81521
2 changed files with 4 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
handlers/gemini.py
View File

@ -22,16 +22,10 @@ generation_config = {
} }
safety_settings = [ safety_settings = [
{"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"}, {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
{"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_MEDIUM_AND_ABOVE"}, {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
{ {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
"threshold": "BLOCK_MEDIUM_AND_ABOVE",
},
{
"category": "HARM_CATEGORY_DANGEROUS_CONTENT",
"threshold": "BLOCK_MEDIUM_AND_ABOVE",
},
] ]
# Global history cache # Global history cache

2
handlers/yi.py
View File

@ -53,7 +53,6 @@ def yi_handler(message: Message, bot: TeleBot) -> None:
player_message.pop() player_message.pop()
r = client.chat.completions.create(messages=player_message, model=YI_MODEL) r = client.chat.completions.create(messages=player_message, model=YI_MODEL)
content = r.choices[0].message.content.encode("utf8").decode() content = r.choices[0].message.content.encode("utf8").decode()
print(content)
if not content: if not content:
yi_reply_text = "yi did not answer." yi_reply_text = "yi did not answer."
player_message.pop() player_message.pop()
@ -140,7 +139,6 @@ def yi_photo_handler(message: Message, bot: TeleBot) -> None:
headers=headers, headers=headers,
json=payload, json=payload,
).json() ).json()
print(response)
try: try:
text = response["choices"][0]["message"]["content"].encode("utf8").decode() text = response["choices"][0]["message"]["content"].encode("utf8").decode()
bot.reply_to(message, "yi vision answer:\n" + text) bot.reply_to(message, "yi vision answer:\n" + text)