github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-02-18 18:44:42 -05:00
Code Issues Packages Projects Releases Wiki Activity

Address feedback

Browse Source
This commit is contained in:
Krzysztof Czerwinski
2026-02-18 18:30:47 +09:00
parent d63f2ead40
commit be0184d14f
5 changed files with 41 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
autogpt_platform/backend/backend/blocks/telegram/__init__.py
View File

@@ -1 +0,0 @@
# Telegram Bot blocks for AutoGPT Platform

2
autogpt_platform/backend/backend/blocks/telegram/_api.py
View File

@@ -16,7 +16,7 @@ logger = logging.getLogger(__name__)
TELEGRAM_API_BASE = "https://api.telegram.org"
class TelegramAPIException(Exception):
class TelegramAPIException(ValueError):
"""Exception raised for Telegram API errors."""
def __init__(self, message: str, error_code: int = 0):

75
autogpt_platform/backend/backend/blocks/telegram/blocks.py
View File

@@ -4,8 +4,10 @@ Telegram action blocks for sending messages, media, and managing chat content.
import base64
import logging
import mimetypes
from enum import Enum
from typing import Optional
from pathlib import Path
from typing import Any, Optional
from backend.data.block import (
Block,
@@ -104,7 +106,7 @@ class SendTelegramMessageBlock(Block):
reply_to_message_id: Optional[int],
disable_notification: bool,
) -> dict:
data: dict = {
data: dict[str, Any] = {
"chat_id": chat_id,
"text": text,
}
@@ -132,7 +134,7 @@ class SendTelegramMessageBlock(Block):
yield "message_id", result.get("message_id", 0)
yield "status", "Message sent"
except Exception as e:
raise ValueError(f"Failed to send message: {e}")
raise ValueError(f"Failed to send message: {e}") from e
class SendTelegramPhotoBlock(Block):
@@ -194,7 +196,7 @@ class SendTelegramPhotoBlock(Block):
parse_mode: str,
reply_to_message_id: Optional[int],
) -> dict:
data: dict = {
data: dict[str, Any] = {
"chat_id": chat_id,
"photo": photo_url,
}
@@ -216,11 +218,9 @@ class SendTelegramPhotoBlock(Block):
parse_mode: str,
reply_to_message_id: Optional[int],
) -> dict:
from pathlib import Path
path = Path(file_path)
file_bytes = path.read_bytes()
data: dict = {"chat_id": str(chat_id)}
data: dict[str, Any] = {"chat_id": str(chat_id)}
if caption:
data["caption"] = caption
if parse_mode and parse_mode != "none":
@@ -228,13 +228,14 @@ class SendTelegramPhotoBlock(Block):
if reply_to_message_id:
data["reply_to_message_id"] = str(reply_to_message_id)
mime_type = mimetypes.guess_type(path.name)[0] or "image/jpeg"
return await call_telegram_api_with_file(
credentials,
"sendPhoto",
file_field="photo",
file_data=file_bytes,
filename=path.name,
content_type="image/jpeg",
content_type=mime_type,
data=data,
)
@@ -284,7 +285,7 @@ class SendTelegramPhotoBlock(Block):
yield "message_id", result.get("message_id", 0)
yield "status", "Photo sent"
except Exception as e:
raise ValueError(f"Failed to send photo: {e}")
raise ValueError(f"Failed to send photo: {e}") from e
class SendTelegramVoiceBlock(Block):
@@ -349,7 +350,7 @@ class SendTelegramVoiceBlock(Block):
duration: Optional[int],
reply_to_message_id: Optional[int],
) -> dict:
data: dict = {
data: dict[str, Any] = {
"chat_id": chat_id,
"voice": voice_url,
}
@@ -371,11 +372,9 @@ class SendTelegramVoiceBlock(Block):
duration: Optional[int],
reply_to_message_id: Optional[int],
) -> dict:
from pathlib import Path
path = Path(file_path)
file_bytes = path.read_bytes()
data: dict = {"chat_id": str(chat_id)}
data: dict[str, Any] = {"chat_id": str(chat_id)}
if caption:
data["caption"] = caption
if duration:
@@ -383,13 +382,14 @@ class SendTelegramVoiceBlock(Block):
if reply_to_message_id:
data["reply_to_message_id"] = str(reply_to_message_id)
mime_type = mimetypes.guess_type(path.name)[0] or "audio/ogg"
return await call_telegram_api_with_file(
credentials,
"sendVoice",
file_field="voice",
file_data=file_bytes,
filename=path.name,
content_type="audio/ogg",
content_type=mime_type,
data=data,
)
@@ -439,7 +439,7 @@ class SendTelegramVoiceBlock(Block):
yield "message_id", result.get("message_id", 0)
yield "status", "Voice sent"
except Exception as e:
raise ValueError(f"Failed to send voice: {e}")
raise ValueError(f"Failed to send voice: {e}") from e
class ReplyToTelegramMessageBlock(Block):
@@ -489,7 +489,7 @@ class ReplyToTelegramMessageBlock(Block):
text: str,
parse_mode: str,
) -> dict:
data: dict = {
data: dict[str, Any] = {
"chat_id": chat_id,
"text": text,
"reply_to_message_id": reply_to_message_id,
@@ -513,7 +513,7 @@ class ReplyToTelegramMessageBlock(Block):
yield "message_id", result.get("message_id", 0)
yield "status", "Reply sent"
except Exception as e:
raise ValueError(f"Failed to send reply: {e}")
raise ValueError(f"Failed to send reply: {e}") from e
class GetTelegramFileBlock(Block):
@@ -590,7 +590,7 @@ class GetTelegramFileBlock(Block):
yield "file", file_result
yield "status", "File downloaded"
except Exception as e:
raise ValueError(f"Failed to download file: {e}")
raise ValueError(f"Failed to download file: {e}") from e
class DeleteTelegramMessageBlock(Block):
@@ -649,7 +649,7 @@ class DeleteTelegramMessageBlock(Block):
)
yield "status", "Message deleted"
except Exception as e:
raise ValueError(f"Failed to delete message: {e}")
raise ValueError(f"Failed to delete message: {e}") from e
class EditTelegramMessageBlock(Block):
@@ -701,7 +701,7 @@ class EditTelegramMessageBlock(Block):
text: str,
parse_mode: str,
) -> dict:
data: dict = {
data: dict[str, Any] = {
"chat_id": chat_id,
"message_id": message_id,
"text": text,
@@ -725,7 +725,7 @@ class EditTelegramMessageBlock(Block):
yield "message_id", result.get("message_id", 0)
yield "status", "Message edited"
except Exception as e:
raise ValueError(f"Failed to edit message: {e}")
raise ValueError(f"Failed to edit message: {e}") from e
class SendTelegramAudioBlock(Block):
@@ -801,7 +801,7 @@ class SendTelegramAudioBlock(Block):
duration: Optional[int],
reply_to_message_id: Optional[int],
) -> dict:
data: dict = {
data: dict[str, Any] = {
"chat_id": chat_id,
"audio": audio_url,
}
@@ -829,11 +829,9 @@ class SendTelegramAudioBlock(Block):
duration: Optional[int],
reply_to_message_id: Optional[int],
) -> dict:
from pathlib import Path
path = Path(file_path)
file_bytes = path.read_bytes()
data: dict = {"chat_id": str(chat_id)}
data: dict[str, Any] = {"chat_id": str(chat_id)}
if caption:
data["caption"] = caption
if title:
@@ -845,13 +843,14 @@ class SendTelegramAudioBlock(Block):
if reply_to_message_id:
data["reply_to_message_id"] = str(reply_to_message_id)
mime_type = mimetypes.guess_type(path.name)[0] or "audio/mpeg"
return await call_telegram_api_with_file(
credentials,
"sendAudio",
file_field="audio",
file_data=file_bytes,
filename=path.name,
content_type="audio/mpeg",
content_type=mime_type,
data=data,
)
@@ -902,7 +901,7 @@ class SendTelegramAudioBlock(Block):
yield "message_id", result.get("message_id", 0)
yield "status", "Audio sent"
except Exception as e:
raise ValueError(f"Failed to send audio: {e}")
raise ValueError(f"Failed to send audio: {e}") from e
class SendTelegramDocumentBlock(Block):
@@ -972,7 +971,7 @@ class SendTelegramDocumentBlock(Block):
parse_mode: str,
reply_to_message_id: Optional[int],
) -> dict:
data: dict = {
data: dict[str, Any] = {
"chat_id": chat_id,
"document": document_url,
}
@@ -995,11 +994,9 @@ class SendTelegramDocumentBlock(Block):
parse_mode: str,
reply_to_message_id: Optional[int],
) -> dict:
from pathlib import Path
path = Path(file_path)
file_bytes = path.read_bytes()
data: dict = {"chat_id": str(chat_id)}
data: dict[str, Any] = {"chat_id": str(chat_id)}
if caption:
data["caption"] = caption
if parse_mode and parse_mode != "none":
@@ -1007,13 +1004,14 @@ class SendTelegramDocumentBlock(Block):
if reply_to_message_id:
data["reply_to_message_id"] = str(reply_to_message_id)
mime_type = mimetypes.guess_type(path.name)[0] or "application/octet-stream"
return await call_telegram_api_with_file(
credentials,
"sendDocument",
file_field="document",
file_data=file_bytes,
filename=display_filename or path.name,
content_type="application/octet-stream",
content_type=mime_type,
data=data,
)
@@ -1061,7 +1059,7 @@ class SendTelegramDocumentBlock(Block):
yield "message_id", result.get("message_id", 0)
yield "status", "Document sent"
except Exception as e:
raise ValueError(f"Failed to send document: {e}")
raise ValueError(f"Failed to send document: {e}") from e
class SendTelegramVideoBlock(Block):
@@ -1129,7 +1127,7 @@ class SendTelegramVideoBlock(Block):
duration: Optional[int],
reply_to_message_id: Optional[int],
) -> dict:
data: dict = {
data: dict[str, Any] = {
"chat_id": chat_id,
"video": video_url,
}
@@ -1154,11 +1152,9 @@ class SendTelegramVideoBlock(Block):
duration: Optional[int],
reply_to_message_id: Optional[int],
) -> dict:
from pathlib import Path
path = Path(file_path)
file_bytes = path.read_bytes()
data: dict = {"chat_id": str(chat_id)}
data: dict[str, Any] = {"chat_id": str(chat_id)}
if caption:
data["caption"] = caption
if parse_mode and parse_mode != "none":
@@ -1168,13 +1164,14 @@ class SendTelegramVideoBlock(Block):
if reply_to_message_id:
data["reply_to_message_id"] = str(reply_to_message_id)
mime_type = mimetypes.guess_type(path.name)[0] or "video/mp4"
return await call_telegram_api_with_file(
credentials,
"sendVideo",
file_field="video",
file_data=file_bytes,
filename=path.name,
content_type="video/mp4",
content_type=mime_type,
data=data,
)
@@ -1223,4 +1220,4 @@ class SendTelegramVideoBlock(Block):
yield "message_id", result.get("message_id", 0)
yield "status", "Video sent"
except Exception as e:
raise ValueError(f"Failed to send video: {e}")
raise ValueError(f"Failed to send video: {e}") from e

