main.go

package main

import (
	"context"
	"fmt"
	"os"

	"github.com/devinyf/dashscopego"
	"github.com/devinyf/dashscopego/qwen"
)

// 定义工具 tools.
// nolint:all
var tools = []qwen.Tool{
	{
		Type: "function",
		Function: qwen.ToolFunction{
			Name:        "get_current_weather",
			Description: "当你想查询指定城市的天气时非常有用。",
			Parameters: qwen.ToolCallParameter{
				Type: "object",
				Properties: map[string]qwen.PropertieDefinition{
					"location": {
						Type:        "string",
						Description: "城市名称",
					},
				},
			},
			Required: []string{"location"},
		},
	},
}

// 定义工具调用的函数. e.g. 天气查询.
func getCurrentWeather(cityName string) string {
	return fmt.Sprintf("%v今天有钞票雨。 ", cityName)
}

func main() {
	model := qwen.QwenTurbo
	token := os.Getenv("DASHSCOPE_API_KEY")

	if token == "" {
		panic("token is empty")
	}

	cli := dashscopego.NewTongyiClient(model, token)

	content := qwen.TextContent{Text: "青岛今天的天气怎么样?"}
	messageHistory := []dashscopego.TextMessage{
		{Role: qwen.RoleUser, Content: &content},
	}

NEXT_ROUND:
	input := dashscopego.TextInput{
		Messages: messageHistory,
	}

	req := &dashscopego.TextRequest{
		Input: input,
		Tools: tools,
		// StreamingFn: // function_call 目前好像还不支持streaming模式.
	}

	ctx := context.TODO()
	resp, err := cli.CreateCompletion(ctx, req)
	if err != nil {
		panic(err)
	}

	// 判断是否需要调用工具
	if resp.HasToolCallInput() {
		// 使用 tools.
		toolCalls := *resp.Output.Choices[0].Message.ToolCalls

		for _, toolCall := range toolCalls {
			fnName := toolCall.Function.Name
			if fnName == "get_current_weather" {
				argMap := toolCall.Function.GetArguments()
				cityName := argMap["properties"]["location"]
				toolAnswer := getCurrentWeather(cityName)
				// fmt.Println("tool answer: ", tool_answer)

				toolMessage := dashscopego.TextMessage{
					Role: qwen.RoleTool,
					Content: &qwen.TextContent{
						Text: toolAnswer,
					},
					Name: &fnName,
				}

				// 添加 assistant 的回答到消息记录(如果漏填接口会报错)
				assistantOutput := resp.Output.Choices[0].Message
				messageHistory = append(messageHistory, assistantOutput)
				// 添加 tool 的回答
				messageHistory = append(messageHistory, toolMessage)

				// 继续下一轮对话
				goto NEXT_ROUND
			}
		}
	}
	// Final result
	fmt.Println("\nnon-stream Final Result: ")                     // nolint:all
	fmt.Println(resp.Output.Choices[0].Message.Content.ToString()) // nolint:all
}