from fi_instrumentation import register, Transport
from fi_instrumentation.fi_types import ProjectType, EvalName, EvalTag, EvalTagType, EvalSpanKind, ModelChoices
# Setup OTel via our register function
trace_provider = register(
project_type=ProjectType.EXPERIMENT,
project_name="FUTURE_AGI", # Your project name
project_version_name="openai-exp", # Version identifier for this prototype
transport=Transport.HTTP, # Transport mechanism for your traces
eval_tags = [
EvalTag(
eval_name=EvalName.TONE,
value=EvalSpanKind.LLM,
type=EvalTagType.OBSERVATION_SPAN,
model=ModelChoices.TURING_LARGE,
mapping={
'input' : 'llm.input_messages'
},
custom_eval_name="<custom_eval_name2>",
),
]
)