Private
Public Access
1
0

Store ChromaHighlighter in lmcli.Context and use it

In preparation for TUI
This commit is contained in:
2024-03-12 07:19:16 +00:00
parent 0a27b9a8d3
commit f2c7d2bdd0
2 changed files with 7 additions and 14 deletions

View File

@@ -8,6 +8,7 @@ import (
"git.mlow.ca/mlow/lmcli/pkg/lmcli/provider"
"git.mlow.ca/mlow/lmcli/pkg/lmcli/provider/anthropic"
"git.mlow.ca/mlow/lmcli/pkg/lmcli/provider/openai"
"git.mlow.ca/mlow/lmcli/pkg/util/tty"
"gorm.io/driver/sqlite"
"gorm.io/gorm"
)
@@ -15,6 +16,8 @@ import (
type Context struct {
Config Config
Store ConversationStore
Chroma *tty.ChromaHighlighter
}
func NewContext() (*Context, error) {
@@ -34,7 +37,9 @@ func NewContext() (*Context, error) {
Fatal("%v\n", err)
}
return &Context{*config, s}, nil
hl := tty.NewChromaHighlighter("markdown", *config.Chroma.Formatter, *config.Chroma.Style)
return &Context{*config, s, hl}, nil
}
func (c *Context) GetCompletionProvider(model string) (provider.ChatCompletionClient, error) {