langfun 0.1.2.dev202505090804__py3-none-any.whl → 0.1.2.dev202505100803__py3-none-any.whl

langfun/core/llms/vertexai_test.py

@@ -17,6 +17,7 @@ import os
 import unittest
 from unittest import mock
 
+from google import auth
 from google.auth import exceptions
 import langfun.core as lf
 from langfun.core.llms import rest
@@ -61,7 +62,11 @@ class VertexAITest(unittest.TestCase):
         lf.concurrent.RetryError,
         'Failed to refresh Google authentication credentials'
     ):
-      with mock.patch.object(rest.REST, '_sample_single') as mock_sample_single:
+      with (
+          mock.patch.object(auth, 'default') as mock_auth,
+          mock.patch.object(rest.REST, '_sample_single') as mock_sample_single
+      ):
+        mock_auth.return_value = mock.MagicMock(), None
         mock_sample_single.side_effect = _auth_refresh_error
         model = vertexai.VertexAIGemini15Pro(
             project='abc', location='us-central1', max_attempts=1

{langfun-0.1.2.dev202505090804.dist-info → langfun-0.1.2.dev202505100803.dist-info}/METADATA

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

{langfun-0.1.2.dev202505090804.dist-info → langfun-0.1.2.dev202505100803.dist-info}/RECORD

@@ -107,7 +107,7 @@ langfun/core/llms/openai_test.py,sha256=gwuO6aoa296iM2welWV9ua4KF8gEVGsEPakgbtkW
 langfun/core/llms/rest.py,sha256=MCybcHApJcf49lubLnDzScN9Oc2IWY_JnMHIGdbDOuU,4474
 langfun/core/llms/rest_test.py,sha256=_zM7nV8DEVyoXNiQOnuwJ917mWjki0614H88rNmDboE,5020
 langfun/core/llms/vertexai.py,sha256=nPWwv1C-wVpWpE4Vyl4vJoAEMB519-XNzPjyLY4N3tM,18816
-langfun/core/llms/vertexai_test.py,sha256=0M4jsPOXGagdzPfEdJixmyLdhmmERePZWSFfTwnaYCQ,4875
+langfun/core/llms/vertexai_test.py,sha256=_e-acnNBAf9C3WO6i1b2J_mhRzdDdYQTorD9hIVZKOg,5034
 langfun/core/llms/cache/__init__.py,sha256=QAo3InUMDM_YpteNnVCSejI4zOsnjSMWKJKzkb3VY64,993
 langfun/core/llms/cache/base.py,sha256=rt3zwmyw0y9jsSGW-ZbV1vAfLxQ7_3AVk0l2EySlse4,3918
 langfun/core/llms/cache/in_memory.py,sha256=i58oiQL28RDsq37dwqgVpC2mBETJjIEFS20yHiV5MKU,5185
@@ -156,8 +156,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.dev202505090804.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202505090804.dist-info/METADATA,sha256=m_vknRs5lJe6eWbqgQG2REHAaNbJn8P1nPpgpxorCsc,8178
-langfun-0.1.2.dev202505090804.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-langfun-0.1.2.dev202505090804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202505090804.dist-info/RECORD,,
+langfun-0.1.2.dev202505100803.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202505100803.dist-info/METADATA,sha256=M52b_XjFPeZNK9ZFRbXrVjBPlT9LyI3NThMyeLjKaQ4,8178
+langfun-0.1.2.dev202505100803.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
+langfun-0.1.2.dev202505100803.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202505100803.dist-info/RECORD,,

{langfun-0.1.2.dev202505090804.dist-info → langfun-0.1.2.dev202505100803.dist-info}/WHEEL

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