mirror of
https://github.com/foss42/apidash.git
synced 2025-08-26 14:24:03 +08:00
28 lines
927 B
Dart
28 lines
927 B
Dart
import 'dart:convert';
|
|
import 'package:ollama_dart/ollama_dart.dart';
|
|
import '../features/explain.dart';
|
|
import 'package:apidash/models/request_model.dart';
|
|
|
|
class DashBotService {
|
|
final OllamaClient _client;
|
|
late final ExplainFeature _explainFeature;
|
|
|
|
DashBotService() : _client = OllamaClient(baseUrl: 'http://127.0.0.1:11434/api') {
|
|
_explainFeature = ExplainFeature(this);
|
|
}
|
|
|
|
Future<String> generateResponse(String prompt) async {
|
|
final response = await _client.generateCompletion(
|
|
request: GenerateCompletionRequest(model: 'llama3.2:3b', prompt: prompt),
|
|
);
|
|
return response.response.toString();
|
|
}
|
|
|
|
Future<String> handleRequest(String input, RequestModel? requestModel, dynamic responseModel) async {
|
|
if (input == "Explain API") {
|
|
return _explainFeature.explainLatestApi(requestModel: requestModel, responseModel: responseModel);
|
|
}
|
|
return generateResponse(input);
|
|
}
|
|
}
|