mirror of
https://github.com/laoxong/nofx.git
synced 2026-06-04 09:58:22 +08:00
shinchan-zhai
2cdc3d0cd8
* feat: add X-Client-ID header for claw402 monitoring * fix: set temperature=1 for kimi provider (k2.5 model restriction)
75 lines
1.9 KiB
Go
75 lines
1.9 KiB
Go
package provider
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"nofx/mcp"
|
|
)
|
|
|
|
const (
|
|
DefaultKimiBaseURL = "https://api.moonshot.ai/v1" // Global endpoint (use api.moonshot.cn for China)
|
|
DefaultKimiModel = "moonshot-v1-auto"
|
|
)
|
|
|
|
func init() {
|
|
mcp.RegisterProvider(mcp.ProviderKimi, func(opts ...mcp.ClientOption) mcp.AIClient {
|
|
return NewKimiClientWithOptions(opts...)
|
|
})
|
|
}
|
|
|
|
type KimiClient struct {
|
|
*mcp.Client
|
|
}
|
|
|
|
func (c *KimiClient) BaseClient() *mcp.Client { return c.Client }
|
|
|
|
// NewKimiClient creates Kimi (Moonshot) client (backward compatible)
|
|
func NewKimiClient() mcp.AIClient {
|
|
return NewKimiClientWithOptions()
|
|
}
|
|
|
|
// NewKimiClientWithOptions creates Kimi client (supports options pattern)
|
|
func NewKimiClientWithOptions(opts ...mcp.ClientOption) mcp.AIClient {
|
|
kimiOpts := []mcp.ClientOption{
|
|
mcp.WithProvider(mcp.ProviderKimi),
|
|
mcp.WithModel(DefaultKimiModel),
|
|
mcp.WithBaseURL(DefaultKimiBaseURL),
|
|
mcp.WithTemperature(1.0), // Kimi K2.5 only allows temperature=1
|
|
}
|
|
|
|
allOpts := append(kimiOpts, opts...)
|
|
baseClient := mcp.NewClient(allOpts...).(*mcp.Client)
|
|
|
|
kimiClient := &KimiClient{
|
|
Client: baseClient,
|
|
}
|
|
|
|
baseClient.Hooks = kimiClient
|
|
return kimiClient
|
|
}
|
|
|
|
func (c *KimiClient) SetAPIKey(apiKey string, customURL string, customModel string) {
|
|
c.APIKey = apiKey
|
|
|
|
if len(apiKey) > 8 {
|
|
c.Log.Infof("🔧 [MCP] Kimi API Key: %s...%s", apiKey[:4], apiKey[len(apiKey)-4:])
|
|
}
|
|
if customURL != "" {
|
|
c.BaseURL = customURL
|
|
c.Log.Infof("🔧 [MCP] Kimi using custom BaseURL: %s", customURL)
|
|
} else {
|
|
c.Log.Infof("🔧 [MCP] Kimi using default BaseURL: %s", c.BaseURL)
|
|
}
|
|
if customModel != "" {
|
|
c.Model = customModel
|
|
c.Log.Infof("🔧 [MCP] Kimi using custom Model: %s", customModel)
|
|
} else {
|
|
c.Log.Infof("🔧 [MCP] Kimi using default Model: %s", c.Model)
|
|
}
|
|
}
|
|
|
|
// Kimi uses standard OpenAI-compatible API
|
|
func (c *KimiClient) SetAuthHeader(reqHeaders http.Header) {
|
|
c.Client.SetAuthHeader(reqHeaders)
|
|
}
|