langfun 0.0.2.dev20240520__py3-none-any.whl → 0.0.2.dev20240522__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.

Files changed (7) hide show
  1. langfun/core/llms/__init__.py +1 -0
  2. langfun/core/llms/vertexai.py +8 -0
  3. {langfun-0.0.2.dev20240520.dist-info → langfun-0.0.2.dev20240522.dist-info}/METADATA +1 -1
  4. {langfun-0.0.2.dev20240520.dist-info → langfun-0.0.2.dev20240522.dist-info}/RECORD +7 -7
  5. {langfun-0.0.2.dev20240520.dist-info → langfun-0.0.2.dev20240522.dist-info}/LICENSE +0 -0
  6. {langfun-0.0.2.dev20240520.dist-info → langfun-0.0.2.dev20240522.dist-info}/WHEEL +0 -0
  7. {langfun-0.0.2.dev20240520.dist-info → langfun-0.0.2.dev20240522.dist-info}/top_level.txt +0 -0
langfun/core/llms/__init__.py CHANGED
@@ -90,6 +90,7 @@ from langfun.core.llms.groq import GroqGemma7B_IT
90
90
 
91
91
  from langfun.core.llms.vertexai import VertexAI
92
92
  from langfun.core.llms.vertexai import VertexAIGeminiPro1_5
93
+ from langfun.core.llms.vertexai import VertexAIGeminiFlash1_5
93
94
  from langfun.core.llms.vertexai import VertexAIGeminiPro1
94
95
  from langfun.core.llms.vertexai import VertexAIGeminiPro1Vision
95
96
  from langfun.core.llms.vertexai import VertexAIPalm2
langfun/core/llms/vertexai.py CHANGED
@@ -25,6 +25,7 @@ import pyglove as pg
25
25
 
