mirror of
https://github.com/tiennm99/DocsGPT.git
synced 2026-10-04 22:13:08 +00:00
Alex
636d4d35d1
Rewrite the default/creative/strict presets (classic + agentic) into
structured sections: grounding and cite-by-title guidance, insufficient-
context behavior, current date, respond-in-user-language, scoped mermaid
usage, an untrusted-content guardrail, and a conditional XML-tagged
document context block. A memory directory listing is injected at render
time via the template prefetch mechanism so the model starts oriented
without burning a tool call.
Fixes along the way:
- Agentic preset swap was dead code: _get_prompt_content cached the
classic preset before create_agent's swap check ran, so agentic and
research agents always got the classic preset. The swap now happens
inside _get_prompt_content.
- Jinja autoescape corrupted document content in custom prompts
(< -> <); prompts are not HTML, autoescape is now off.
- Literal {summaries} leaked into the prompt when no docs were
retrieved; the placeholder is now stripped.
- Agentic/research prompts referenced tool names from a dropped naming
scheme (search_internal, reason_think); they now reference the real
names (search, reason).
- The strict preset told the model to "be very creative and use your
imagination" right after "never make up information".
- extract_tool_usages recorded intermediate attribute chains as
bare-tool usages, which meant "run all actions" at prefetch; only
maximal chains are recorded now.
- Headless runs retrieved docs but never rendered them into the
prompt; the prompt is now rendered like the streaming path.
- Default tools were unreachable by name in prompt templates
(prefetch results were keyed by synthetic id only); defaults now
claim the name key unless an explicit row shadows it.
Tool layer: memory/notes/todo actions are namespaced (memory_view,
note_overwrite, todo_create, ...) with legacy unprefixed names still
accepted via prefix stripping; duplicate action names across tools are
disambiguated with the owning tool's name instead of numeric suffixes;
thin tool descriptions rewritten (brave, duckduckgo, telegram, ntfy,
cryptoprice, read_webpage, internal_search, think).
Docs are now wrapped per chunk in <document index>/<source>/<content>
tags for citation-by-title support.
103 lines
3.7 KiB
Python
103 lines
3.7 KiB
Python
import logging
|
|
|
|
import requests
|
|
|
|
from application.agents.tools.base import Tool
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class TelegramTool(Tool):
|
|
"""
|
|
Telegram Bot
|
|
A flexible Telegram tool for performing various actions (e.g., sending messages, images).
|
|
Requires a bot token and chat ID for configuration
|
|
"""
|
|
|
|
def __init__(self, config):
|
|
self.config = config
|
|
self.token = config.get("token", "")
|
|
|
|
def execute_action(self, action_name, **kwargs):
|
|
actions = {
|
|
"telegram_send_message": self._send_message,
|
|
"telegram_send_image": self._send_image,
|
|
}
|
|
if action_name not in actions:
|
|
raise ValueError(f"Unknown action: {action_name}")
|
|
return actions[action_name](**kwargs)
|
|
|
|
def _send_message(self, text, chat_id):
|
|
logger.debug("Sending Telegram message to chat_id=%s", chat_id)
|
|
url = f"https://api.telegram.org/bot{self.token}/sendMessage"
|
|
payload = {"chat_id": chat_id, "text": text}
|
|
response = requests.post(url, data=payload, timeout=100)
|
|
return {"status_code": response.status_code, "message": "Message sent"}
|
|
|
|
def _send_image(self, image_url, chat_id):
|
|
logger.debug("Sending Telegram image to chat_id=%s", chat_id)
|
|
url = f"https://api.telegram.org/bot{self.token}/sendPhoto"
|
|
payload = {"chat_id": chat_id, "photo": image_url}
|
|
response = requests.post(url, data=payload, timeout=100)
|
|
return {"status_code": response.status_code, "message": "Image sent"}
|
|
|
|
def get_actions_metadata(self):
|
|
return [
|
|
{
|
|
"name": "telegram_send_message",
|
|
"description": (
|
|
"Send a text message to the configured Telegram chat via "
|
|
"the bot. Compose the final message text before sending."
|
|
),
|
|
"parameters": {
|
|
"type": "object",
|
|
"properties": {
|
|
"text": {
|
|
"type": "string",
|
|
"description": "Text to send in the notification",
|
|
},
|
|
"chat_id": {
|
|
"type": "string",
|
|
"description": "Chat ID to send the notification to",
|
|
},
|
|
},
|
|
"required": ["text"],
|
|
"additionalProperties": False,
|
|
},
|
|
},
|
|
{
|
|
"name": "telegram_send_image",
|
|
"description": (
|
|
"Send an image to the configured Telegram chat. Requires "
|
|
"a publicly accessible image URL."
|
|
),
|
|
"parameters": {
|
|
"type": "object",
|
|
"properties": {
|
|
"image_url": {
|
|
"type": "string",
|
|
"description": "URL of the image to send",
|
|
},
|
|
"chat_id": {
|
|
"type": "string",
|
|
"description": "Chat ID to send the image to",
|
|
},
|
|
},
|
|
"required": ["image_url"],
|
|
"additionalProperties": False,
|
|
},
|
|
},
|
|
]
|
|
|
|
def get_config_requirements(self):
|
|
return {
|
|
"token": {
|
|
"type": "string",
|
|
"label": "Bot Token",
|
|
"description": "Telegram bot token for authentication",
|
|
"required": True,
|
|
"secret": True,
|
|
"order": 1,
|
|
},
|
|
}
|