File size: 476 Bytes
fa3ab28 8127077 dfd6d8a 8127077 fa3ab28 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | from google import genai
from smolagents import LiteLLMModel
def gemini_model_liteLLM(
model: str,
response_format=None,
) -> LiteLLMModel:
"""model for example, gemini-2.0-flash"""
return LiteLLMModel(
model_id=f"gemini/{model}",
response_format=response_format,
temperature=0.0,
seed=42,
thinking={"type": "enabled", "budget_tokens": 1024},
)
def gemini_client():
client = genai.Client()
return client
|