diff --git a/agentneo/tracing/llm_utils.py b/agentneo/tracing/llm_utils.py index b12c567e..7c6cdec1 100644 --- a/agentneo/tracing/llm_utils.py +++ b/agentneo/tracing/llm_utils.py @@ -13,10 +13,14 @@ with open("agentneo/configs/model_costs.json", "r") as file: config = json.load(file) except FileNotFoundError: - with resources.open_text("agentneo", "configs/model_costs.json") as file: + # with resources.open_text("agentneo", "configs/model_costs.json") as file: + # config = json.load(file) + from importlib.resources import files + with (files("agentneo.configs") / "model_costs.json").open("r") as file: config = json.load(file) + def extract_llm_output(result): # import pdb