paygent-sdk 1.0.0__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.
paygent_sdk/models.py ADDED
@@ -0,0 +1,485 @@
1
+ """
2
+ Data models for the Paygent SDK.
3
+ """
4
+
5
+ from dataclasses import dataclass
6
+ from typing import Dict
7
+
8
+ # Import constants
9
+ from .constants import (
10
+ OpenAIModels,
11
+ AnthropicModels,
12
+ GoogleDeepMindModels,
13
+ MetaModels,
14
+ AWSModels,
15
+ MistralAIModels,
16
+ CohereModels,
17
+ DeepSeekModels,
18
+ )
19
+
20
+
21
+ @dataclass
22
+ class UsageData:
23
+ """Represents the usage data structure."""
24
+ service_provider: str
25
+ model: str
26
+ prompt_tokens: int
27
+ completion_tokens: int
28
+ total_tokens: int
29
+
30
+
31
+ @dataclass
32
+ class UsageDataWithStrings:
33
+ """Represents the usage data structure with prompt and output strings."""
34
+ service_provider: str
35
+ model: str
36
+ prompt_string: str
37
+ output_string: str
38
+
39
+
40
+ @dataclass
41
+ class APIRequest:
42
+ """Represents the request body for the API call."""
43
+ agent_id: str
44
+ customer_id: str
45
+ indicator: str
46
+ amount: float
47
+
48
+
49
+ @dataclass
50
+ class ModelPricing:
51
+ """Represents pricing information for different models."""
52
+ prompt_tokens_cost: float
53
+ completion_tokens_cost: float
54
+
55
+
56
+ # Default model pricing (cost per 1000 tokens in USD)
57
+ MODEL_PRICING: Dict[str, ModelPricing] = {
58
+ # OpenAI Models (pricing per 1000 tokens)
59
+ OpenAIModels.GPT_5: ModelPricing(
60
+ prompt_tokens_cost=0.00125, # $0.00125 per 1000 tokens
61
+ completion_tokens_cost=0.01 # $0.01 per 1000 tokens
62
+ ),
63
+ OpenAIModels.GPT_5_MINI: ModelPricing(
64
+ prompt_tokens_cost=0.00025, # $0.00025 per 1000 tokens
65
+ completion_tokens_cost=0.002 # $0.002 per 1000 tokens
66
+ ),
67
+ OpenAIModels.GPT_5_NANO: ModelPricing(
68
+ prompt_tokens_cost=0.00005, # $0.00005 per 1000 tokens
69
+ completion_tokens_cost=0.0004 # $0.0004 per 1000 tokens
70
+ ),
71
+ OpenAIModels.GPT_5_CHAT_LATEST: ModelPricing(
72
+ prompt_tokens_cost=0.00125, # $0.00125 per 1000 tokens
73
+ completion_tokens_cost=0.01 # $0.01 per 1000 tokens
74
+ ),
75
+ OpenAIModels.GPT_5_CODEX: ModelPricing(
76
+ prompt_tokens_cost=0.00125, # $0.00125 per 1000 tokens
77
+ completion_tokens_cost=0.01 # $0.01 per 1000 tokens
78
+ ),
79
+ OpenAIModels.GPT_5_PRO: ModelPricing(
80
+ prompt_tokens_cost=0.015, # $0.015 per 1000 tokens
81
+ completion_tokens_cost=0.12 # $0.12 per 1000 tokens
82
+ ),
83
+ OpenAIModels.GPT_5_SEARCH_API: ModelPricing(
84
+ prompt_tokens_cost=0.00125, # $0.00125 per 1000 tokens
85
+ completion_tokens_cost=0.01 # $0.01 per 1000 tokens
86
+ ),
87
+ OpenAIModels.GPT_4_1: ModelPricing(
88
+ prompt_tokens_cost=0.002, # $0.002 per 1000 tokens
89
+ completion_tokens_cost=0.008 # $0.008 per 1000 tokens
90
+ ),
91
+ OpenAIModels.GPT_4_1_MINI: ModelPricing(
92
+ prompt_tokens_cost=0.0004, # $0.0004 per 1000 tokens
93
+ completion_tokens_cost=0.0016 # $0.0016 per 1000 tokens
94
+ ),
95
+ OpenAIModels.GPT_4_1_NANO: ModelPricing(
96
+ prompt_tokens_cost=0.0001, # $0.0001 per 1000 tokens
97
+ completion_tokens_cost=0.0004 # $0.0004 per 1000 tokens
98
+ ),
99
+ OpenAIModels.GPT_4O: ModelPricing(
100
+ prompt_tokens_cost=0.0025, # $0.0025 per 1000 tokens
101
+ completion_tokens_cost=0.01 # $0.01 per 1000 tokens
102
+ ),
103
+ OpenAIModels.GPT_4O_2024_05_13: ModelPricing(
104
+ prompt_tokens_cost=0.005, # $0.005 per 1000 tokens
105
+ completion_tokens_cost=0.015 # $0.015 per 1000 tokens
106
+ ),
107
+ OpenAIModels.GPT_4O_MINI: ModelPricing(
108
+ prompt_tokens_cost=0.00015, # $0.00015 per 1000 tokens
109
+ completion_tokens_cost=0.0006 # $0.0006 per 1000 tokens
110
+ ),
111
+ OpenAIModels.GPT_REALTIME: ModelPricing(
112
+ prompt_tokens_cost=0.004, # $0.004 per 1000 tokens
113
+ completion_tokens_cost=0.016 # $0.016 per 1000 tokens
114
+ ),
115
+ OpenAIModels.GPT_REALTIME_MINI: ModelPricing(
116
+ prompt_tokens_cost=0.0006, # $0.0006 per 1000 tokens
117
+ completion_tokens_cost=0.0024 # $0.0024 per 1000 tokens
118
+ ),
119
+ OpenAIModels.GPT_4O_REALTIME_PREVIEW: ModelPricing(
120
+ prompt_tokens_cost=0.005, # $0.005 per 1000 tokens
121
+ completion_tokens_cost=0.02 # $0.02 per 1000 tokens
122
+ ),
123
+ OpenAIModels.GPT_4O_MINI_REALTIME_PREVIEW: ModelPricing(
124
+ prompt_tokens_cost=0.0006, # $0.0006 per 1000 tokens
125
+ completion_tokens_cost=0.0024 # $0.0024 per 1000 tokens
126
+ ),
127
+ OpenAIModels.GPT_AUDIO: ModelPricing(
128
+ prompt_tokens_cost=0.0025, # $0.0025 per 1000 tokens
129
+ completion_tokens_cost=0.01 # $0.01 per 1000 tokens
130
+ ),
131
+ OpenAIModels.GPT_AUDIO_MINI: ModelPricing(
132
+ prompt_tokens_cost=0.0006, # $0.0006 per 1000 tokens
133
+ completion_tokens_cost=0.0024 # $0.0024 per 1000 tokens
134
+ ),
135
+ OpenAIModels.GPT_4O_AUDIO_PREVIEW: ModelPricing(
136
+ prompt_tokens_cost=0.0025, # $0.0025 per 1000 tokens
137
+ completion_tokens_cost=0.01 # $0.01 per 1000 tokens
138
+ ),
139
+ OpenAIModels.GPT_4O_MINI_AUDIO_PREVIEW: ModelPricing(
140
+ prompt_tokens_cost=0.00015, # $0.00015 per 1000 tokens
141
+ completion_tokens_cost=0.0006 # $0.0006 per 1000 tokens
142
+ ),
143
+ OpenAIModels.O1: ModelPricing(
144
+ prompt_tokens_cost=0.015, # $0.015 per 1000 tokens
145
+ completion_tokens_cost=0.06 # $0.06 per 1000 tokens
146
+ ),
147
+ OpenAIModels.O1_PRO: ModelPricing(
148
+ prompt_tokens_cost=0.15, # $0.15 per 1000 tokens
149
+ completion_tokens_cost=0.6 # $0.6 per 1000 tokens
150
+ ),
151
+ OpenAIModels.O3_PRO: ModelPricing(
152
+ prompt_tokens_cost=0.02, # $0.02 per 1000 tokens
153
+ completion_tokens_cost=0.08 # $0.08 per 1000 tokens
154
+ ),
155
+ OpenAIModels.O3: ModelPricing(
156
+ prompt_tokens_cost=0.002, # $0.002 per 1000 tokens
157
+ completion_tokens_cost=0.008 # $0.008 per 1000 tokens
158
+ ),
159
+ OpenAIModels.O3_DEEP_RESEARCH: ModelPricing(
160
+ prompt_tokens_cost=0.01, # $0.01 per 1000 tokens
161
+ completion_tokens_cost=0.04 # $0.04 per 1000 tokens
162
+ ),
163
+ OpenAIModels.O4_MINI: ModelPricing(
164
+ prompt_tokens_cost=0.0011, # $0.0011 per 1000 tokens
165
+ completion_tokens_cost=0.0044 # $0.0044 per 1000 tokens
166
+ ),
167
+ OpenAIModels.O4_MINI_DEEP_RESEARCH: ModelPricing(
168
+ prompt_tokens_cost=0.002, # $0.002 per 1000 tokens
169
+ completion_tokens_cost=0.008 # $0.008 per 1000 tokens
170
+ ),
171
+ OpenAIModels.O3_MINI: ModelPricing(
172
+ prompt_tokens_cost=0.0011, # $0.0011 per 1000 tokens
173
+ completion_tokens_cost=0.0044 # $0.0044 per 1000 tokens
174
+ ),
175
+ OpenAIModels.O1_MINI: ModelPricing(
176
+ prompt_tokens_cost=0.0011, # $0.0011 per 1000 tokens
177
+ completion_tokens_cost=0.0044 # $0.0044 per 1000 tokens
178
+ ),
179
+ OpenAIModels.CODEX_MINI_LATEST: ModelPricing(
180
+ prompt_tokens_cost=0.0015, # $0.0015 per 1000 tokens
181
+ completion_tokens_cost=0.006 # $0.006 per 1000 tokens
182
+ ),
183
+ OpenAIModels.GPT_4O_MINI_SEARCH_PREVIEW: ModelPricing(
184
+ prompt_tokens_cost=0.00015, # $0.00015 per 1000 tokens
185
+ completion_tokens_cost=0.0006 # $0.0006 per 1000 tokens
186
+ ),
187
+ OpenAIModels.GPT_4O_SEARCH_PREVIEW: ModelPricing(
188
+ prompt_tokens_cost=0.0025, # $0.0025 per 1000 tokens
189
+ completion_tokens_cost=0.01 # $0.01 per 1000 tokens
190
+ ),
191
+ OpenAIModels.COMPUTER_USE_PREVIEW: ModelPricing(
192
+ prompt_tokens_cost=0.003, # $0.003 per 1000 tokens
193
+ completion_tokens_cost=0.012 # $0.012 per 1000 tokens
194
+ ),
195
+ OpenAIModels.CHATGPT_4O_LATEST: ModelPricing(
196
+ prompt_tokens_cost=0.005, # $0.005 per 1000 tokens
197
+ completion_tokens_cost=0.015 # $0.015 per 1000 tokens
198
+ ),
199
+ OpenAIModels.GPT_4_TURBO_2024_04_09: ModelPricing(
200
+ prompt_tokens_cost=0.01, # $0.01 per 1000 tokens
201
+ completion_tokens_cost=0.03 # $0.03 per 1000 tokens
202
+ ),
203
+ OpenAIModels.GPT_4_0125_PREVIEW: ModelPricing(
204
+ prompt_tokens_cost=0.01, # $0.01 per 1000 tokens
205
+ completion_tokens_cost=0.03 # $0.03 per 1000 tokens
206
+ ),
207
+ OpenAIModels.GPT_4_1106_PREVIEW: ModelPricing(
208
+ prompt_tokens_cost=0.01, # $0.01 per 1000 tokens
209
+ completion_tokens_cost=0.03 # $0.03 per 1000 tokens
210
+ ),
211
+ OpenAIModels.GPT_4_1106_VISION_PREVIEW: ModelPricing(
212
+ prompt_tokens_cost=0.01, # $0.01 per 1000 tokens
213
+ completion_tokens_cost=0.03 # $0.03 per 1000 tokens
214
+ ),
215
+ OpenAIModels.GPT_4_0613: ModelPricing(
216
+ prompt_tokens_cost=0.03, # $0.03 per 1000 tokens
217
+ completion_tokens_cost=0.06 # $0.06 per 1000 tokens
218
+ ),
219
+ OpenAIModels.GPT_4_0314: ModelPricing(
220
+ prompt_tokens_cost=0.03, # $0.03 per 1000 tokens
221
+ completion_tokens_cost=0.06 # $0.06 per 1000 tokens
222
+ ),
223
+ OpenAIModels.GPT_4_32K: ModelPricing(
224
+ prompt_tokens_cost=0.06, # $0.06 per 1000 tokens
225
+ completion_tokens_cost=0.12 # $0.12 per 1000 tokens
226
+ ),
227
+ OpenAIModels.GPT_3_5_TURBO: ModelPricing(
228
+ prompt_tokens_cost=0.0005, # $0.0005 per 1000 tokens
229
+ completion_tokens_cost=0.0015 # $0.0015 per 1000 tokens
230
+ ),
231
+ OpenAIModels.GPT_3_5_TURBO_0125: ModelPricing(
232
+ prompt_tokens_cost=0.0005, # $0.0005 per 1000 tokens
233
+ completion_tokens_cost=0.0015 # $0.0015 per 1000 tokens
234
+ ),
235
+ OpenAIModels.GPT_3_5_TURBO_1106: ModelPricing(
236
+ prompt_tokens_cost=0.001, # $0.001 per 1000 tokens
237
+ completion_tokens_cost=0.002 # $0.002 per 1000 tokens
238
+ ),
239
+ OpenAIModels.GPT_3_5_TURBO_0613: ModelPricing(
240
+ prompt_tokens_cost=0.0015, # $0.0015 per 1000 tokens
241
+ completion_tokens_cost=0.002 # $0.002 per 1000 tokens
242
+ ),
243
+ OpenAIModels.GPT_3_5_0301: ModelPricing(
244
+ prompt_tokens_cost=0.0015, # $0.0015 per 1000 tokens
245
+ completion_tokens_cost=0.002 # $0.002 per 1000 tokens
246
+ ),
247
+ OpenAIModels.GPT_3_5_TURBO_INSTRUCT: ModelPricing(
248
+ prompt_tokens_cost=0.0015, # $0.0015 per 1000 tokens
249
+ completion_tokens_cost=0.002 # $0.002 per 1000 tokens
250
+ ),
251
+ OpenAIModels.GPT_3_5_TURBO_16K_0613: ModelPricing(
252
+ prompt_tokens_cost=0.003, # $0.003 per 1000 tokens
253
+ completion_tokens_cost=0.004 # $0.004 per 1000 tokens
254
+ ),
255
+ OpenAIModels.DAVINCI_002: ModelPricing(
256
+ prompt_tokens_cost=0.002, # $0.002 per 1000 tokens
257
+ completion_tokens_cost=0.002 # $0.002 per 1000 tokens
258
+ ),
259
+ OpenAIModels.BABBAGE_002: ModelPricing(
260
+ prompt_tokens_cost=0.0004, # $0.0004 per 1000 tokens
261
+ completion_tokens_cost=0.0004 # $0.0004 per 1000 tokens
262
+ ),
263
+
264
+ # Anthropic Models (pricing per 1000 tokens)
265
+ AnthropicModels.SONNET_4_5: ModelPricing(
266
+ prompt_tokens_cost=0.003, # $0.003 per 1000 tokens
267
+ completion_tokens_cost=0.015 # $0.015 per 1000 tokens
268
+ ),
269
+ AnthropicModels.HAIKU_4_5: ModelPricing(
270
+ prompt_tokens_cost=0.001, # $0.001 per 1000 tokens
271
+ completion_tokens_cost=0.005 # $0.005 per 1000 tokens
272
+ ),
273
+ AnthropicModels.OPUS_4_1: ModelPricing(
274
+ prompt_tokens_cost=0.015, # $0.015 per 1000 tokens
275
+ completion_tokens_cost=0.075 # $0.075 per 1000 tokens
276
+ ),
277
+ AnthropicModels.SONNET_4: ModelPricing(
278
+ prompt_tokens_cost=0.003, # $0.003 per 1000 tokens
279
+ completion_tokens_cost=0.015 # $0.015 per 1000 tokens
280
+ ),
281
+ AnthropicModels.OPUS_4: ModelPricing(
282
+ prompt_tokens_cost=0.015, # $0.015 per 1000 tokens
283
+ completion_tokens_cost=0.075 # $0.075 per 1000 tokens
284
+ ),
285
+ AnthropicModels.SONNET_3_7: ModelPricing(
286
+ prompt_tokens_cost=0.003, # $0.003 per 1000 tokens
287
+ completion_tokens_cost=0.015 # $0.015 per 1000 tokens
288
+ ),
289
+ AnthropicModels.HAIKU_3_5: ModelPricing(
290
+ prompt_tokens_cost=0.0008, # $0.0008 per 1000 tokens
291
+ completion_tokens_cost=0.004 # $0.004 per 1000 tokens
292
+ ),
293
+ AnthropicModels.OPUS_3: ModelPricing(
294
+ prompt_tokens_cost=0.015, # $0.015 per 1000 tokens
295
+ completion_tokens_cost=0.075 # $0.075 per 1000 tokens
296
+ ),
297
+ AnthropicModels.HAIKU_3: ModelPricing(
298
+ prompt_tokens_cost=0.00025, # $0.00025 per 1000 tokens
299
+ completion_tokens_cost=0.00125 # $0.00125 per 1000 tokens
300
+ ),
301
+
302
+ # Google DeepMind Models (pricing per 1000 tokens)
303
+ GoogleDeepMindModels.GEMINI_2_5_PRO: ModelPricing(
304
+ prompt_tokens_cost=0.00125, # $0.00125 per 1000 tokens
305
+ completion_tokens_cost=0.01 # $0.01 per 1000 tokens
306
+ ),
307
+ GoogleDeepMindModels.GEMINI_2_5_FLASH: ModelPricing(
308
+ prompt_tokens_cost=0.00015, # $0.00015 per 1000 tokens
309
+ completion_tokens_cost=0.0006 # $0.0006 per 1000 tokens
310
+ ),
311
+ GoogleDeepMindModels.GEMINI_2_5_FLASH_PREVIEW: ModelPricing(
312
+ prompt_tokens_cost=0.3, # $0.30 per 1000 tokens
313
+ completion_tokens_cost=2.5 # $2.50 per 1000 tokens
314
+ ),
315
+ GoogleDeepMindModels.GEMINI_2_5_FLASH_LITE: ModelPricing(
316
+ prompt_tokens_cost=0.0001, # $0.0001 per 1000 tokens
317
+ completion_tokens_cost=0.0004 # $0.0004 per 1000 tokens
318
+ ),
319
+ GoogleDeepMindModels.GEMINI_2_5_FLASH_LITE_PREVIEW: ModelPricing(
320
+ prompt_tokens_cost=0.0001, # $0.0001 per 1000 tokens
321
+ completion_tokens_cost=0.0004 # $0.0004 per 1000 tokens
322
+ ),
323
+ GoogleDeepMindModels.GEMINI_2_5_FLASH_NATIVE_AUDIO: ModelPricing(
324
+ prompt_tokens_cost=0.0005, # $0.0005 per 1000 tokens
325
+ completion_tokens_cost=0.002 # $0.002 per 1000 tokens
326
+ ),
327
+ GoogleDeepMindModels.GEMINI_2_5_FLASH_IMAGE: ModelPricing(
328
+ prompt_tokens_cost=0.0003, # $0.0003 per 1000 tokens
329
+ completion_tokens_cost=0.03 # $0.03 per 1000 tokens
330
+ ),
331
+ GoogleDeepMindModels.GEMINI_2_5_FLASH_PREVIEW_TTS: ModelPricing(
332
+ prompt_tokens_cost=0.0005, # $0.0005 per 1000 tokens
333
+ completion_tokens_cost=0.01 # $0.01 per 1000 tokens
334
+ ),
335
+ GoogleDeepMindModels.GEMINI_2_5_PRO_PREVIEW_TTS: ModelPricing(
336
+ prompt_tokens_cost=0.001, # $0.001 per 1000 tokens
337
+ completion_tokens_cost=0.02 # $0.02 per 1000 tokens
338
+ ),
339
+ GoogleDeepMindModels.GEMINI_2_5_COMPUTER_USE_PREVIEW: ModelPricing(
340
+ prompt_tokens_cost=0.00125, # $0.00125 per 1000 tokens
341
+ completion_tokens_cost=0.01 # $0.01 per 1000 tokens
342
+ ),
343
+
344
+ # Meta Models (pricing per 1000 tokens)
345
+ MetaModels.LLAMA_4_MAVERICK: ModelPricing(
346
+ prompt_tokens_cost=0.00027, # $0.00027 per 1000 tokens
347
+ completion_tokens_cost=0.00085 # $0.00085 per 1000 tokens
348
+ ),
349
+ MetaModels.LLAMA_4_SCOUT: ModelPricing(
350
+ prompt_tokens_cost=0.00018, # $0.00018 per 1000 tokens
351
+ completion_tokens_cost=0.00059 # $0.00059 per 1000 tokens
352
+ ),
353
+ MetaModels.LLAMA_3_3_70B_INSTRUCT_TURBO: ModelPricing(
354
+ prompt_tokens_cost=0.00088, # $0.00088 per 1000 tokens
355
+ completion_tokens_cost=0.00088 # $0.00088 per 1000 tokens
356
+ ),
357
+ MetaModels.LLAMA_3_2_3B_INSTRUCT_TURBO: ModelPricing(
358
+ prompt_tokens_cost=0.00006, # $0.00006 per 1000 tokens
359
+ completion_tokens_cost=0.00006 # $0.00006 per 1000 tokens
360
+ ),
361
+ MetaModels.LLAMA_3_1_405B_INSTRUCT_TURBO: ModelPricing(
362
+ prompt_tokens_cost=0.0035, # $0.0035 per 1000 tokens
363
+ completion_tokens_cost=0.0035 # $0.0035 per 1000 tokens
364
+ ),
365
+ MetaModels.LLAMA_3_1_70B_INSTRUCT_TURBO: ModelPricing(
366
+ prompt_tokens_cost=0.00088, # $0.00088 per 1000 tokens
367
+ completion_tokens_cost=0.00088 # $0.00088 per 1000 tokens
368
+ ),
369
+ MetaModels.LLAMA_3_1_8B_INSTRUCT_TURBO: ModelPricing(
370
+ prompt_tokens_cost=0.00018, # $0.00018 per 1000 tokens
371
+ completion_tokens_cost=0.00018 # $0.00018 per 1000 tokens
372
+ ),
373
+ MetaModels.LLAMA_3_70B_INSTRUCT_TURBO: ModelPricing(
374
+ prompt_tokens_cost=0.00088, # $0.00088 per 1000 tokens
375
+ completion_tokens_cost=0.00088 # $0.00088 per 1000 tokens
376
+ ),
377
+ MetaModels.LLAMA_3_70B_INSTRUCT_REFERENCE: ModelPricing(
378
+ prompt_tokens_cost=0.00088, # $0.00088 per 1000 tokens
379
+ completion_tokens_cost=0.00088 # $0.00088 per 1000 tokens
380
+ ),
381
+ MetaModels.LLAMA_3_8B_INSTRUCT_LITE: ModelPricing(
382
+ prompt_tokens_cost=0.0001, # $0.0001 per 1000 tokens
383
+ completion_tokens_cost=0.0001 # $0.0001 per 1000 tokens
384
+ ),
385
+ MetaModels.LLAMA_2: ModelPricing(
386
+ prompt_tokens_cost=0.0009, # $0.0009 per 1000 tokens
387
+ completion_tokens_cost=0.0009 # $0.0009 per 1000 tokens
388
+ ),
389
+ MetaModels.LLAMA_GUARD_4_12B: ModelPricing(
390
+ prompt_tokens_cost=0.0002, # $0.0002 per 1000 tokens
391
+ completion_tokens_cost=0.0002 # $0.0002 per 1000 tokens
392
+ ),
393
+ MetaModels.LLAMA_GUARD_3_11B_VISION_TURBO: ModelPricing(
394
+ prompt_tokens_cost=0.00018, # $0.00018 per 1000 tokens
395
+ completion_tokens_cost=0.00018 # $0.00018 per 1000 tokens
396
+ ),
397
+ MetaModels.LLAMA_GUARD_3_8B: ModelPricing(
398
+ prompt_tokens_cost=0.0002, # $0.0002 per 1000 tokens
399
+ completion_tokens_cost=0.0002 # $0.0002 per 1000 tokens
400
+ ),
401
+ MetaModels.LLAMA_GUARD_2_8B: ModelPricing(
402
+ prompt_tokens_cost=0.0002, # $0.0002 per 1000 tokens
403
+ completion_tokens_cost=0.0002 # $0.0002 per 1000 tokens
404
+ ),
405
+ MetaModels.SALESFORCE_LLAMA_RANK_V1_8B: ModelPricing(
406
+ prompt_tokens_cost=0.0001, # $0.0001 per 1000 tokens
407
+ completion_tokens_cost=0.0001 # $0.0001 per 1000 tokens
408
+ ),
409
+
410
+ # AWS Models (pricing per 1000 tokens)
411
+ AWSModels.AMAZON_NOVA_MICRO: ModelPricing(
412
+ prompt_tokens_cost=0.035, # $0.035 per 1000 tokens
413
+ completion_tokens_cost=0.14 # $0.14 per 1000 tokens
414
+ ),
415
+ AWSModels.AMAZON_NOVA_LITE: ModelPricing(
416
+ prompt_tokens_cost=0.06, # $0.06 per 1000 tokens
417
+ completion_tokens_cost=0.24 # $0.24 per 1000 tokens
418
+ ),
419
+ AWSModels.AMAZON_NOVA_PRO: ModelPricing(
420
+ prompt_tokens_cost=0.8, # $0.8 per 1000 tokens
421
+ completion_tokens_cost=3.2 # $3.2 per 1000 tokens
422
+ ),
423
+
424
+ # Mistral AI Models (pricing per 1000 tokens)
425
+ MistralAIModels.MISTRAL_7B_INSTRUCT: ModelPricing(
426
+ prompt_tokens_cost=0.028, # $0.028 per 1000 tokens
427
+ completion_tokens_cost=0.054 # $0.054 per 1000 tokens
428
+ ),
429
+ MistralAIModels.MISTRAL_LARGE: ModelPricing(
430
+ prompt_tokens_cost=2.0, # $2.00 per 1000 tokens
431
+ completion_tokens_cost=6.0 # $6.00 per 1000 tokens
432
+ ),
433
+ MistralAIModels.MISTRAL_SMALL: ModelPricing(
434
+ prompt_tokens_cost=0.2, # $0.20 per 1000 tokens
435
+ completion_tokens_cost=0.6 # $0.60 per 1000 tokens
436
+ ),
437
+ MistralAIModels.MISTRAL_MEDIUM: ModelPricing(
438
+ prompt_tokens_cost=0.4, # $0.40 per 1000 tokens
439
+ completion_tokens_cost=2.0 # $2.00 per 1000 tokens
440
+ ),
441
+
442
+ # Cohere Models (pricing per 1000 tokens)
443
+ CohereModels.COMMAND_R7B: ModelPricing(
444
+ prompt_tokens_cost=0.0000375, # $0.0000375 per 1000 tokens
445
+ completion_tokens_cost=0.00015 # $0.00015 per 1000 tokens
446
+ ),
447
+ CohereModels.COMMAND_R: ModelPricing(
448
+ prompt_tokens_cost=0.00015, # $0.00015 per 1000 tokens
449
+ completion_tokens_cost=0.0006 # $0.0006 per 1000 tokens
450
+ ),
451
+ CohereModels.COMMAND_R_PLUS: ModelPricing(
452
+ prompt_tokens_cost=0.00250, # $0.00250 per 1000 tokens
453
+ completion_tokens_cost=0.01 # $0.01 per 1000 tokens
454
+ ),
455
+ CohereModels.COMMAND_A: ModelPricing(
456
+ prompt_tokens_cost=0.001, # $0.001 per 1000 tokens
457
+ completion_tokens_cost=0.002 # $0.002 per 1000 tokens
458
+ ),
459
+ CohereModels.AYA_EXPANSE_8B_32B: ModelPricing(
460
+ prompt_tokens_cost=0.00050, # $0.00050 per 1000 tokens
461
+ completion_tokens_cost=0.00150 # $0.00150 per 1000 tokens
462
+ ),
463
+
464
+ # DeepSeek Models (pricing per 1000 tokens)
465
+ DeepSeekModels.DEEPSEEK_CHAT: ModelPricing(
466
+ prompt_tokens_cost=0.00007, # $0.00007 per 1000 tokens
467
+ completion_tokens_cost=0.00027 # $0.00027 per 1000 tokens
468
+ ),
469
+ DeepSeekModels.DEEPSEEK_REASONER: ModelPricing(
470
+ prompt_tokens_cost=0.00014, # $0.00014 per 1000 tokens
471
+ completion_tokens_cost=0.00219 # $0.00219 per 1000 tokens
472
+ ),
473
+ DeepSeekModels.DEEPSEEK_R1_GLOBAL: ModelPricing(
474
+ prompt_tokens_cost=0.00135, # $0.00135 per 1000 tokens
475
+ completion_tokens_cost=0.0054 # $0.0054 per 1000 tokens
476
+ ),
477
+ DeepSeekModels.DEEPSEEK_R1_DATAZONE: ModelPricing(
478
+ prompt_tokens_cost=0.001485, # $0.001485 per 1000 tokens
479
+ completion_tokens_cost=0.00594 # $0.00594 per 1000 tokens
480
+ ),
481
+ DeepSeekModels.DEEPSEEK_V3_2_EXP: ModelPricing(
482
+ prompt_tokens_cost=0.000028, # $0.000028 per 1000 tokens
483
+ completion_tokens_cost=0.00042 # $0.00042 per 1000 tokens
484
+ ),
485
+ }