thread.go

// Code generated by Fern. DO NOT EDIT.

package zep

import (
	json "encoding/json"
	fmt "fmt"
	internal "github.com/getzep/zep-go/v3/internal"
)

type CreateThreadRequest struct {
	// The unique identifier of the thread.
	ThreadID string `json:"thread_id" url:"-"`
	// The unique identifier of the user associated with the thread
	UserID string `json:"user_id" url:"-"`
}

type ThreadGetRequest struct {
	// Limit the number of results returned
	Limit *int `json:"-" url:"limit,omitempty"`
	// Cursor for pagination
	Cursor *int `json:"-" url:"cursor,omitempty"`
	// Number of most recent messages to return (overrides limit and cursor)
	Lastn *int `json:"-" url:"lastn,omitempty"`
}

type ThreadGetUserContextRequest struct {
	// Optional template ID to use for custom context rendering.
	TemplateID *string `json:"-" url:"template_id,omitempty"`
}

type ThreadListAllRequest struct {
	// Page number for pagination, starting from 1
	PageNumber *int `json:"-" url:"page_number,omitempty"`
	// Number of threads to retrieve per page.
	PageSize *int `json:"-" url:"page_size,omitempty"`
	// Field to order the results by: created_at, updated_at, user_id, thread_id.
	OrderBy *string `json:"-" url:"order_by,omitempty"`
	// Order direction: true for ascending, false for descending.
	Asc *bool `json:"-" url:"asc,omitempty"`
}

type AddThreadMessagesRequest struct {
	// Optional list of role types to ignore when adding messages to graph memory.
	// The message itself will still be added, retained and used as context for messages
	// that are added to a user's graph.
	IgnoreRoles []RoleType `json:"ignore_roles,omitempty" url:"ignore_roles,omitempty"`
	// A list of message objects, where each message contains a role and content.
	Messages []*Message `json:"messages" url:"messages"`
	// Optionally return context block relevant to the most recent messages.
	ReturnContext *bool `json:"return_context,omitempty" url:"return_context,omitempty"`
	// When true, prevents extraction of generic Entity nodes that do not match the configured ontology.
	StrictOntology *bool `json:"strict_ontology,omitempty" url:"strict_ontology,omitempty"`

	extraProperties map[string]interface{}
	rawJSON         json.RawMessage
}

func (a *AddThreadMessagesRequest) GetIgnoreRoles() []RoleType {
	if a == nil {
		return nil
	}
	return a.IgnoreRoles
}

func (a *AddThreadMessagesRequest) GetMessages() []*Message {
	if a == nil {
		return nil
	}
	return a.Messages
}

func (a *AddThreadMessagesRequest) GetReturnContext() *bool {
	if a == nil {
		return nil
	}
	return a.ReturnContext
}

func (a *AddThreadMessagesRequest) GetStrictOntology() *bool {
	if a == nil {
		return nil
	}
	return a.StrictOntology
}

func (a *AddThreadMessagesRequest) GetExtraProperties() map[string]interface{} {
	return a.extraProperties
}

func (a *AddThreadMessagesRequest) UnmarshalJSON(data []byte) error {
	type unmarshaler AddThreadMessagesRequest
	var value unmarshaler
	if err := json.Unmarshal(data, &value); err != nil {
		return err
	}
	*a = AddThreadMessagesRequest(value)
	extraProperties, err := internal.ExtractExtraProperties(data, *a)
	if err != nil {
		return err
	}
	a.extraProperties = extraProperties
	a.rawJSON = json.RawMessage(data)
	return nil
}

func (a *AddThreadMessagesRequest) String() string {
	if len(a.rawJSON) > 0 {
		if value, err := internal.StringifyJSON(a.rawJSON); err == nil {
			return value
		}
	}
	if value, err := internal.StringifyJSON(a); err == nil {
		return value
	}
	return fmt.Sprintf("%#v", a)
}

type AddThreadMessagesResponse struct {
	Context      *string  `json:"context,omitempty" url:"context,omitempty"`
	MessageUUIDs []string `json:"message_uuids,omitempty" url:"message_uuids,omitempty"`
	TaskID       *string  `json:"task_id,omitempty" url:"task_id,omitempty"`

	extraProperties map[string]interface{}
	rawJSON         json.RawMessage
}

func (a *AddThreadMessagesResponse) GetContext() *string {
	if a == nil {
		return nil
	}
	return a.Context
}

func (a *AddThreadMessagesResponse) GetMessageUUIDs() []string {
	if a == nil {
		return nil
	}
	return a.MessageUUIDs
}

func (a *AddThreadMessagesResponse) GetTaskID() *string {
	if a == nil {
		return nil
	}
	return a.TaskID
}

func (a *AddThreadMessagesResponse) GetExtraProperties() map[string]interface{} {
	return a.extraProperties
}

func (a *AddThreadMessagesResponse) UnmarshalJSON(data []byte) error {
	type unmarshaler AddThreadMessagesResponse
	var value unmarshaler
	if err := json.Unmarshal(data, &value); err != nil {
		return err
	}
	*a = AddThreadMessagesResponse(value)
	extraProperties, err := internal.ExtractExtraProperties(data, *a)
	if err != nil {
		return err
	}
	a.extraProperties = extraProperties
	a.rawJSON = json.RawMessage(data)
	return nil
}

func (a *AddThreadMessagesResponse) String() string {
	if len(a.rawJSON) > 0 {
		if value, err := internal.StringifyJSON(a.rawJSON); err == nil {
			return value
		}
	}
	if value, err := internal.StringifyJSON(a); err == nil {
		return value
	}
	return fmt.Sprintf("%#v", a)
}

