Update ollama_model.dart

Mobile-Artificial-Intelligence · May 6, 2024 · e99786d · e99786d
1 parent 2479541
commit e99786d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lib/classes/ollama_model.dart b/lib/classes/ollama_model.dart
@@ -39,6 +39,7 @@ class OllamaModel extends LargeLanguageModel {
  super.listener, 
  super.name,
  super.uri,
+ super.token,
  super.useDefault,
  super.penalizeNewline,
  super.seed,
@@ -198,6 +199,7 @@ class OllamaModel extends LargeLanguageModel {
  } else {
  chat = ChatOllama(
  baseUrl: '$uri/api',
+ headers: { 'Authorization': 'Bearer: $token' },
  defaultOptions: ChatOllamaOptions(
  model: name,
  numKeep: nKeep,