langfun 0.1.2.dev202502110804__py3-none-any.whl → 0.1.2.dev202502120804__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.
Files changed (27) hide show
  1. langfun/core/__init__.py +6 -2
  2. langfun/core/language_model.py +365 -22
  3. langfun/core/language_model_test.py +123 -35
  4. langfun/core/llms/__init__.py +50 -57
  5. langfun/core/llms/anthropic.py +434 -163
  6. langfun/core/llms/anthropic_test.py +20 -1
  7. langfun/core/llms/deepseek.py +90 -51
  8. langfun/core/llms/deepseek_test.py +15 -16
  9. langfun/core/llms/fake.py +6 -0
  10. langfun/core/llms/gemini.py +480 -390
  11. langfun/core/llms/gemini_test.py +27 -7
  12. langfun/core/llms/google_genai.py +80 -50
  13. langfun/core/llms/google_genai_test.py +11 -4
  14. langfun/core/llms/groq.py +268 -167
  15. langfun/core/llms/groq_test.py +9 -3
  16. langfun/core/llms/openai.py +839 -328
  17. langfun/core/llms/openai_compatible.py +3 -18
  18. langfun/core/llms/openai_compatible_test.py +20 -5
  19. langfun/core/llms/openai_test.py +14 -4
  20. langfun/core/llms/rest.py +11 -6
  21. langfun/core/llms/vertexai.py +238 -240
  22. langfun/core/llms/vertexai_test.py +35 -8
  23. {langfun-0.1.2.dev202502110804.dist-info → langfun-0.1.2.dev202502120804.dist-info}/METADATA +1 -1
  24. {langfun-0.1.2.dev202502110804.dist-info → langfun-0.1.2.dev202502120804.dist-info}/RECORD +27 -27
  25. {langfun-0.1.2.dev202502110804.dist-info → langfun-0.1.2.dev202502120804.dist-info}/LICENSE +0 -0
  26. {langfun-0.1.2.dev202502110804.dist-info → langfun-0.1.2.dev202502120804.dist-info}/WHEEL +0 -0
  27. {langfun-0.1.2.dev202502110804.dist-info → langfun-0.1.2.dev202502120804.dist-info}/top_level.txt +0 -0
langfun/core/llms/__init__.py CHANGED
@@ -40,35 +40,43 @@ from langfun.core.llms.groq import Groq
40
40
  # Gemini models.
41
41
  from langfun.core.llms.google_genai import GenAI
42
42
  from langfun.core.llms.google_genai import Gemini2Flash
43
+ from langfun.core.llms.google_genai import Gemini2Flash_001
44
+ from langfun.core.llms.google_genai import Gemini2FlashLitePreview_20250205
45
+ from langfun.core.llms.google_genai import Gemini15Pro
46
+ from langfun.core.llms.google_genai import Gemini15Pro_002
47
+ from langfun.core.llms.google_genai import Gemini15Pro_001
48
+ from langfun.core.llms.google_genai import Gemini15Flash
49
+ from langfun.core.llms.google_genai import Gemini15Flash_002
50
+ from langfun.core.llms.google_genai import Gemini15Flash_001
51
+ from langfun.core.llms.google_genai import Gemini15Flash8B
52
+ from langfun.core.llms.google_genai import Gemini15Flash8B_001
53
+
43
54
  from langfun.core.llms.google_genai import Gemini2ProExp_20250205
44
55
  from langfun.core.llms.google_genai import Gemini2FlashThinkingExp_20250121
45
- from langfun.core.llms.google_genai import GeminiFlash2_0ThinkingExp_20241219
46
- from langfun.core.llms.google_genai import GeminiFlash2_0Exp
47
56
  from langfun.core.llms.google_genai import GeminiExp_20241206
48
- from langfun.core.llms.google_genai import GeminiExp_20241114
49
- from langfun.core.llms.google_genai import GeminiPro1_5
50
- from langfun.core.llms.google_genai import GeminiPro1_5_002
51
- from langfun.core.llms.google_genai import GeminiPro1_5_001
52
- from langfun.core.llms.google_genai import GeminiFlash1_5
53
- from langfun.core.llms.google_genai import GeminiFlash1_5_002
54
- from langfun.core.llms.google_genai import GeminiFlash1_5_001
55
- from langfun.core.llms.google_genai import GeminiPro1
56
57
 
57
58
  from langfun.core.llms.vertexai import VertexAIGemini
58
59
  from langfun.core.llms.vertexai import VertexAIGemini2Flash
