Skip to content

Commit 8312061

Browse files
feat: register schema type (#141)
1 parent a897050 commit 8312061

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

internal/serialization/serialization.go

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,13 @@ func init() {
5151

5252
_ = GenericRegister[schema.Message]("_eino_message")
5353
_ = GenericRegister[schema.Document]("_eino_document")
54+
_ = GenericRegister[schema.RoleType]("_eino_role_type")
55+
_ = GenericRegister[schema.ChatMessagePart]("_eino_chat_message_type")
56+
_ = GenericRegister[schema.ToolCall]("_eino_tool_call")
57+
_ = GenericRegister[schema.FunctionCall]("_eino_function_call")
58+
_ = GenericRegister[schema.ResponseMeta]("_eino_response_meta")
59+
_ = GenericRegister[schema.TokenUsage]("_eino_token_usage")
60+
_ = GenericRegister[schema.LogProbs]("_eino_log_probs")
5461
}
5562

5663
func GenericRegister[T any](key string) error {

0 commit comments

Comments
 (0)