Skip to content

Commit a220717

Browse files
committed
add log level option, code reuse
1 parent a0c80ca commit a220717

File tree

3 files changed

+23
-20
lines changed

3 files changed

+23
-20
lines changed

Editor/LLMCharacterEditor.cs

+3-9
Original file line numberDiff line numberDiff line change
@@ -60,21 +60,15 @@ public override void OnInspectorGUI()
6060
{
6161
LLMCharacter llmScript = (LLMCharacter)target;
6262
SerializedObject llmScriptSO = new SerializedObject(llmScript);
63-
llmScriptSO.Update();
6463

65-
GUI.enabled = false;
66-
AddScript(llmScriptSO);
67-
GUI.enabled = true;
68-
EditorGUI.BeginChangeCheck();
64+
OnInspectorGUIStart(llmScriptSO);
6965
AddOptionsToggles(llmScriptSO);
66+
7067
AddClientSettings(llmScriptSO);
7168
AddChatSettings(llmScriptSO);
7269
AddModelSettings(llmScriptSO, llmScript);
7370

74-
if (EditorGUI.EndChangeCheck())
75-
Repaint();
76-
77-
llmScriptSO.ApplyModifiedProperties();
71+
OnInspectorGUIEnd(llmScriptSO);
7872
}
7973
}
8074

Editor/LLMEditor.cs

+2-11
Original file line numberDiff line numberDiff line change
@@ -126,15 +126,9 @@ void ShowProgress(float progress, string progressText)
126126
public override void OnInspectorGUI()
127127
{
128128
LLM llmScript = (LLM)target;
129-
// LLM llmScript = (LLM)target;
130129
SerializedObject llmScriptSO = new SerializedObject(llmScript);
131-
llmScriptSO.Update();
132130

133-
GUI.enabled = false;
134-
AddScript(llmScriptSO);
135-
GUI.enabled = true;
136-
137-
EditorGUI.BeginChangeCheck();
131+
OnInspectorGUIStart(llmScriptSO);
138132

139133
ShowProgress(LLMUnitySetup.libraryProgress, "Setup Library");
140134
ShowProgress(llmScript.modelProgress, "Model Downloading");
@@ -147,10 +141,7 @@ public override void OnInspectorGUI()
147141
GUI.enabled = true;
148142
AddChatSettings(llmScriptSO);
149143

150-
if (EditorGUI.EndChangeCheck())
151-
Repaint();
152-
153-
llmScriptSO.ApplyModifiedProperties();
144+
OnInspectorGUIEnd(llmScriptSO);
154145
}
155146
}
156147
}

Editor/PropertyEditor.cs

+18
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ public void AddOptionsToggle(SerializedObject llmScriptSO, string propertyName,
3131

3232
public void AddOptionsToggles(SerializedObject llmScriptSO)
3333
{
34+
LLMUnitySetup.SetDebugMode((LLMUnitySetup.DebugModeType)EditorGUILayout.EnumPopup("Log Level", LLMUnitySetup.DebugMode));
3435
EditorGUILayout.BeginHorizontal();
3536
AddOptionsToggle(llmScriptSO, "advancedOptions", "Advanced Options");
3637
EditorGUILayout.EndHorizontal();
@@ -127,5 +128,22 @@ public Attribute GetPropertyAttribute(SerializedProperty prop, Type attributeCla
127128
}
128129
return null;
129130
}
131+
132+
public void OnInspectorGUIStart(SerializedObject scriptSO)
133+
{
134+
scriptSO.Update();
135+
GUI.enabled = false;
136+
AddScript(scriptSO);
137+
GUI.enabled = true;
138+
EditorGUI.BeginChangeCheck();
139+
}
140+
141+
public void OnInspectorGUIEnd(SerializedObject scriptSO)
142+
{
143+
if (EditorGUI.EndChangeCheck())
144+
Repaint();
145+
146+
scriptSO.ApplyModifiedProperties();
147+
}
130148
}
131149
}

0 commit comments

Comments
 (0)