better usecount

huggingface · Jul 3, 2024 · 3c274f6 · 3c274f6
1 parent d70a925
commit 3c274f6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/lib/server/textGeneration/tools.ts b/src/lib/server/textGeneration/tools.ts
@@ -99,6 +99,8 @@ async function* callTool(
 			Date.now() - startTime
 		);
 
+		await collections.tools.findOneAndUpdate({ _id: tool._id }, { $inc: { useCount: 1 } });
+
 		return { ...toolResult, call } as ToolResult;
 	} catch (error) {
 		MetricsServer.getMetrics().tool.toolUseCountError.inc({ tool: call.name });

diff --git a/src/routes/tools/[toolId]/+page.svelte b/src/routes/tools/[toolId]/+page.svelte
@@ -71,9 +71,7 @@
 								{data.tool?.createdByName}
 							</a>
 							<span class="text-gray-300">•</span>
-							text -> image
-							<span class="text-gray-300">•</span>
-							1.2M runs
+							{data.tool.useCount} runs
 						</p>
 					{/if}