LLMIFY THIS MOFO

2026-09-01 01:09:59 -04:00
parent dad09d1109
commit c89ff2b68e
18 changed files with 288 additions and 41 deletions
@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"log"
+	"strings"
 	"time"

 	"github.com/sashabaranov/go-openai"
@@ -17,6 +18,11 @@ type OpenAIAdapter struct {

 // NewOpenAIAdapter creates a new OpenAI LLM adapter
 func NewOpenAIAdapter(cfg Config) *OpenAIAdapter {
+	if cfg.BaseURL != "" {
+		clientConfig := openai.DefaultConfig(cfg.APIKey)
+		clientConfig.BaseURL = strings.TrimRight(cfg.BaseURL, "/") + "/v1"
+		return &OpenAIAdapter{client: openai.NewClientWithConfig(clientConfig), config: cfg}
+	}
 	return &OpenAIAdapter{
 		client: openai.NewClient(cfg.APIKey),
 		config: cfg,