traceloop

type BackoffConfig struct {
	MaxRetries uint64
}

type Completion struct {
	Model    string    `json:"model"`
	Messages []Message `json:"messages"`
}

type Config struct {
	BaseURL         string
	APIKey          string
	TracerName      string
	ServiceName     string
	PollingInterval time.Duration
	BackoffConfig   BackoffConfig
}

type LLMSpan struct {
	// contains filtered or unexported fields
}

type Message struct {
	Index   int    `json:"index"`
	Role    string `json:"role"`
	Content string `json:"content"`
}

type Prompt struct {
	Vendor           string    `json:"vendor"`
	Model            string    `json:"model"`
	Mode             string    `json:"mode"`
	Temperature      float32   `json:"temperature"`
	TopP             float32   `json:"top_p"`
	Stop             []string  `json:"stop"`
	FrequencyPenalty float32   `json:"frequency_penalty"`
	PresencePenalty  float32   `json:"presence_penalty"`
	Messages         []Message `json:"messages"`
}

type PromptsResponse struct {
	Prompts     []model.Prompt `json:"prompts"`
	Environment string         `json:"environment"`
}

type Task struct {
	Name string `json:"name"`
	// contains filtered or unexported fields
}

type Traceloop struct {
	http.Client
	// contains filtered or unexported fields
}

type Usage struct {
	TotalTokens      int `json:"total_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	PromptTokens     int `json:"prompt_tokens"`
}

type Workflow struct {
	Attributes WorkflowAttributes `json:"workflow_attributes"`
	// contains filtered or unexported fields
}

type WorkflowAttributes struct {
	Name                  string            `json:"workflow_name"`
	AssociationProperties map[string]string `json:"association_properties"`
}