diff options
Diffstat (limited to 'models/consts.go')
| -rw-r--r-- | models/consts.go | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/models/consts.go b/models/consts.go index 14f1a49..da664f9 100644 --- a/models/consts.go +++ b/models/consts.go @@ -1,5 +1,7 @@ package models +import "regexp" + const ( LoadedMark = "(loaded) " ToolRespMultyType = "multimodel_content" @@ -13,3 +15,17 @@ const ( APITypeChat APIType = iota APITypeCompletion ) + +var ( + ToolCallRE = regexp.MustCompile(`__tool_call__\s*([\s\S]*?)__tool_call__`) + QuotesRE = regexp.MustCompile(`(".*?")`) + StarRE = regexp.MustCompile(`(\*.*?\*)`) + ThinkRE = regexp.MustCompile(`(?s)<think>.*?</think>`) + CodeBlockRE = regexp.MustCompile(`(?s)\x60{3}(?:.*?)\n(.*?)\n\s*\x60{3}\s*`) + SingleBacktickRE = regexp.MustCompile(`\x60([^\x60]*)\x60`) + RoleRE = regexp.MustCompile(`^(\w+):`) +) + +var ( + SysLabels = []string{"assistant"} +) |