langfun 0.1.2.dev202409270804__tar.gz → 0.1.2.dev202409280804__tar.gz

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.
Files changed (131) hide show
  1. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/PKG-INFO +1 -1
  2. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/__init__.py +6 -0
  3. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/vertexai.py +49 -2
  4. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun.egg-info/PKG-INFO +1 -1
  5. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/LICENSE +0 -0
  6. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/README.md +0 -0
  7. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/__init__.py +0 -0
  8. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/__init__.py +0 -0
  9. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/coding/__init__.py +0 -0
  10. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/coding/python/__init__.py +0 -0
  11. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/coding/python/correction.py +0 -0
  12. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/coding/python/correction_test.py +0 -0
  13. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/coding/python/errors.py +0 -0
  14. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/coding/python/errors_test.py +0 -0
  15. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/coding/python/execution.py +0 -0
  16. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/coding/python/execution_test.py +0 -0
  17. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/coding/python/generation.py +0 -0
  18. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/coding/python/generation_test.py +0 -0
  19. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/coding/python/parsing.py +0 -0
  20. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/coding/python/parsing_test.py +0 -0
  21. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/coding/python/permissions.py +0 -0
  22. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/coding/python/permissions_test.py +0 -0
  23. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/component.py +0 -0
  24. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/component_test.py +0 -0
  25. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/concurrent.py +0 -0
  26. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/concurrent_test.py +0 -0
  27. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/console.py +0 -0
  28. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/console_test.py +0 -0
  29. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/eval/__init__.py +0 -0
  30. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/eval/base.py +0 -0
  31. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/eval/base_test.py +0 -0
  32. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/eval/matching.py +0 -0
  33. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/eval/matching_test.py +0 -0
  34. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/eval/patching.py +0 -0
  35. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/eval/patching_test.py +0 -0
  36. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/eval/scoring.py +0 -0
  37. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/eval/scoring_test.py +0 -0
  38. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/langfunc.py +0 -0
  39. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/langfunc_test.py +0 -0
  40. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/language_model.py +0 -0
  41. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/language_model_test.py +0 -0
  42. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/anthropic.py +0 -0
  43. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/anthropic_test.py +0 -0
  44. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/cache/__init__.py +0 -0
  45. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/cache/base.py +0 -0
  46. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/cache/in_memory.py +0 -0
  47. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/cache/in_memory_test.py +0 -0
  48. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/fake.py +0 -0
  49. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/fake_test.py +0 -0
  50. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/google_genai.py +0 -0
  51. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/google_genai_test.py +0 -0
  52. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/groq.py +0 -0
  53. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/groq_test.py +0 -0
  54. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/llama_cpp.py +0 -0
  55. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/llama_cpp_test.py +0 -0
  56. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/openai.py +0 -0
  57. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/openai_test.py +0 -0
  58. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/rest.py +0 -0
  59. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/rest_test.py +0 -0
  60. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/llms/vertexai_test.py +0 -0
  61. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/logging.py +0 -0
  62. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/logging_test.py +0 -0
  63. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/memories/__init__.py +0 -0
  64. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/memories/conversation_history.py +0 -0
  65. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/memories/conversation_history_test.py +0 -0
  66. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/memory.py +0 -0
  67. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/message.py +0 -0
  68. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/message_test.py +0 -0
  69. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/modalities/__init__.py +0 -0
  70. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/modalities/audio.py +0 -0
  71. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/modalities/audio_test.py +0 -0
  72. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/modalities/image.py +0 -0
  73. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/modalities/image_test.py +0 -0
  74. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/modalities/mime.py +0 -0
  75. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/modalities/mime_test.py +0 -0
  76. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/modalities/ms_office.py +0 -0
  77. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/modalities/ms_office_test.py +0 -0
  78. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/modalities/pdf.py +0 -0
  79. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/modalities/pdf_test.py +0 -0
  80. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/modalities/video.py +0 -0
  81. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/modalities/video_test.py +0 -0
  82. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/modality.py +0 -0
  83. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/modality_test.py +0 -0
  84. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/natural_language.py +0 -0
  85. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/natural_language_test.py +0 -0
  86. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/repr_utils.py +0 -0
  87. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/repr_utils_test.py +0 -0
  88. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/sampling.py +0 -0
  89. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/sampling_test.py +0 -0
  90. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/__init__.py +0 -0
  91. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/completion.py +0 -0
  92. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/completion_test.py +0 -0
  93. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/description.py +0 -0
  94. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/description_test.py +0 -0
  95. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/function_generation.py +0 -0
  96. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/function_generation_test.py +0 -0
  97. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/mapping.py +0 -0
  98. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/mapping_test.py +0 -0
  99. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/parsing.py +0 -0
  100. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/parsing_test.py +0 -0
  101. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/prompting.py +0 -0
  102. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/prompting_test.py +0 -0
  103. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/schema.py +0 -0
  104. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/schema_generation.py +0 -0
  105. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/schema_generation_test.py +0 -0
  106. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/schema_test.py +0 -0
  107. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/scoring.py +0 -0
  108. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/scoring_test.py +0 -0
  109. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/tokenization.py +0 -0
  110. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/structured/tokenization_test.py +0 -0
  111. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/subscription.py +0 -0
  112. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/subscription_test.py +0 -0
  113. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/template.py +0 -0
  114. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/template_test.py +0 -0
  115. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/templates/__init__.py +0 -0
  116. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/templates/completion.py +0 -0
  117. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/templates/completion_test.py +0 -0
  118. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/templates/conversation.py +0 -0
  119. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/templates/conversation_test.py +0 -0
  120. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/templates/demonstration.py +0 -0
  121. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/templates/demonstration_test.py +0 -0
  122. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/templates/selfplay.py +0 -0
  123. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/templates/selfplay_test.py +0 -0
  124. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/text_formatting.py +0 -0
  125. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun/core/text_formatting_test.py +0 -0
  126. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun.egg-info/SOURCES.txt +0 -0
  127. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun.egg-info/dependency_links.txt +0 -0
  128. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun.egg-info/requires.txt +0 -0
  129. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/langfun.egg-info/top_level.txt +0 -0
  130. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/setup.cfg +0 -0
  131. {langfun-0.1.2.dev202409270804 → langfun-0.1.2.dev202409280804}/setup.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: langfun
