Add TODO for FalkorDB LLM API cost calculation

ag2ai · Dec 5, 2024 · 83650b3 · 83650b3
1 parent 2b1481d
commit 83650b3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/autogen/agentchat/contrib/graph_rag/falkor_graph_query_engine.py b/autogen/agentchat/contrib/graph_rag/falkor_graph_query_engine.py
@@ -36,6 +36,8 @@ def __init__(
         Initialize a FalkorDB knowledge graph.
         Please also refer to https://github.com/FalkorDB/GraphRAG-SDK/blob/main/graphrag_sdk/kg.py
 
+        TODO: Fix LLM API cost calculation for FalkorDB useages.
+
         Args:
             name (str): Knowledge graph name.
             host (str): FalkorDB hostname.