Merge pull request stitionai#202 from rajakumar05032000/fix/token-usa…

…ge-log1 Fixed - token usage in UI & logs
tojaish · Mar 28, 2024 · de007c6 · de007c6
2 parents 5dd119b + ab2aa32
commit de007c6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ui/src/lib/api.js b/ui/src/lib/api.js
@@ -120,7 +120,8 @@ export async function executeAgent(prompt) {
 }
 
 export async function getTokenUsage() {
-  const response = await fetch(`${API_BASE_URL}/api/token-usage`);
+  const projectName = localStorage.getItem("selectedProject");
+  const response = await fetch(`${API_BASE_URL}/api/token-usage?project_name=${projectName}`);
   const data = await response.json();
   return data.token_usage;
 }