Function protoToContents

examples/docker_agent/main.go:165–239

protoToContents converts history to Gemini conversation format.

(inputs []*proto.Message)

}
// protoToContents converts history to Gemini conversation format.
func protoToContents(inputs []*proto.Message) []*genai.Content {
	var contents []*genai.Content
	for _, msg := range inputs {
		if msg.GetInternalOnly() {
			continue
		}
		role := msg.Role
		if role != "user" {
			role = "model"
		}

		content := msg.GetContent()
		if content == nil {
			continue
		}

		switch m := content.Type.(type) {
		case *proto.Content_Text:
			contents = append(contents, &genai.Content{
				Role: role,
				Parts: []*genai.Part{
					{
						Text: m.Text.Text,
					},
				},
			})
		case *proto.Content_Confirmation:
			// shouldn't be sent to Gemini
			switch m.Confirmation.Decision.(type) {
			case *proto.ConfirmationContent_Decline:
				// shouldn't be sent to Gemini
			case *proto.ConfirmationContent_Approval:
				// shouldn't be sent to Gemini
			}
		case *proto.Content_ToolCall:
			tc := m.ToolCall
			if fc := tc.GetFunctionCall(); fc != nil {
				contents = append(contents, &genai.Content{
					Role: "model",
					Parts: []*genai.Part{
						{
							ThoughtSignature: tc.Signature,
							FunctionCall: &genai.FunctionCall{
								ID:   tc.Id,
								Name: fc.Name,
								Args: fc.Arguments.AsMap(),
							},
						},
					},
				})
			}
		case *proto.Content_ToolResult:
			tr := m.ToolResult
			if fr := tr.GetFunctionResult(); fr != nil {
				var respMap map[string]any
				if fr.GetResponse() != nil {
					respMap = fr.GetResponse().AsMap()
				}