also display consumption in middle of the tool-calling process

Author: rainu

frontend/src/views/Chat.vue

@@ -160,6 +160,7 @@ export default {
 				},
 			] as LLMMessageContentPart[],
 			outputStreamRole: Role.Bot,
+			runningAskingConsumption: {} as HistoryEntryConsumption,
 			error: null as { title: string; message: string } | null,
 			userScroll: false,
 			minimized: false,
@@ -201,7 +202,9 @@ export default {
 			return result
 		},
 		totalChatConsumption(): HistoryEntryConsumption | null {
-			let result = {} as HistoryEntryConsumption
+			let result = {
+				...this.runningAskingConsumption
+			} as HistoryEntryConsumption
 
 			for (let msg of this.chatHistory) {
 				if(msg.Consumption === undefined){
@@ -298,6 +301,7 @@ export default {
 				this.progress = true
 				this.error = null
 				this.userScroll = false
+				this.runningAskingConsumption = {}
 
 				const setInput = () => {
 					this.pushHistory({
@@ -353,6 +357,7 @@ export default {
 			} finally {
 				this.progress = false
 				this.outputStream[0].Content = ''
+				this.runningAskingConsumption = {}
 			}
 		},
 		async onSubmit(input: ChatInputType) {
@@ -393,6 +398,9 @@ export default {
 		EventsOn('llm:message:update', (message: LLMMessage) => {
 			this.updateHistoryMessage(message)
 		})
+		EventsOn('llm:consumption:update', (consumption: HistoryEntryConsumption) => {
+			this.runningAskingConsumption = consumption
+		})
 		EventsOn('system:restart', () => {
 			// backend requested a restart
 			// so we have to save the current state and restart the app

internal/controller/llm.go

@@ -225,6 +225,7 @@ func (c *Controller) LLMAsk(args LLMAskArgs) (result LLMAskResult, err error) {
 		}
 
 		consumption.Add(c.aiModel.ConsumptionOf(resp))
+		RuntimeEventsEmit(c.ctx, EventNameLLMConsumptionUpdate, consumption.Summary())
 
 		tcMessage, err := c.handleToolCall(resp)
 		if err != nil {

internal/controller/llm_tool.go

@@ -15,8 +15,9 @@ import (
 )
 
 const (
-	EventNameLLMMessageAdd    = "llm:message:add"
-	EventNameLLMMessageUpdate = "llm:message:update"
+	EventNameLLMMessageAdd        = "llm:message:add"
+	EventNameLLMMessageUpdate     = "llm:message:update"
+	EventNameLLMConsumptionUpdate = "llm:consumption:update"
 )
 
 func (c *Controller) handleToolCall(resp *llms.ContentResponse) (result LLMMessages, err error) {