forked from phoenix-oss/llama-stack-mirror
c23363d561
This PR adds two new methods to the telemetry API: 1) Gives the ability to query spans directly instead of first querying traces and then using that to get spans 2) Another method save_spans_to_dataset, which builds on the query spans to save it on dataset. This give the ability to saves spans that are part of an agent session to a dataset. The unique aspect of this API is that we dont require each provider of telemetry to implement this method. Hence, its implemented in the protocol class itself. This required the protocol check to be slightly modified.
18 lines
642 B
Python
18 lines
642 B
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the terms described in the LICENSE file in
|
|
# the root directory of this source tree.
|
|
from pydantic import BaseModel
|
|
|
|
from llama_stack.distribution.utils.config_dirs import RUNTIME_BASE_DIR
|
|
from llama_stack.providers.utils.kvstore.config import (
|
|
KVStoreConfig,
|
|
SqliteKVStoreConfig,
|
|
)
|
|
|
|
|
|
class MetaReferenceEvalConfig(BaseModel):
|
|
kvstore: KVStoreConfig = SqliteKVStoreConfig(
|
|
db_path=(RUNTIME_BASE_DIR / "meta_reference_eval.db").as_posix()
|
|
) # Uses SQLite config specific to Meta Reference Eval storage
|