mirror of
https://github.com/di-sukharev/opencommit.git
synced 2026-04-20 03:02:51 -04:00
32f3e176f0
* 2.0.18 * patch * 2.0.19 * style(.prettierrc): reorder properties to follow alphabetical order and improve readability * feat(generateCommitMessageFromGitDiff.ts): changing logic of MAX_REQ_TOKENS fix(api.ts): add missing import for GenerateCommitMessageErrorEnum The token count validation is added to prevent the request from exceeding the default model token limit. The validation is done by counting the tokens in each message and adding 4 to each count to account for the additional tokens added by the API. If the total token count exceeds the limit, an error is thrown. The missing import for GenerateCommitMessageErrorEnum is also added. feat: add support for splitting long line-diffs into smaller pieces This change adds support for splitting long line-diffs into smaller pieces to avoid exceeding the maximum commit message length. The `splitDiff` function splits a single line into multiple lines if it exceeds the maximum length. It also splits the diff into smaller pieces if adding the next line would exceed the maximum length. This change improves the readability of commit messages and makes them more consistent. refactor: improve code readability by adding whitespace and reformatting code This commit improves the readability of the code by adding whitespace and reformatting the code. The changes do not affect the functionality of the code. Additionally, a new function `delay` has been added to the file. --------- Co-authored-by: di-sukharev <dim.sukharev@gmail.com>
112 lines
2.9 KiB
TypeScript
112 lines
2.9 KiB
TypeScript
import { intro, outro } from '@clack/prompts';
|
|
import axios from 'axios';
|
|
import chalk from 'chalk';
|
|
import {
|
|
ChatCompletionRequestMessage,
|
|
Configuration as OpenAiApiConfiguration,
|
|
OpenAIApi
|
|
} from 'openai';
|
|
|
|
import {CONFIG_MODES, DEFAULT_MODEL_TOKEN_LIMIT, getConfig} from './commands/config';
|
|
import {tokenCount} from './utils/tokenCount';
|
|
import {GenerateCommitMessageErrorEnum} from './generateCommitMessageFromGitDiff';
|
|
|
|
const config = getConfig();
|
|
|
|
let maxTokens = config?.OCO_OPENAI_MAX_TOKENS;
|
|
let basePath = config?.OCO_OPENAI_BASE_PATH;
|
|
let apiKey = config?.OCO_OPENAI_API_KEY;
|
|
|
|
const [command, mode] = process.argv.slice(2);
|
|
|
|
if (!apiKey && command !== 'config' && mode !== CONFIG_MODES.set) {
|
|
intro('opencommit');
|
|
|
|
outro(
|
|
'OCO_OPENAI_API_KEY is not set, please run `oco config set OCO_OPENAI_API_KEY=<your token>. Make sure you add payment details, so API works.`'
|
|
);
|
|
outro(
|
|
'For help look into README https://github.com/di-sukharev/opencommit#setup'
|
|
);
|
|
|
|
process.exit(1);
|
|
}
|
|
|
|
const MODEL = config?.OCO_MODEL || 'gpt-3.5-turbo';
|
|
|
|
class OpenAi {
|
|
private openAiApiConfiguration = new OpenAiApiConfiguration({
|
|
apiKey: apiKey
|
|
});
|
|
private openAI!: OpenAIApi;
|
|
|
|
constructor() {
|
|
if (basePath) {
|
|
this.openAiApiConfiguration.basePath = basePath;
|
|
}
|
|
this.openAI = new OpenAIApi(this.openAiApiConfiguration);
|
|
}
|
|
|
|
public generateCommitMessage = async (
|
|
messages: Array<ChatCompletionRequestMessage>
|
|
): Promise<string | undefined> => {
|
|
const params = {
|
|
model: MODEL,
|
|
messages,
|
|
temperature: 0,
|
|
top_p: 0.1,
|
|
max_tokens: maxTokens || 500
|
|
};
|
|
try {
|
|
const REQUEST_TOKENS = messages.map(
|
|
(msg) => tokenCount(msg.content) + 4
|
|
).reduce((a, b) => a + b, 0);
|
|
|
|
if (REQUEST_TOKENS > (DEFAULT_MODEL_TOKEN_LIMIT - maxTokens)) {
|
|
throw new Error(GenerateCommitMessageErrorEnum.tooMuchTokens);
|
|
}
|
|
|
|
const { data } = await this.openAI.createChatCompletion(params);
|
|
|
|
const message = data.choices[0].message;
|
|
|
|
return message?.content;
|
|
} catch (error) {
|
|
outro(`${chalk.red('✖')} ${JSON.stringify(params)}`);
|
|
|
|
const err = error as Error;
|
|
outro(`${chalk.red('✖')} ${err?.message || err}`);
|
|
|
|
if (
|
|
axios.isAxiosError<{ error?: { message: string } }>(error) &&
|
|
error.response?.status === 401
|
|
) {
|
|
const openAiError = error.response.data.error;
|
|
|
|
if (openAiError?.message) outro(openAiError.message);
|
|
outro(
|
|
'For help look into README https://github.com/di-sukharev/opencommit#setup'
|
|
);
|
|
}
|
|
|
|
throw err;
|
|
}
|
|
};
|
|
}
|
|
|
|
export const getOpenCommitLatestVersion = async (): Promise<
|
|
string | undefined
|
|
> => {
|
|
try {
|
|
const { data } = await axios.get(
|
|
'https://unpkg.com/opencommit/package.json'
|
|
);
|
|
return data.version;
|
|
} catch (_) {
|
|
outro('Error while getting the latest version of opencommit');
|
|
return undefined;
|
|
}
|
|
};
|
|
|
|
export const api = new OpenAi();
|