type MessageListResponse struct {
	// A list of message objects.
	Messages []*Message `json:"messages,omitempty" url:"messages,omitempty"`
	// The number of messages returned.
	RowCount *int `json:"row_count,omitempty" url:"row_count,omitempty"`
	// The thread creation timestamp.
	ThreadCreatedAt *string `json:"thread_created_at,omitempty" url:"thread_created_at,omitempty"`
	// The total number of messages.
	TotalCount *int `json:"total_count,omitempty" url:"total_count,omitempty"`
	// The user ID associated with this thread.
	UserID *string `json:"user_id,omitempty" url:"user_id,omitempty"`
	// The opaque user identifier used by dashboard routes.
	UserUUID *string `json:"user_uuid,omitempty" url:"user_uuid,omitempty"`

	extraProperties map[string]interface{}
	rawJSON         json.RawMessage
}

func (m *MessageListResponse) GetMessages() []*Message {
	if m == nil {
		return nil
	}
	return m.Messages
}

func (m *MessageListResponse) GetRowCount() *int {
	if m == nil {
		return nil
	}
	return m.RowCount
}

func (m *MessageListResponse) GetThreadCreatedAt() *string {
	if m == nil {
		return nil
	}
	return m.ThreadCreatedAt
}

func (m *MessageListResponse) GetTotalCount() *int {
	if m == nil {
		return nil
	}
	return m.TotalCount
}

func (m *MessageListResponse) GetUserID() *string {
	if m == nil {
		return nil
	}
	return m.UserID
}

func (m *MessageListResponse) GetUserUUID() *string {
	if m == nil {
		return nil
	}
	return m.UserUUID
}

func (m *MessageListResponse) GetExtraProperties() map[string]interface{} {
	return m.extraProperties
}

func (m *MessageListResponse) UnmarshalJSON(data []byte) error {
	type unmarshaler MessageListResponse
	var value unmarshaler
	if err := json.Unmarshal(data, &value); err != nil {
		return err
	}
	*m = MessageListResponse(value)
	extraProperties, err := internal.ExtractExtraProperties(data, *m)
	if err != nil {
		return err
	}
	m.extraProperties = extraProperties
	m.rawJSON = json.RawMessage(data)
	return nil
}

func (m *MessageListResponse) String() string {
	if len(m.rawJSON) > 0 {
		if value, err := internal.StringifyJSON(m.rawJSON); err == nil {
			return value
		}
	}
	if value, err := internal.StringifyJSON(m); err == nil {
		return value
	}
	return fmt.Sprintf("%#v", m)
}

type ThreadContextResponse struct {
	// Context block containing relevant facts, entities, and messages/episodes from the user graph. Meant to be replaced in the system prompt on every chat turn.
	Context *string `json:"context,omitempty" url:"context,omitempty"`

	extraProperties map[string]interface{}
	rawJSON         json.RawMessage
}

func (t *ThreadContextResponse) GetContext() *string {
	if t == nil {
		return nil
	}
	return t.Context
}

func (t *ThreadContextResponse) GetExtraProperties() map[string]interface{} {
	return t.extraProperties
}

func (t *ThreadContextResponse) UnmarshalJSON(data []byte) error {
	type unmarshaler ThreadContextResponse
	var value unmarshaler
	if err := json.Unmarshal(data, &value); err != nil {
		return err
	}
	*t = ThreadContextResponse(value)
	extraProperties, err := internal.ExtractExtraProperties(data, *t)
	if err != nil {
		return err
	}
	t.extraProperties = extraProperties
	t.rawJSON = json.RawMessage(data)
	return nil
}

func (t *ThreadContextResponse) String() string {
	if len(t.rawJSON) > 0 {
		if value, err := internal.StringifyJSON(t.rawJSON); err == nil {
			return value
		}
	}
	if value, err := internal.StringifyJSON(t); err == nil {
		return value
	}
	return fmt.Sprintf("%#v", t)
}

type ThreadListResponse struct {
	ResponseCount *int      `json:"response_count,omitempty" url:"response_count,omitempty"`
	Threads       []*Thread `json:"threads,omitempty" url:"threads,omitempty"`
	TotalCount    *int      `json:"total_count,omitempty" url:"total_count,omitempty"`

	extraProperties map[string]interface{}
	rawJSON         json.RawMessage
}

func (t *ThreadListResponse) GetResponseCount() *int {
	if t == nil {
		return nil
	}
	return t.ResponseCount
}

func (t *ThreadListResponse) GetThreads() []*Thread {
	if t == nil {
		return nil
	}
	return t.Threads
}

func (t *ThreadListResponse) GetTotalCount() *int {
	if t == nil {
		return nil
	}
	return t.TotalCount
}

func (t *ThreadListResponse) GetExtraProperties() map[string]interface{} {
	return t.extraProperties
}

func (t *ThreadListResponse) UnmarshalJSON(data []byte) error {
	type unmarshaler ThreadListResponse
	var value unmarshaler
	if err := json.Unmarshal(data, &value); err != nil {
		return err
	}
	*t = ThreadListResponse(value)
	extraProperties, err := internal.ExtractExtraProperties(data, *t)
	if err != nil {
		return err
	}
	t.extraProperties = extraProperties
	t.rawJSON = json.RawMessage(data)
	return nil
}

func (t *ThreadListResponse) String() string {
	if len(t.rawJSON) > 0 {
		if value, err := internal.StringifyJSON(t.rawJSON); err == nil {
			return value
		}
	}
	if value, err := internal.StringifyJSON(t); err == nil {
		return value
	}
	return fmt.Sprintf("%#v", t)
}