openllmetry-ruby/semantic_conventions_ai/lib/opentelemetry/semantic_conventions.rb at 365e69e26f094cf5358dad043322dd80c36f5f3c · traceloop/openllmetry-ruby · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
module OpenTelemetry
  module SemanticConventionsAi
    module SpanAttributes
      # LLM
      LLM_VENDOR = "llm.vendor"
      LLM_REQUEST_TYPE = "llm.request.type"
      LLM_REQUEST_MODEL = "llm.request.model"
      LLM_RESPONSE_MODEL = "llm.response.model"
      LLM_REQUEST_MAX_TOKENS = "llm.request.max_tokens"
      LLM_USAGE_TOTAL_TOKENS = "llm.usage.total_tokens"
      LLM_USAGE_COMPLETION_TOKENS = "llm.usage.completion_tokens"
      LLM_USAGE_PROMPT_TOKENS = "llm.usage.prompt_tokens"
      LLM_TEMPERATURE = "llm.temperature"
      LLM_TOP_P = "llm.top_p"
      LLM_FREQUENCY_PENALTY = "llm.frequency_penalty"
      LLM_PRESENCE_PENALTY = "llm.presence_penalty"
      LLM_PROMPTS = "llm.prompts"
      LLM_COMPLETIONS = "llm.completions"
      LLM_CHAT_STOP_SEQUENCES = "llm.chat.stop_sequences"

      # Vector DB
      VECTOR_DB_VENDOR = "vector_db.vendor"
      VECTOR_DB_QUERY_TOP_K = "vector_db.query.top_k"

      # LLM Workflows
      TRACELOOP_SPAN_KIND = "traceloop.span.kind"
      TRACELOOP_WORKFLOW_NAME = "traceloop.workflow.name"
      TRACELOOP_ENTITY_NAME = "traceloop.entity.name"
      TRACELOOP_ASSOCIATION_PROPERTIES = "traceloop.association.properties"

      # Deprecated
      TRACELOOP_CORRELATION_ID = "traceloop.correlation.id"

      # Gen AI
      GEN_AI_REQUEST_MODEL = "gen_ai.request.model"
      GEN_AI_RESPONSE_MODEL = "gen_ai.response.model"
      GEN_AI_USAGE_OUTPUT_TOKENS = "gen_ai.usage.output_tokens"
      GEN_AI_USAGE_INPUT_TOKENS = "gen_ai.usage.input_tokens"
      GEN_AI_COMPLETIONS = "gen_ai.completion"
      GEN_AI_PROMPTS = "gen_ai.prompt"
      GEN_AI_SYSTEM = "gen_ai.system"
      GEN_AI_PROVIDER = "gen_ai.provider.name"
      GEN_AI_CONVERSATION_ID = "gen_ai.conversation.id"
      GEN_AI_BEDROCK_GUARDRAILS = "gen_ai.bedrock.guardrail"
    end

    module LLMRequestTypeValues
        COMPLETION = "completion"
        CHAT = "chat"
        RERANK = "rerank"
        UNKNOWN = "unknown"
    end

    module TraceloopSpanKindValues
        WORKFLOW = "workflow"
        TASK = "task"
        AGENT = "agent"
        TOOL = "tool"
        UNKNOWN = "unknown"
    end
  end
end