Update answerer.py

answerer.py

@@ -1,4 +1,5 @@
 import os, gc
+from typing import AsyncGenerator
 from rwkv.model import RWKV
 from rwkv.utils import PIPELINE, PIPELINE_ARGS
 from asyncio import sleep
@@ -20,7 +21,7 @@ class Answerer:
     repetitiveness = .3,
     diversity = 0,
     _count_penalty = 1,
-  ):
+  ) -> AsyncGenerator[str, None]:
     args = PIPELINE_ARGS(
       temperature=chaos,
       top_p=repetitiveness,
@@ -63,7 +64,7 @@ class Answerer:
       else: 
         occurrences[current_token] = 1
       
-      tmp = self.__pipeline.decode(tokens)
+      tmp: str = self.__pipeline.decode(tokens)
       if "\ufffd" not in tmp:
         tokens.clear()
         result += tmp