|
@@ -1,6 +1,8 @@
|
|
package requests
|
|
package requests
|
|
|
|
|
|
import (
|
|
import (
|
|
|
|
+ "encoding/json"
|
|
|
|
+
|
|
"github.com/langgenius/dify-plugin-daemon/pkg/entities/model_entities"
|
|
"github.com/langgenius/dify-plugin-daemon/pkg/entities/model_entities"
|
|
)
|
|
)
|
|
|
|
|
|
@@ -133,6 +135,18 @@ type RequestGetLLMNumTokens struct {
|
|
Tools []model_entities.PromptMessageTool `json:"tools" validate:"omitempty,dive"`
|
|
Tools []model_entities.PromptMessageTool `json:"tools" validate:"omitempty,dive"`
|
|
}
|
|
}
|
|
|
|
|
|
|
|
+func (r RequestGetLLMNumTokens) MarshalJSON() ([]byte, error) {
|
|
|
|
+ type alias RequestGetLLMNumTokens
|
|
|
|
+ p := alias(r)
|
|
|
|
+ if p.PromptMessages == nil {
|
|
|
|
+ p.PromptMessages = []model_entities.PromptMessage{}
|
|
|
|
+ }
|
|
|
|
+ if p.Tools == nil {
|
|
|
|
+ p.Tools = []model_entities.PromptMessageTool{}
|
|
|
|
+ }
|
|
|
|
+ return json.Marshal(p)
|
|
|
|
+}
|
|
|
|
+
|
|
type RequestGetAIModelSchema struct {
|
|
type RequestGetAIModelSchema struct {
|
|
BaseRequestInvokeModel
|
|
BaseRequestInvokeModel
|
|
Credentials
|
|
Credentials
|