fix (#250)

openshieldai · Nov 20, 2024 · 1fa1a65 · 1fa1a65
2 parents fbe37b4 + d371a80
commit 1fa1a65
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/lib/config.go b/lib/config.go
@@ -33,11 +33,14 @@ type Providers struct {
 }
 
 type Services struct {
-	LlamaGuard *ServiceLlamaGuard `mapstructure:"llamaguard"`
+	LlamaGuard  *ServiceLlamaGuard  `mapstructure:"llamaguard"`
+	PromptGuard *ServicePromptGuard `mapstructure:"promptguard"`
 }
 type ServiceLlamaGuard struct {
-	PromptGuard *ServicePromptGuard `mapstructure:"promptguard"`
+	Enabled bool   `mapstructure:"enabled"`
+	BaseUrl string `mapstructure:"url"`
 }
+
 type ServicePromptGuard struct {
 	Enabled bool   `mapstructure:"enabled"`
 	BaseUrl string `mapstructure:"url"`