添加项目文件。

2025-03-12 20:02:52 +08:00
parent 4d34907fa7
commit 3ccd6d9a39
32 changed files with 1833 additions and 0 deletions
--- a/ViewModels/SettingViewModel.cs
+++ b/ViewModels/SettingViewModel.cs
@@ -0,0 +1,216 @@
+using Microsoft.Win32;
+using OllamaSharp;
+using System.Collections.ObjectModel;
+using System.ComponentModel;
+using System.Diagnostics;
+using System.IO;
+using System.Runtime.CompilerServices;
+using System.Windows;
+using System.Windows.Input;
+using System.Windows.Media;
+using YwxApp.AiChat.Commands;
+using YwxApp.AiChat.Models;
+namespace YwxApp.AiChat.ViewModels
+{
+
+
+
+    /// <summary>
+    /// 0、Current class:
+    /// </summary>
+    public class SettingViewModel : INotifyPropertyChanged
+    {
+        #region Field | Property | Collection | Command
+
+        #region Field
+        private string _selectedModel;                  //select model
+        private string _modelInfo;                      //model info
+        private SolidColorBrush _labelBackgroundColor;  //color style
+        private readonly ShareOllamaObject _ollama;     //OllamaAPI object.
+        #endregion
+
+        #region Property
+        public string OllamaAppPath
+        {
+            get { return _ollama.OllamaAppPath; }
+            set { _ollama.OllamaAppPath = value; OnPropertyChanged(); }
+        }
+        public string SelectedModel
+        {
+            get => _selectedModel;
+            set
+            {
+                if (_selectedModel != value)
+                {
+                    _selectedModel = value;
+                    ResetModelName();
+                }
+                OnPropertyChanged();
+            }
+        }
+        public string ModelInformation
+        {
+            get => _modelInfo;
+            set
+            {
+                _modelInfo = value;
+                OnPropertyChanged();
+            }
+        }
+        public SolidColorBrush LabelBackgroundColor
+        {
+            get => _labelBackgroundColor;
+            set
+            {
+                if (_labelBackgroundColor != value)
+                {
+                    _labelBackgroundColor = value;
+                    OnPropertyChanged();
+                }
+            }
+        }
+        #endregion
+
+        #region Collection
+        public ObservableCollection<string> ModelList
+        {
+            get { return _ollama.ModelList; }
+            set { _ollama.ModelList = value; OnPropertyChanged(); }
+        }
+        #endregion
+
+        #region Command
+        public ICommand OpenFileDialogCommand { get; }      //select Ollama application file path command.
+        public ICommand GetModelListCommand { get; }        //get model list command.
+        public ICommand ModelListUpdateCommand { get; }     //model list update command.
+        public ICommand StartOllamaServerCommand { get; }   //start ollam server command.
+        #endregion
+
+        #endregion
+
+        #region Constructor
+        public SettingViewModel(ShareOllamaObject ollama)
+        {
+            _ollama = ollama;
+            Task task = OnGetModelList();
+            OpenFileDialogCommand = new ParameterlessCommand(() => OnSelectOllamaAppPathDialog());
+            GetModelListCommand = new ParameterlessCommand(async () => await OnGetModelList());
+            ModelListUpdateCommand = new ParameterlessCommand(async () => await OnModelListUpdate());
+            StartOllamaServerCommand = new ParameterlessCommand(async () => OnStartOllamaServer());
+            SetConnected();
+        }
+        #endregion
+
+        #region other method
+        ///set  ollama model server application object.
+        public void SetOllamaApiClient(OllamaApiClient ollama)
+        {
+            _ollama.Ollama = ollama;
+        }
+
+        // set the connection states color
+        public void SetConnected()
+        {
+            if (_ollama.OllamaEnabled)
+            {
+                LabelBackgroundColor = Brushes.Green;
+            }
+            else
+            {
+                LabelBackgroundColor = Brushes.Red;
+            }
+        }
+        /// <summary>
+        /// reset the model
+        /// </summary>
+        private void ResetModelName()
+        {
+            _ollama.OllamaEnabled = false;
+            _ollama.Ollama.SelectedModel = SelectedModel;
+            ModelInformationChanged();
+            _ollama.OllamaEnabled = true;
+        }
+        /// <summary>
+        /// model info changed
+        /// </summary>
+        public void ModelInformationChanged()
+        {
+            string modelName = SelectedModel.Split(':')[0].ToLower();
+            string modelInfoPath = $"{Environment.CurrentDirectory}\\model introduction\\{modelName}.txt";
+            string info = string.Empty;
+            if (File.Exists(modelInfoPath))
+            {
+                info = File.ReadAllText(modelInfoPath);
+            }
+            //MessageBox.Show(modelInfoPath);
+            switch (modelName)
+            {
+                case ModelDescription.Llama32:
+                    ModelInformation = info;
+                    break;
+                case ModelDescription.CodeGemma:
+                    ModelInformation = info;
+                    break;
+                default:
+                    ModelInformation = "";
+                    break;
+            }
+        }
+        #endregion
+
+        #region command trigger method
+        private void OnStartOllamaServer()
+        {
+            if (!_ollama.OllamaEnabled)
+            {
+                _ollama.StartOllama(OllamaAppPath, SelectedModel);
+            }
+        }
+        private void OnSelectOllamaAppPathDialog()
+        {
+            OpenFileDialog openFileDialog = new();
+            if (openFileDialog.ShowDialog() == true)
+            {
+                OllamaAppPath = openFileDialog.FileName;
+            }
+        }
+        /// <summary>
+        /// get the model list
+        /// </summary>
+        private async Task OnGetModelList()
+        {
+            try
+            {
+                //ModelList.Clear();
+                ModelList = (ObservableCollection<string>)_ollama.GetModelList();
+                Debug.Print($"ModelList count: {ModelList.Count}");
+                SelectedModel = _ollama.Ollama.SelectedModel;
+                var modelName = ModelList.FirstOrDefault(name => name.Equals(SelectedModel));
+                if (ModelList.Count > 0 && modelName != null)
+                {
+                    SelectedModel = ModelList[ModelList.Count - 1];
+                }
+            }
+            catch (Exception ex)
+            {
+                MessageBox.Show($"Error: {ex.Message}", "Error", MessageBoxButton.OK, MessageBoxImage.Error);
+            }
+        }
+        /// <summary>
+        /// update the model list
+        /// </summary>
+        private async Task OnModelListUpdate()
+        {
+            MessageBox.Show($"List Update");
+        }
+        #endregion
+
+        #region property changed event
+        public event PropertyChangedEventHandler PropertyChanged;
+        protected virtual void OnPropertyChanged([CallerMemberName] string propertyName = null)
+        {
+            PropertyChanged?.Invoke(this, new PropertyChangedEventArgs(propertyName));
+        }
+        #endregion
+    }
+}