26
26
  SUPPORTED_MODELS_AND_SETTINGS = {
27
27
  'gemini-1.5-pro-preview-0409': pg.Dict(api='gemini', rpm=5),
28
+ 'gemini-1.5-flash-preview-0514': pg.Dict(api='gemini', rpm=5),
28
29
  'gemini-1.0-pro': pg.Dict(api='gemini', rpm=300),
29
30
  'gemini-1.0-pro-vision': pg.Dict(api='gemini', rpm=100),
30
31
  # PaLM APIs.
@@ -266,6 +267,13 @@ class VertexAIGeminiPro1_5(VertexAI): # pylint: disable=invalid-name
266
267
  multimodal = True
267
268
 
268
269
 
270
+ class VertexAIGeminiFlash1_5(VertexAI): # pylint: disable=invalid-name
271
+ """Vertex AI Gemini 1.5 Flash model."""
272
+
273
+ model = 'gemini-1.5-flash-preview-0514'
274
+ multimodal = True
275
+
276
+
269
277
  class VertexAIGeminiPro1(VertexAI): # pylint: disable=invalid-name
270
278
  """Vertex AI Gemini 1.0 Pro model."""
271
279
 
{langfun-0.0.2.dev20240520.dist-info → langfun-0.0.2.dev20240522.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: langfun
3
- Version: 0.0.2.dev20240520
3
+ Version: 0.0.2.dev20240522
4
4
  Summary: Langfun: Language as Functions.
5
5
  Home-page: https://github.com/google/langfun
6
6
  Author: Langfun Authors
{langfun-0.0.2.dev20240520.dist-info → langfun-0.0.2.dev20240522.dist-info}/RECORD RENAMED
@@ -48,7 +48,7 @@ langfun/core/eval/patching.py,sha256=R0s2eAd1m97exQt06dmUL0V_MBG0W2Hxg7fhNB7cXW0
48
48
  langfun/core/eval/patching_test.py,sha256=8kCd54Egjju22FMgtJuxEsrXkW8ifs-UUBHtrCG1L6w,4775
49
49
  langfun/core/eval/scoring.py,sha256=1J7IATo-8FXUR0SBqk9icztHiM0lWkBFcWUo-vUURgQ,6376
50
50
  langfun/core/eval/scoring_test.py,sha256=O8olHbrUEg60gMxwOkWzKBJZpZoUlmVnBANX5Se2SXM,4546
51
- langfun/core/llms/__init__.py,sha256=h_kam-0fjWISAQ90KZ_ydBhwADVCzrhLPXmAki3GfU0,4175
51
+ langfun/core/llms/__init__.py,sha256=1z66CXeIrou48-YwHrLmixBepEFXwZv8DtwSOspq-sE,4237
52
52
  langfun/core/llms/anthropic.py,sha256=7W9YdPN3SlAFhAIQlihMkrpo7tTY_4NvD0KIlCrqcsk,8505
53
53
  langfun/core/llms/anthropic_test.py,sha256=TMM30myyEhwF99Le4RvJEXOn8RYl0q1FRkt9Q9nl1jk,5540
54
54
  langfun/core/llms/fake.py,sha256=Dd7-6ka9pFf3fcWZyczamjOqQ91MOI-m7We3Oc9Ffmo,2927
@@ -61,7 +61,7 @@ langfun/core/llms/llama_cpp.py,sha256=Y_KkMUf3Xfac49koMUtUslKl3h-HWp3-ntq7Jaa3bd
61
61
  langfun/core/llms/llama_cpp_test.py,sha256=ZxC6defGd_HX9SFRU9U4cJiQnBKundbOrchbXuC1Z2M,1683
62
62
  langfun/core/llms/openai.py,sha256=IN46gIqfY6aEEfxCPNmyH1hrep3oWBhJDwVFilfqNkM,13657
63
63
  langfun/core/llms/openai_test.py,sha256=asSA1sVy_7hnXioD_2HTxtSDpVTKBUO_EjZuyHpwbn0,14854
64
- langfun/core/llms/vertexai.py,sha256=DBH7WzrTL5oIyKWoZTuIR8B_gtgRi7hlS9GteSY0r4E,9317
64
+ langfun/core/llms/vertexai.py,sha256=tOhaiiam1zELOuQ7q4P2Q-V0dsWEf29GOI0B-JyEVI0,9563
65
65
  langfun/core/llms/vertexai_test.py,sha256=1u813e2esr8HVbOpLTOhIgzaX-GqCRTBouJ1doRn80Q,7642
66
66
  langfun/core/llms/cache/__init__.py,sha256=QAo3InUMDM_YpteNnVCSejI4zOsnjSMWKJKzkb3VY64,993
67
67
  langfun/core/llms/cache/base.py,sha256=cFfYvOIUae842pncqCAsRvqXCk2AnAsRYVx0mcIoAeY,3338
@@ -105,8 +105,8 @@ langfun/core/templates/demonstration.py,sha256=vCrgYubdZM5Umqcgp8NUVGXgr4P_c-fik
105
105
  langfun/core/templates/demonstration_test.py,sha256=SafcDQ0WgI7pw05EmPI2S4v1t3ABKzup8jReCljHeK4,2162
106
106
  langfun/core/templates/selfplay.py,sha256=yhgrJbiYwq47TgzThmHrDQTF4nDrTI09CWGhuQPNv-s,2273
107
107
  langfun/core/templates/selfplay_test.py,sha256=DYVrkk7uNKCqJGEHH31HssU2BPuMItU1vJLzfcXIlYg,2156
108
- langfun-0.0.2.dev20240520.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
109
- langfun-0.0.2.dev20240520.dist-info/METADATA,sha256=ndSicgG9hzjEnpZrs3WR8qP-CxdoXFlTIhGXCbvCaho,3452
110
- langfun-0.0.2.dev20240520.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
111
- langfun-0.0.2.dev20240520.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
112
- langfun-0.0.2.dev20240520.dist-info/RECORD,,
108
+ langfun-0.0.2.dev20240522.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
109
+ langfun-0.0.2.dev20240522.dist-info/METADATA,sha256=RKvADXja1HiFMDOiogNo4itCqJjah6ysTsN3dTQfcwY,3452
110
+ langfun-0.0.2.dev20240522.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
111
+ langfun-0.0.2.dev20240522.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
112
+ langfun-0.0.2.dev20240522.dist-info/RECORD,,