3
- Version: 0.1.2.dev202409270804
3
+ Version: 0.1.2.dev202409280804
4
4
  Summary: Langfun: Language as Functions.
5
5
  Home-page: https://github.com/google/langfun
6
6
  Author: Langfun Authors
@@ -103,9 +103,15 @@ from langfun.core.llms.groq import GroqGemma7B_IT
103
103
 
104
104
  from langfun.core.llms.vertexai import VertexAI
105
105
  from langfun.core.llms.vertexai import VertexAIGeminiPro1_5
106
+ from langfun.core.llms.vertexai import VertexAIGeminiPro1_5_Latest
107
+ from langfun.core.llms.vertexai import VertexAIGeminiPro1_5_001
108
+ from langfun.core.llms.vertexai import VertexAIGeminiPro1_5_002
106
109
  from langfun.core.llms.vertexai import VertexAIGeminiPro1_5_0514
107
110
  from langfun.core.llms.vertexai import VertexAIGeminiPro1_5_0409
111
+ from langfun.core.llms.vertexai import VertexAIGeminiFlash1_5_Latest
108
112
  from langfun.core.llms.vertexai import VertexAIGeminiFlash1_5
113
+ from langfun.core.llms.vertexai import VertexAIGeminiFlash1_5_001
114
+ from langfun.core.llms.vertexai import VertexAIGeminiFlash1_5_002
109
115
  from langfun.core.llms.vertexai import VertexAIGeminiFlash1_5_0514
110
116
  from langfun.core.llms.vertexai import VertexAIGeminiPro1
111
117
  from langfun.core.llms.vertexai import VertexAIGeminiPro1Vision
@@ -41,8 +41,14 @@ except ImportError:
41
41
 
42
42
 
