diff --git a/core/providers/bedrock/bedrock_test.go b/core/providers/bedrock/bedrock_test.go index 428d77e7a..9ad550358 100644 --- a/core/providers/bedrock/bedrock_test.go +++ b/core/providers/bedrock/bedrock_test.go @@ -69,6 +69,16 @@ func TestBedrock(t *testing.T) { Embedding: true, ListModels: true, Reasoning: true, + BatchCreate: true, + BatchList: true, + BatchRetrieve: true, + BatchCancel: true, + BatchResults: true, + FileUpload: true, + FileList: true, + FileRetrieve: true, + FileDelete: true, + FileContent: true, }, } diff --git a/core/providers/gemini/gemini_test.go b/core/providers/gemini/gemini_test.go index 0ecbe2204..a39645016 100644 --- a/core/providers/gemini/gemini_test.go +++ b/core/providers/gemini/gemini_test.go @@ -54,6 +54,16 @@ func TestGemini(t *testing.T) { SpeechSynthesisStream: true, Reasoning: true, ListModels: true, + BatchCreate: true, + BatchList: true, + BatchRetrieve: true, + BatchCancel: true, + BatchResults: true, + FileUpload: true, + FileList: true, + FileRetrieve: true, + FileDelete: true, + FileContent: true, }, } diff --git a/core/providers/openai/openai_test.go b/core/providers/openai/openai_test.go index 31a0fa02e..ff40595b1 100644 --- a/core/providers/openai/openai_test.go +++ b/core/providers/openai/openai_test.go @@ -60,6 +60,16 @@ func TestOpenAI(t *testing.T) { Embedding: true, Reasoning: true, ListModels: true, + BatchCreate: true, + BatchList: true, + BatchRetrieve: true, + BatchCancel: true, + BatchResults: true, + FileUpload: true, + FileList: true, + FileRetrieve: true, + FileDelete: true, + FileContent: true, }, }