@aige/core
Version:
AI Game Engine
81 lines (80 loc) • 3.06 kB
JavaScript
;
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const openai_1 = __importDefault(require("openai"));
class Client {
constructor(options = {}, game) {
this.tokens = 0;
const defaultClientOptions = {
clientOptions: {
apiKey: process.env['OPENAI_API_KEY'],
timeout: 30000,
maxRetries: 0
}
};
options.clientOptions = options.clientOptions || {};
this.client = new openai_1.default({ ...defaultClientOptions, ...options.clientOptions });
this.model = options.model || 'gpt-3.5-turbo-1106';
this.imageModel = options.imageModel || 'dall-e-2';
this.contextWindow = options.contextWindow || 16385;
this.game = game;
}
async call(tool, parameters) {
const tools = [];
if (Array.isArray(tool)) {
for (const t of tool)
tools.push(t.schema);
}
else {
tools.push(tool.schema);
}
if (parameters.debug)
console.dir(tools, { depth: null });
const response = await this.client.chat.completions.create({
model: this.model,
messages: parameters.messages || [],
tool_choice: parameters.tool_choice,
tools
});
const message = response.choices[0]?.message;
const usage = response.usage?.total_tokens;
if (!message)
throw new Error('No message found');
this.game.history.push(message);
if (usage && !isNaN(usage))
this.tokens += usage;
if (!message?.tool_calls) {
const directMessage = message?.content;
if (!directMessage)
throw new Error('No direct message found');
return { scene_emoji: '🤖', scene: directMessage };
}
else {
for (const tool_call of message.tool_calls) {
const called = Array.isArray(tool) ? tool.find(t => t.schema.function.name === tool_call.function.name) : tool;
if (called) {
try {
const parsedArgs = JSON.parse(tool_call.function.arguments);
return called.execute(parsedArgs);
}
catch (err) {
throw new Error(`Failed to parse ${tool_call.function.name} arguments: ${err}`);
}
}
}
}
}
async image(prompt, options = {}) {
const response = await this.client.images.generate({
size: options.size || '512x512',
style: options.style || 'vivid',
model: options.model || this.imageModel,
response_format: options.response_format || 'url',
prompt
});
return response.data[0];
}
}
exports.default = Client;