43
43
  SUPPORTED_MODELS_AND_SETTINGS = {
44
- 'gemini-1.5-pro-001': pg.Dict(api='gemini', rpm=50),
45
- 'gemini-1.5-flash-001': pg.Dict(api='gemini', rpm=200),
44
+ 'gemini-1.5-pro-001': pg.Dict(api='gemini', rpm=500),
45
+ 'gemini-1.5-pro-002': pg.Dict(api='gemini', rpm=500),
46
+ 'gemini-1.5-flash-002': pg.Dict(api='gemini', rpm=500),
47
+ 'gemini-1.5-flash-001': pg.Dict(api='gemini', rpm=500),
48
+ 'gemini-1.5-pro': pg.Dict(api='gemini', rpm=500),
49
+ 'gemini-1.5-flash': pg.Dict(api='gemini', rpm=500),
50
+ 'gemini-1.5-pro-latest': pg.Dict(api='gemini', rpm=500),
51
+ 'gemini-1.5-flash-latest': pg.Dict(api='gemini', rpm=500),
46
52
  'gemini-1.5-pro-preview-0514': pg.Dict(api='gemini', rpm=50),
47
53
  'gemini-1.5-pro-preview-0409': pg.Dict(api='gemini', rpm=50),
48
54
  'gemini-1.5-flash-preview-0514': pg.Dict(api='gemini', rpm=200),
@@ -417,9 +423,30 @@ _PDF = [
417
423
  ]
418
424
 
419
425
 
426
+ class VertexAIGeminiPro1_5_Latest(VertexAI): # pylint: disable=invalid-name
427
+ """Vertex AI Gemini 1.5 Pro model."""
428
+
429
+ model = 'gemini-1.5-pro-latest'
430
+ supported_modalities = _PDF + _IMAGE_TYPES + _AUDIO_TYPES + _VIDEO_TYPES
431
+
432
+
420
433
  class VertexAIGeminiPro1_5(VertexAI): # pylint: disable=invalid-name
421
434
  """Vertex AI Gemini 1.5 Pro model."""
422
435
 
436
+ model = 'gemini-1.5-pro'
437
+ supported_modalities = _PDF + _IMAGE_TYPES + _AUDIO_TYPES + _VIDEO_TYPES
438
+
439
+
440
+ class VertexAIGeminiPro1_5_002(VertexAI): # pylint: disable=invalid-name
441
+ """Vertex AI Gemini 1.5 Pro model."""
442
+
443
+ model = 'gemini-1.5-pro-002'
444
+ supported_modalities = _PDF + _IMAGE_TYPES + _AUDIO_TYPES + _VIDEO_TYPES
445
+
446
+
447
+ class VertexAIGeminiPro1_5_001(VertexAI): # pylint: disable=invalid-name
448
+ """Vertex AI Gemini 1.5 Pro model."""
449
+
423
450
  model = 'gemini-1.5-pro-001'
424
451
  supported_modalities = _PDF + _IMAGE_TYPES + _AUDIO_TYPES + _VIDEO_TYPES
425
452
 
@@ -438,7 +465,27 @@ class VertexAIGeminiPro1_5_0409(VertexAI): # pylint: disable=invalid-name
438
465
  supported_modalities = _PDF + _IMAGE_TYPES + _AUDIO_TYPES + _VIDEO_TYPES
439
466
 
440
467
 
468
+ class VertexAIGeminiFlash1_5_Latest(VertexAI): # pylint: disable=invalid-name
469
+ """Vertex AI Gemini 1.5 Flash model."""
470
+
471
+ model = 'gemini-1.5-flash-latest'
472
+ supported_modalities = _PDF + _IMAGE_TYPES + _AUDIO_TYPES + _VIDEO_TYPES
473
+
474
+
441
475
  class VertexAIGeminiFlash1_5(VertexAI): # pylint: disable=invalid-name
476
+ """Vertex AI Gemini 1.5 Flash model."""
477
+ model = 'gemini-1.5-flash'
478
+ supported_modalities = _PDF + _IMAGE_TYPES + _AUDIO_TYPES + _VIDEO_TYPES
479
+
480
+
481
+ class VertexAIGeminiFlash1_5_002(VertexAI): # pylint: disable=invalid-name
482
+ """Vertex AI Gemini 1.5 Flash model."""
483
+
484
+ model = 'gemini-1.5-flash-002'
485
+ supported_modalities = _PDF + _IMAGE_TYPES + _AUDIO_TYPES + _VIDEO_TYPES
486
+
487
+
488
+ class VertexAIGeminiFlash1_5_001(VertexAI): # pylint: disable=invalid-name
442
489
  """Vertex AI Gemini 1.5 Flash model."""
443
490
  model = 'gemini-1.5-flash-001'
444
491
  supported_modalities = _PDF + _IMAGE_TYPES + _AUDIO_TYPES + _VIDEO_TYPES
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: langfun
3
- Version: 0.1.2.dev202409270804
3
+ Version: 0.1.2.dev202409280804
4
4
  Summary: Langfun: Language as Functions.
5
5
  Home-page: https://github.com/google/langfun
6
6
  Author: Langfun Authors