[Go] fix system role capability (#603)

Fixes #541.
firebase · Jul 12, 2024 · 2e83e9b · 2e83e9b
1 parent 61a83ca
commit 2e83e9b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/go/plugins/googleai/googleai.go b/go/plugins/googleai/googleai.go
@@ -39,14 +39,14 @@ var (
 	basicText = ai.ModelCapabilities{
 		Multiturn:  true,
 		Tools:      true,
-		SystemRole: false,
+		SystemRole: true,
 		Media:      false,
 	}
 
 	multimodal = ai.ModelCapabilities{
 		Multiturn:  true,
 		Tools:      true,
-		SystemRole: false,
+		SystemRole: true,
 		Media:      true,
 	}
 

diff --git a/go/plugins/vertexai/vertexai.go b/go/plugins/vertexai/vertexai.go
@@ -35,14 +35,14 @@ var (
 	basicText = ai.ModelCapabilities{
 		Multiturn:  true,
 		Tools:      true,
-		SystemRole: false,
+		SystemRole: true,
 		Media:      false,
 	}
 
 	multimodal = ai.ModelCapabilities{
 		Multiturn:  true,
 		Tools:      true,
-		SystemRole: false,
+		SystemRole: true,
 		Media:      true,
 	}