FormImport

mili-tan · Mar 20, 2024 · a2158d0 · a2158d0
1 parent 7b4ef28
commit a2158d0
Show file tree

Hide file tree

Showing 8 changed files with 479 additions and 13 deletions.
diff --git a/Onllama.Tiny/Form1.cs b/Onllama.Tiny/Form1.cs
@@ -280,12 +280,12 @@ private void dropdown1_SelectedValueChanged(object sender, object value)
         {
             if (value.ToString() == "导入模型")
             {
-                MessageBox.Show("imp");
+                new FormImport().ShowDialog();
+                ListModels();
             }
             else if (value.ToString() == "Ollama 设置")
             {
                 new FormSettings().ShowDialog();
-                Notification.info(this, "设置已更改", "请手动重启 Ollama 核心以使配置生效。");
             }
             else if (value.ToString() == "刷新模型列表")
             {

diff --git a/Onllama.Tiny/FormImport.Designer.cs b/Onllama.Tiny/FormImport.Designer.cs
diff --git a/Onllama.Tiny/FormImport.cs b/Onllama.Tiny/FormImport.cs
@@ -0,0 +1,69 @@
+using AntdUI;
+using OllamaSharp.Models;
+using OllamaSharp.Streamer;
+using Onllama.Tiny.Properties;
+
+namespace Onllama.Tiny
+{
+    public partial class FormImport : Form
+    {
+        public FormImport()
+        {
+            InitializeComponent();
+        }
+
+        private void buttonOpen_Click(object sender, EventArgs e)
+        {
+            var f = new OpenFileDialog();
+            if (f.ShowDialog() == DialogResult.OK) input1.Text = f.FileName;
+            inputName.Text = f.SafeFileNames.Last().Split('.', '-').First();
+            foreach (var item in select1.Items)
+                if (input1.Text.Contains(item.ToString()))
+                    select1.SelectedValue = item.ToString();
+        }
+
+        private void buttonSave_Click(object sender, EventArgs e)
+        {
+            AntdUI.Modal.open(new AntdUI.Modal.Config(this, "您确定要导入模型吗？",
+                new[]
+                {
+                    new Modal.TextLine(inputName.Text, Style.Db.Primary)
+                }, TType.Success)
+            {
+                OkType = TTypeMini.Success,
+                OkText = "导入",
+                OnOk = _ =>
+                {
+                    try
+                    {
+                        Task.Run(() => Form1.OllamaApi.CreateModel(
+                            new CreateModelRequest
+                                {ModelFileContent = inputMf.Text, Name = inputName.Text, Stream = true},
+                            new ActionResponseStreamer<CreateStatus>(x => Invoke(() => Text = x.Status)))).Wait();
+                    }
+                    catch (Exception exception)
+                    {
+                        Console.WriteLine(exception);
+                    }
+
+                    Invoke(Close);
+                    return true;
+                }
+            });
+        }
+
+        private void select1_SelectedValueChanged(object sender, object value)
+        {
+            inputMf.Text = "FROM " + input1.Text + Environment.NewLine;
+            if (value.ToString() == "qwen") inputMf.Text += Resources.qwenTmp;
+            if (value.ToString() == "yi") inputMf.Text += Resources.yiTmp;
+            if (value.ToString() == "gemma") inputMf.Text += Resources.gemmaTmp;
+            if (value.ToString() == "mistral") inputMf.Text += Resources.mistralTmp;
+        }
+
+        private void FormImport_Load(object sender, EventArgs e)
+        {
+
+        }
+    }
+}