resolved bugs in json loading

raga-ai-hub · Oct 17, 2024 · 81984fd · 81984fd
1 parent bd04693
commit 81984fd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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