Feat: code block copy; model update; [WIP:broke tables]

1 files changed, 7 insertions, 5 deletions

diff --git a/tools.go b/tools.go
index 8fda0ab..94c1b16 100644
--- a/tools.go
+++ b/tools.go
@@ -9,11 +9,13 @@ import (
 )
 
 var (
-	toolCallRE = regexp.MustCompile(`__tool_call__\s*([\s\S]*?)__tool_call__`)
-	quotesRE   = regexp.MustCompile(`(".*?")`)
-	starRE     = regexp.MustCompile(`(\*.*?\*)`)
-	thinkRE    = regexp.MustCompile(`(<think>.*?</think>)`)
-	// codeBlokRE = regexp.MustCompile(`(\x60\x60\x60.*?\x60\x60\x60)`)
+	toolCallRE  = regexp.MustCompile(`__tool_call__\s*([\s\S]*?)__tool_call__`)
+	quotesRE    = regexp.MustCompile(`(".*?")`)
+	starRE      = regexp.MustCompile(`(\*.*?\*)`)
+	thinkRE     = regexp.MustCompile(`(<think>.*?</think>)`)
+	codeBlockRE = regexp.MustCompile(`(?s)\x60{3}(?:.*?)\n(.*?)\n\s*\x60{3}\s*`)
+	// codeBlockRE = regexp.MustCompile("```\s*([\s\S]*?)```")
+	// codeBlockRE = regexp.MustCompile(`(\x60\x60\x60.*?\x60\x60\x60)`)
 	basicSysMsg = `Large Language Model that helps user with any of his requests.`
 	toolSysMsg  = `You're a helpful assistant.
 # Tools