`
class TextGenerationModel {
final GenerativeModel _generativeModel;
TextGenerationModel(String apiKey, {int timeout = 10})
: _generativeModel = GenerativeModel(
apiKey: apiKey, model: 'gemini-pro',
);
Future<String> generateText(String prompt) async {
try {
// final content = [Content.text(prompt)];
// final response = await _generativeModel.generateContent(content);
// return response.generatedText;
final response = await _generativeModel.generateText(prompt: prompt);
return response.generatedText;
} on GenerativeAIException catch (error) {
// Handle specific Generative AI errors
return Future.error(error.message);
} catch (error) {
// Handle general errors
return Future.error('An error occurred');
}
}
}`
I'm getting an error in `final response = await _generativeModel.generateText(prompt: prompt);`,
i tried different ways but could not get a solution. Please help in this. Thanks in advance.