Merge pull request #48 from traPtitech/feat/deepwiki-mcp-config

feat: deep wiki mcpを追加

Author: kaitoyama

internal/gpt/gpt.go

@@ -9,10 +9,12 @@ import (
 	"time"
 
 	"github.com/traPtitech/BOT_GPT/internal/bot"
+	"github.com/traPtitech/BOT_GPT/internal/gpt/tooling"
 	"github.com/traPtitech/BOT_GPT/internal/repository"
 
 	"github.com/openai/openai-go/v2"
 	"github.com/openai/openai-go/v2/option"
+	"github.com/openai/openai-go/v2/responses"
 )
 
 type FinishReason int
@@ -26,15 +28,25 @@ const (
 var (
 	blobs                    = [...]string{":blob_bongo:", ":blob_crazy_happy:", ":blob_grin:", ":blob_hype:", ":blob_love:", ":blob_lurk:", ":blob_pyon:", ":blob_pyon_inverse:", ":blob_slide:", ":blob_snowball_1:", ":blob_snowball_2:", ":blob_speedy_roll:", ":blob_speedy_roll_inverse:", ":blob_thinking:", ":blob_thinking_fast:", ":blob_thinking_portal:", ":blob_thinking_upsidedown:", ":blob_thonkang:", ":blob_thumbs_up:", ":blobblewobble:", ":blobenjoy:", ":blobglitch:", ":blobbass:", ":blobjam:", ":blobkeyboard:", ":bloblamp:", ":blobmaracas:", ":blobmicrophone:", ":blobthinksmart:", ":blobwobwork:", ":conga_party_thinking_blob:", ":Hyperblob:", ":party_blob:", ":partyparrot_blob:", ":partyparrot_blob_cat:"}
 	amazed                   = [...]string{":amazed_fuzzy:", ":amazed_amazed_fuzzy:", ":amazed_god_enel:", ":amazed_hamutaro:"}
-	blobsAndAmazed           = append(blobs[:], amazed[:]...)
 	warnings                 = [...]string{":warning:", ":ikura-hamu_shooting_warning:"}
 	apiKey                   string
 	baseURL                  string
-	DefaultSystemRoleMessage = "あなたは日本の学生サークルである東京科学大学デジタル創作同好会traPの部内SNS「traQ」のユーザーを、楽しませる娯楽用途や勉強するための学習用途として、作られた対話型AIです。身内しかいないSNSで、ユーザーに緩く接してください。そして、ユーザーの言う事に出来る限り従うようにしてください。特定の指示がなければ、数式は\\[は使わずに$$で括った上で、\n - \\begin{align}(やequation,eqnarray,split等)は\\[は使わずに$$で括った上で、\\begin{aligned}を使う\n - \\newlineは\\\\等を使う\n - \\mboxは\\textを使う\n - \\(は使わずに$を使う\nようにしてください。"
-	ChannelMessages          = make(map[string][]Message)
+	DefaultSystemRoleMessage                  = "あなたは日本の学生サークルである東京科学大学デジタル創作同好会traPの部内SNS「traQ」のユーザーを、楽しませる娯楽用途や勉強するための学習用途として、作られた対話型AIです。身内しかいないSNSで、ユーザーに緩く接してください。そして、ユーザーの言う事に出来る限り従うようにしてください。特定の指示がなければ、数式は\\[は使わずに$$で括った上で、\n - \\begin{align}(やequation,eqnarray,split等)は\\[は使わずに$$で括った上で、\\begin{aligned}を使う\n - \\newlineは\\\\等を使う\n - \\mboxは\\textを使う\n - \\(は使わずに$を使う\nようにしてください。"
+	ChannelMessages                           = make(map[string]Message)
+	toolProvider             tooling.Provider = tooling.NewStaticProvider(tooling.DefaultSpecs())
 )
 
-type Message = openai.ChatCompletionMessageParamUnion
+func SetToolProvider(p tooling.Provider) {
+	if p == nil {
+		toolProvider = tooling.NewStaticProvider(tooling.DefaultSpecs())
+
+		return
+	}
+
+	toolProvider = p
+}
+
+type Message = []responses.ResponseInputItemUnionParam
 
 const SystemString = "FirstSystemMessageを変更しました。/gptsys showで確認できます。\nFirstSystemMessageとは、常に履歴の一番最初に入り、最初にgptに情報や状況を説明するのに使用する文字列です"
 
@@ -51,11 +63,46 @@ func InitGPT() {
 		if err != nil {
 			log.Printf("Failed to load messages for channel %s: %v", channelID, err)
 			// Initialize empty slice on error
-			ChannelMessages[channelID] = make([]Message, 0)
+			ChannelMessages[channelID] = make(Message, 0)
 		} else {
-			ChannelMessages[channelID] = messages
+			// 保存されたメッセージをResponse API形式に変換
+			convertedMessages := convertChatMessagesToResponseItems(messages)
+			ChannelMessages[channelID] = convertedMessages
+		}
+	}
+}
+
+// convertChatMessagesToResponseItems converts v2 ChatCompletionMessageParamUnion to Response API format
+func convertChatMessagesToResponseItems(messages []openai.ChatCompletionMessageParamUnion) []responses.ResponseInputItemUnionParam {
+	var result []responses.ResponseInputItemUnionParam
+
+	for _, msg := range messages {
+		if userMsg := msg.OfUser; userMsg != nil {
+			if textContent := userMsg.Content.OfString; textContent.Valid() {
+				result = append(result, responses.ResponseInputItemParamOfMessage(textContent.Value, "user"))
+			} else if len(userMsg.Content.OfArrayOfContentParts) > 0 {
+				var contentParams responses.ResponseInputMessageContentListParam
+				for _, part := range userMsg.Content.OfArrayOfContentParts {
+					if textPart := part.OfText; textPart != nil {
+						contentParams = append(contentParams, responses.ResponseInputContentParamOfInputText(textPart.Text))
+					} else if imagePart := part.OfImageURL; imagePart != nil {
+						contentParams = append(contentParams, responses.ResponseInputContentParamOfInputText("[image]"))
+					}
+				}
+				result = append(result, responses.ResponseInputItemParamOfMessage(contentParams, "user"))
+			}
+		} else if assistantMsg := msg.OfAssistant; assistantMsg != nil {
+			if textContent := assistantMsg.Content.OfString; textContent.Valid() {
+				result = append(result, responses.ResponseInputItemParamOfMessage(textContent.Value, "assistant"))
+			}
+		} else if systemMsg := msg.OfSystem; systemMsg != nil {
+			if textContent := systemMsg.Content.OfString; textContent.Valid() {
+				result = append(result, responses.ResponseInputItemParamOfMessage(textContent.Value, "system"))
+			}
 		}
 	}
+
+	return result
 }
 
 func getAPIKey() string {
@@ -84,68 +131,71 @@ func getRandomAmazed() string {
 	return amazed[rand.Intn(len(amazed))]
 }
 
-func getRandomBlobAndAmazed() string {
-	return blobsAndAmazed[rand.Intn(len(blobsAndAmazed))]
-}
-
 func getRandomWarning() string {
 	return warnings[rand.Intn(len(warnings))]
 }
 
-func OpenAIStream(messages []Message, model string, do func(string)) (responseMessage string, finishReason FinishReason, err error) {
+func OpenAIStream(messages Message, model string, do func(string)) (responseMessage string, finishReason FinishReason, err error) {
 	c := openai.NewClient(
 		option.WithAPIKey(apiKey),
 		option.WithBaseURL(baseURL),
 	)
 	ctx := context.Background()
 
-	req := openai.ChatCompletionNewParams{
-		Model:    openai.ChatModel(model),
-		Messages: messages,
+	tools, toolErr := toolProvider.Tools(ctx)
+	if toolErr != nil {
+		return "", errorHappen, fmt.Errorf("resolve tools: %w", toolErr)
+	}
+
+	// Response APIで全メッセージ履歴を使用
+	req := responses.ResponseNewParams{
+		Model: openai.ChatModel(model),
+		Input: responses.ResponseNewParamsInputUnion{
+			OfInputItemList: responses.ResponseInputParam(messages),
+		},
+		Tools: tools,
 	}
-	stream := c.Chat.Completions.NewStreaming(ctx, req)
+	stream := c.Responses.NewStreaming(ctx, req)
 	if stream.Err() != nil {
 		return "", errorHappen, stream.Err()
 	}
 	defer stream.Close()
 
-	deltaTime := 500 * time.Millisecond
-	lastDoTime := time.Now()
 	for stream.Next() {
-		response := stream.Current()
+		ev := stream.Current()
 		if stream.Err() != nil {
 			do(responseMessage + getRandomWarning() + ":blobglitch: Error: " + fmt.Sprint(stream.Err()))
 			finishReason = errorHappen
 
 			break
 		}
 
-		if len(response.Choices) > 0 {
-			choice := response.Choices[0]
-
-			if choice.FinishReason != "" {
-				if choice.FinishReason == "stop" {
-					time.Sleep(200 * time.Millisecond)
-					do(responseMessage)
-					finishReason = stop
-
-					break
-				} else if choice.FinishReason == "length" {
-					do(responseMessage + "\n" + getRandomAmazed() + "トークン(履歴を含む文字数)が上限に達しました。/resetを実行してください。")
-					finishReason = length
-
-					break
-				}
-			}
-
-			if choice.Delta.Content != "" {
-				responseMessage += choice.Delta.Content
+		switch ev.Type {
+		case "response.output_text.delta":
+			textDelta := ev.AsResponseOutputTextDelta()
+			responseMessage += textDelta.Delta
+			do(responseMessage)
+		case "response.completed":
+			finishReason = stop
+			do(responseMessage)
+		case "response.failed":
+			failedEvent := ev.AsResponseFailed()
+			do(responseMessage + getRandomAmazed() + ":blobglitch: Failed: " + failedEvent.Response.Error.Message)
+			finishReason = errorHappen
+		case "response.incomplete":
+			incompleteEvent := ev.AsResponseIncomplete()
+			// トークン上限の場合 (max_output_tokensはトークン上限を示す)
+			if incompleteEvent.Response.IncompleteDetails.Reason == "max_output_tokens" {
+				do(responseMessage + "\n" + getRandomAmazed() + "トークン(履歴を含む文字数)が上限に達しました。/resetを実行してください。")
+				finishReason = length
+			} else {
+				do(responseMessage + getRandomWarning() + ":blobglitch: Incomplete: " + incompleteEvent.Response.IncompleteDetails.Reason)
+				finishReason = errorHappen
 			}
-		}
-
-		if time.Since(lastDoTime) >= deltaTime {
-			lastDoTime = time.Now()
-			do(getRandomBlobAndAmazed() + responseMessage + ":loading:")
+		case "error":
+			errorEvent := ev.AsError()
+			do(responseMessage + getRandomWarning() + ":blobglitch: Error: " + errorEvent.Message)
+			finishReason = errorHappen
 		}
 	}
 
@@ -160,7 +210,7 @@ func OpenAIStream(messages []Message, model string, do func(string)) (responseMe
 func Chat(channelID, newMessageText string, imageBase64 []string) {
 	_, exist := ChannelMessages[channelID]
 	if !exist {
-		ChannelMessages[channelID] = make([]Message, 0)
+		ChannelMessages[channelID] = make(Message, 0)
 	}
 	addSystemMessageIfNotExist(channelID, DefaultSystemRoleMessage)
 
@@ -210,7 +260,7 @@ func ChatShowSystemMessage(channelID string) {
 
 func ChatReset(channelID string) {
 	msg := bot.PostMessage(channelID, ":blobnom::loading:")
-	ChannelMessages[channelID] = make([]Message, 0)
+	ChannelMessages[channelID] = make(Message, 0)
 	err := bot.EditMessageWithErr(msg.Id, ":done:")
 	if err != nil {
 		bot.EditMessage(msg.Id, "Error: "+fmt.Sprint(err))
@@ -222,16 +272,29 @@ func ChatReset(channelID string) {
 }
 
 func addMessageAsUser(channelID, message string) {
-	newMessage := openai.UserMessage(message)
-	ChannelMessages[channelID] = append(ChannelMessages[channelID], newMessage)
+	userMessage := responses.ResponseInputItemParamOfMessage(message, "user")
+	ChannelMessages[channelID] = append(ChannelMessages[channelID], userMessage)
 
 	index := len(ChannelMessages[channelID]) - 1
-	if err := repository.SaveMessage(channelID, index, newMessage); err != nil {
+	if err := repository.SaveMessage(channelID, index, openai.UserMessage(message)); err != nil {
 		fmt.Printf("Failed to save user message: %v\n", err)
 	}
 }
 
 func addImageAndTextAsUser(channelID, message string, imageDataBase64 []string) {
+	// 現時点では画像もテキストとして扱う
+	var content responses.ResponseInputMessageContentListParam
+	content = append(content, responses.ResponseInputContentParamOfInputText(message))
+	if len(imageDataBase64) > 0 {
+		imageText := fmt.Sprintf("[%d images attached]", len(imageDataBase64))
+		content = append(content, responses.ResponseInputContentParamOfInputText(imageText))
+	}
+
+	userMessage := responses.ResponseInputItemParamOfMessage(content, "user")
+	ChannelMessages[channelID] = append(ChannelMessages[channelID], userMessage)
+
+	index := len(ChannelMessages[channelID]) - 1
+	// repository用に旧形式で保存
 	var parts []openai.ChatCompletionContentPartUnionParam
 
 	parts = append(parts, openai.TextContentPart(message))
@@ -243,39 +306,35 @@ func addImageAndTextAsUser(channelID, message string, imageDataBase64 []string)
 		})
 		parts = append(parts, imagePart)
 	}
-
-	newMessage := openai.UserMessage(parts)
-	ChannelMessages[channelID] = append(ChannelMessages[channelID], newMessage)
-
-	index := len(ChannelMessages[channelID]) - 1
-	if err := repository.SaveMessage(channelID, index, newMessage); err != nil {
+	if err := repository.SaveMessage(channelID, index, openai.UserMessage(parts)); err != nil {
 		fmt.Printf("Failed to save image message: %v\n", err)
 	}
 }
 
 func addMessageAsAssistant(channelID, message string) {
-	newMessage := openai.AssistantMessage(message)
-	ChannelMessages[channelID] = append(ChannelMessages[channelID], newMessage)
+	assistantMessage := responses.ResponseInputItemParamOfMessage(message, "assistant")
+	ChannelMessages[channelID] = append(ChannelMessages[channelID], assistantMessage)
 
 	index := len(ChannelMessages[channelID]) - 1
-	if err := repository.SaveMessage(channelID, index, newMessage); err != nil {
+	if err := repository.SaveMessage(channelID, index, openai.AssistantMessage(message)); err != nil {
 		fmt.Printf("Failed to save assistant message: %v\n", err)
 	}
 }
 
 func addSystemMessageIfNotExist(channelID, message string) {
+	// システムメッセージが既に存在するかチェック
 	for _, msg := range ChannelMessages[channelID] {
-		if msg.OfSystem != nil {
+		if msg.GetRole() != nil && *msg.GetRole() == "system" {
 			return
 		}
 	}
 
-	systemMessage := openai.SystemMessage(message)
-
-	ChannelMessages[channelID] = append([]Message{systemMessage}, ChannelMessages[channelID]...)
+	// システムメッセージが存在しない場合のみ先頭に追加
+	systemMessage := responses.ResponseInputItemParamOfMessage(message, "system")
+	ChannelMessages[channelID] = append([]responses.ResponseInputItemUnionParam{systemMessage}, ChannelMessages[channelID]...)
 
 	index := 0
-	if err := repository.SaveMessage(channelID, index, systemMessage); err != nil {
+	if err := repository.SaveMessage(channelID, index, openai.SystemMessage(message)); err != nil {
 		fmt.Printf("Failed to save system message: %v\n", err)
 	}
 }