mirror of
https://github.com/kevinthedang/discord-ollama.git
synced 2026-09-23 17:08:58 -04:00
Eric Curtin
0beaed85aa
llmman (https://github.com/llmmanorg/llmman) serves the Ollama API on port 17434 and works with the bot as-is, so document it in setup-local and link it from the README. Since the backend is no longer Ollama-only, rename the env vars to LLM_ENDPOINT / LLM_PORT as requested in review. OLLAMA_IP / OLLAMA_PORT are kept as deprecated aliases in keys.ts and docker-compose.yml so existing deployments keep working.
75 lines
2.6 KiB
TypeScript
75 lines
2.6 KiB
TypeScript
import { describe, expect, it } from 'vitest'
|
|
import { getEnvVar } from '../src/utils/index.js'
|
|
|
|
/**
|
|
* getEnvVar test suite, tests the getEnvVar function
|
|
*
|
|
* @param name name of the test suite
|
|
* @param fn function holding tests to run
|
|
*/
|
|
describe('Environment Setup', () => {
|
|
// dummy set of keys
|
|
const keys = {
|
|
clientToken: 'CLIENT_TOKEN',
|
|
}
|
|
|
|
// set keys in environment
|
|
process.env['clientToken'] = keys.clientToken
|
|
|
|
// test for non-empty string
|
|
it('returns a non-empty string', () => {
|
|
expect(getEnvVar('CLIENT_TOKEN')).not.toBe('')
|
|
})
|
|
|
|
// test for string type
|
|
it('returns a string', () => {
|
|
expect(typeof getEnvVar('CLIENT_TOKEN')).toBe('string')
|
|
})
|
|
|
|
// test for distinct key
|
|
it('returns a distinct key', () => {
|
|
expect(getEnvVar('CLIENT_TOKEN')).toEqual(process.env[keys.clientToken])
|
|
})
|
|
|
|
// test for fallback case
|
|
it('returns a fallback', () => {
|
|
expect(getEnvVar('NON_EXISTENT_KEY', 'fallback')).toBe('fallback')
|
|
})
|
|
|
|
// test that all keys are consistently found
|
|
it('returns all keys found', () => {
|
|
for (const key in keys) {
|
|
expect(getEnvVar(key)).toEqual(keys[key])
|
|
}
|
|
})
|
|
|
|
// test that an error is thrown if key is not found
|
|
it('throws an error if key is not found', () => {
|
|
expect(() => getEnvVar('NON_EXISTENT_KEY')).toThrowError()
|
|
})
|
|
|
|
// test that *_ENDPOINT keys are validated as IPv4 addresses like *_IP keys
|
|
it('validates *_ENDPOINT keys as IPv4', () => {
|
|
expect(getEnvVar('UNSET_TEST_ENDPOINT', '127.0.0.1')).toBe('127.0.0.1')
|
|
expect(() => getEnvVar('UNSET_TEST_ENDPOINT', 'not-an-ip')).toThrowError()
|
|
})
|
|
|
|
// test the deprecated alias lookup used in src/keys.ts
|
|
it('falls back to the deprecated OLLAMA_IP / OLLAMA_PORT aliases', () => {
|
|
const saved = { LLM_ENDPOINT: process.env.LLM_ENDPOINT, LLM_PORT: process.env.LLM_PORT }
|
|
delete process.env.LLM_ENDPOINT
|
|
delete process.env.LLM_PORT
|
|
process.env.OLLAMA_IP = '10.0.0.5'
|
|
process.env.OLLAMA_PORT = '17434'
|
|
try {
|
|
expect(getEnvVar('LLM_ENDPOINT', process.env.OLLAMA_IP ?? '127.0.0.1')).toBe('10.0.0.5')
|
|
expect(getEnvVar('LLM_PORT', process.env.OLLAMA_PORT ?? '11434')).toBe('17434')
|
|
} finally {
|
|
delete process.env.OLLAMA_IP
|
|
delete process.env.OLLAMA_PORT
|
|
if (saved.LLM_ENDPOINT !== undefined) process.env.LLM_ENDPOINT = saved.LLM_ENDPOINT
|
|
if (saved.LLM_PORT !== undefined) process.env.LLM_PORT = saved.LLM_PORT
|
|
}
|
|
})
|
|
})
|