シンプルエージェントの構築

package design

import (
    . "goa.design/goa-ai/dsl"
    . "goa.design/goa/v3/dsl"
)

var _ = Service("assistant", func() {
    Description("シンプルなアシスタントサービス")

    // アシスタント用のエージェントを定義
    Agent("helper", "リクエストを支援する有能なアシスタント", func() {
        // このエージェントにツールセットをアタッチ
        UseToolset("utilities")
    })
})

// 現在時刻ツールを持つユーティリティツールセット
var _ = Toolset("utilities", "汎用ユーティリティツール", func() {
    Tool("get_time", "現在の日付と時刻を取得する", func() {
        Result(String, "HH:MM:SS形式の現在時刻")
    })
})

goa gen <your-module>/design -o .

package main

import (
    "context"
    "time"

    "goa.design/goa-ai/runtime/agent"
    utilities "your-module/gen/assistant/toolsets/utilities"
)

func registerTools(rt *agent.Runtime) error {
    reg := utilities.NewRegistration()
    
    reg.GetTime = func(ctx context.Context, payload *utilities.GetTimePayload) (*utilities.GetTimeResult, error) {
        return &utilities.GetTimeResult{
            Time: time.Now().Format("15:04:05"),
        }, nil
    }
    
    return rt.RegisterToolset(reg)
}

package main

import (
    "context"
    "log"

    "goa.design/goa-ai/features/model/bedrock"
    "goa.design/goa-ai/runtime/agent"
    helper "your-module/gen/assistant/agents/helper"
)

func main() {
    // ランタイムを作成
    rt, err := agent.NewRuntime(context.Background(), nil, nil)
    if err != nil {
        log.Fatal(err)
    }

    // モデルクライアントを設定（例：Bedrock）
    modelClient, err := bedrock.New(/* AWS config */, bedrock.Options{
        DefaultModel: "anthropic.claude-3-5-sonnet-20241022-v2:0",
        MaxTokens:    1024,
    }, nil)
    if err != nil {
        log.Fatal(err)
    }

    // ツールを登録
    if err := registerTools(rt); err != nil {
        log.Fatal(err)
    }

    // エージェントを登録
    if err := helper.Register(rt, modelClient); err != nil {
        log.Fatal(err)
    }

    // エージェントランを実行
    result, err := rt.Run(context.Background(), helper.AgentID, "今何時ですか？")
    if err != nil {
        log.Fatal(err)
    }

    log.Printf("エージェント応答: %s", result.Reply)
}