feat: manually mirror opencoze's code from bytedance
Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf
This commit is contained in:
32
backend/domain/conversation/message/entity/const.go
Normal file
32
backend/domain/conversation/message/entity/const.go
Normal file
@@ -0,0 +1,32 @@
|
||||
/*
|
||||
* Copyright 2025 coze-dev Authors
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package entity
|
||||
|
||||
type ScrollPageDirection string
|
||||
|
||||
const (
|
||||
ScrollPageDirectionPrev ScrollPageDirection = "up"
|
||||
ScrollPageDirectionNext ScrollPageDirection = "down"
|
||||
)
|
||||
|
||||
type MessageStatus int32
|
||||
|
||||
const (
|
||||
MessageStatusAvailable MessageStatus = 1
|
||||
MessageStatusDeleted MessageStatus = 2
|
||||
MessageStatusBroken MessageStatus = 4
|
||||
)
|
||||
64
backend/domain/conversation/message/entity/knowledge.go
Normal file
64
backend/domain/conversation/message/entity/knowledge.go
Normal file
@@ -0,0 +1,64 @@
|
||||
/*
|
||||
* Copyright 2025 coze-dev Authors
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package entity
|
||||
|
||||
type VerboseInfo struct {
|
||||
MessageType string `json:"msg_type"`
|
||||
Data string `json:"data"`
|
||||
}
|
||||
|
||||
type VerboseData struct {
|
||||
Chunks []RecallDataInfo `json:"chunks"`
|
||||
OriReq string `json:"ori_req"`
|
||||
StatusCode int `json:"status_code"`
|
||||
}
|
||||
|
||||
type RecallDataInfo struct {
|
||||
Slice string `json:"slice"`
|
||||
Score float64 `json:"score"`
|
||||
Meta MetaInfo `json:"meta"`
|
||||
}
|
||||
|
||||
type MetaInfo struct {
|
||||
Dataset DatasetInfo `json:"dataset"`
|
||||
Document DocumentInfo `json:"document"`
|
||||
Link LinkInfo `json:"link"`
|
||||
Card CardInfo `json:"card"`
|
||||
}
|
||||
|
||||
type DatasetInfo struct {
|
||||
ID string `json:"id"`
|
||||
Name string `json:"name"`
|
||||
}
|
||||
|
||||
type DocumentInfo struct {
|
||||
ID string `json:"id"`
|
||||
Name string `json:"name"`
|
||||
FormatType int32 `json:"format_type"`
|
||||
SourceType int32 `json:"source_type"`
|
||||
}
|
||||
|
||||
type LinkInfo struct {
|
||||
Title string `json:"title"`
|
||||
URL string `json:"url"`
|
||||
}
|
||||
|
||||
type CardInfo struct {
|
||||
Title string `json:"title"`
|
||||
Con string `json:"con"`
|
||||
Index string `json:"index"`
|
||||
}
|
||||
55
backend/domain/conversation/message/entity/message.go
Normal file
55
backend/domain/conversation/message/entity/message.go
Normal file
@@ -0,0 +1,55 @@
|
||||
/*
|
||||
* Copyright 2025 coze-dev Authors
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package entity
|
||||
|
||||
import "github.com/coze-dev/coze-studio/backend/api/model/crossdomain/message"
|
||||
|
||||
type Message = message.Message
|
||||
|
||||
type ListMeta struct {
|
||||
ConversationID int64 `json:"conversation_id"`
|
||||
RunID []*int64 `json:"run_id"`
|
||||
UserID string `json:"user_id"`
|
||||
AgentID int64 `json:"agent_id"`
|
||||
OrderBy *string `json:"order_by"`
|
||||
Limit int `json:"limit"`
|
||||
Cursor int64 `json:"cursor"` // message id
|
||||
Direction ScrollPageDirection `json:"direction"` // "prev" "Next"
|
||||
}
|
||||
|
||||
type ListResult struct {
|
||||
Messages []*Message `json:"messages"`
|
||||
PrevCursor int64 `json:"prev_cursor"`
|
||||
NextCursor int64 `json:"next_cursor"`
|
||||
HasMore bool `json:"has_more"`
|
||||
Direction ScrollPageDirection `json:"direction"`
|
||||
}
|
||||
|
||||
type GetByRunIDsRequest struct {
|
||||
ConversationID int64 `json:"conversation_id"`
|
||||
RunID []int64 `json:"run_id"`
|
||||
}
|
||||
|
||||
type DeleteMeta struct {
|
||||
MessageIDs []int64 `json:"message_ids"`
|
||||
RunIDs []int64 `json:"run_ids"`
|
||||
}
|
||||
|
||||
type BrokenMeta struct {
|
||||
ID int64 `json:"id"`
|
||||
Position *int32 `json:"position"`
|
||||
}
|
||||
55
backend/domain/conversation/message/entity/message_ext.go
Normal file
55
backend/domain/conversation/message/entity/message_ext.go
Normal file
@@ -0,0 +1,55 @@
|
||||
/*
|
||||
* Copyright 2025 coze-dev Authors
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package entity
|
||||
|
||||
type MessageExtKey string
|
||||
|
||||
const (
|
||||
MessageExtKeyInputTokens MessageExtKey = "input_tokens"
|
||||
MessageExtKeyOutputTokens MessageExtKey = "output_tokens"
|
||||
MessageExtKeyToken MessageExtKey = "token"
|
||||
MessageExtKeyPluginStatus MessageExtKey = "plugin_status"
|
||||
MessageExtKeyTimeCost MessageExtKey = "time_cost"
|
||||
MessageExtKeyWorkflowTokens MessageExtKey = "workflow_tokens"
|
||||
MessageExtKeyBotState MessageExtKey = "bot_state"
|
||||
MessageExtKeyPluginRequest MessageExtKey = "plugin_request"
|
||||
MessageExtKeyToolName MessageExtKey = "tool_name"
|
||||
MessageExtKeyPlugin MessageExtKey = "plugin"
|
||||
MessageExtKeyMockHitInfo MessageExtKey = "mock_hit_info"
|
||||
MessageExtKeyMessageTitle MessageExtKey = "message_title"
|
||||
MessageExtKeyStreamPluginRunning MessageExtKey = "stream_plugin_running"
|
||||
MessageExtKeyExecuteDisplayName MessageExtKey = "execute_display_name"
|
||||
MessageExtKeyTaskType MessageExtKey = "task_type"
|
||||
MessageExtKeyCallID MessageExtKey = "call_id"
|
||||
ExtKeyResumeInfo MessageExtKey = "resume_info"
|
||||
ExtKeyBreakPoint MessageExtKey = "break_point"
|
||||
ExtKeyToolCallsIDs MessageExtKey = "tool_calls_ids"
|
||||
ExtKeyRequiresAction MessageExtKey = "requires_action"
|
||||
)
|
||||
|
||||
type BotStateExt struct {
|
||||
BotID string `json:"bot_id"`
|
||||
AgentName string `json:"agent_name"`
|
||||
AgentID string `json:"agent_id"`
|
||||
Awaiting string `json:"awaiting"`
|
||||
}
|
||||
|
||||
type UsageExt struct {
|
||||
TotalCount int64 `json:"total_count"`
|
||||
InputTokens int64 `json:"input_tokens"`
|
||||
OutputTokens int64 `json:"output_tokens"`
|
||||
}
|
||||
Reference in New Issue
Block a user