Run gofmt/goimports on go sources
This commit is contained in:
parent
4590f1db38
commit
200ec57f29
@ -4,6 +4,7 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -83,8 +84,8 @@ var newCmd = &cobra.Command{
|
|||||||
|
|
||||||
messages := []Message{
|
messages := []Message{
|
||||||
{
|
{
|
||||||
OriginalContent: messageContents,
|
|
||||||
Role: "user",
|
Role: "user",
|
||||||
|
OriginalContent: messageContents,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -111,8 +112,8 @@ var promptCmd = &cobra.Command{
|
|||||||
|
|
||||||
messages := []Message{
|
messages := []Message{
|
||||||
{
|
{
|
||||||
OriginalContent: message,
|
|
||||||
Role: "user",
|
Role: "user",
|
||||||
|
OriginalContent: message,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -128,5 +129,5 @@ var promptCmd = &cobra.Command{
|
|||||||
|
|
||||||
func NewRootCmd() *cobra.Command {
|
func NewRootCmd() *cobra.Command {
|
||||||
rootCmd.AddCommand(newCmd, promptCmd)
|
rootCmd.AddCommand(newCmd, promptCmd)
|
||||||
return rootCmd;
|
return rootCmd
|
||||||
}
|
}
|
||||||
|
@ -37,7 +37,6 @@ func InitializeConfig() *Config {
|
|||||||
defaultConfig := &Config{}
|
defaultConfig := &Config{}
|
||||||
defaultConfig.OpenAI.APIKey = "your_key_here"
|
defaultConfig.OpenAI.APIKey = "your_key_here"
|
||||||
|
|
||||||
|
|
||||||
file, err := os.Create(configFile)
|
file, err := os.Create(configFile)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
Fatal("Could not open config file for writing: %v", err)
|
Fatal("Could not open config file for writing: %v", err)
|
||||||
|
@ -5,10 +5,11 @@ import (
|
|||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
|
|
||||||
openai "github.com/sashabaranov/go-openai"
|
openai "github.com/sashabaranov/go-openai"
|
||||||
)
|
)
|
||||||
|
|
||||||
func CreateChatCompletionRequest(system string, messages []Message) (*openai.ChatCompletionRequest) {
|
func CreateChatCompletionRequest(system string, messages []Message) *openai.ChatCompletionRequest {
|
||||||
chatCompletionMessages := []openai.ChatCompletionMessage{
|
chatCompletionMessages := []openai.ChatCompletionMessage{
|
||||||
{
|
{
|
||||||
Role: "system",
|
Role: "system",
|
||||||
@ -16,7 +17,7 @@ func CreateChatCompletionRequest(system string, messages []Message) (*openai.Cha
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, m := range(messages) {
|
for _, m := range messages {
|
||||||
chatCompletionMessages = append(chatCompletionMessages, openai.ChatCompletionMessage{
|
chatCompletionMessages = append(chatCompletionMessages, openai.ChatCompletionMessage{
|
||||||
Role: m.Role,
|
Role: m.Role,
|
||||||
Content: m.OriginalContent,
|
Content: m.OriginalContent,
|
||||||
|
@ -4,9 +4,10 @@ import (
|
|||||||
"database/sql"
|
"database/sql"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"gorm.io/gorm"
|
|
||||||
"gorm.io/driver/sqlite"
|
|
||||||
sqids "github.com/sqids/sqids-go"
|
sqids "github.com/sqids/sqids-go"
|
||||||
|
"gorm.io/driver/sqlite"
|
||||||
|
"gorm.io/gorm"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Store struct {
|
type Store struct {
|
||||||
@ -28,9 +29,8 @@ type Conversation struct {
|
|||||||
Title string
|
Title string
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
func getDataDir() string {
|
func getDataDir() string {
|
||||||
var dataDir string;
|
var dataDir string
|
||||||
|
|
||||||
xdgDataHome := os.Getenv("XDG_DATA_HOME")
|
xdgDataHome := os.Getenv("XDG_DATA_HOME")
|
||||||
if xdgDataHome != "" {
|
if xdgDataHome != "" {
|
||||||
@ -57,7 +57,7 @@ func InitializeStore() *Store {
|
|||||||
&Message{},
|
&Message{},
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, x := range(models) {
|
for _, x := range models {
|
||||||
err := db.AutoMigrate(x)
|
err := db.AutoMigrate(x)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
Fatal("Could not perform database migrations: %v\n", err)
|
Fatal("Could not perform database migrations: %v\n", err)
|
||||||
@ -76,7 +76,7 @@ func (s *Store) SaveConversation(conversation *Conversation) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if !conversation.ShortName.Valid {
|
if !conversation.ShortName.Valid {
|
||||||
shortName, _ := s.sqids.Encode([]uint64{ uint64(conversation.ID) })
|
shortName, _ := s.sqids.Encode([]uint64{uint64(conversation.ID)})
|
||||||
conversation.ShortName = sql.NullString{String: shortName, Valid: true}
|
conversation.ShortName = sql.NullString{String: shortName, Valid: true}
|
||||||
err = s.db.Save(&conversation).Error
|
err = s.db.Save(&conversation).Error
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user