telegram: lazy-init tool traces, add typing indicator, remove budget

Three changes to the Telegram integration:

1. Typing indicator (SendChatAction every 4s) replaces the
   premature '🤔 Thinking...' message. Telegram shows the
   native '...' status while the agent runs.

2. Tool trace message is now created lazily — only when the
   first tool_call fires — not at conversation start.

3. Remove daily token budget enforcement. The bot package
   retains the CheckDailyBudget API for programmatic use but
   the Telegram integration no longer blocks users mid-
   conversation after accumulated usage crosses the threshold.

Author: jkyberneees

cmd/odek/telegram.go

@@ -51,7 +51,6 @@ func telegramCmd(args []string) error {
 
 	// 4. Create bot client.
 	bot := telegram.NewBot(cfg.Token)
-	bot.SetDailyTokenBudget(cfg.DailyTokenBudget)
 
 	// 4b. Create logger.
 	level := telegram.ParseLogLevel(cfg.LogLevel)
@@ -294,17 +293,32 @@ func handleChatMessage(
 
 	rend := render.New(os.Stderr, false).WithModel(modelLabel)
 
+	// ── Typing Indicator ────────────────────────────────────────────
+	// Send "typing" action every 4s while the agent runs (Telegram shows
+	// it for ~5s). Stops when the goroutine's context is cancelled.
+	typingDone := make(chan struct{})
+	defer close(typingDone)
+	go func() {
+		ticker := time.NewTicker(4 * time.Second)
+		defer ticker.Stop()
+		for {
+			select {
+			case <-ticker.C:
+				bot.SendChatAction(chatID, "typing")
+			case <-typingDone:
+				return
+			}
+		}
+	}()
+
 	// ── Tool Tracing ───────────────────────────────────────────────
 	// Single editable message showing live tool execution progress.
+	// The message is created lazily — only when the first tool call
+	// fires, not before. This avoids the premature "🤔 Thinking…" spam.
 	var traceMsgID int
 	var traceMu sync.Mutex
 	traceLines := make([]string, 0, 8)
 
-	// Send initial thinking message.
-	if initMsg, err := bot.SendMessage(chatID, "🤔 Thinking...", nil); err == nil {
-		traceMsgID = initMsg.ID
-	}
-
 	// truncate shortens a string for display, appending "…" if trimmed.
 	truncate := func(s string, max int) string {
 		if len(s) > max {
@@ -331,6 +345,15 @@ func handleChatMessage(
 		ToolEventHandler: func(event string, name string, data string) {
 			traceMu.Lock()
 			defer traceMu.Unlock()
+
+			// Lazy-init: create the trace message on the first tool call.
+			if traceMsgID == 0 && event == "tool_call" {
+				if msg, err := bot.SendMessage(chatID, "🔧 …", nil); err == nil {
+					traceMsgID = msg.ID
+				} else {
+					return
+				}
+			}
 			if traceMsgID == 0 {
 				return
 			}
@@ -343,8 +366,6 @@ func handleChatMessage(
 				bot.EditMessageText(chatID, traceMsgID, strings.Join(traceLines, "\n"), nil)
 
 			case "tool_result":
-				// Replace the last line's ⏳ with a completion marker
-				// and the result size instead of the actual content.
 				sizeLabel := fmt.Sprintf("%dB", len(data))
 				if len(data) > 1024 {
 					sizeLabel = fmt.Sprintf("%dKB", len(data)/1024)
@@ -386,20 +407,6 @@ func handleChatMessage(
 		return
 	}
 
-	// Check daily token budget.
-	totalTokens := int64(agent.TotalInputTokens() + agent.TotalOutputTokens())
-	if err := bot.CheckDailyBudget(totalTokens); err != nil {
-		fmt.Fprintf(os.Stderr, "odek telegram: %v\n", err)
-		reportError(bot, chatID, "Daily token budget exceeded. Usage for today has been tracked and will be enforced going forward.")
-		// Still save the session so the conversation isn't lost.
-		cs.Messages = updatedMessages
-		cs.TurnCount++
-		if err := sessionManager.Save(chatID, cs.Messages); err != nil {
-			fmt.Fprintf(os.Stderr, "odek telegram: session save: %v\n", err)
-		}
-		return
-	}
-
 	// Save the updated session messages.
 	cs.Messages = updatedMessages
 	cs.TurnCount++

coverage2.out

@@ -168,6 +168,33 @@ func TestResetTrust(t *testing.T) {
 
 // ── Test newID uniqueness ──────────────────────────────────────────────────
 
+// ── Test SetLogger ────────────────────────────────────────────────────────
+
+func TestTelegramApprover_SetLogger_Nil(t *testing.T) {
+	ts := testServer(t, nil)
+	defer ts.Close()
+	bot := testBot(t, ts)
+
+	a := NewTelegramApprover(bot, 1)
+	// Initially uses NopLogger.
+	a.SetLogger(nil)
+	// After nil, should use NopLogger (no panic).
+	a.SetLogger(nil)
+}
+
+func TestTelegramApprover_SetLogger_Valid(t *testing.T) {
+	ts := testServer(t, nil)
+	defer ts.Close()
+	bot := testBot(t, ts)
+
+	a := NewTelegramApprover(bot, 1)
+	logger := NewFileLogger(LogDebug, "")
+	a.SetLogger(logger)
+	// Just verify no panic — the logger is set internally.
+}
+
+// ── Test newID uniqueness ──────────────────────────────────────────────────
+
 func TestNewID_Unique(t *testing.T) {
 	ts := testServer(t, nil)
 	defer ts.Close()

coverage3.out

@@ -420,3 +420,15 @@ func (b *Bot) GetMe() (*User, error) {
 	}
 	return &user, nil
 }
+
+// SendChatAction tells the user that the bot is doing something on their
+// behalf (e.g., "typing"). The action is shown as a status in the chat for
+// ~5 seconds or until the next message is sent. Callers should re-send every
+// 4 seconds for long-running operations.
+func (b *Bot) SendChatAction(chatID int64, action string) error {
+	params := map[string]any{
+		"chat_id": chatID,
+		"action":  action,
+	}
+	return b.doJSON("sendChatAction", params, nil)
+}