2
autogpt_platform/backend/backend/blocks/telegram/triggers.py
View File

@@ -191,7 +191,7 @@ class TelegramMessageTriggerBlock(TelegramTriggerBase, Block):
elif "photo" in message:
# Get the largest photo (last in array)
photos = message.get("photo", [])
photo_fid = photos[-1]["file_id"] if photos else ""
photo_fid = photos[-1].get("file_id", "") if photos else ""
yield "event", "photo"
yield "text", ""
yield "photo_file_id", photo_fid

5
autogpt_platform/backend/backend/integrations/webhooks/telegram.py
View File

@@ -4,6 +4,7 @@ Telegram Bot API Webhooks Manager.
Handles webhook registration and validation for Telegram bots.
"""
import hmac
import logging
from fastapi import HTTPException, Request
@@ -23,7 +24,7 @@ class TelegramWebhookType(StrEnum):
BOT = "bot"
class TelegramWebhooksManager(BaseWebhooksManager[TelegramWebhookType]):
class TelegramWebhooksManager(BaseWebhooksManager):
"""
Manages Telegram bot webhooks.
@@ -54,7 +55,7 @@ class TelegramWebhooksManager(BaseWebhooksManager[TelegramWebhookType]):
"""
# Verify secret token header
secret_header = request.headers.get("X-Telegram-Bot-Api-Secret-Token")
if not secret_header or secret_header != webhook.secret:
if not secret_header or not hmac.compare_digest(secret_header, webhook.secret):
raise HTTPException(
status_code=403,
detail="Invalid or missing X-Telegram-Bot-Api-Secret-Token",