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.
- langfun/core/llms/__init__.py +6 -0
- langfun/core/llms/vertexai.py +49 -2
- {langfun-0.1.2.dev202409270804.dist-info → langfun-0.1.2.dev202409280804.dist-info}/METADATA +1 -1
- {langfun-0.1.2.dev202409270804.dist-info → langfun-0.1.2.dev202409280804.dist-info}/RECORD +7 -7
- {langfun-0.1.2.dev202409270804.dist-info → langfun-0.1.2.dev202409280804.dist-info}/LICENSE +0 -0
- {langfun-0.1.2.dev202409270804.dist-info → langfun-0.1.2.dev202409280804.dist-info}/WHEEL +0 -0
- {langfun-0.1.2.dev202409270804.dist-info → langfun-0.1.2.dev202409280804.dist-info}/top_level.txt +0 -0
langfun/core/llms/__init__.py
CHANGED
@@ -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
|
langfun/core/llms/vertexai.py
CHANGED
@@ -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=
|
45
|
-
'gemini-1.5-
|
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
|
@@ -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=
|
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=
|
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.
|
123
|
-
langfun-0.1.2.
|
124
|
-
langfun-0.1.2.
|
125
|
-
langfun-0.1.2.
|
126
|
-
langfun-0.1.2.
|
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,,
|
File without changes
|
File without changes
|
{langfun-0.1.2.dev202409270804.dist-info → langfun-0.1.2.dev202409280804.dist-info}/top_level.txt
RENAMED
File without changes
|