60
+ from langfun.core.llms.vertexai import VertexAIGemini2Flash_001
61
+ from langfun.core.llms.vertexai import VertexAIGemini2FlashLitePreview_20250205
62
+ from langfun.core.llms.vertexai import VertexAIGemini15Pro
63
+ from langfun.core.llms.vertexai import VertexAIGemini15Pro_002
64
+ from langfun.core.llms.vertexai import VertexAIGemini15Pro_001
65
+ from langfun.core.llms.vertexai import VertexAIGemini15Flash
66
+ from langfun.core.llms.vertexai import VertexAIGemini15Flash_002
67
+ from langfun.core.llms.vertexai import VertexAIGemini15Flash_001
68
+ from langfun.core.llms.vertexai import VertexAIGemini15Flash8B
69
+ from langfun.core.llms.vertexai import VertexAIGemini15Flash8B_001
70
+
59
71
  from langfun.core.llms.vertexai import VertexAIGemini2ProExp_20250205
60
72
  from langfun.core.llms.vertexai import VertexAIGemini2FlashThinkingExp_20250121
61
- from langfun.core.llms.vertexai import VertexAIGeminiFlash2_0ThinkingExp_20241219
62
- from langfun.core.llms.vertexai import VertexAIGeminiFlash2_0Exp
63
73
  from langfun.core.llms.vertexai import VertexAIGeminiExp_20241206
64
- from langfun.core.llms.vertexai import VertexAIGeminiExp_20241114
65
- from langfun.core.llms.vertexai import VertexAIGeminiPro1_5
66
- from langfun.core.llms.vertexai import VertexAIGeminiPro1_5_002
67
- from langfun.core.llms.vertexai import VertexAIGeminiPro1_5_001
68
- from langfun.core.llms.vertexai import VertexAIGeminiFlash1_5
69
- from langfun.core.llms.vertexai import VertexAIGeminiFlash1_5_002
70
- from langfun.core.llms.vertexai import VertexAIGeminiFlash1_5_001
71
- from langfun.core.llms.vertexai import VertexAIGeminiPro1
74
+
75
+ # For backward compatibility.
76
+ GeminiPro1_5 = Gemini15Pro
77
+ GeminiFlash1_5 = Gemini15Flash
78
+ VertexAIGeminiPro1_5 = VertexAIGemini15Pro
79
+ VertexAIGeminiFlash1_5 = VertexAIGemini15Flash
72
80
 
73
81
  # OpenAI models.
74
82
  from langfun.core.llms.openai import OpenAI
@@ -92,8 +100,6 @@ from langfun.core.llms.openai import Gpt4Turbo_20240409
92
100
  from langfun.core.llms.openai import Gpt4TurboPreview
93
101
  from langfun.core.llms.openai import Gpt4TurboPreview_20240125
94
102
  from langfun.core.llms.openai import Gpt4TurboPreview_20231106
95
- from langfun.core.llms.openai import Gpt4VisionPreview
96
- from langfun.core.llms.openai import Gpt4VisionPreview_20231106
97
103
  from langfun.core.llms.openai import Gpt4
98
104
  from langfun.core.llms.openai import Gpt4_20230613
99
105
 
@@ -110,7 +116,6 @@ from langfun.core.llms.openai import Gpt35Turbo16K_20230613
110
116
  # For backward compatibility.
111
117
  Gpt4TurboPreview_0125 = Gpt4TurboPreview_20240125
112
118
  Gpt4TurboPreview_1106 = Gpt4TurboPreview_20231106
113
- Gpt4VisionPreview_1106 = Gpt4VisionPreview_20231106
114
119
  Gpt4_0613 = Gpt4_20230613
115
120
  Gpt4_32K_0613 = Gpt4_32K_20230613
116
121
  Gpt35Turbo_0125 = Gpt35Turbo_20240125
@@ -120,70 +125,58 @@ Gpt35Turbo16K_0613 = Gpt35Turbo16K_20230613
120
125
 
121
126
  from langfun.core.llms.openai import Gpt35
122
127
 
123
- from langfun.core.llms.openai import Gpt3
124
- from langfun.core.llms.openai import Gpt3Curie
125
- from langfun.core.llms.openai import Gpt3Babbage
126
- from langfun.core.llms.openai import Gpt3Ada
127
-
128
128
  # Anthropic models.
129
129
 
130
130
  from langfun.core.llms.anthropic import Claude35Sonnet
131
- from langfun.core.llms.anthropic import Claude35Sonnet20241022
132
- from langfun.core.llms.anthropic import Claude35Sonnet20240620
131
+ from langfun.core.llms.anthropic import Claude35Sonnet_20241022
132
+ from langfun.core.llms.anthropic import Claude35Haiku
133
+ from langfun.core.llms.anthropic import Claude35Haiku_20241022
133
134
  from langfun.core.llms.anthropic import Claude3Opus
