fix: fix HTML entity handling and improve env prefix consistency

- Update `RecommendedEnvPrefix` from `"AI"` to `"AI_"` for consistency.
- Use `RecommendedEnvPrefix` variable instead of hardcoding `"AI_"` in `env.ParseWithOptions`.
- Add conditional check to print raw output directly if `c.config.Raw` is true in the chat UI.
- Replace HTML entities `&gt;` and `&lt;` with `>` and `<` respectively in `findOriginalUpdateBlocks`.

Signed-off-by: codiing-hui <wecoding@yeah.net>

Author: coding-hui

internal/options/config.go

@@ -22,7 +22,7 @@ var configTemplate string
 
 const (
 	// RecommendedEnvPrefix defines the ENV prefix used by all iam service.
-	RecommendedEnvPrefix = "AI"
+	RecommendedEnvPrefix = "AI_"
 
 	defaultMarkdownFormatText = "Format the response as markdown without enclosing backticks."
 	defaultJSONFormatText     = "Format the response as json without enclosing backticks."
@@ -252,7 +252,7 @@ func EnsureConfig() (Config, error) {
 	}
 	c.Models = ms
 
-	if err := env.ParseWithOptions(&c, env.Options{Prefix: "AI_"}); err != nil {
+	if err := env.ParseWithOptions(&c, env.Options{Prefix: RecommendedEnvPrefix}); err != nil {
 		return c, errbook.Wrap("Could not parse environment into settings file.", err)
 	}
 

internal/ui/chat/chat.go

@@ -95,11 +95,15 @@ func (c *Chat) Run() error {
 	}
 
 	if term.IsOutputTTY() {
-		switch {
-		case c.GlamOutput != "":
-			fmt.Print(c.GlamOutput)
-		case c.Output != "":
+		if c.config.Raw && c.Output != "" {
 			fmt.Print(c.Output)
+		} else {
+			switch {
+			case c.GlamOutput != "":
+				fmt.Print(c.GlamOutput)
+			case c.Output != "":
+				fmt.Print(c.Output)
+			}
 		}
 		fmt.Println()
 	}

internal/ui/coders/diff_block_editor.go

@@ -243,6 +243,8 @@ func (e *EditBlockCoder) Execute(ctx context.Context, messages []llms.ChatMessag
 }
 
 func findOriginalUpdateBlocks(content string, fence []string) ([]PartialCodeBlock, error) {
+	content = strings.ReplaceAll(content, ">", ">")
+	content = strings.ReplaceAll(content, "&lt;", "<")
 	edits := findAllCodeBlocks(content, fence)
 	result := make([]PartialCodeBlock, 0, len(edits))