langfun 0.1.2.dev202503170804__py3-none-any.whl → 0.1.2.dev202503190804__py3-none-any.whl

langfun/core/llms/gemini.py

@@ -640,7 +640,10 @@ class Gemini(rest.REST):
     ]
     usage = json['usageMetadata']
     input_tokens = usage['promptTokenCount']
-    output_tokens = usage['candidatesTokenCount']
+    # NOTE(daiyip): We saw cases that `candidatesTokenCount` is not present.
+    # Therefore, we use 0 as the default value.
+    output_tokens = usage.get('candidatesTokenCount', 0)
+
     return lf.LMSamplingResult(
         [lf.LMSample(message) for message in messages],
         usage=lf.LMSamplingUsage(

{langfun-0.1.2.dev202503170804.dist-info → langfun-0.1.2.dev202503190804.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: langfun
-Version: 0.1.2.dev202503170804
+Version: 0.1.2.dev202503190804
 Summary: Langfun: Language as Functions.
 Home-page: https://github.com/google/langfun
 Author: Langfun Authors

{langfun-0.1.2.dev202503170804.dist-info → langfun-0.1.2.dev202503190804.dist-info}/RECORD

@@ -84,7 +84,7 @@ langfun/core/llms/deepseek.py,sha256=jvTxdXPr-vH6HNakn_Ootx1heDg8Fen2FUkUW36bpCs
 langfun/core/llms/deepseek_test.py,sha256=DvROWPlDuow5E1lfoSkhyGt_ELA19JoQoDsTnRgDtTg,1847
 langfun/core/llms/fake.py,sha256=xmgCkk9y0I4x0IT32SZ9_OT27aLadXH8PRiYNo5VTd4,3265
 langfun/core/llms/fake_test.py,sha256=2h13qkwEz_JR0mtUDPxdAhQo7MueXaFSwsD2DIRDW9g,7653
-langfun/core/llms/gemini.py,sha256=EJwZZYXrh1ykXsIeq9w_VNdJyCfBnBz3x7_jEx6WvQw,22962
+langfun/core/llms/gemini.py,sha256=msTaDZcPBwZUMIHBXZUshexUJayIr3I7g-7z_uqtpqA,23095
 langfun/core/llms/gemini_test.py,sha256=ZzQmgB5bxJdtD2nVOz0WQ1zjNCea_jM5Yuqs8kchnGw,6621
 langfun/core/llms/google_genai.py,sha256=mzUdWAShRzz2k_foUo0eIdNOyHs_xjEtbewqYyuD0k0,4768
 langfun/core/llms/google_genai_test.py,sha256=NKNtpebArQ9ZR7Qsnhd2prFIpMjleojy6o6VMXkJ1zY,1502
@@ -148,8 +148,8 @@ langfun/core/templates/demonstration.py,sha256=vCrgYubdZM5Umqcgp8NUVGXgr4P_c-fik
 langfun/core/templates/demonstration_test.py,sha256=SafcDQ0WgI7pw05EmPI2S4v1t3ABKzup8jReCljHeK4,2162
 langfun/core/templates/selfplay.py,sha256=yhgrJbiYwq47TgzThmHrDQTF4nDrTI09CWGhuQPNv-s,2273
 langfun/core/templates/selfplay_test.py,sha256=Ot__1P1M8oJfoTp-M9-PQ6HUXqZKyMwvZ5f7yQ3yfyM,2326
-langfun-0.1.2.dev202503170804.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202503170804.dist-info/METADATA,sha256=DrJ6cM9QnyxYnZMKfJCJhwdVbykLIbcFZqsOUPO_Y1E,7670
-langfun-0.1.2.dev202503170804.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-langfun-0.1.2.dev202503170804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202503170804.dist-info/RECORD,,
+langfun-0.1.2.dev202503190804.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202503190804.dist-info/METADATA,sha256=TuxWfA0ais-zZDUyETYdlj44fWJ27D8A5FjRK63MjBk,7670
+langfun-0.1.2.dev202503190804.dist-info/WHEEL,sha256=beeZ86-EfXScwlR_HKu4SllMC9wUEj_8Z_4FJ3egI2w,91
+langfun-0.1.2.dev202503190804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202503190804.dist-info/RECORD,,

{langfun-0.1.2.dev202503170804.dist-info → langfun-0.1.2.dev202503190804.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (76.0.0)
+Generator: setuptools (76.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any