add media

This commit is contained in:
lpf
2026-02-19 20:25:20 +08:00
parent 876d9d66e6
commit 764b2d3b79
6 changed files with 345 additions and 14 deletions

View File

@@ -89,7 +89,7 @@ func (p *HTTPProvider) Chat(ctx context.Context, messages []Message, tools []Too
func (p *HTTPProvider) callChatCompletions(ctx context.Context, messages []Message, tools []ToolDefinition, model string, options map[string]interface{}) ([]byte, int, string, error) {
requestBody := map[string]interface{}{
"model": model,
"messages": messages,
"messages": toChatCompletionsMessages(messages),
}
if len(tools) > 0 {
requestBody["tools"] = tools
@@ -138,10 +138,85 @@ func (p *HTTPProvider) callResponses(ctx context.Context, messages []Message, to
return p.postJSON(ctx, endpointFor(p.apiBase, "/responses"), requestBody)
}
func toChatCompletionsMessages(messages []Message) []map[string]interface{} {
out := make([]map[string]interface{}, 0, len(messages))
for _, msg := range messages {
entry := map[string]interface{}{
"role": msg.Role,
}
content := toChatCompletionsContent(msg)
if len(content) > 0 {
entry["content"] = content
} else {
entry["content"] = msg.Content
}
if len(msg.ToolCalls) > 0 {
entry["tool_calls"] = msg.ToolCalls
}
if strings.TrimSpace(msg.ToolCallID) != "" {
entry["tool_call_id"] = msg.ToolCallID
}
out = append(out, entry)
}
return out
}
func toChatCompletionsContent(msg Message) []map[string]interface{} {
if len(msg.ContentParts) == 0 {
return nil
}
content := make([]map[string]interface{}, 0, len(msg.ContentParts))
for _, part := range msg.ContentParts {
switch strings.ToLower(strings.TrimSpace(part.Type)) {
case "input_text":
if strings.TrimSpace(part.Text) == "" {
continue
}
content = append(content, map[string]interface{}{
"type": "text",
"text": part.Text,
})
case "input_image":
if strings.TrimSpace(part.ImageURL) == "" {
continue
}
content = append(content, map[string]interface{}{
"type": "image_url",
"image_url": map[string]interface{}{
"url": part.ImageURL,
},
})
case "input_file":
fileLabel := strings.TrimSpace(part.Filename)
if fileLabel == "" {
fileLabel = "attached file"
}
mimeType := strings.TrimSpace(part.MIMEType)
if mimeType == "" {
mimeType = "application/octet-stream"
}
content = append(content, map[string]interface{}{
"type": "text",
"text": fmt.Sprintf("[file attachment: %s, mime=%s]", fileLabel, mimeType),
})
}
}
return content
}
func toResponsesInputItems(msg Message) []map[string]interface{} {
role := strings.ToLower(strings.TrimSpace(msg.Role))
switch role {
case "system", "developer", "user":
if content := responsesMessageContent(msg); len(content) > 0 {
return []map[string]interface{}{{
"type": "message",
"role": role,
"content": content,
}}
}
return []map[string]interface{}{responsesMessageItem(role, msg.Content, "input_text")}
case "assistant":
items := make([]map[string]interface{}, 0, 1+len(msg.ToolCalls))
@@ -197,6 +272,43 @@ func toResponsesInputItems(msg Message) []map[string]interface{} {
}
}
func responsesMessageContent(msg Message) []map[string]interface{} {
content := make([]map[string]interface{}, 0, len(msg.ContentParts))
for _, part := range msg.ContentParts {
switch strings.ToLower(strings.TrimSpace(part.Type)) {
case "input_text":
if strings.TrimSpace(part.Text) == "" {
continue
}
content = append(content, map[string]interface{}{
"type": "input_text",
"text": part.Text,
})
case "input_image":
if strings.TrimSpace(part.ImageURL) == "" {
continue
}
content = append(content, map[string]interface{}{
"type": "input_image",
"image_url": part.ImageURL,
})
case "input_file":
if strings.TrimSpace(part.FileData) == "" {
continue
}
entry := map[string]interface{}{
"type": "input_file",
"file_data": part.FileData,
}
if strings.TrimSpace(part.Filename) != "" {
entry["filename"] = part.Filename
}
content = append(content, entry)
}
}
return content
}
func responsesMessageItem(role, text, contentType string) map[string]interface{} {
ct := strings.TrimSpace(contentType)
if ct == "" {

View File

@@ -29,10 +29,20 @@ type UsageInfo struct {
}
type Message struct {
Role string `json:"role"`
Content string `json:"content"`
ToolCalls []ToolCall `json:"tool_calls,omitempty"`
ToolCallID string `json:"tool_call_id,omitempty"`
Role string `json:"role"`
Content string `json:"content"`
ContentParts []MessageContentPart `json:"content_parts,omitempty"`
ToolCalls []ToolCall `json:"tool_calls,omitempty"`
ToolCallID string `json:"tool_call_id,omitempty"`
}
type MessageContentPart struct {
Type string `json:"type"`
Text string `json:"text,omitempty"`
ImageURL string `json:"image_url,omitempty"`
MIMEType string `json:"mime_type,omitempty"`
Filename string `json:"filename,omitempty"`
FileData string `json:"file_data,omitempty"`
}
type LLMProvider interface {