mateus-ai-agent-sdk
Version:
A simple SDK to interact with OpenAI models
52 lines (51 loc) • 2.27 kB
JavaScript
;
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
var _a;
Object.defineProperty(exports, "__esModule", { value: true });
const axios_1 = __importDefault(require("axios"));
const dotenv_1 = __importDefault(require("dotenv"));
dotenv_1.default.config();
const API_KEY = (_a = process.env.OPENAI_API_KEY) !== null && _a !== void 0 ? _a : '';
if (!API_KEY) {
throw new Error('Missing OpenAI API Key. Set OPENAI_API_KEY in .env');
}
class OpenAIService {
constructor(apiKey = API_KEY, baseURL = 'https://api.openai.com/v1') {
this.api = axios_1.default.create({
baseURL,
headers: {
'Authorization': `Bearer ${apiKey}`,
'Content-Type': 'application/json'
}
});
}
generateText(prompt_1) {
return __awaiter(this, arguments, void 0, function* (prompt, model = 'gpt-3.5-turbo', maxTokens = 200, temperature = 0.7) {
try {
const response = yield this.api.post('/chat/completions', {
model,
messages: [{ role: 'user', content: prompt }],
max_tokens: maxTokens,
temperature: temperature,
});
return response.data.choices[0].message.content.trim();
}
catch (error) {
console.error('Error generating text:', error);
throw error;
}
});
}
}
exports.default = OpenAIService;