Skip to content

Commit 3247c61

Browse files
committed
fixed image calling
1 parent d7eec91 commit 3247c61

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

lib/providers/gemini_chat_provider.dart

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ class GeminiChatProvider extends StateNotifier<GeminiChatState> {
7979

8080
getPrompt(String prompt, XFile? result) async {
8181
var model = GenerativeModel(
82-
model: 'gemini-pro',
82+
model: result == null ? 'gemini-pro' : 'gemini-pro-vision',
8383
apiKey: ref.read(appSettingsProvider).geminiApiKey);
8484
var filteredMessage = state.messages.whereType<types.TextMessage>();
8585
// return;
@@ -118,12 +118,14 @@ class GeminiChatProvider extends StateNotifier<GeminiChatState> {
118118
try {
119119
if (prompt.isEmpty && result == null) return;
120120
if (result != null) {}
121-
var res = await chat.sendMessage(result == null
122-
? Content.text(prompt)
123-
: Content.multi([
124-
TextPart(prompt),
125-
...[DataPart("image/jpeg", await result.readAsBytes())]
126-
]));
121+
var res = result == null
122+
? await chat.sendMessage(Content.text(prompt))
123+
: await model.generateContent([
124+
Content.multi([
125+
TextPart(prompt),
126+
...[DataPart("image/jpeg", await result.readAsBytes())]
127+
])
128+
]);
127129
log(res.text!);
128130
addMessage(types.TextMessage(
129131
author: state.users[1],

0 commit comments

Comments
 (0)