Fix caption (#4)

* refactor: Set caption to enhancedPrompt in HandleCommand * fix: Ensure caption length is within Telegram limits * meme bot v0.1.34
azalio · Dec 31, 2024 · 7eb5bdd · 7eb5bdd
1 parent b6e63b5
commit 7eb5bdd
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/internal/service/prompt_enhancer.go b/internal/service/prompt_enhancer.go
@@ -42,6 +42,15 @@ func (p *PromptEnhancer) EnhancePrompt(ctx context.Context, originalPrompt strin
 		return originalPrompt, "", fmt.Errorf("enhancing prompt: %w", err)
 	}
 
+	if caption == "" {
+		caption = originalPrompt
+	}
+
+	// Ensure caption length is within Telegram limits
+	if len(caption) > 1024 {
+		caption = caption[:1024]
+	}
+
 	p.logger.Debug(ctx, "Successfully enhanced prompt", map[string]interface{}{
 		"original_prompt": originalPrompt,
 		"enhanced_prompt": enhancedPrompt,