minor fix: removes the duplicated handling logic for TracingProviderEnum.ARIZE and TracingProviderEnum.PHOENIX from the OpsTraceProviderConfigMap (#21927)
Signed-off-by: neatguycoding <15627489+NeatGuyCoding@users.noreply.github.com>
This commit is contained in:
@@ -41,28 +41,6 @@ from tasks.ops_trace_task import process_trace_tasks
|
|||||||
class OpsTraceProviderConfigMap(dict[str, dict[str, Any]]):
|
class OpsTraceProviderConfigMap(dict[str, dict[str, Any]]):
|
||||||
def __getitem__(self, provider: str) -> dict[str, Any]:
|
def __getitem__(self, provider: str) -> dict[str, Any]:
|
||||||
match provider:
|
match provider:
|
||||||
case TracingProviderEnum.ARIZE:
|
|
||||||
from core.ops.arize_phoenix_trace.arize_phoenix_trace import ArizePhoenixDataTrace
|
|
||||||
from core.ops.entities.config_entity import ArizeConfig
|
|
||||||
|
|
||||||
return {
|
|
||||||
"config_class": ArizeConfig,
|
|
||||||
"secret_keys": ["api_key", "space_id"],
|
|
||||||
"other_keys": ["project", "endpoint"],
|
|
||||||
"trace_instance": ArizePhoenixDataTrace,
|
|
||||||
}
|
|
||||||
|
|
||||||
case TracingProviderEnum.PHOENIX:
|
|
||||||
from core.ops.arize_phoenix_trace.arize_phoenix_trace import ArizePhoenixDataTrace
|
|
||||||
from core.ops.entities.config_entity import PhoenixConfig
|
|
||||||
|
|
||||||
return {
|
|
||||||
"config_class": PhoenixConfig,
|
|
||||||
"secret_keys": ["api_key"],
|
|
||||||
"other_keys": ["project", "endpoint"],
|
|
||||||
"trace_instance": ArizePhoenixDataTrace,
|
|
||||||
}
|
|
||||||
|
|
||||||
case TracingProviderEnum.LANGFUSE:
|
case TracingProviderEnum.LANGFUSE:
|
||||||
from core.ops.entities.config_entity import LangfuseConfig
|
from core.ops.entities.config_entity import LangfuseConfig
|
||||||
from core.ops.langfuse_trace.langfuse_trace import LangFuseDataTrace
|
from core.ops.langfuse_trace.langfuse_trace import LangFuseDataTrace
|
||||||
|
Reference in New Issue
Block a user