langfun 0.0.2.dev20240604__py3-none-any.whl → 0.0.2.dev20240605__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of langfun might be problematic. Click here for more details.
- langfun/core/llms/vertexai.py +10 -1
- {langfun-0.0.2.dev20240604.dist-info → langfun-0.0.2.dev20240605.dist-info}/METADATA +1 -1
- {langfun-0.0.2.dev20240604.dist-info → langfun-0.0.2.dev20240605.dist-info}/RECORD +6 -6
- {langfun-0.0.2.dev20240604.dist-info → langfun-0.0.2.dev20240605.dist-info}/LICENSE +0 -0
- {langfun-0.0.2.dev20240604.dist-info → langfun-0.0.2.dev20240605.dist-info}/WHEEL +0 -0
- {langfun-0.0.2.dev20240604.dist-info → langfun-0.0.2.dev20240605.dist-info}/top_level.txt +0 -0
langfun/core/llms/vertexai.py
CHANGED
|
@@ -179,7 +179,16 @@ class VertexAI(lf.LanguageModel):
|
|
|
179
179
|
assert self._api_initialized, 'Vertex AI API is not initialized.'
|
|
180
180
|
# TODO(yifenglu): It seems this exception is due to the instability of the
|
|
181
181
|
# API. We should revisit this later.
|
|
182
|
-
retry_on_errors = [
|
|
182
|
+
retry_on_errors = [
|
|
183
|
+
(Exception, 'InternalServerError'),
|
|
184
|
+
(
|
|
185
|
+
Exception,
|
|
186
|
+
(
|
|
187
|
+
'ValueError: Response candidate content has no parts (and thus'
|
|
188
|
+
' no text).'
|
|
189
|
+
),
|
|
190
|
+
),
|
|
191
|
+
]
|
|
183
192
|
|
|
184
193
|
return lf.concurrent_execute(
|
|
185
194
|
self._sample_single,
|
|
@@ -63,7 +63,7 @@ langfun/core/llms/openai.py,sha256=IN46gIqfY6aEEfxCPNmyH1hrep3oWBhJDwVFilfqNkM,1
|
|
|
63
63
|
langfun/core/llms/openai_test.py,sha256=QWDzTgi8F2Z9u9ip6alK4rDEp_YraVTxWlDX5XOsKJk,14858
|
|
64
64
|
langfun/core/llms/rest.py,sha256=laopuq-zD8V-3Y6eFDngftHEbE66VlUkCD2-rvvRaLU,3388
|
|
65
65
|
langfun/core/llms/rest_test.py,sha256=Zw08Xbl_O2OQuUglLXHsPsY5KW2VEcPGl1gR8PRHuFY,3449
|
|
66
|
-
langfun/core/llms/vertexai.py,sha256=
|
|
66
|
+
langfun/core/llms/vertexai.py,sha256=wIpckH-rMHUBA1vhauQk4LVrSsPQEsVntz7kLDKwm9g,11359
|
|
67
67
|
langfun/core/llms/vertexai_test.py,sha256=G18BG36h5KvmX2zutDTLjtYCRjTuP_nWIFm4FMnLnyY,7651
|
|
68
68
|
langfun/core/llms/cache/__init__.py,sha256=QAo3InUMDM_YpteNnVCSejI4zOsnjSMWKJKzkb3VY64,993
|
|
69
69
|
langfun/core/llms/cache/base.py,sha256=cFfYvOIUae842pncqCAsRvqXCk2AnAsRYVx0mcIoAeY,3338
|
|
@@ -113,8 +113,8 @@ langfun/core/templates/demonstration.py,sha256=vCrgYubdZM5Umqcgp8NUVGXgr4P_c-fik
|
|
|
113
113
|
langfun/core/templates/demonstration_test.py,sha256=SafcDQ0WgI7pw05EmPI2S4v1t3ABKzup8jReCljHeK4,2162
|
|
114
114
|
langfun/core/templates/selfplay.py,sha256=yhgrJbiYwq47TgzThmHrDQTF4nDrTI09CWGhuQPNv-s,2273
|
|
115
115
|
langfun/core/templates/selfplay_test.py,sha256=DYVrkk7uNKCqJGEHH31HssU2BPuMItU1vJLzfcXIlYg,2156
|
|
116
|
-
langfun-0.0.2.
|
|
117
|
-
langfun-0.0.2.
|
|
118
|
-
langfun-0.0.2.
|
|
119
|
-
langfun-0.0.2.
|
|
120
|
-
langfun-0.0.2.
|
|
116
|
+
langfun-0.0.2.dev20240605.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
|
|
117
|
+
langfun-0.0.2.dev20240605.dist-info/METADATA,sha256=NMWv4oYMcRuZXl22coMTShqGC8hj_Y2PGfTk7n-Alt0,3550
|
|
118
|
+
langfun-0.0.2.dev20240605.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
|
|
119
|
+
langfun-0.0.2.dev20240605.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
|
|
120
|
+
langfun-0.0.2.dev20240605.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|