File size: 311 Bytes
1c57e18
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
from google import genai

class GeminiDriver:
    def __init__(self):
        self.client = genai.Client()

    def generate_text(self, prompt: str) -> str:
        response = self.client.models.generate_content(
            model="gemini-3-flash-preview", contents=prompt
        )
        return response.text