feat: correctly implement bitbucket & add OpenAIAdapter
This commit is contained in:
@@ -7,7 +7,7 @@ import (
|
||||
"path/filepath"
|
||||
|
||||
"git.schreifuchs.ch/schreifuchs/pierre-bot/internal/chatter"
|
||||
"git.schreifuchs.ch/schreifuchs/pierre-bot/internal/gitadapters"
|
||||
"git.schreifuchs.ch/schreifuchs/pierre-bot/internal/gitadapters/bitbucket"
|
||||
"git.schreifuchs.ch/schreifuchs/pierre-bot/internal/gitadapters/gitea"
|
||||
"git.schreifuchs.ch/schreifuchs/pierre-bot/internal/pierre"
|
||||
"github.com/alecthomas/kong"
|
||||
@@ -32,7 +32,7 @@ type RepoArgs struct {
|
||||
|
||||
type LLMConfig struct {
|
||||
Provider string `help:"Provider for llm (ollama or gemini)" required:"" env:"LLM_PROVIDER"`
|
||||
Endpoint string `help:"Endpoint for provider (only for ollama)" env:"LLM_ENDPOINT"`
|
||||
BaseURL string `help:"Endpoint for provider (only for ollama)" env:"LLM_BASE_URL"`
|
||||
APIKey string `help:"APIKey for provider" env:"LLM_API_KEY"`
|
||||
Model string `help:"Model to use" env:"LLM_MODEL"`
|
||||
}
|
||||
@@ -84,7 +84,7 @@ func main() {
|
||||
if cfg.Bitbucket.BaseURL == "" {
|
||||
log.Fatal("Bitbucket Base URL is required when using bitbucket provider.")
|
||||
}
|
||||
git = gitadapters.NewBitbucket(cfg.Bitbucket.BaseURL, cfg.Bitbucket.Token)
|
||||
git = bitbucket.NewBitbucket(cfg.Bitbucket.BaseURL, cfg.Bitbucket.Token)
|
||||
case "gitea":
|
||||
if cfg.Gitea.BaseURL == "" {
|
||||
log.Fatal("Gitea Base URL is required when using gitea provider.")
|
||||
@@ -105,7 +105,10 @@ func main() {
|
||||
case "gemini":
|
||||
ai, err = chatter.NewGeminiAdapter(context.Background(), cfg.LLM.APIKey, cfg.LLM.Model)
|
||||
case "ollama":
|
||||
ai, err = chatter.NewOllamaAdapter(cfg.LLM.Endpoint, cfg.LLM.Model)
|
||||
ai, err = chatter.NewOllamaAdapter(cfg.LLM.BaseURL, cfg.LLM.Model)
|
||||
case "openai":
|
||||
ai = chatter.NewOpenAIAdapter(cfg.LLM.APIKey, cfg.LLM.Model, cfg.LLM.BaseURL)
|
||||
|
||||
default:
|
||||
log.Fatalf("%s is not a valid llm provider", cfg.LLM.Provider)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user