express-genix 3.0.0 → 4.1.0

package/templates/services/aiService.js.ejs

@@ -0,0 +1,80 @@
+const { getModel } = require('../config/ai');
+const { HumanMessage, SystemMessage, AIMessage } = require('@langchain/core/messages');
+const { StringOutputParser } = require('@langchain/core/output_parsers');
+const { ChatPromptTemplate } = require('@langchain/core/prompts');
+const logger = require('../utils/logger');
+
+const DEFAULT_SYSTEM_PROMPT = 'You are a helpful AI assistant.';
+
+/**
+ * Send a chat message and get a complete response.
+ */
+const chat = async (message, options = {}) => {
+  const model = getModel(options);
+  const messages = [
+    new SystemMessage(options.systemPrompt || DEFAULT_SYSTEM_PROMPT),
+  ];
+
+  if (Array.isArray(options.history)) {
+    for (const msg of options.history) {
+      if (msg.role === 'user') messages.push(new HumanMessage(msg.content));
+      else if (msg.role === 'assistant') messages.push(new AIMessage(msg.content));
+    }
+  }
+
+  messages.push(new HumanMessage(message));
+
+  const response = await model.invoke(messages);
+
+  logger.info('AI chat completed', {
+    provider: options.provider || process.env.AI_PROVIDER || 'openai',
+  });
+
+  return {
+    content: response.content,
+    model: response.response_metadata?.model || 'unknown',
+    usage: response.usage_metadata || null,
+  };
+};
+
+/**
+ * Stream a chat response. Returns an async generator yielding text chunks.
+ */
+const stream = async function* (message, options = {}) {
+  const model = getModel(options);
+  const messages = [
+    new SystemMessage(options.systemPrompt || DEFAULT_SYSTEM_PROMPT),
+  ];
+
+  if (Array.isArray(options.history)) {
+    for (const msg of options.history) {
+      if (msg.role === 'user') messages.push(new HumanMessage(msg.content));
+      else if (msg.role === 'assistant') messages.push(new AIMessage(msg.content));
+    }
+  }
+
+  messages.push(new HumanMessage(message));
+
+  const streamResponse = await model.stream(messages);
+
+  for await (const chunk of streamResponse) {
+    if (chunk.content) {
+      yield chunk.content;
+    }
+  }
+};
+
+/**
+ * Run a prompt template chain with variable substitution.
+ */
+const chain = async (template, variables = {}, options = {}) => {
+  const model = getModel(options);
+  const prompt = ChatPromptTemplate.fromTemplate(template);
+  const outputParser = new StringOutputParser();
+  const pipeline = prompt.pipe(model).pipe(outputParser);
+
+  const result = await pipeline.invoke(variables);
+  return result;
+};
+
+module.exports = { chat, stream, chain };