Fix/embedding disabled bug (#19)

* fix(config): embedding disabled bug * update version
xbotter · Aug 10, 2023 · 306ce86 · 306ce86
1 parent 8229258
commit 306ce86
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/Directory.Build.props b/Directory.Build.props
@@ -2,7 +2,7 @@
 	<PropertyGroup>
 		<Nullable>enable</Nullable>
 		<AvaloniaVersion>11.0.0</AvaloniaVersion>
-		<Version>0.3.0</Version>
+		<Version>0.3.1</Version>
 	</PropertyGroup>
 
 	<PropertyGroup Condition="'$(Configuration)'=='Release'">

diff --git a/PromptPlayground/ViewModels/ConfigViewModel.cs b/PromptPlayground/ViewModels/ConfigViewModel.cs
@@ -134,14 +134,14 @@ public ILLMConfigViewModel GetLLM()
         return this.SelectedModel;
     }
 
-    public IVectorDbConfigViewModel GetVectorDb()
+    public IVectorDbConfigViewModel? GetVectorDb()
     {
-        return this.SelectedVectorDb;
+        return this.EnableVectorDB ? this.SelectedVectorDb : null;
     }
 
-    public IEmbeddingConfigViewModel GetEmbedding()
+    public IEmbeddingConfigViewModel? GetEmbedding()
     {
-        return this.SelectedEmbedding;
+        return this.EnableVectorDB ? this.SelectedEmbedding : null;
     }
 
     public ConfigViewModel(bool requireLoadConfig = false) : this()

diff --git a/PromptPlayground/ViewModels/ConfigViewModels/IConfigViewModel.cs b/PromptPlayground/ViewModels/ConfigViewModels/IConfigViewModel.cs
@@ -54,8 +54,8 @@ public interface IConfigAttributesProvider
     {
         IList<ConfigAttribute> AllAttributes { get; }
         ILLMConfigViewModel GetLLM();
-        IVectorDbConfigViewModel GetVectorDb();
-        IEmbeddingConfigViewModel GetEmbedding();
+        IVectorDbConfigViewModel? GetVectorDb();
+        IEmbeddingConfigViewModel? GetEmbedding();
     }
 
     public class ConfigAttribute : ObservableObject