langfun 0.1.2.dev202505220804__py3-none-any.whl → 0.1.2.dev202505240803__py3-none-any.whl

langfun/core/llms/__init__.py

@@ -141,6 +141,9 @@ from langfun.core.llms.openai import Gpt35
 
 # Anthropic models.
 
+from langfun.core.llms.anthropic import Claude4
+from langfun.core.llms.anthropic import Claude4Sonnet_20250514
+from langfun.core.llms.anthropic import Claude4Opus_20250514
 from langfun.core.llms.anthropic import Claude37
 from langfun.core.llms.anthropic import Claude37Sonnet_20250219
 from langfun.core.llms.anthropic import Claude35Sonnet
@@ -155,6 +158,8 @@ from langfun.core.llms.anthropic import Claude3Haiku
 from langfun.core.llms.anthropic import Claude3Haiku_20240307
 
 from langfun.core.llms.vertexai import VertexAIAnthropic
+from langfun.core.llms.vertexai import VertexAIClaude4Opus_20250514
+from langfun.core.llms.vertexai import VertexAIClaude4Sonnet_20250514
 from langfun.core.llms.vertexai import VertexAIClaude37Sonnet_20250219
 from langfun.core.llms.vertexai import VertexAIClaude35Sonnet_20241022
 from langfun.core.llms.vertexai import VertexAIClaude35Haiku_20241022

langfun/core/llms/anthropic.py

