cozeloop

type AuthError = consts.AuthError

type Client interface {
	// PromptClient interface of prompt client
	PromptClient
	// TraceClient interface of trace client
	TraceClient

	// GetWorkspaceID return workspace id
	GetWorkspaceID() string
	// Close Close the client. Should be called before program exit.
	Close(ctx context.Context)
}

type GetPromptOption func(option *prompt.GetPromptOptions)

type GetPromptParam = prompt.GetPromptParam

type HttpClient = httpclient.HTTPClient

type LogLevel = logger.LogLevel

type Logger = logger.Logger

type Option func(o *options)

type PromptClient interface {
	// GetPrompt get prompt by prompt key and version.
	// if version is not set,  the latest version of the corresponding prompt will be obtained.
	GetPrompt(ctx context.Context, param GetPromptParam, options ...GetPromptOption) (*entity.Prompt, error)
	// PromptFormat format prompt with variables
	PromptFormat(ctx context.Context, prompt *entity.Prompt, variables map[string]any, options ...PromptFormatOption) (messages []*entity.Message, err error)
}

type PromptFormatOption func(option *prompt.PromptFormatOptions)

type RemoteServiceError = consts.RemoteServiceError

type Span interface {
	SpanContext

	// SetTags sets business custom tags.
	SetTags(ctx context.Context, tagKVs map[string]interface{})

	// SetBaggage sets tags and also passes these tags to other downstream spans (assuming
	//the user uses ToHeader and FromHeader to handle header passing between services).
	SetBaggage(ctx context.Context, baggageItems map[string]string)

	// Finish The span will be reported only after an explicit call to Finish.
	// Under the hood, it is actually placed in an asynchronous queue waiting to be reported.
	Finish(ctx context.Context)

	// GetStartTime returns the start time of the Span.
	GetStartTime() time.Time

	// ToHeader Convert the span to headers. Used for cross-process correlation.
	ToHeader() (map[string]string, error)
	// contains filtered or unexported methods
}

type SpanContext interface {
	GetSpanID() string
	GetTraceID() string
	GetBaggage() map[string]string
}

type StartSpanOption = func(o *startSpanOptions)

type TraceClient interface {
	// StartSpan Generate a span that automatically links to the previous span in the context.
	// The start time of the span starts counting from the call of StartSpan.
	// The generated span will be automatically written into the context.
	// Subsequent spans that need to be chained should call StartSpan based on the new context.
	StartSpan(ctx context.Context, name, spanType string, opts ...StartSpanOption) (context.Context, Span)
	// GetSpanFromContext Get the span from the context.
	GetSpanFromContext(ctx context.Context) Span
	// GetSpanFromHeader Get the span from the header.
	GetSpanFromHeader(ctx context.Context, header map[string]string) SpanContext
	// Flush Force the reporting of spans in the queue.
	Flush(ctx context.Context)
}