mirror of
https://github.com/httprunner/httprunner.git
synced 2026-09-05 23:51:25 +08:00
fix: chat with screenshot
This commit is contained in:
@@ -95,14 +95,12 @@ func GetModelConfig(modelType option.LLMServiceType) (*ModelConfig, error) {
|
||||
"env %s missed", EnvModelName)
|
||||
}
|
||||
|
||||
maxTokens := 4096
|
||||
temperature := float32(0.7)
|
||||
modelConfig := &openai.ChatModelConfig{
|
||||
BaseURL: openaiBaseURL,
|
||||
APIKey: openaiAPIKey,
|
||||
Model: modelName,
|
||||
Timeout: defaultTimeout,
|
||||
MaxTokens: &maxTokens,
|
||||
Temperature: &temperature,
|
||||
}
|
||||
|
||||
|
||||
@@ -100,6 +100,15 @@ func logRequest(messages ConversationHistory) {
|
||||
func logResponse(message *schema.Message) {
|
||||
logger := log.Info().Str("role", string(message.Role)).
|
||||
Str("content", message.Content)
|
||||
|
||||
var toolCalls []string
|
||||
if len(message.ToolCalls) > 0 {
|
||||
for _, toolCall := range message.ToolCalls {
|
||||
toolCalls = append(toolCalls, toolCall.Function.Name)
|
||||
}
|
||||
logger = logger.Strs("tool_calls", toolCalls)
|
||||
}
|
||||
|
||||
if message.ResponseMeta != nil {
|
||||
logger = logger.Str("finish_reason", message.ResponseMeta.FinishReason)
|
||||
// Log usage statistics
|
||||
|
||||
Reference in New Issue
Block a user