@@ -59,6 +59,58 @@ class AnthropicModelInfo(lf.ModelInfo):
 
 
 SUPPORTED_MODELS = [
+    AnthropicModelInfo(
+        model_id='claude-4-opus-20250514',
+        provider='Anthropic',
+        in_service=True,
+        description='Claude 4 Opus model (5/14/2025).',
+        release_date=datetime.datetime(2025, 5, 14),
+        input_modalities=(
+            AnthropicModelInfo.INPUT_IMAGE_TYPES
+            + AnthropicModelInfo.INPUT_DOC_TYPES
+        ),
+        context_length=lf.ModelInfo.ContextLength(
+            max_input_tokens=200_000,
+            max_output_tokens=8_192,
+        ),
+        pricing=lf.ModelInfo.Pricing(
+            cost_per_1m_cached_input_tokens=1.5,
+            cost_per_1m_input_tokens=15,
+            cost_per_1m_output_tokens=75,
+        ),
+        rate_limits=AnthropicModelInfo.RateLimits(
+            # Tier 4 rate limits
+            max_requests_per_minute=2000,
+            max_input_tokens_per_minute=100_000,
+            max_output_tokens_per_minute=80_000,
+        ),
+    ),
+    AnthropicModelInfo(
+        model_id='claude-4-sonnet-20250514',
+        provider='Anthropic',
+        in_service=True,
+        description='Claude 4 Sonnet model (5/14/2025).',
+        release_date=datetime.datetime(2025, 5, 14),
+        input_modalities=(
+            AnthropicModelInfo.INPUT_IMAGE_TYPES
+            + AnthropicModelInfo.INPUT_DOC_TYPES
+        ),
+        context_length=lf.ModelInfo.ContextLength(
+            max_input_tokens=200_000,
+            max_output_tokens=8_192,
+        ),
+        pricing=lf.ModelInfo.Pricing(
+            cost_per_1m_cached_input_tokens=0.3,
+            cost_per_1m_input_tokens=3,
+            cost_per_1m_output_tokens=15,
+        ),
+        rate_limits=AnthropicModelInfo.RateLimits(
+            # Tier 4 rate limits
+            max_requests_per_minute=2000,
+            max_input_tokens_per_minute=100_000,
+            max_output_tokens_per_minute=80_000,
+        ),
+    ),
     # 3.5 Sonnet models.
     AnthropicModelInfo(
         model_id='claude-3-5-sonnet-latest',
@@ -138,6 +190,58 @@ SUPPORTED_MODELS = [
             max_output_tokens_per_minute=80_000,
         ),
     ),
+    AnthropicModelInfo(
+        model_id='claude-opus-4@20250514',
+        alias_for='claude-opus-4-20250514',
+        provider='VertexAI',
+        in_service=True,
+        description='Claude 4 Opus model served on VertexAI (5/14/2025).',
+        release_date=datetime.datetime(2025, 5, 14),
+        input_modalities=(
+            AnthropicModelInfo.INPUT_IMAGE_TYPES
+            + AnthropicModelInfo.INPUT_DOC_TYPES
+        ),
+        context_length=lf.ModelInfo.ContextLength(
+            max_input_tokens=200_000,
+            max_output_tokens=8_192,
+        ),
+        pricing=lf.ModelInfo.Pricing(
+            cost_per_1m_cached_input_tokens=1.5,
+            cost_per_1m_input_tokens=15.0,
+            cost_per_1m_output_tokens=75.0,
+        ),
+        rate_limits=AnthropicModelInfo.RateLimits(
+            max_requests_per_minute=100,
+            max_input_tokens_per_minute=1_000_000,
+            max_output_tokens_per_minute=80_000,
+        ),
+    ),
+    AnthropicModelInfo(
+        model_id='claude-sonnet-4@20250514',
+        alias_for='claude-sonnet-4@20250514',
+        provider='VertexAI',
+        in_service=True,
+        description='Claude 4 Sonnet model served on VertexAI (5/14/2025).',
+        release_date=datetime.datetime(2025, 5, 14),
+        input_modalities=(
+            AnthropicModelInfo.INPUT_IMAGE_TYPES
+            + AnthropicModelInfo.INPUT_DOC_TYPES
+        ),
+        context_length=lf.ModelInfo.ContextLength(
+            max_input_tokens=200_000,
+            max_output_tokens=8_192,
+        ),
+        pricing=lf.ModelInfo.Pricing(
+            cost_per_1m_cached_input_tokens=0.3,
+            cost_per_1m_input_tokens=3.0,
+            cost_per_1m_output_tokens=15.0,
+        ),
+        rate_limits=AnthropicModelInfo.RateLimits(
+            max_requests_per_minute=100,
+            max_input_tokens_per_minute=1_000_000,
+            max_output_tokens_per_minute=80_000,
+        ),
+    ),
     AnthropicModelInfo(
         model_id='claude-3-5-sonnet-v2@20241022',
         alias_for='claude-3-5-sonnet-20241022',
@@ -575,6 +679,24 @@ class Anthropic(rest.REST):
     return super()._error(status_code, content)
 
 
+class Claude4(Anthropic):
+  """Base class for Claude 4 models."""
+
+
+# pylint: disable=invalid-name
+class Claude4Opus_20250514(Claude4):
+  """Claude 4 Opus model 20250514."""
+
+  model = 'claude-4-opus-20250514'
+
+
+# pylint: disable=invalid-name
+class Claude4Sonnet_20250514(Claude4):
+  """Claude 4 Sonnet model 20250514."""
+
+  model = 'claude-4-sonnet-20250514'
+
+
 class Claude37(Anthropic):
   """Base class for Claude 3.7 models."""
 

langfun/core/llms/vertexai.py

@@ -338,6 +338,18 @@ class VertexAIAnthropic(VertexAI, anthropic.Anthropic):
 # pylint: disable=invalid-name
 
 
+class VertexAIClaude4Opus_20250514(VertexAIAnthropic):
+  """Anthropic's Claude 4 Opus model on VertexAI."""
+  model = 'claude-opus-4@20250514'
+  location = 'us-east5'
+
+
+class VertexAIClaude4Sonnet_20250514(VertexAIAnthropic):
+  """Anthropic's Claude 4 Sonnet model on VertexAI."""
+  model = 'claude-sonnet-4@20250514'
+  location = 'us-east5'
+
+
 class VertexAIClaude37Sonnet_20250219(VertexAIAnthropic):
   """Anthropic's Claude 3.7 model on VertexAI."""
   model = 'claude-3-7-sonnet@20250219'

{langfun-0.1.2.dev202505220804.dist-info → langfun-0.1.2.dev202505240803.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langfun
-Version: 0.1.2.dev202505220804
+Version: 0.1.2.dev202505240803
 Summary: Langfun: Language as Functions.
 Home-page: https://github.com/google/langfun
 Author: Langfun Authors

{langfun-0.1.2.dev202505220804.dist-info → langfun-0.1.2.dev202505240803.dist-info}/RECORD

@@ -81,8 +81,8 @@ langfun/core/eval/v2/reporting.py,sha256=yUIPCAMnp7InIzpv1DDWrcLO-75iiOUTpscj7sm
 langfun/core/eval/v2/reporting_test.py,sha256=hcPJJaMtPulqERvHYTpId83WXdqDKnnexmULtK7WKwk,5686
 langfun/core/eval/v2/runners.py,sha256=iqbH4jMtnNMhfuv1eHaxJmk1Vvsrz-sAJJFP8U44-tA,16758
 langfun/core/eval/v2/runners_test.py,sha256=DO3xV0sBNB6n65j41xx2i7gqUCJcPF37DFZLEjrmISg,11987
-langfun/core/llms/__init__.py,sha256=tYMsdEiCJCDGQK8fF2PTZgOQylVTHyPnqP8ytonh2YY,8834
-langfun/core/llms/anthropic.py,sha256=dCkTYvwb_wWWeKQNzMtRfA6kno1E_bO9PBHSNqL0wiw,22080
+langfun/core/llms/__init__.py,sha256=_fdiqQ3s-VxNfd0C2uEnJzNncr8xaPkbQEg9TlOXQ1k,9144
+langfun/core/llms/anthropic.py,sha256=YcQ2VG8iOfXtry_tTpAukmiwXa2hK_9LkpkmXk41Nm0,26226
 langfun/core/llms/anthropic_test.py,sha256=qA9vByp_cwwXNlXzcwHpPWFnO9lfFo8NKfDi5nBNqgI,9052
 langfun/core/llms/azure_openai.py,sha256=-KkSLaR54MlsIqz_XIwv0TnsBnvNTAxnjA2Q2O2u5KM,2733
 langfun/core/llms/azure_openai_test.py,sha256=lkMZkQdJBV97fTM4C4z8qNfvr6spgiN5G4hvVUIVr0M,1735
@@ -106,7 +106,7 @@ langfun/core/llms/openai_compatible_test.py,sha256=KwOMA7tsmOxFBjezltkBDSU77AvOQ
 langfun/core/llms/openai_test.py,sha256=gwuO6aoa296iM2welWV9ua4KF8gEVGsEPakgbtkWkFQ,2687
 langfun/core/llms/rest.py,sha256=MCybcHApJcf49lubLnDzScN9Oc2IWY_JnMHIGdbDOuU,4474
 langfun/core/llms/rest_test.py,sha256=_zM7nV8DEVyoXNiQOnuwJ917mWjki0614H88rNmDboE,5020
-langfun/core/llms/vertexai.py,sha256=PVUMcxLNgYfbKOS2pw4M8aqb0Jq7bn-bOw-VsxQqOf0,19009
+langfun/core/llms/vertexai.py,sha256=OqtSyq2PqQuFHva3ewhU_pt74sKxCAZMRKA5x1KyDcM,19353
 langfun/core/llms/vertexai_test.py,sha256=_e-acnNBAf9C3WO6i1b2J_mhRzdDdYQTorD9hIVZKOg,5034
 langfun/core/llms/cache/__init__.py,sha256=QAo3InUMDM_YpteNnVCSejI4zOsnjSMWKJKzkb3VY64,993
 langfun/core/llms/cache/base.py,sha256=rt3zwmyw0y9jsSGW-ZbV1vAfLxQ7_3AVk0l2EySlse4,3918
@@ -156,8 +156,8 @@ langfun/core/templates/demonstration.py,sha256=vCrgYubdZM5Umqcgp8NUVGXgr4P_c-fik
 langfun/core/templates/demonstration_test.py,sha256=SafcDQ0WgI7pw05EmPI2S4v1t3ABKzup8jReCljHeK4,2162
 langfun/core/templates/selfplay.py,sha256=yhgrJbiYwq47TgzThmHrDQTF4nDrTI09CWGhuQPNv-s,2273
 langfun/core/templates/selfplay_test.py,sha256=Ot__1P1M8oJfoTp-M9-PQ6HUXqZKyMwvZ5f7yQ3yfyM,2326
-langfun-0.1.2.dev202505220804.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202505220804.dist-info/METADATA,sha256=MhW6I0xv1Qb9incj5YrOGVnXXoZlj_Le1Ac3uZlZgnw,8178
-langfun-0.1.2.dev202505220804.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-langfun-0.1.2.dev202505220804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202505220804.dist-info/RECORD,,
+langfun-0.1.2.dev202505240803.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202505240803.dist-info/METADATA,sha256=PkWFG3l7Pk1iOQm2DfB9OzuTk28f5er8rQdoBDiE22M,8178
+langfun-0.1.2.dev202505240803.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+langfun-0.1.2.dev202505240803.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202505240803.dist-info/RECORD,,