feat: solution for 'Агент с RAG-памятью'

src/utils.js

@@ -0,0 +1,33 @@
+const dotenv = require('dotenv');
+dotenv.config();
+
+const { Configuration, OpenAIApi } = require('openai');
+
+const config = new Configuration({
+  apiKey: process.env.OPENAI_API_KEY,
+});
+
+const openaiClient = new OpenAIApi(config);
+
+async function getEmbedding(text) {
+  const response = await openaiClient.createEmbedding({
+    model: 'text-embedding-ada-002',
+    input: text,
+  });
+  return response.data.data[0].embedding;
+}
+
+async function getChatCompletion(prompt) {
+  const response = await openaiClient.createChatCompletion({
+    model: 'gpt-3.5-turbo',
+    messages: [{ role: 'user', content: prompt }],
+    temperature: 0.7,
+    max_tokens: 500,
+  });
+  return response.data.choices[0].message.content.trim();
+}
+
+module.exports = {
+  getEmbedding,
+  getChatCompletion,
+};