| 162 | } |
| 163 | |
| 164 | func sanitizePluginRequest(request any) any { |
| 165 | switch req := request.(type) { |
| 166 | case pluginapi.AuthLoginStartRequest: |
| 167 | req.HTTPClient = nil |
| 168 | return req |
| 169 | case pluginapi.AuthLoginPollRequest: |
| 170 | req.HTTPClient = nil |
| 171 | return req |
| 172 | case pluginapi.AuthRefreshRequest: |
| 173 | req.HTTPClient = nil |
| 174 | return req |
| 175 | case pluginapi.AuthModelRequest: |
| 176 | req.HTTPClient = nil |
| 177 | return req |
| 178 | case pluginapi.SchedulerPickRequest: |
| 179 | req.Options.Metadata = sanitizePluginMetadata(req.Options.Metadata) |
| 180 | for index := range req.Candidates { |
| 181 | req.Candidates[index].Metadata = sanitizePluginMetadata(req.Candidates[index].Metadata) |
| 182 | } |
| 183 | return req |
| 184 | case pluginapi.ModelRouteRequest: |
| 185 | req.Metadata = sanitizePluginMetadata(req.Metadata) |
| 186 | return req |
| 187 | case pluginapi.ExecutorRequest: |
| 188 | req.HTTPClient = nil |
| 189 | req.Metadata = sanitizePluginMetadata(req.Metadata) |
| 190 | return req |
| 191 | case pluginapi.RequestInterceptRequest: |
| 192 | req.Metadata = sanitizePluginMetadata(req.Metadata) |
| 193 | return req |
| 194 | case pluginapi.ResponseInterceptRequest: |
| 195 | req.Metadata = sanitizePluginMetadata(req.Metadata) |
| 196 | return req |
| 197 | case pluginapi.StreamChunkInterceptRequest: |
| 198 | req.Metadata = sanitizePluginMetadata(req.Metadata) |
| 199 | return req |
| 200 | case rpcRequestInterceptRequest: |
| 201 | req.Metadata = sanitizePluginMetadata(req.Metadata) |
| 202 | return req |
| 203 | case rpcModelRouteRequest: |
| 204 | req.Metadata = sanitizePluginMetadata(req.Metadata) |
| 205 | return req |
| 206 | case rpcResponseInterceptRequest: |
| 207 | req.Metadata = sanitizePluginMetadata(req.Metadata) |
| 208 | return req |
| 209 | case rpcStreamChunkInterceptRequest: |
| 210 | req.Metadata = sanitizePluginMetadata(req.Metadata) |
| 211 | return req |
| 212 | case pluginapi.ExecutorHTTPRequest: |
| 213 | req.HTTPClient = nil |
| 214 | return req |
| 215 | case rpcExecutorRequest: |
| 216 | req.HTTPClient = nil |
| 217 | req.Metadata = sanitizePluginMetadata(req.Metadata) |
| 218 | return req |
| 219 | default: |
| 220 | return request |
| 221 | } |