hjxdl 0.3.26__py3-none-any.whl → 0.3.28__py3-none-any.whl

hdl/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.3.26'
-__version_tuple__ = version_tuple = (0, 3, 26)
+__version__ = version = '0.3.28'
+__version_tuple__ = version_tuple = (0, 3, 28)

hdl/utils/decorators/llm.py

@@ -13,8 +13,10 @@ def measure_stream_performance(func):
         # 初始化计时和统计变量
         first_char_time = None
         total_chars = 0
+        total_tokens = 0
 
         for i in stream:
+            total_tokens += 1
             if i['type'] == 'text' and i['content']:
                 if first_char_time is None:
                     # 第一次输出字符的时间
@@ -31,12 +33,16 @@ def measure_stream_performance(func):
 
         # 每秒输出的字符数
         chars_per_second = total_chars / total_time if total_time > 0 else 0
+        tokens_per_second = total_tokens / total_time if total_time > 0 else 0
 
         # 打印统计信息
         print("\n--- Statistics ---")
         print(f"Time to first character: {time_to_first_char:.2f} seconds")
         print(f"Total time: {total_time:.2f} seconds")
+        print(f"Total characters: {total_chars}")
+        print(f"Total tokens: {total_tokens}")
         print(f"Characters per second: {chars_per_second:.2f}")
+        print(f"Tokens per second: {tokens_per_second:.2f}")
 
     return wrapper
 

{hjxdl-0.3.26.dist-info → hjxdl-0.3.28.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: hjxdl
-Version: 0.3.26
+Version: 0.3.28
 Summary: A collection of functions for Jupyter notebooks
 Home-page: https://github.com/huluxiaohuowa/hdl
 Author: Jianxing Hu

{hjxdl-0.3.26.dist-info → hjxdl-0.3.28.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
-hdl/_version.py,sha256=30eq06siTU79TQsQ2EEqu52177bco5xynk3YSelgFpk,413
+hdl/_version.py,sha256=UN5ualVWmw8HnAwzrBSVto-hh1hWU3cwc0AuXsdYiWI,413
 hdl/args/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/args/loss_args.py,sha256=s7YzSdd7IjD24rZvvOrxLLFqMZQb9YylxKeyelSdrTk,70
 hdl/controllers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -122,7 +122,7 @@ hdl/utils/database_tools/connect.py,sha256=xCacGucKxlQUXs6AsNddpeECvdqT1180V1ZWq
 hdl/utils/database_tools/datetime.py,sha256=xqE2xNiOpADzX-R8_bM0bioJRF3Ay9Jp1CAG6dy6uVI,1202
 hdl/utils/database_tools/web.py,sha256=awJ8lafL-2KRjf3V1uuij8JIvX9U5fI8fLZKOkOvqtk,5771
 hdl/utils/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-hdl/utils/decorators/llm.py,sha256=RRdixVVWHopx476fVgmr9-w7yBqZMBdJMrkjZZKFjdk,1559
+hdl/utils/decorators/llm.py,sha256=QCinGJ6eFlG-xMtfH6I2C5OI5-Vm0mJU8BVJ3UQCvz0,1851
 hdl/utils/desc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/utils/desc/func_desc.py,sha256=sHmVZZmV7Zgii--gnHqMs6fTb7HVkqTOf8Pl_0F6qlI,3808
 hdl/utils/desc/template.py,sha256=Kf_tbL-XkDCKNQ3UncbCuYEeUgXEa7kRVCf9TD2b8og,2526
@@ -143,8 +143,8 @@ hdl/utils/schedulers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 hdl/utils/schedulers/norm_lr.py,sha256=bDwCmdEK-WkgxQMFBiMuchv8Mm7C0-GZJ6usm-PQk14,4461
 hdl/utils/weather/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/utils/weather/weather.py,sha256=k11o6wM15kF8b9NMlEfrg68ak-SfSYLN3nOOflFUv-I,4381
-hjxdl-0.3.26.dist-info/LICENSE,sha256=lkMiSbeZHBQLB9LJEkS9-L3Z-LBC4yGnKrzHSG8RkPM,2599
-hjxdl-0.3.26.dist-info/METADATA,sha256=Y7g6KkHdFw_eroB_ynzGyMI9MN3Z4YypdWjf6YVnd_Y,1336
-hjxdl-0.3.26.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-hjxdl-0.3.26.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.3.26.dist-info/RECORD,,
+hjxdl-0.3.28.dist-info/LICENSE,sha256=lkMiSbeZHBQLB9LJEkS9-L3Z-LBC4yGnKrzHSG8RkPM,2599
+hjxdl-0.3.28.dist-info/METADATA,sha256=1XGiJCOBRzJdzAhry_kGvDLvyLeXdr37Ri_4Y7VvDFc,1336
+hjxdl-0.3.28.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+hjxdl-0.3.28.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.3.28.dist-info/RECORD,,