mirror of
https://github.com/alibaba/higress.git
synced 2026-05-22 03:37:26 +08:00
28 lines
1.1 KiB
Go
28 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/alibaba/higress/plugins/wasm-go/extensions/ai-proxy/config"
|
|
"github.com/higress-group/wasm-go/pkg/wrapper"
|
|
"github.com/tidwall/gjson"
|
|
)
|
|
|
|
// NeedsClaudeResponseConversionForTest exposes needsClaudeResponseConversion for unit tests.
|
|
func NeedsClaudeResponseConversionForTest(ctx wrapper.HttpContext) bool {
|
|
return needsClaudeResponseConversion(ctx)
|
|
}
|
|
|
|
// ParseGlobalConfigForTest exposes parseGlobalConfig for unit tests.
|
|
func ParseGlobalConfigForTest(json gjson.Result, pluginConfig *config.PluginConfig) error {
|
|
return parseGlobalConfig(json, pluginConfig)
|
|
}
|
|
|
|
// ParseOverrideRuleConfigForTest exposes parseOverrideRuleConfig for unit tests.
|
|
func ParseOverrideRuleConfigForTest(json gjson.Result, global config.PluginConfig, pluginConfig *config.PluginConfig) error {
|
|
return parseOverrideRuleConfig(json, global, pluginConfig)
|
|
}
|
|
|
|
// OnStreamingResponseBodyForTest exposes onStreamingResponseBody for unit tests.
|
|
func OnStreamingResponseBodyForTest(ctx wrapper.HttpContext, pluginConfig config.PluginConfig, chunk []byte, isLastChunk bool) []byte {
|
|
return onStreamingResponseBody(ctx, pluginConfig, chunk, isLastChunk)
|
|
}
|