Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
gempir committed Mar 3, 2024
1 parent 12ca062 commit ff01e7a
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 6 deletions.
14 changes: 9 additions & 5 deletions internal/george/ollama.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ type Ollama struct {
helixClient helixclient.Client
}

func NewOllama(emoteservice *emoteservice.SevenTvClient) *Ollama {
func NewOllama(emoteservice *emoteservice.SevenTvClient, helixClient helixclient.Client) *Ollama {
llm, err := ollama.New(ollama.WithModel("mistral"))
if err != nil {
log.Fatal(err)
Expand All @@ -26,6 +26,7 @@ func NewOllama(emoteservice *emoteservice.SevenTvClient) *Ollama {
return &Ollama{
llm: llm,
emoteservice: emoteservice,
helixClient: helixClient,
}
}

Expand All @@ -47,10 +48,13 @@ func (o *Ollama) AnalyzeUser(query string, channel string, username string, mont
return fmt.Errorf("failed to get user data: %w", err)
}

user, err := o.emoteservice.GetUser(userDataMap[username].ID)
if err != nil {
log.Errorf("failed to get user data from 7tv: %s", err)
return nil
var user emoteservice.User
if _, ok := userDataMap[username]; ok {
user, err = o.emoteservice.GetUser(userDataMap[username].ID)
if err != nil {
log.Errorf("failed to get user data from 7tv: %s", err)
return nil
}
}

fullQuery += "\nlogs:```\n"
Expand Down
2 changes: 1 addition & 1 deletion main.go
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@ func main() {

seventvClient := emoteservice.NewSevenTvClient(db)

george := george.NewOllama(seventvClient)
george := george.NewOllama(seventvClient, helixClient)
emoteChief := emotechief.NewEmoteChief(cfg, db, helixClient, seventvClient)
channelPointManager := channelpoint.NewChannelPointManager(cfg, helixClient, db)
wsHandler := ws.NewWsHandler(authClient)
Expand Down

0 comments on commit ff01e7a

Please sign in to comment.