Compare commits
1 Commits
jmorganca/
...
brucemacd/
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
f82d00af74 |
@@ -209,9 +209,6 @@ func main() {
|
|||||||
|
|
||||||
st := &store.Store{}
|
st := &store.Store{}
|
||||||
|
|
||||||
// Initialize native settings with store
|
|
||||||
SetSettingsStore(st)
|
|
||||||
|
|
||||||
// Enable CORS in development mode
|
// Enable CORS in development mode
|
||||||
if devMode {
|
if devMode {
|
||||||
os.Setenv("OLLAMA_CORS", "1")
|
os.Setenv("OLLAMA_CORS", "1")
|
||||||
@@ -256,27 +253,22 @@ func main() {
|
|||||||
done <- osrv.Run(octx)
|
done <- osrv.Run(octx)
|
||||||
}()
|
}()
|
||||||
|
|
||||||
restartServer := func() {
|
|
||||||
ocancel()
|
|
||||||
<-done
|
|
||||||
octx, ocancel = context.WithCancel(ctx)
|
|
||||||
go func() {
|
|
||||||
done <- osrv.Run(octx)
|
|
||||||
}()
|
|
||||||
}
|
|
||||||
|
|
||||||
uiServer := ui.Server{
|
uiServer := ui.Server{
|
||||||
Token: token,
|
Token: token,
|
||||||
Restart: restartServer,
|
Restart: func() {
|
||||||
|
ocancel()
|
||||||
|
<-done
|
||||||
|
octx, ocancel = context.WithCancel(ctx)
|
||||||
|
go func() {
|
||||||
|
done <- osrv.Run(octx)
|
||||||
|
}()
|
||||||
|
},
|
||||||
Store: st,
|
Store: st,
|
||||||
ToolRegistry: toolRegistry,
|
ToolRegistry: toolRegistry,
|
||||||
Dev: devMode,
|
Dev: devMode,
|
||||||
Logger: slog.Default(),
|
Logger: slog.Default(),
|
||||||
}
|
}
|
||||||
|
|
||||||
// Set restart callback for native settings
|
|
||||||
SetRestartCallback(restartServer)
|
|
||||||
|
|
||||||
srv := &http.Server{
|
srv := &http.Server{
|
||||||
Handler: uiServer.Handler(),
|
Handler: uiServer.Handler(),
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,5 @@
|
|||||||
#import "app_darwin.h"
|
#import "app_darwin.h"
|
||||||
#import "menu.h"
|
#import "menu.h"
|
||||||
#import "settings_darwin.h"
|
|
||||||
#import "../../updater/updater_darwin.h"
|
#import "../../updater/updater_darwin.h"
|
||||||
#import <AppKit/AppKit.h>
|
#import <AppKit/AppKit.h>
|
||||||
#import <Cocoa/Cocoa.h>
|
#import <Cocoa/Cocoa.h>
|
||||||
@@ -253,7 +252,7 @@ bool firstTimeRun,startHidden; // Set in run before initialization
|
|||||||
}
|
}
|
||||||
|
|
||||||
- (void)settingsUI {
|
- (void)settingsUI {
|
||||||
openNativeSettings();
|
[self uiRequest:@"/settings"];
|
||||||
}
|
}
|
||||||
|
|
||||||
- (void)openUI {
|
- (void)openUI {
|
||||||
|
|||||||
@@ -1,438 +0,0 @@
|
|||||||
//go:build darwin
|
|
||||||
|
|
||||||
package main
|
|
||||||
|
|
||||||
/*
|
|
||||||
#cgo CFLAGS: -x objective-c
|
|
||||||
#cgo LDFLAGS: -framework Cocoa
|
|
||||||
|
|
||||||
#include <stdlib.h>
|
|
||||||
#include "settings_darwin.h"
|
|
||||||
*/
|
|
||||||
import "C"
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"crypto/ed25519"
|
|
||||||
"crypto/rand"
|
|
||||||
"encoding/json"
|
|
||||||
"encoding/pem"
|
|
||||||
"fmt"
|
|
||||||
"log/slog"
|
|
||||||
"net/http"
|
|
||||||
"os"
|
|
||||||
"os/exec"
|
|
||||||
"path/filepath"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
|
||||||
"time"
|
|
||||||
"unsafe"
|
|
||||||
|
|
||||||
"golang.org/x/crypto/ssh"
|
|
||||||
|
|
||||||
appauth "github.com/ollama/ollama/app/auth"
|
|
||||||
"github.com/ollama/ollama/app/store"
|
|
||||||
"github.com/ollama/ollama/auth"
|
|
||||||
"github.com/ollama/ollama/envconfig"
|
|
||||||
)
|
|
||||||
|
|
||||||
// settingsStore is a reference to the app's store for settings
|
|
||||||
var settingsStore *store.Store
|
|
||||||
|
|
||||||
// SetSettingsStore sets the store reference for settings callbacks
|
|
||||||
func SetSettingsStore(s *store.Store) {
|
|
||||||
settingsStore = s
|
|
||||||
}
|
|
||||||
|
|
||||||
//export getSettingsExpose
|
|
||||||
func getSettingsExpose() C.bool {
|
|
||||||
if settingsStore == nil {
|
|
||||||
return C.bool(false)
|
|
||||||
}
|
|
||||||
settings, err := settingsStore.Settings()
|
|
||||||
if err != nil {
|
|
||||||
slog.Error("failed to get settings", "error", err)
|
|
||||||
return C.bool(false)
|
|
||||||
}
|
|
||||||
return C.bool(settings.Expose)
|
|
||||||
}
|
|
||||||
|
|
||||||
//export setSettingsExpose
|
|
||||||
func setSettingsExpose(expose C.bool) {
|
|
||||||
if settingsStore == nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
settings, err := settingsStore.Settings()
|
|
||||||
if err != nil {
|
|
||||||
slog.Error("failed to get settings", "error", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
settings.Expose = bool(expose)
|
|
||||||
if err := settingsStore.SetSettings(settings); err != nil {
|
|
||||||
slog.Error("failed to save settings", "error", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//export getSettingsBrowser
|
|
||||||
func getSettingsBrowser() C.bool {
|
|
||||||
if settingsStore == nil {
|
|
||||||
return C.bool(false)
|
|
||||||
}
|
|
||||||
settings, err := settingsStore.Settings()
|
|
||||||
if err != nil {
|
|
||||||
slog.Error("failed to get settings", "error", err)
|
|
||||||
return C.bool(false)
|
|
||||||
}
|
|
||||||
return C.bool(settings.Browser)
|
|
||||||
}
|
|
||||||
|
|
||||||
//export setSettingsBrowser
|
|
||||||
func setSettingsBrowser(browser C.bool) {
|
|
||||||
if settingsStore == nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
settings, err := settingsStore.Settings()
|
|
||||||
if err != nil {
|
|
||||||
slog.Error("failed to get settings", "error", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
settings.Browser = bool(browser)
|
|
||||||
if err := settingsStore.SetSettings(settings); err != nil {
|
|
||||||
slog.Error("failed to save settings", "error", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//export getSettingsModels
|
|
||||||
func getSettingsModels() *C.char {
|
|
||||||
if settingsStore == nil {
|
|
||||||
return C.CString(envconfig.Models())
|
|
||||||
}
|
|
||||||
settings, err := settingsStore.Settings()
|
|
||||||
if err != nil {
|
|
||||||
slog.Error("failed to get settings", "error", err)
|
|
||||||
return C.CString(envconfig.Models())
|
|
||||||
}
|
|
||||||
if settings.Models == "" {
|
|
||||||
return C.CString(envconfig.Models())
|
|
||||||
}
|
|
||||||
return C.CString(settings.Models)
|
|
||||||
}
|
|
||||||
|
|
||||||
//export setSettingsModels
|
|
||||||
func setSettingsModels(path *C.char) {
|
|
||||||
if settingsStore == nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
settings, err := settingsStore.Settings()
|
|
||||||
if err != nil {
|
|
||||||
slog.Error("failed to get settings", "error", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
settings.Models = C.GoString(path)
|
|
||||||
if err := settingsStore.SetSettings(settings); err != nil {
|
|
||||||
slog.Error("failed to save settings", "error", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//export getSettingsContextLength
|
|
||||||
func getSettingsContextLength() C.int {
|
|
||||||
if settingsStore == nil {
|
|
||||||
return C.int(4096)
|
|
||||||
}
|
|
||||||
settings, err := settingsStore.Settings()
|
|
||||||
if err != nil {
|
|
||||||
slog.Error("failed to get settings", "error", err)
|
|
||||||
return C.int(4096)
|
|
||||||
}
|
|
||||||
if settings.ContextLength <= 0 {
|
|
||||||
return C.int(4096)
|
|
||||||
}
|
|
||||||
return C.int(settings.ContextLength)
|
|
||||||
}
|
|
||||||
|
|
||||||
//export setSettingsContextLength
|
|
||||||
func setSettingsContextLength(length C.int) {
|
|
||||||
if settingsStore == nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
settings, err := settingsStore.Settings()
|
|
||||||
if err != nil {
|
|
||||||
slog.Error("failed to get settings", "error", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
settings.ContextLength = int(length)
|
|
||||||
if err := settingsStore.SetSettings(settings); err != nil {
|
|
||||||
slog.Error("failed to save settings", "error", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// restartCallback is set by the app to restart the ollama server
|
|
||||||
var restartCallback func()
|
|
||||||
|
|
||||||
// SetRestartCallback sets the function to call when settings change requires a restart
|
|
||||||
func SetRestartCallback(cb func()) {
|
|
||||||
restartCallback = cb
|
|
||||||
}
|
|
||||||
|
|
||||||
//export restartOllamaServer
|
|
||||||
func restartOllamaServer() {
|
|
||||||
if restartCallback != nil {
|
|
||||||
slog.Info("restarting ollama server due to settings change")
|
|
||||||
go restartCallback()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// hasOllamaKey checks if the user has an Ollama key file
|
|
||||||
func hasOllamaKey() bool {
|
|
||||||
home, err := os.UserHomeDir()
|
|
||||||
if err != nil {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
keyPath := filepath.Join(home, ".ollama", "id_ed25519")
|
|
||||||
_, err = os.Stat(keyPath)
|
|
||||||
return err == nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// ensureKeypair generates a new keypair if one doesn't exist
|
|
||||||
func ensureKeypair() error {
|
|
||||||
home, err := os.UserHomeDir()
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
privKeyPath := filepath.Join(home, ".ollama", "id_ed25519")
|
|
||||||
|
|
||||||
// Check if key already exists
|
|
||||||
if _, err := os.Stat(privKeyPath); err == nil {
|
|
||||||
return nil // Key exists
|
|
||||||
}
|
|
||||||
|
|
||||||
// Generate new keypair
|
|
||||||
slog.Info("generating new keypair for ollama account")
|
|
||||||
|
|
||||||
pubKey, privKey, err := ed25519.GenerateKey(rand.Reader)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("failed to generate key: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Marshal private key
|
|
||||||
privKeyBytes, err := ssh.MarshalPrivateKey(privKey, "")
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("failed to marshal private key: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Ensure directory exists
|
|
||||||
if err := os.MkdirAll(filepath.Dir(privKeyPath), 0o755); err != nil {
|
|
||||||
return fmt.Errorf("failed to create directory: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Write private key
|
|
||||||
if err := os.WriteFile(privKeyPath, pem.EncodeToMemory(privKeyBytes), 0o600); err != nil {
|
|
||||||
return fmt.Errorf("failed to write private key: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Write public key
|
|
||||||
sshPubKey, err := ssh.NewPublicKey(pubKey)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("failed to create ssh public key: %w", err)
|
|
||||||
}
|
|
||||||
pubKeyBytes := ssh.MarshalAuthorizedKey(sshPubKey)
|
|
||||||
pubKeyPath := filepath.Join(home, ".ollama", "id_ed25519.pub")
|
|
||||||
if err := os.WriteFile(pubKeyPath, pubKeyBytes, 0o644); err != nil {
|
|
||||||
return fmt.Errorf("failed to write public key: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
slog.Info("keypair generated successfully")
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// userResponse matches the API response from ollama.com/api/me
|
|
||||||
type userResponse struct {
|
|
||||||
Name string `json:"name"`
|
|
||||||
Email string `json:"email"`
|
|
||||||
Plan string `json:"plan"`
|
|
||||||
AvatarURL string `json:"avatarurl"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// fetchUserFromAPI fetches user data from ollama.com using signed request
|
|
||||||
func fetchUserFromAPI() (*userResponse, error) {
|
|
||||||
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
||||||
defer cancel()
|
|
||||||
|
|
||||||
timestamp := strconv.FormatInt(time.Now().Unix(), 10)
|
|
||||||
signString := fmt.Sprintf("POST,/api/me?ts=%s", timestamp)
|
|
||||||
signature, err := auth.Sign(ctx, []byte(signString))
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("failed to sign request: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
endpoint := fmt.Sprintf("https://ollama.com/api/me?ts=%s", timestamp)
|
|
||||||
req, err := http.NewRequestWithContext(ctx, http.MethodPost, endpoint, nil)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("failed to create request: %w", err)
|
|
||||||
}
|
|
||||||
req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", signature))
|
|
||||||
|
|
||||||
client := &http.Client{Timeout: 10 * time.Second}
|
|
||||||
resp, err := client.Do(req)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("failed to call ollama.com: %w", err)
|
|
||||||
}
|
|
||||||
defer resp.Body.Close()
|
|
||||||
|
|
||||||
if resp.StatusCode != http.StatusOK {
|
|
||||||
return nil, fmt.Errorf("unexpected status: %d", resp.StatusCode)
|
|
||||||
}
|
|
||||||
|
|
||||||
var user userResponse
|
|
||||||
if err := json.NewDecoder(resp.Body).Decode(&user); err != nil {
|
|
||||||
return nil, fmt.Errorf("failed to parse response: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Make avatar URL absolute
|
|
||||||
if user.AvatarURL != "" && !strings.HasPrefix(user.AvatarURL, "http") {
|
|
||||||
user.AvatarURL = "https://ollama.com/" + user.AvatarURL
|
|
||||||
}
|
|
||||||
|
|
||||||
// Cache the avatar URL
|
|
||||||
cachedAvatarURL = user.AvatarURL
|
|
||||||
|
|
||||||
// Cache the user data
|
|
||||||
if settingsStore != nil {
|
|
||||||
storeUser := store.User{
|
|
||||||
Name: user.Name,
|
|
||||||
Email: user.Email,
|
|
||||||
Plan: user.Plan,
|
|
||||||
}
|
|
||||||
if err := settingsStore.SetUser(storeUser); err != nil {
|
|
||||||
slog.Warn("failed to cache user", "error", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return &user, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
//export getAccountName
|
|
||||||
func getAccountName() *C.char {
|
|
||||||
// Only return cached data - never block on network
|
|
||||||
if settingsStore == nil {
|
|
||||||
return C.CString("")
|
|
||||||
}
|
|
||||||
user, err := settingsStore.User()
|
|
||||||
if err != nil || user == nil {
|
|
||||||
return C.CString("")
|
|
||||||
}
|
|
||||||
return C.CString(user.Name)
|
|
||||||
}
|
|
||||||
|
|
||||||
// cachedAvatarURL stores the avatar URL from the last API fetch
|
|
||||||
var cachedAvatarURL string
|
|
||||||
|
|
||||||
//export getAccountAvatarURL
|
|
||||||
func getAccountAvatarURL() *C.char {
|
|
||||||
return C.CString(cachedAvatarURL)
|
|
||||||
}
|
|
||||||
|
|
||||||
//export getAccountEmail
|
|
||||||
func getAccountEmail() *C.char {
|
|
||||||
if settingsStore != nil {
|
|
||||||
user, err := settingsStore.User()
|
|
||||||
if err == nil && user != nil {
|
|
||||||
return C.CString(user.Email)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return C.CString("")
|
|
||||||
}
|
|
||||||
|
|
||||||
//export getAccountPlan
|
|
||||||
func getAccountPlan() *C.char {
|
|
||||||
if settingsStore != nil {
|
|
||||||
user, err := settingsStore.User()
|
|
||||||
if err == nil && user != nil {
|
|
||||||
return C.CString(user.Plan)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return C.CString("")
|
|
||||||
}
|
|
||||||
|
|
||||||
//export signOutAccount
|
|
||||||
func signOutAccount() {
|
|
||||||
if settingsStore != nil {
|
|
||||||
if err := settingsStore.ClearUser(); err != nil {
|
|
||||||
slog.Error("failed to clear user", "error", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Also remove the key file
|
|
||||||
home, err := os.UserHomeDir()
|
|
||||||
if err != nil {
|
|
||||||
slog.Error("failed to get home dir", "error", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
keyPath := filepath.Join(home, ".ollama", "id_ed25519")
|
|
||||||
if err := os.Remove(keyPath); err != nil && !os.IsNotExist(err) {
|
|
||||||
slog.Error("failed to remove key file", "error", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//export openConnectUrl
|
|
||||||
func openConnectUrl() {
|
|
||||||
// Ensure keypair exists (generate if needed)
|
|
||||||
if err := ensureKeypair(); err != nil {
|
|
||||||
slog.Error("failed to ensure keypair", "error", err)
|
|
||||||
// Fallback to basic connect page
|
|
||||||
cmd := exec.Command("open", "https://ollama.com/connect")
|
|
||||||
cmd.Start()
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// Build connect URL with public key
|
|
||||||
connectURL, err := appauth.BuildConnectURL("https://ollama.com")
|
|
||||||
if err != nil {
|
|
||||||
slog.Error("failed to build connect URL", "error", err)
|
|
||||||
// Fallback to basic connect page
|
|
||||||
connectURL = "https://ollama.com/connect"
|
|
||||||
}
|
|
||||||
|
|
||||||
cmd := exec.Command("open", connectURL)
|
|
||||||
if err := cmd.Start(); err != nil {
|
|
||||||
slog.Error("failed to open connect URL", "error", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//export refreshAccountFromAPI
|
|
||||||
func refreshAccountFromAPI() {
|
|
||||||
if !hasOllamaKey() {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
_, err := fetchUserFromAPI()
|
|
||||||
if err != nil {
|
|
||||||
slog.Debug("failed to refresh account", "error", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//export prefetchAccountData
|
|
||||||
func prefetchAccountData() {
|
|
||||||
// Run in background goroutine to not block app startup
|
|
||||||
go func() {
|
|
||||||
if !hasOllamaKey() {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
_, err := fetchUserFromAPI()
|
|
||||||
if err != nil {
|
|
||||||
slog.Debug("failed to prefetch account data", "error", err)
|
|
||||||
} else {
|
|
||||||
slog.Debug("prefetched account data successfully")
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
}
|
|
||||||
|
|
||||||
// OpenNativeSettings opens the native settings window
|
|
||||||
func OpenNativeSettings() {
|
|
||||||
C.openNativeSettings()
|
|
||||||
}
|
|
||||||
|
|
||||||
// Ensure the CString is freed (caller must free)
|
|
||||||
func freeCString(s *C.char) {
|
|
||||||
C.free(unsafe.Pointer(s))
|
|
||||||
}
|
|
||||||
@@ -1,38 +0,0 @@
|
|||||||
#import <Cocoa/Cocoa.h>
|
|
||||||
|
|
||||||
@interface SettingsWindowController : NSWindowController <NSWindowDelegate>
|
|
||||||
|
|
||||||
// General tab
|
|
||||||
@property(nonatomic, strong) NSButton *exposeCheckbox;
|
|
||||||
@property(nonatomic, strong) NSButton *browserCheckbox;
|
|
||||||
@property(nonatomic, strong) NSSlider *contextLengthSlider;
|
|
||||||
|
|
||||||
// Models tab
|
|
||||||
@property(nonatomic, strong) NSPathControl *modelsPathControl;
|
|
||||||
@property(nonatomic, strong) NSButton *modelsPathButton;
|
|
||||||
|
|
||||||
// Account tab
|
|
||||||
@property(nonatomic, strong) NSView *avatarView;
|
|
||||||
@property(nonatomic, strong) NSTextField *avatarInitialLabel;
|
|
||||||
@property(nonatomic, strong) NSImageView *avatarImageView;
|
|
||||||
@property(nonatomic, strong) NSTextField *accountNameLabel;
|
|
||||||
@property(nonatomic, strong) NSTextField *accountEmailLabel;
|
|
||||||
@property(nonatomic, strong) NSButton *manageButton;
|
|
||||||
@property(nonatomic, strong) NSButton *signOutButton;
|
|
||||||
@property(nonatomic, strong) NSButton *signInButton;
|
|
||||||
@property(nonatomic, strong) NSView *signedInContainer;
|
|
||||||
@property(nonatomic, strong) NSView *signedOutContainer;
|
|
||||||
|
|
||||||
// Plan section
|
|
||||||
@property(nonatomic, strong) NSView *planContainer;
|
|
||||||
@property(nonatomic, strong) NSTextField *planNameLabel;
|
|
||||||
@property(nonatomic, strong) NSButton *upgradeButton;
|
|
||||||
@property(nonatomic, strong) NSButton *viewUsageButton;
|
|
||||||
|
|
||||||
+ (instancetype)sharedController;
|
|
||||||
- (void)showSettings;
|
|
||||||
|
|
||||||
@end
|
|
||||||
|
|
||||||
// Go callbacks for settings
|
|
||||||
void openNativeSettings(void);
|
|
||||||
File diff suppressed because it is too large
Load Diff
@@ -1,16 +0,0 @@
|
|||||||
//go:build windows
|
|
||||||
|
|
||||||
package main
|
|
||||||
|
|
||||||
import "github.com/ollama/ollama/app/store"
|
|
||||||
|
|
||||||
// SetSettingsStore sets the store reference for settings callbacks (stub for Windows)
|
|
||||||
func SetSettingsStore(s *store.Store) {
|
|
||||||
// TODO: Implement Windows native settings
|
|
||||||
}
|
|
||||||
|
|
||||||
// SetRestartCallback sets the function to call when settings change requires a restart (stub for Windows)
|
|
||||||
func SetRestartCallback(cb func()) {
|
|
||||||
// TODO: Implement Windows native settings
|
|
||||||
}
|
|
||||||
|
|
||||||
@@ -524,13 +524,8 @@ func (s *llamaServer) Load(ctx context.Context, systemInfo ml.SystemInfo, system
|
|||||||
// Use the size of one layer as a buffer
|
// Use the size of one layer as a buffer
|
||||||
layers := s.ggml.Tensors().GroupLayers()
|
layers := s.ggml.Tensors().GroupLayers()
|
||||||
if blk0, ok := layers["blk.0"]; ok {
|
if blk0, ok := layers["blk.0"]; ok {
|
||||||
buffer := blk0.Size() + kv[0]
|
|
||||||
for i := range gpus {
|
for i := range gpus {
|
||||||
if gpus[i].FreeMemory > buffer {
|
gpus[i].FreeMemory -= blk0.Size() + kv[0]
|
||||||
gpus[i].FreeMemory -= buffer
|
|
||||||
} else {
|
|
||||||
gpus[i].FreeMemory = 0
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
slog.Warn("model missing blk.0 layer size")
|
slog.Warn("model missing blk.0 layer size")
|
||||||
@@ -580,11 +575,7 @@ func (s *llamaServer) Load(ctx context.Context, systemInfo ml.SystemInfo, system
|
|||||||
projectorGPU = firstIntegrated
|
projectorGPU = firstIntegrated
|
||||||
}
|
}
|
||||||
|
|
||||||
if gpus[projectorGPU].FreeMemory > projectorWeights {
|
gpus[projectorGPU].FreeMemory -= projectorWeights
|
||||||
gpus[projectorGPU].FreeMemory -= projectorWeights
|
|
||||||
} else {
|
|
||||||
gpus[projectorGPU].FreeMemory = 0
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
var kvTotal uint64
|
var kvTotal uint64
|
||||||
|
|||||||
@@ -463,6 +463,8 @@ func FromChatRequest(r ChatCompletionRequest) (*api.ChatRequest, error) {
|
|||||||
}
|
}
|
||||||
messages = append(messages, api.Message{Role: msg.Role, Content: content, Thinking: msg.Reasoning, ToolCalls: toolCalls, ToolName: toolName, ToolCallID: msg.ToolCallID})
|
messages = append(messages, api.Message{Role: msg.Role, Content: content, Thinking: msg.Reasoning, ToolCalls: toolCalls, ToolName: toolName, ToolCallID: msg.ToolCallID})
|
||||||
case []any:
|
case []any:
|
||||||
|
var texts []string
|
||||||
|
var images []api.ImageData
|
||||||
for _, c := range content {
|
for _, c := range content {
|
||||||
data, ok := c.(map[string]any)
|
data, ok := c.(map[string]any)
|
||||||
if !ok {
|
if !ok {
|
||||||
@@ -474,7 +476,7 @@ func FromChatRequest(r ChatCompletionRequest) (*api.ChatRequest, error) {
|
|||||||
if !ok {
|
if !ok {
|
||||||
return nil, errors.New("invalid message format")
|
return nil, errors.New("invalid message format")
|
||||||
}
|
}
|
||||||
messages = append(messages, api.Message{Role: msg.Role, Content: text})
|
texts = append(texts, text)
|
||||||
case "image_url":
|
case "image_url":
|
||||||
var url string
|
var url string
|
||||||
if urlMap, ok := data["image_url"].(map[string]any); ok {
|
if urlMap, ok := data["image_url"].(map[string]any); ok {
|
||||||
@@ -492,23 +494,24 @@ func FromChatRequest(r ChatCompletionRequest) (*api.ChatRequest, error) {
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
messages = append(messages, api.Message{Role: msg.Role, Images: []api.ImageData{img}})
|
images = append(images, img)
|
||||||
default:
|
default:
|
||||||
return nil, errors.New("invalid message format")
|
return nil, errors.New("invalid message format")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
// since we might have added multiple messages above, if we have tools
|
toolCalls, err := FromCompletionToolCall(msg.ToolCalls)
|
||||||
// calls we'll add them to the last message
|
if err != nil {
|
||||||
if len(messages) > 0 && len(msg.ToolCalls) > 0 {
|
return nil, err
|
||||||
toolCalls, err := FromCompletionToolCall(msg.ToolCalls)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
messages[len(messages)-1].ToolCalls = toolCalls
|
|
||||||
messages[len(messages)-1].ToolName = toolName
|
|
||||||
messages[len(messages)-1].ToolCallID = msg.ToolCallID
|
|
||||||
messages[len(messages)-1].Thinking = msg.Reasoning
|
|
||||||
}
|
}
|
||||||
|
messages = append(messages, api.Message{
|
||||||
|
Role: msg.Role,
|
||||||
|
Content: strings.Join(texts, "\n\n"),
|
||||||
|
Images: images,
|
||||||
|
Thinking: msg.Reasoning,
|
||||||
|
ToolCalls: toolCalls,
|
||||||
|
ToolName: toolName,
|
||||||
|
ToolCallID: msg.ToolCallID,
|
||||||
|
})
|
||||||
default:
|
default:
|
||||||
// content is only optional if tool calls are present
|
// content is only optional if tool calls are present
|
||||||
if msg.ToolCalls == nil {
|
if msg.ToolCalls == nil {
|
||||||
|
|||||||
@@ -41,7 +41,7 @@ func TestFromChatRequest_Basic(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestFromChatRequest_WithImage(t *testing.T) {
|
func TestFromChatRequest_MultiPartContent(t *testing.T) {
|
||||||
imgData, _ := base64.StdEncoding.DecodeString(image)
|
imgData, _ := base64.StdEncoding.DecodeString(image)
|
||||||
|
|
||||||
req := ChatCompletionRequest{
|
req := ChatCompletionRequest{
|
||||||
@@ -50,7 +50,12 @@ func TestFromChatRequest_WithImage(t *testing.T) {
|
|||||||
{
|
{
|
||||||
Role: "user",
|
Role: "user",
|
||||||
Content: []any{
|
Content: []any{
|
||||||
map[string]any{"type": "text", "text": "Hello"},
|
map[string]any{"type": "text", "text": "First part."},
|
||||||
|
map[string]any{"type": "text", "text": "Second part."},
|
||||||
|
map[string]any{
|
||||||
|
"type": "image_url",
|
||||||
|
"image_url": map[string]any{"url": prefix + image},
|
||||||
|
},
|
||||||
map[string]any{
|
map[string]any{
|
||||||
"type": "image_url",
|
"type": "image_url",
|
||||||
"image_url": map[string]any{"url": prefix + image},
|
"image_url": map[string]any{"url": prefix + image},
|
||||||
@@ -65,20 +70,31 @@ func TestFromChatRequest_WithImage(t *testing.T) {
|
|||||||
t.Fatalf("unexpected error: %v", err)
|
t.Fatalf("unexpected error: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(result.Messages) != 2 {
|
// Multi-part content array should produce a single message per OpenAI spec
|
||||||
t.Fatalf("expected 2 messages, got %d", len(result.Messages))
|
if len(result.Messages) != 1 {
|
||||||
|
t.Fatalf("expected 1 message, got %d", len(result.Messages))
|
||||||
}
|
}
|
||||||
|
|
||||||
if result.Messages[0].Content != "Hello" {
|
msg := result.Messages[0]
|
||||||
t.Errorf("expected first message content 'Hello', got %q", result.Messages[0].Content)
|
if msg.Role != "user" {
|
||||||
|
t.Errorf("expected role 'user', got %q", msg.Role)
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(result.Messages[1].Images) != 1 {
|
// Multiple text parts should be joined
|
||||||
t.Fatalf("expected 1 image, got %d", len(result.Messages[1].Images))
|
expectedContent := "First part.\n\nSecond part."
|
||||||
|
if msg.Content != expectedContent {
|
||||||
|
t.Errorf("expected content %q, got %q", expectedContent, msg.Content)
|
||||||
}
|
}
|
||||||
|
|
||||||
if string(result.Messages[1].Images[0]) != string(imgData) {
|
// Multiple images should be in the same message
|
||||||
t.Error("image data mismatch")
|
if len(msg.Images) != 2 {
|
||||||
|
t.Fatalf("expected 2 images, got %d", len(msg.Images))
|
||||||
|
}
|
||||||
|
|
||||||
|
for i, img := range msg.Images {
|
||||||
|
if string(img) != string(imgData) {
|
||||||
|
t.Errorf("image %d data mismatch", i)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user