feat: register support model

completion.go

@@ -166,6 +166,10 @@ func checkEndpointSupportsModel(endpoint, model string) bool {
 	return !disabledModelsForEndpoints[endpoint][model]
 }
 
+func RegisterSupportsModel(endpoint, model string) {
+	disabledModelsForEndpoints[endpoint][model] = true
+}
+
 func checkPromptType(prompt any) bool {
 	_, isString := prompt.(string)
 	_, isStringSlice := prompt.([]string)

completion_test.go

@@ -300,3 +300,27 @@ func TestCompletionWithGPT4oModels(t *testing.T) {
 		})
 	}
 }
+
+func TestRegisterSupportsModel(t *testing.T) {
+	type args struct {
+		endpoint string
+		model    string
+	}
+	tests := []struct {
+		name string
+		args args
+	}{
+		{
+			name: "Register model ",
+			args: args{
+				endpoint: "/chat/completions",
+				model:    "local-model-3.5",
+			},
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(_ *testing.T) {
+			openai.RegisterSupportsModel(tt.args.endpoint, tt.args.model)
+		})
+	}
+}