Slide 8
Slide 8 text
8
Datadog LLM Observabilityの設定
Inline Method方式
from ddtrace.llmobs import LLMObs
from ddtrace.llmobs.decorators import llm
@llm(model_name="gemini", model_provider="google", name="call_gemini")
def call_gemini(...):
# Spanに付与するアノテーションを設定
LLMObs.annotate(
input_data=parts[0].text if parts else None,
output_data=output_text,
metadata={"model": model, "temperature": temperature, "top_p": top_p, "max_output_tokens":
max_output_tokens},
metrics={
"input_tokens": response.usage_metadata.prompt_token_count,
"output_tokens": response.usage_metadata.candidates_token_count,
"total_tokens": response.usage_metadata.total_token_count,
},
tags={"version_prompt": version},
)
※ Auto Instrumentation機能もあるが、当時はgoogle-genaiパッケージが未対応だったので上記方法で設定🥲