Add gemma models and token statistics
Browse files- routers/powerpoint.py +5 -1
routers/powerpoint.py
CHANGED
|
@@ -42,6 +42,8 @@ class MultimodalModel(str, Enum):
|
|
| 42 |
gemini_20_flash = "gemini-2.0-flash"
|
| 43 |
gemini_20_flash_lite = "gemini-2.0-flash-lite"
|
| 44 |
gemini_25_pro = "gemini-2.5-pro"
|
|
|
|
|
|
|
| 45 |
|
| 46 |
class TemplateType(str, Enum):
|
| 47 |
single_graph = "single_graph"
|
|
@@ -506,7 +508,7 @@ async def image_to_pptx(
|
|
| 506 |
"""
|
| 507 |
try:
|
| 508 |
# Get available model
|
| 509 |
-
model = get_available_model()
|
| 510 |
|
| 511 |
# Get template generator
|
| 512 |
template_generator = TEMPLATE_GENERATORS.get(template)
|
|
@@ -531,6 +533,8 @@ async def image_to_pptx(
|
|
| 531 |
'response_schema': template_generator.get_response_schema(),
|
| 532 |
}
|
| 533 |
)
|
|
|
|
|
|
|
| 534 |
|
| 535 |
# Parse the response
|
| 536 |
presentation_data = template_generator.get_response_schema().model_validate(json.loads(response.text))
|
|
|
|
| 42 |
gemini_20_flash = "gemini-2.0-flash"
|
| 43 |
gemini_20_flash_lite = "gemini-2.0-flash-lite"
|
| 44 |
gemini_25_pro = "gemini-2.5-pro"
|
| 45 |
+
gemma_3_4b = "gemma-3-4b-it"
|
| 46 |
+
gemma_3_12b = "gemma-3-12b-it"
|
| 47 |
|
| 48 |
class TemplateType(str, Enum):
|
| 49 |
single_graph = "single_graph"
|
|
|
|
| 508 |
"""
|
| 509 |
try:
|
| 510 |
# Get available model
|
| 511 |
+
#model = get_available_model()
|
| 512 |
|
| 513 |
# Get template generator
|
| 514 |
template_generator = TEMPLATE_GENERATORS.get(template)
|
|
|
|
| 533 |
'response_schema': template_generator.get_response_schema(),
|
| 534 |
}
|
| 535 |
)
|
| 536 |
+
# Display input/output tokens
|
| 537 |
+
print(response.usage_metadata)
|
| 538 |
|
| 539 |
# Parse the response
|
| 540 |
presentation_data = template_generator.get_response_schema().model_validate(json.loads(response.text))
|