135
+ from langfun.core.llms.anthropic import Claude3Opus_20240229
134
136
  from langfun.core.llms.anthropic import Claude3Sonnet
137
+ from langfun.core.llms.anthropic import Claude3Sonnet_20240229
135
138
  from langfun.core.llms.anthropic import Claude3Haiku
139
+ from langfun.core.llms.anthropic import Claude3Haiku_20240307
136
140
 
137
141
  from langfun.core.llms.vertexai import VertexAIAnthropic
138
- from langfun.core.llms.vertexai import VertexAIClaude3_5_Sonnet_20241022
139
- from langfun.core.llms.vertexai import VertexAIClaude3_5_Sonnet_20240620
140
- from langfun.core.llms.vertexai import VertexAIClaude3_5_Haiku_20241022
141
- from langfun.core.llms.vertexai import VertexAIClaude3_Opus_20240229
142
+ from langfun.core.llms.vertexai import VertexAIClaude35Sonnet_20241022
143
+ from langfun.core.llms.vertexai import VertexAIClaude35Haiku_20241022
144
+ from langfun.core.llms.vertexai import VertexAIClaude3Opus_20240229
142
145
 
143
146
  # Misc open source models.
144
147
 
145
148
  # Gemma models.
146
149
  from langfun.core.llms.groq import GroqGemma2_9B_IT
147
- from langfun.core.llms.groq import GroqGemma_7B_IT
148
150
 
149
151
  # Llama models.
150
152
  from langfun.core.llms.vertexai import VertexAILlama
151
- from langfun.core.llms.vertexai import VertexAILlama3_2_90B
152
- from langfun.core.llms.vertexai import VertexAILlama3_1_405B
153
- from langfun.core.llms.vertexai import VertexAILlama3_1_70B
154
- from langfun.core.llms.vertexai import VertexAILlama3_1_8B
155
-
156
- from langfun.core.llms.groq import GroqLlama3_2_3B
157
- from langfun.core.llms.groq import GroqLlama3_2_1B
158
- from langfun.core.llms.groq import GroqLlama3_1_70B
159
- from langfun.core.llms.groq import GroqLlama3_1_8B
160
- from langfun.core.llms.groq import GroqLlama3_70B
161
- from langfun.core.llms.groq import GroqLlama3_8B
162
- from langfun.core.llms.groq import GroqLlama2_70B
153
+ from langfun.core.llms.vertexai import VertexAILlama32_90B
154
+ from langfun.core.llms.vertexai import VertexAILlama31_405B
155
+ from langfun.core.llms.vertexai import VertexAILlama31_70B
156
+ from langfun.core.llms.vertexai import VertexAILlama31_8B
157
+
158
+ from langfun.core.llms.groq import GroqLlama33_70B_Versatile
159
+ from langfun.core.llms.groq import GroqLlama33_70B_SpecDec
160
+ from langfun.core.llms.groq import GroqLlama32_1B
161
+ from langfun.core.llms.groq import GroqLlama32_3B
162
+ from langfun.core.llms.groq import GroqLlama32_11B_Vision
163
+ from langfun.core.llms.groq import GroqLlama32_90B_Vision
163
164
 
164
165
  # Mistral models.
165
166
  from langfun.core.llms.vertexai import VertexAIMistral
166
167
  from langfun.core.llms.vertexai import VertexAIMistralLarge_20241121
167
- from langfun.core.llms.vertexai import VertexAIMistralLarge_20240724
168
- from langfun.core.llms.vertexai import VertexAIMistralNemo_20240724
169
168
  from langfun.core.llms.vertexai import VertexAICodestral_20250113
170
- from langfun.core.llms.vertexai import VertexAICodestral_20240529
171
169
 
172
170
  from langfun.core.llms.groq import GroqMistral_8x7B
173
171
 
174
172
  # DeepSeek models.
175
173
  from langfun.core.llms.deepseek import DeepSeek
176
- from langfun.core.llms.deepseek import DeepSeekChat
177
- from langfun.core.llms.deepseek import DeepSeekReasoner
178
-
179
- # Whisper models.
180
- from langfun.core.llms.groq import GroqWhisper_Large_v3
181
- from langfun.core.llms.groq import GroqWhisper_Large_v3Turbo
174
+ from langfun.core.llms.deepseek import DeepSeekV3
175
+ from langfun.core.llms.deepseek import DeepSeekR1
182
176
 
183
177
  # LLaMA C++ models.
184
178
  from langfun.core.llms.llama_cpp import LlamaCppRemote
185
179
 
186
-
187
180
  # Placeholder for Google-internal imports.
188
181
 
189
182
  # Include cache as sub-module.