lint

Author: nina-kollman

traceloop-sdk/llm_span.go

@@ -31,4 +31,4 @@ func (llmSpan *LLMSpan) LogCompletion(ctx context.Context, completion Completion
 	setMessagesAttribute(llmSpan.span, "llm.completions", completion.Messages)
 
 	defer llmSpan.span.End()
-}
+}

traceloop-sdk/model/types.go

@@ -7,4 +7,4 @@ const (
 	SpanKindAgent SpanKind = "agent"
 	SpanKindTask SpanKind = "task"
 	SpanKindWorkflow SpanKind = "workflow"
-)
+)

traceloop-sdk/sdk.go

@@ -21,10 +21,6 @@ import (
 
 const PromptsPath = "/v1/traceloop/prompts"
 
-const (
-	ToolType = "tool"
-	AgentType = "agent"
-)
 
 type Traceloop struct {
 	config         Config
@@ -179,7 +175,7 @@ func (instance *Traceloop) LogToolCall(ctx context.Context, attrs ToolCallAttrib
 
 	spanAttrs := []attribute.KeyValue{
 		semconvai.TraceloopWorkflowName.String(workflowAttrs.Name),
-		semconvai.TraceloopSpanKind.String(ToolType),
+		semconvai.TraceloopSpanKind.String(string(model.SpanKindTool)),
 		semconvai.TraceloopEntityName.String(attrs.Name),
 	}
 
@@ -202,7 +198,7 @@ func (instance *Traceloop) LogAgent(ctx context.Context, attrs AgentAttributes,
 
 	spanAttrs := []attribute.KeyValue{
 		semconvai.TraceloopWorkflowName.String(workflowAttrs.Name),
-		semconvai.TraceloopSpanKind.String(AgentType),
+		semconvai.TraceloopSpanKind.String(string(model.SpanKindAgent)),
 		semconvai.LLMAgentName.String(attrs.Name),
 	}
 

traceloop-sdk/task.go

@@ -19,11 +19,3 @@ func (task *Task) End() {
 func (task *Task) LogPrompt(prompt Prompt) LLMSpan {
 	return task.workflow.sdk.LogPrompt(task.ctx, prompt, task.workflow.Attributes)
 }
-
-func (task *Task) LogAgent(attrs AgentAttributes) LLMSpan {
-	return task.workflow.sdk.LogAgent(task.ctx, attrs, task.workflow.Attributes)
-}
-
-func (task *Task) LogToolCall(attrs ToolCallAttributes) LLMSpan {
-	return task.workflow.sdk.LogToolCall(task.ctx, attrs, task.workflow.Attributes)
-}