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

fix: file not close

Browse Source 
Signed-off-by: yihong0618 <zouzou0208@gmail.com>
This commit is contained in:
yihong0618 2024-03-05 17:04:24 +08:00
parent 66e3160f17
commit 6f6771e1e8
2 changed files with 27 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
handlers/claude.py
View File

@ -201,7 +201,9 @@ def claude_photo_handler(message: Message, bot: TeleBot) -> None:
with open("claude_temp.jpg", "wb") as temp_file: with open("claude_temp.jpg", "wb") as temp_file:
temp_file.write(downloaded_file) temp_file.write(downloaded_file)
f = Path("claude_temp.jpg")
try: try:
with f:
r = client.messages.create( r = client.messages.create(
max_tokens=1024, max_tokens=1024,
messages=[ messages=[
@ -217,13 +219,13 @@ def claude_photo_handler(message: Message, bot: TeleBot) -> None:
"source": { "source": {
"type": "base64", "type": "base64",
"media_type": "image/jpeg", "media_type": "image/jpeg",
"data": Path("claude_temp.jpg"), "data": f,
}, },
}, },
], ],
}, },
], ],
model="claude-3-opus-20240229", model=ANTHROPIC_MODEL,
) )
bot.reply_to(message, "Claude vision answer:\n" + r.content[0].text) bot.reply_to(message, "Claude vision answer:\n" + r.content[0].text)
except Exception as e: except Exception as e:

4
handlers/gemini.py
View File

@ -210,8 +210,8 @@ def gemini_photo_handler(message: Message, bot: TeleBot) -> None:
temp_file.write(downloaded_file) temp_file.write(downloaded_file)
model = genai.GenerativeModel("gemini-pro-vision") model = genai.GenerativeModel("gemini-pro-vision")
image_path = Path("gemini_temp.jpg") with open("gemini_temp.jpg", "rb") as image_file:
image_data = image_path.read_bytes() image_data = image_file.read()
contents = { contents = {
"parts": [{"mime_type": "image/jpeg", "data": image_data}, {"text": prompt}] "parts": [{"mime_type": "image/jpeg", "data": image_data}, {"text": prompt}]
} }