langfun 0.1.2.dev202409270804__py3-none-any.whl → 0.1.2.dev202409280804__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.
@@ -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
@@ -52,7 +52,7 @@ langfun/core/eval/patching.py,sha256=R0s2eAd1m97exQt06dmUL0V_MBG0W2Hxg7fhNB7cXW0
52
52
  langfun/core/eval/patching_test.py,sha256=8kCd54Egjju22FMgtJuxEsrXkW8ifs-UUBHtrCG1L6w,4775
53
53
  langfun/core/eval/scoring.py,sha256=wZz90Iw5Sco3cAiA1T71cJEWhD6qmvMeE1Ai-pez_aY,6210
54
54
  langfun/core/eval/scoring_test.py,sha256=O8olHbrUEg60gMxwOkWzKBJZpZoUlmVnBANX5Se2SXM,4546
55
- langfun/core/llms/__init__.py,sha256=ZgnzZSjI37LhUVNvgPGNSLS-HMCqPjaCGdwPd2Ij5Rs,5031
55
+ langfun/core/llms/__init__.py,sha256=lmOl2HoUYDeAnSJjObd9DfhEuchfvmrJ-GeKTbVFqds,5427
56
56
  langfun/core/llms/anthropic.py,sha256=Gon3fOi31RhZFgNd0ijyTnKnUdp9hrWrCoSXyO4UaLw,7316
57
57
  langfun/core/llms/anthropic_test.py,sha256=T-swuMkfnlgs8Fpif4rtXs579exGk0TsbLMirXDZCkg,5533
58
58
  langfun/core/llms/fake.py,sha256=gCHBYBLvBCsC78HI1hpoqXCS-p1FMTgY1P1qh_sGBPk,3070
@@ -67,7 +67,7 @@ langfun/core/llms/openai.py,sha256=vnDrKuD-pli0AtDIDq_TmlltOk7z7_PQ-xpU4K1ARdU,1
67
67
  langfun/core/llms/openai_test.py,sha256=UcBFW_7RkkMEo47Tn5RuVRK_DryTN7bb9ITphlzthE8,17762
68
68
  langfun/core/llms/rest.py,sha256=laopuq-zD8V-3Y6eFDngftHEbE66VlUkCD2-rvvRaLU,3388
69
69
  langfun/core/llms/rest_test.py,sha256=NZ3Nf0XQVpT9kLP5cBVo_yBHLI7vWTYhWQxYEJVMGs4,3472
70
- langfun/core/llms/vertexai.py,sha256=hQR8eghtvPCH9l4fvuFkfV51UOiACH705Le5bmwEBDc,15164
70
+ langfun/core/llms/vertexai.py,sha256=3w3EhGt-mF1JJmHK8X5I0PAPqLSEDIj520OayZdr-nc,16876
71
71
  langfun/core/llms/vertexai_test.py,sha256=EPR-mB2hNUpvpf7E8m_k5bh04epdQTVUuYU6hPgZyu8,10321
72
72
  langfun/core/llms/cache/__init__.py,sha256=QAo3InUMDM_YpteNnVCSejI4zOsnjSMWKJKzkb3VY64,993
73
73
  langfun/core/llms/cache/base.py,sha256=rt3zwmyw0y9jsSGW-ZbV1vAfLxQ7_3AVk0l2EySlse4,3918
@@ -119,8 +119,8 @@ langfun/core/templates/demonstration.py,sha256=vCrgYubdZM5Umqcgp8NUVGXgr4P_c-fik
119
119
  langfun/core/templates/demonstration_test.py,sha256=SafcDQ0WgI7pw05EmPI2S4v1t3ABKzup8jReCljHeK4,2162
120
120
  langfun/core/templates/selfplay.py,sha256=yhgrJbiYwq47TgzThmHrDQTF4nDrTI09CWGhuQPNv-s,2273
121
121
  langfun/core/templates/selfplay_test.py,sha256=rBW2Qr8yi-aWYwoTwRR-n1peKyMX9QXPZXURjLgoiRs,2264
122
- langfun-0.1.2.dev202409270804.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
123
- langfun-0.1.2.dev202409270804.dist-info/METADATA,sha256=1eE5H7baelcluzk2Shlet9VWxivsCDIaWICenplnM3k,8890
124
- langfun-0.1.2.dev202409270804.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
125
- langfun-0.1.2.dev202409270804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
126
- langfun-0.1.2.dev202409270804.dist-info/RECORD,,
122
+ langfun-0.1.2.dev202409280804.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
123
+ langfun-0.1.2.dev202409280804.dist-info/METADATA,sha256=_wBVve9ZKBMFFLBTmeY-LwjHbpJHKVaejKyZv9dGVsM,8890
124
+ langfun-0.1.2.dev202409280804.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
125
+ langfun-0.1.2.dev202409280804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
126
+ langfun-0.1.2.dev202409280804.dist-info/RECORD,,