feat: hostname config to support other api services

src/commands/aicommits.ts

@@ -63,6 +63,7 @@ export default async (
 			config['max-length'],
 			config.type,
 			config.timeout,
+			config.hostname,
 			config.proxy,
 		);
 	} finally {

src/commands/prepare-commit-msg-hook.ts

@@ -48,6 +48,7 @@ export default () => (async () => {
 			config['max-length'],
 			config.type,
 			config.timeout,
+			config.hostname,
 			config.proxy,
 		);
 	} finally {

src/utils/config.ts

@@ -104,6 +104,16 @@ const configParsers = {
 
 		return parsed;
 	},
+	hostname(hostname?: string) {
+		if (!hostname) {
+			return 'api.openai.com';
+		}
+
+		// eslint-disable-next-line regexp/no-unused-capturing-group
+		parseAssert('hostname', /^([a-zA-Z\d-]+\.)*[a-zA-Z\d-]+\.[a-zA-Z]+$/.test(hostname), 'Must be an hostname');
+
+		return hostname;
+	},
 } as const;
 
 type ConfigKeys = keyof typeof configParsers;

src/utils/openai.ts

@@ -67,10 +67,11 @@ const createChatCompletion = async (
 	apiKey: string,
 	json: CreateChatCompletionRequest,
 	timeout: number,
+	hostname: string,
 	proxy?: string,
 ) => {
 	const { response, data } = await httpsPost(
-		'api.openai.com',
+		hostname,
 		'/v1/chat/completions',
 		{
 			Authorization: `Bearer ${apiKey}`,
@@ -131,6 +132,7 @@ export const generateCommitMessage = async (
 	maxLength: number,
 	type: CommitType,
 	timeout: number,
+	hostname: string,
 	proxy?: string,
 ) => {
 	try {
@@ -157,6 +159,7 @@ export const generateCommitMessage = async (
 				n: completions,
 			},
 			timeout,
+			hostname,
 			proxy,
 		);
 

tests/specs/config.ts

@@ -101,6 +101,30 @@ export default testSuite(({ describe }) => {
 			});
 		});
 
+		await describe('hostname', ({ test }) => {
+			test('must be an hostname', async () => {
+				const { stderr } = await aicommits(['config', 'set', 'hostname=https://api.openai.com'], {
+					reject: false,
+				});
+
+				expect(stderr).toMatch('Must be an hostname');
+			});
+
+			test('updates config', async () => {
+				const defaultConfig = await aicommits(['config', 'get', 'hostname']);
+				expect(defaultConfig.stdout).toBe('hostname=api.openai.com');
+
+				const hostname = 'hostname=api.chatanywhere.com.cn';
+				await aicommits(['config', 'set', hostname]);
+
+				const configFile = await fs.readFile(configPath, 'utf8');
+				expect(configFile).toMatch(hostname);
+
+				const get = await aicommits(['config', 'get', 'hostname']);
+				expect(get.stdout).toBe(hostname);
+			});
+		});
+
 		await test('set config file', async () => {
 			await aicommits(['config', 'set', openAiToken]);
 

tests/specs/openai/conventional-commits.ts

@@ -137,7 +137,7 @@ export default testSuite(({ describe }) => {
 				'max-length': 50,
 				...configOverrides,
 			} as ValidConfig;
-			const commitMessages = await generateCommitMessage(OPENAI_KEY!, 'gpt-3.5-turbo', config.locale, gitDiff, config.generate, config['max-length'], config.type, 7000);
+			const commitMessages = await generateCommitMessage(OPENAI_KEY!, 'gpt-3.5-turbo', config.locale, gitDiff, config.generate, config['max-length'], config.type, 7000, config.hostname);
 
 			return commitMessages[0];
 		}