fix: ai cache count update not in time

Author: Anxcye

lib/service/ai/index.dart

@@ -1,13 +1,16 @@
 import 'package:anx_reader/config/shared_preference_provider.dart';
+import 'package:anx_reader/providers/ai_cache_count.dart';
 import 'package:anx_reader/service/ai/ai_cache.dart';
 import 'package:anx_reader/service/ai/ai_dio.dart';
 import 'package:anx_reader/service/ai/claude.dart';
 import 'package:anx_reader/service/ai/deepseek.dart';
 import 'package:anx_reader/service/ai/gemini.dart';
 import 'package:anx_reader/service/ai/openai.dart';
 import 'package:anx_reader/utils/log/common.dart';
+import 'package:flutter_riverpod/flutter_riverpod.dart';
 
 Stream<String> aiGenerateStream(
+  WidgetRef ref,
   String prompt, {
   String? identifier,
   Map<String, String>? config,
@@ -52,16 +55,18 @@ Stream<String> aiGenerateStream(
   }
   AiDio.instance.cancel();
   await AiCache.setAiCache(hash, buffer, identifier);
+  ref.read(aiCacheCountProvider.notifier).refresh();
 }
-
+  
 Future<String> aiGenerate(
+  WidgetRef ref,
   String prompt, {
   String? identifier,
   Map<String, String>? config,
 }) async {
   final buffer = StringBuffer();
   await for (final chunk
-      in aiGenerateStream(prompt, identifier: identifier, config: config)) {
+      in aiGenerateStream(ref, prompt, identifier: identifier, config: config)) {
     buffer.write(chunk);
   }
   return buffer.toString();

lib/widgets/ai_stream.dart

@@ -5,8 +5,9 @@ import 'package:flutter/foundation.dart';
 import 'package:flutter/material.dart';
 import 'package:flutter/services.dart';
 import 'package:flutter_markdown/flutter_markdown.dart';
+import 'package:flutter_riverpod/flutter_riverpod.dart';
 
-class AiStream extends StatefulWidget {
+class AiStream extends ConsumerStatefulWidget {
   final String prompt;
   final String? identifier;
   final Map<String, String>? config;
@@ -24,13 +25,14 @@ class AiStream extends StatefulWidget {
   AiStreamState createState() => AiStreamState();
 }
 
-class AiStreamState extends State<AiStream> {
+class AiStreamState extends ConsumerState<AiStream> {
   late Stream stream;
 
   @override
   void initState() {
     super.initState();
     stream = aiGenerateStream(
+      ref,
       widget.prompt,
       identifier: widget.identifier,
       config: widget.config,
@@ -75,6 +77,7 @@ class AiStreamState extends State<AiStream> {
                       onPressed: () {
                         setState(() {
                           stream = aiGenerateStream(
+                            ref,
                             widget.prompt,
                             identifier: widget.identifier,
                             config: widget.config,