Feat multipart (#18)

Feat multipart (#18)

Author: ming845378603

entity/prompt.go

@@ -29,8 +29,9 @@ const (
 )
 
 type Message struct {
-	Role    Role    `json:"role"`
-	Content *string `json:"content,omitempty"`
+	Role    Role           `json:"role"`
+	Content *string        `json:"content,omitempty"`
+	Parts   []*ContentPart `json:"parts,omitempty"`
 }
 
 type Role string
@@ -43,6 +44,20 @@ const (
 	RolePlaceholder Role = "placeholder"
 )
 
+type ContentPart struct {
+	Type     ContentType `json:"type"`
+	Text     *string     `json:"text,omitempty"`
+	ImageURL *string     `json:"image_url,omitempty"`
+}
+
+type ContentType string
+
+const (
+	ContentTypeText              ContentType = "text"
+	ContentTypeImageURL          ContentType = "image_url"
+	ContentTypeMultiPartVariable ContentType = "multi_part_variable"
+)
+
 type ToolType string
 
 const (
@@ -69,6 +84,7 @@ const (
 	VariableTypeArrayInteger VariableType = "array<integer>"
 	VariableTypeArrayFloat   VariableType = "array<float>"
 	VariableTypeArrayObject  VariableType = "array<object>"
+	VariableTypeMultiPart    VariableType = "multi_part"
 )
 
 type ToolChoiceType string
@@ -142,6 +158,35 @@ func (m *Message) DeepCopy() *Message {
 	if m.Content != nil {
 		copied.Content = util.Ptr(*m.Content)
 	}
+	if m.Parts != nil {
+		copied.Parts = deepCopyContentParts(m.Parts)
+	}
+	return copied
+}
+
+func deepCopyContentParts(parts []*ContentPart) []*ContentPart {
+	if parts == nil {
+		return nil
+	}
+
+	copied := make([]*ContentPart, len(parts))
+	for i, part := range parts {
+		copied[i] = part.DeepCopy()
+	}
+	return copied
+}
+
+func (cp *ContentPart) DeepCopy() *ContentPart {
+	if cp == nil {
+		return nil
+	}
+	copied := &ContentPart{
+		Type:     cp.Type,
+		ImageURL: cp.ImageURL,
+	}
+	if cp.Text != nil {
+		copied.Text = util.Ptr(*cp.Text)
+	}
 	return copied
 }
 

entity/prompt_test.go

@@ -7,6 +7,8 @@ import (
 	"testing"
 
 	. "github.com/smartystreets/goconvey/convey"
+
+	"github.com/coze-dev/cozeloop-go/internal/util"
 )
 
 func TestPromptDeepCopy(t *testing.T) {
@@ -210,6 +212,75 @@ func TestMessageDeepCopy(t *testing.T) {
 			So(copied.Role, ShouldEqual, RoleTool)
 			So(copied.Content, ShouldBeNil)
 		})
+
+		Convey("Test message with Role and Parts", func() {
+			url := "http://example.com/image.png"
+			parts := []*ContentPart{
+				{
+					Type: "text",
+					Text: util.Ptr("Hello"),
+				},
+				{
+					Type:     "image",
+					ImageURL: &url,
+				},
+			}
+			msg := &Message{
+				Role:  "user",
+				Parts: parts,
+			}
+			copied := msg.DeepCopy()
+			So(copied, ShouldNotBeNil)
+			So(copied.Role, ShouldEqual, msg.Role)
+			So(copied.Content, ShouldBeNil)
+			So(copied.Parts, ShouldNotBeNil)
+			So(len(copied.Parts), ShouldEqual, len(msg.Parts))
+			for i, part := range copied.Parts {
+				So(part.Type, ShouldEqual, msg.Parts[i].Type)
+				if part.Text != nil {
+					So(*part.Text, ShouldEqual, *msg.Parts[i].Text)
+				}
+				if part.ImageURL != nil {
+					So(part.ImageURL, ShouldEqual, msg.Parts[i].ImageURL)
+				}
+			}
+		})
+
+		Convey("Test message with Role, Content, and Parts", func() {
+			content := "Hello, World!"
+			url := "http://example.com/image.png"
+			parts := []*ContentPart{
+				{
+					Type: "text",
+					Text: util.Ptr("Hello"),
+				},
+				{
+					Type:     "image",
+					ImageURL: &url,
+				},
+			}
+			msg := &Message{
+				Role:    "user",
+				Content: &content,
+				Parts:   parts,
+			}
+			copied := msg.DeepCopy()
+			So(copied, ShouldNotBeNil)
+			So(copied.Role, ShouldEqual, msg.Role)
+			So(copied.Content, ShouldNotBeNil)
+			So(*copied.Content, ShouldEqual, *msg.Content)
+			So(copied.Parts, ShouldNotBeNil)
+			So(len(copied.Parts), ShouldEqual, len(msg.Parts))
+			for i, part := range copied.Parts {
+				So(part.Type, ShouldEqual, msg.Parts[i].Type)
+				if part.Text != nil {
+					So(*part.Text, ShouldEqual, *msg.Parts[i].Text)
+				}
+				if part.ImageURL != nil {
+					So(part.ImageURL, ShouldEqual, msg.Parts[i].ImageURL)
+				}
+			}
+		})
 	})
 }