Merge pull request #75 from undreamai/feature/embedding

Embedding and search functionality (RAG system)

.github/setup.sh

@@ -0,0 +1 @@
+git config core.hooksPath .github/hooks

.gitignore

@@ -5,4 +5,5 @@ obj/
 obj.meta
 bin/
 bin.meta
-
+*.api
+*.api.meta

Editor/EmbeddingEditor.cs

@@ -0,0 +1,36 @@
+using UnityEditor;
+
+namespace LLMUnity
+{
+    [CustomEditor(typeof(Embedding))]
+    public class EmbeddingEditor : Editor
+    {
+
+        void ShowProgress(float progress, string progressText)
+        {
+            if (progress != 1) EditorGUI.ProgressBar(EditorGUILayout.GetControlRect(), progress, progressText);
+        }
+
+        public override void OnInspectorGUI()
+        {
+            Embedding embdeddingScript = (Embedding)target;
+            SerializedObject embdeddingScriptSO = new SerializedObject(embdeddingScript);
+            embdeddingScriptSO.Update();
+
+            embdeddingScriptSO.ApplyModifiedProperties();
+            string[] options = new string[embdeddingScript.options.Length];
+            for (int i = 0; i < embdeddingScript.options.Length; i++)
+            {
+                options[i] = embdeddingScript.options[i].Item1;
+            }
+
+            int newIndex = EditorGUILayout.Popup("Model", embdeddingScript.SelectedOption, options);
+            if (newIndex != embdeddingScript.SelectedOption)
+            {
+                embdeddingScript.SelectModel(newIndex);
+            }
+            ShowProgress(embdeddingScript.downloadProgress, "Downloading model");
+            embdeddingScriptSO.ApplyModifiedProperties();
+        }
+    }
+}

Editor/LLMEditor.cs

@@ -9,10 +9,19 @@ public class LLMEditor : LLMClientEditor
         public void AddModelLoaders(SerializedObject llmScriptSO, LLM llmScript)
         {
             EditorGUILayout.BeginHorizontal();
-            if (GUILayout.Button("Download model", GUILayout.Width(buttonWidth)))
+
+            string[] options = new string[llmScript.modelOptions.Length];
+            for (int i = 0; i < llmScript.modelOptions.Length; i++)
             {
-                llmScript.DownloadModel();
+                options[i] = llmScript.modelOptions[i].Item1;
             }
+
+            int newIndex = EditorGUILayout.Popup("Model", llmScript.SelectedOption, options);
+            if (newIndex != llmScript.SelectedOption)
+            {
+                llmScript.DownloadModel(newIndex);
+            }
+
             if (GUILayout.Button("Load model", GUILayout.Width(buttonWidth)))
             {
                 EditorApplication.delayCall += () =>