fix

openshieldai · Nov 20, 2024 · d371a80 · d371a80
1 parent fbe37b4
commit d371a80
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"`