langfun 0.1.2.dev202501310804__tar.gz → 0.1.2.dev202502010803__tar.gz

{langfun-0.1.2.dev202501310804 → langfun-0.1.2.dev202502010803}/PKG-INFO

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

{langfun-0.1.2.dev202501310804 → langfun-0.1.2.dev202502010803}/langfun/core/llms/__init__.py

@@ -67,6 +67,7 @@ from langfun.core.llms.vertexai import VertexAIGeminiPro1
 # OpenAI models.
 from langfun.core.llms.openai import OpenAI
 
+from langfun.core.llms.openai import GptO3Mini
 from langfun.core.llms.openai import GptO1
 from langfun.core.llms.openai import GptO1Preview
 from langfun.core.llms.openai import GptO1Preview_20240912
@@ -167,6 +168,7 @@ from langfun.core.llms.groq import GroqMistral_8x7B
 # DeepSeek models.
 from langfun.core.llms.deepseek import DeepSeek
 from langfun.core.llms.deepseek import DeepSeekChat
+from langfun.core.llms.deepseek import DeepSeekReasoner
 
 # Whisper models.
 from langfun.core.llms.groq import GroqWhisper_Large_v3

{langfun-0.1.2.dev202501310804 → langfun-0.1.2.dev202502010803}/langfun/core/llms/deepseek.py

@@ -25,6 +25,13 @@ SUPPORTED_MODELS_AND_SETTINGS = {
     # TODO(yifenglu): The RPM and TPM are arbitrary numbers. Update them once DeepSeek provides concrete guidelines.
     # DeepSeek doesn't control the rate limit at the moment: https://api-docs.deepseek.com/quick_start/rate_limit
     # The cost is based on: https://api-docs.deepseek.com/quick_start/pricing
+    'deepseek-reasoner': pg.Dict(
+        in_service=True,
+        rpm=100,
+        tpm=1000000,
+        cost_per_1k_input_tokens=0.00055,
+        cost_per_1k_output_tokens=0.00219,
+    ),
     'deepseek-chat': pg.Dict(
         in_service=True,
         rpm=100,
@@ -107,6 +114,16 @@ class DeepSeek(openai_compatible.OpenAICompatible):
     return [k for k, v in SUPPORTED_MODELS_AND_SETTINGS.items() if v.in_service]
 
 
+class DeepSeekReasoner(DeepSeek):
+  """DeepSeek Reasoner model.
+
+  Currently it is powered by DeepSeek-R1 model, 64k input context, 8k max
+  output, 32k max CoT output.
+  """
+
+  model = 'deepseek-reasoner'
+
+
 class DeepSeekChat(DeepSeek):
   """DeepSeek Chat model.
 

{langfun-0.1.2.dev202501310804 → langfun-0.1.2.dev202502010803}/langfun/core/llms/openai.py

@@ -31,6 +31,20 @@ SUPPORTED_MODELS_AND_SETTINGS = {
     # o1 (preview) models.
     # Pricing in US dollars, from https://openai.com/api/pricing/
     # as of 2024-10-10.
+    'o3-mini-2025-01-31': pg.Dict(
+        in_service=True,
+        rpm=10000,
+        tpm=5000000,
+        cost_per_1k_input_tokens=0.0011,
+        cost_per_1k_output_tokens=0.0044,
+    ),
+    'o3-mini': pg.Dict(
+        in_service=True,
+        rpm=10000,
+        tpm=5000000,
+        cost_per_1k_input_tokens=0.0011,
+        cost_per_1k_output_tokens=0.0044,
+    ),
     'o1': pg.Dict(
         in_service=True,
         rpm=10000,
@@ -410,6 +424,14 @@ class OpenAI(openai_compatible.OpenAICompatible):
     return super()._request_args(options)
 
 
+class GptO3Mini(OpenAI):
+  """GPT-O3-mini."""
+
+  model = 'o3-mini'
+  multimodal = True
+  timeout = None
+
+
 class GptO1(OpenAI):
   """GPT-O1."""
 

{langfun-0.1.2.dev202501310804 → langfun-0.1.2.dev202502010803}/langfun.egg-info/PKG-INFO

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