55 lines
1.5 KiB
Dart
55 lines
1.5 KiB
Dart
|
import 'package:halkyon_llm/halkyon_llm.dart';
|
||
|
|
import 'package:test/test.dart';
|
||
|
|
|
||
|
|
LlmEndpoint _at(String baseUrl) => LlmEndpoint(
|
||
|
|
protocol: LlmProtocol.openai,
|
||
|
|
model: 'm',
|
||
|
|
apiKey: '',
|
||
|
|
baseUrl: baseUrl,
|
||
|
|
);
|
||
|
|
|
||
|
|
void main() {
|
||
|
|
test("takes the protocol's own address where none is given", () {
|
||
|
|
final endpoint = LlmEndpoint(
|
||
|
|
protocol: LlmProtocol.anthropic,
|
||
|
|
model: 'm',
|
||
|
|
apiKey: 'k',
|
||
|
|
);
|
||
|
|
|
||
|
|
expect(endpoint.host, 'api.anthropic.com');
|
||
|
|
expect(endpoint.onThisMachine, isFalse);
|
||
|
|
});
|
||
|
|
|
||
|
|
test('refuses http beyond the same machine or network', () {
|
||
|
|
expect(() => _at('http://example.com/v1'), throwsFormatException);
|
||
|
|
expect(_at('http://192.168.1.20:11434/v1').host, '192.168.1.20');
|
||
|
|
});
|
||
|
|
|
||
|
|
test('knows a model on this machine: a loopback address or localhost '
|
||
|
|
'(ADR 0021, D9)', () {
|
||
|
|
for (final url in [
|
||
|
|
'http://127.0.0.1:11434/v1',
|
||
|
|
'http://127.8.0.1/v1',
|
||
|
|
'http://[::1]:8080/v1',
|
||
|
|
'http://localhost:11434/v1',
|
||
|
|
'http://ollama.localhost/v1',
|
||
|
|
'https://LOCALHOST/v1',
|
||
|
|
]) {
|
||
|
|
expect(_at(url).onThisMachine, isTrue, reason: url);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
|
||
|
|
test('takes nothing else for this machine: the network may be shared', () {
|
||
|
|
for (final url in [
|
||
|
|
'http://192.168.1.20:11434/v1',
|
||
|
|
'http://10.0.0.2/v1',
|
||
|
|
'http://[fd00::2]/v1',
|
||
|
|
'http://ollama/v1',
|
||
|
|
'http://gpu.local/v1',
|
||
|
|
'https://localhost.example.com/v1',
|
||
|
|
'https://api.openai.com/v1',
|
||
|
|
]) {
|
||
|
|
expect(_at(url).onThisMachine, isFalse, reason: url);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|