feat: introduce usability functions to allow quick creation of user content and model content.

PiperOrigin-RevId: 728792689

Author: yyyu-google

models.go

@@ -298,3 +298,225 @@ func TestNewPartFromCodeExecutionResult(t *testing.T) {
 		t.Fatalf("expected %v, got %v", expected, result)
 	}
 }
+
+func TestNewUserContentFromParts(t *testing.T) {
+	parts := []*Part{
+		{Text: "Hello, world!"},
+		{Text: "This is a test."},
+	}
+	expected := &Content{
+		Parts: parts,
+		Role:  "user",
+	}
+
+	result := NewUserContentFromParts(parts)
+	if !reflect.DeepEqual(result, expected) {
+		t.Fatalf("expected %v, got %v", expected, result)
+	}
+}
+
+func TestNewUserContentFromText(t *testing.T) {
+	text := "Hello, world!"
+	expected := &Content{
+		Parts: []*Part{
+			{Text: "Hello, world!"},
+		},
+		Role: "user",
+	}
+
+	result := NewUserContentFromText(text)
+	if !reflect.DeepEqual(result, expected) {
+		t.Fatalf("expected %v, got %v", expected, result)
+	}
+}
+
+func TestNewUserContentFromBytes(t *testing.T) {
+	data := []byte{0x01, 0x02, 0x03}
+	mimeType := "application/octet-stream"
+	expected := &Content{
+		Parts: []*Part{
+			{InlineData: &Blob{Data: data, MIMEType: mimeType}},
+		},
+		Role: "user",
+	}
+
+	result := NewUserContentFromBytes(data, mimeType)
+	if !reflect.DeepEqual(result, expected) {
+		t.Fatalf("expected %v, got %v", expected, result)
+	}
+}
+
+func TestNewUserContentFromURI(t *testing.T) {
+	fileURI := "http://example.com/video.mp4"
+	mimeType := "video/mp4"
+	expected := &Content{
+		Parts: []*Part{
+			{FileData: &FileData{FileURI: fileURI, MIMEType: mimeType}},
+		},
+		Role: "user",
+	}
+
+	result := NewUserContentFromURI(fileURI, mimeType)
+	if !reflect.DeepEqual(result, expected) {
+		t.Fatalf("expected %v, got %v", expected, result)
+	}
+}
+
+func TestNewUserContentFromFunctionResponse(t *testing.T) {
+	funcName := "myFunction"
+	response := map[string]any{"result": "success"}
+	expected := &Content{
+		Parts: []*Part{
+			{FunctionResponse: &FunctionResponse{Name: funcName, Response: response}},
+		},
+		Role: "user",
+	}
+
+	result := NewUserContentFromFunctionResponse(funcName, response)
+	if !reflect.DeepEqual(result, expected) {
+		t.Fatalf("expected %v, got %v", expected, result)
+	}
+}
+
+func TestNewUserContentFromExecutableCode(t *testing.T) {
+	code := "print('Hello, world!')"
+	language := LanguagePython
+	expected := &Content{
+		Parts: []*Part{
+			{ExecutableCode: &ExecutableCode{Code: code, Language: language}},
+		},
+		Role: "user",
+	}
+
+	result := NewUserContentFromExecutableCode(code, language)
+	if !reflect.DeepEqual(result, expected) {
+		t.Fatalf("expected %v, got %v", expected, result)
+	}
+}
+
+func TestNewUserContentFromCodeExecutionResult(t *testing.T) {
+	outcome := OutcomeOK
+	output := "Execution output"
+	expected := &Content{
+		Parts: []*Part{
+			{CodeExecutionResult: &CodeExecutionResult{Outcome: outcome, Output: output}},
+		},
+		Role: "user",
+	}
+
+	result := NewUserContentFromCodeExecutionResult(outcome, output)
+	if !reflect.DeepEqual(result, expected) {
+		t.Fatalf("expected %v, got %v", expected, result)
+	}
+}
+
+func TestNewModelContentFromParts(t *testing.T) {
+	parts := []*Part{
+		{Text: "Hello, world!"},
+		{Text: "This is a test."},
+	}
+	expected := &Content{
+		Parts: parts,
+		Role:  "model",
+	}
+
+	result := NewModelContentFromParts(parts)
+	if !reflect.DeepEqual(result, expected) {
+		t.Fatalf("expected %v, got %v", expected, result)
+	}
+}
+
+func TestNewModelContentFromText(t *testing.T) {
+	text := "Hello, world!"
+	expected := &Content{
+		Parts: []*Part{
+			{Text: "Hello, world!"},
+		},
+		Role: "model",
+	}
+
+	result := NewModelContentFromText(text)
+	if !reflect.DeepEqual(result, expected) {
+		t.Fatalf("expected %v, got %v", expected, result)
+	}
+}
+
+func TestNewModelContentFromBytes(t *testing.T) {
+	data := []byte{0x01, 0x02, 0x03}
+	mimeType := "application/octet-stream"
+	expected := &Content{
+		Parts: []*Part{
+			{InlineData: &Blob{Data: data, MIMEType: mimeType}},
+		},
+		Role: "model",
+	}
+
+	result := NewModelContentFromBytes(data, mimeType)
+	if !reflect.DeepEqual(result, expected) {
+		t.Fatalf("expected %v, got %v", expected, result)
+	}
+}
+
+func TestNewModelContentFromURI(t *testing.T) {
+	fileURI := "http://example.com/video.mp4"
+	mimeType := "video/mp4"
+	expected := &Content{
+		Parts: []*Part{
+			{FileData: &FileData{FileURI: fileURI, MIMEType: mimeType}},
+		},
+		Role: "model",
+	}
+
+	result := NewModelContentFromURI(fileURI, mimeType)
+	if !reflect.DeepEqual(result, expected) {
+		t.Fatalf("expected %v, got %v", expected, result)
+	}
+}
+
+func TestNewModelContentFromFunctionCall(t *testing.T) {
+	funcName := "myFunction"
+	args := map[string]any{"arg1": "value1"}
+	expected := &Content{
+		Parts: []*Part{
+			{FunctionCall: &FunctionCall{Name: funcName, Args: args}},
+		},
+		Role: "model",
+	}
+
+	result := NewModelContentFromFunctionCall(funcName, args)
+	if !reflect.DeepEqual(result, expected) {
+		t.Fatalf("expected %v, got %v", expected, result)
+	}
+}
+
+func TestNewModelContentFromExecutableCode(t *testing.T) {
+	code := "print('Hello, world!')"
+	language := LanguagePython
+	expected := &Content{
+		Parts: []*Part{
+			{ExecutableCode: &ExecutableCode{Code: code, Language: language}},
+		},
+		Role: "model",
+	}
+
+	result := NewModelContentFromExecutableCode(code, language)
+	if !reflect.DeepEqual(result, expected) {
+		t.Fatalf("expected %v, got %v", expected, result)
+	}
+}
+
+func TestNewModelContentFromCodeExecutionResult(t *testing.T) {
+	outcome := OutcomeOK
+	output := "Execution output"
+	expected := &Content{
+		Parts: []*Part{
+			{CodeExecutionResult: &CodeExecutionResult{Outcome: outcome, Output: output}},
+		},
+		Role: "model",
+	}
+
+	result := NewModelContentFromCodeExecutionResult(outcome, output)
+	if !reflect.DeepEqual(result, expected) {
+		t.Fatalf("expected %v, got %v", expected, result)
+	}
+}