n8n-io/n8n
1
1
Fork 0
mirror of https://github.com/n8n-io/n8n.git synced 2024-08-16 16:40:30 +03:00
Code Issues Projects Releases Wiki Activity

Use AI Assistant Service

Browse Source
This commit is contained in:
Tomi Turtiainen 2024-06-20 18:25:19 +03:00
parent 0c18c1e93f
commit b5bc89cdda
No known key found for this signature in database
5 changed files with 66 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
packages/cli/package.json
View File

@ -68,6 +68,7 @@
"@types/convict": "^6.1.1",
"@types/cookie-parser": "^1.4.2",
"@types/express": "^4.17.21",
"@types/express-http-proxy": "^1.6.6",
"@types/formidable": "^3.4.5",
"@types/json-diff": "^1.0.0",
"@types/jsonwebtoken": "^9.0.6",
@ -125,6 +126,7 @@
"express": "4.19.2",
"express-async-errors": "3.1.1",
"express-handlebars": "7.1.2",
"express-http-proxy": "^2.0.0",
"express-openapi-validator": "5.1.6",
"express-prom-bundle": "6.6.0",
"express-rate-limit": "7.2.0",

7
packages/cli/src/AbstractServer.ts
View File

@ -92,6 +92,10 @@ export abstract class AbstractServer {
// Additional configuration in derived classes
}
async configureProxyEndpoints(): Promise<void> {
// Additional configuration in derived classes
}
private async setupErrorHandlers() {
const { app } = this;
@ -233,6 +237,9 @@ export abstract class AbstractServer {
);
}
// Setup proxy endpoints before body parsing
await this.configureProxyEndpoints();
// Setup body parsing middleware after the webhook handlers are setup
this.app.use(bodyParser);

6
packages/cli/src/Server.ts
View File

@ -191,6 +191,12 @@ export class Server extends AbstractServer {
controllers.forEach((controller) => registerController(app, controller));
}
async configureProxyEndpoints(): Promise<void> {
const aiController = Container.get(AIController);
this.app.use('/rest/ai', aiController.debugWithAssistant);
registerController(this.app, AIController);
}
async configure(): Promise<void> {
if (config.getEnv('endpoints.metrics.enable')) {
const { MetricsService } = await import('@/services/metrics.service');

69
packages/cli/src/controllers/ai.controller.ts
View File

@ -19,7 +19,11 @@ import { z } from 'zod';
import zodToJsonSchema from 'zod-to-json-schema';
import { ChatMessageHistory } from 'langchain/stores/message/in_memory';
import { ApplicationError } from 'n8n-workflow';
import { NO_MORE_SUGGESTIONS_PROMPT, QUICK_ACTIONS, REACT_DEBUG_PROMPT } from '@/aiAssistant/prompts/debug_prompts';
import {
NO_MORE_SUGGESTIONS_PROMPT,
QUICK_ACTIONS,
REACT_DEBUG_PROMPT,
} from '@/aiAssistant/prompts/debug_prompts';
import {
addConversationToHistory,
chatHistory,
@ -35,6 +39,7 @@ import { createInternetSearchTool } from '@/aiAssistant/tools/internet_search.to
import { prepareDebugUserPrompt } from '@/aiAssistant/utils';
import { N8N_BLOG, N8N_COMMUNITY, N8N_MARKETING_WEBSITE } from '@/aiAssistant/constants';
import { getNextUserPrompt } from '@/aiAssistant/intent_detector';
import proxy from 'express-http-proxy';
const errorSuggestionSchema = z.object({
suggestion: z.object({
@ -97,22 +102,41 @@ export class AIController {
await this.askAssistant(message, res);
}
debugWithAssistant = proxy('http://127.0.0.1:3000/ai-assistant', {});
/**
* Debug n8n error using the agent that has access to different tools.
* THIS IS THE DEBUG MODE
*/
@Post('/debug-with-assistant', { skipAuth: true })
async debugWithAssistant(req: AIRequest.AssistantDebug, res: express.Response) {
const { nodeType, error, errorNode, authType, message, userTraits, nodeInputData, referencedNodesData } = req.body;
resetToolHistory();
if (message) {
await this.askAssistant(`${message}\n`, res, true);
return;
}
clearChatHistory();
const userPrompt = prepareDebugUserPrompt(nodeType, error, errorNode, authType, userTraits, nodeInputData, referencedNodesData);
await this.askAssistant(userPrompt, res, true);
}
// @Post('/debug-with-assistant', { skipAuth: true })
// async debugWithAssistant(req: AIRequest.AssistantDebug, res: express.Response) {
// const {
// nodeType,
// error,
// errorNode,
// authType,
// message,
// userTraits,
// nodeInputData,
// referencedNodesData,
// } = req.body;
// resetToolHistory();
// if (message) {
// await this.askAssistant(`${message}\n`, res, true);
// return;
// }
// clearChatHistory();
// const userPrompt = prepareDebugUserPrompt(
// nodeType,
// error,
// errorNode,
// authType,
// userTraits,
// nodeInputData,
// referencedNodesData,
// );
// await this.askAssistant(userPrompt, res, true);
// }
/**
* Chat with pinecone vector store that contains n8n documentation.
@ -170,9 +194,14 @@ export class AIController {
// ----------------- Tools -----------------
// Internet search tool setup:
// - In debug mode, use only forum to search for answers, while in free-chat mode use all websites (and more results)
const internetToolWebsites = debug ? [N8N_COMMUNITY] : [N8N_MARKETING_WEBSITE, N8N_BLOG, N8N_COMMUNITY];
const internetToolWebsites = debug
? [N8N_COMMUNITY]
: [N8N_MARKETING_WEBSITE, N8N_BLOG, N8N_COMMUNITY];
const internetToolMaxResults = debug ? 5 : 10;
const internetSearchTool = createInternetSearchTool(internetToolWebsites, internetToolMaxResults);
const internetSearchTool = createInternetSearchTool(
internetToolWebsites,
internetToolMaxResults,
);
const tools = [n8nInfoTool, internetSearchTool];
const toolNames = tools.map((tool) => tool.name);
@ -182,7 +211,6 @@ export class AIController {
? ChatPromptTemplate.fromTemplate(REACT_DEBUG_PROMPT)
: ChatPromptTemplate.fromTemplate(REACT_CHAT_PROMPT);
const agent = await createReactAgent({
llm: assistantModel,
tools,
@ -202,14 +230,14 @@ export class AIController {
noMoreHelp = checkIfAllHelpUsed();
// Hard-stop if human asks for too many suggestions
if (noMoreHelp) {
userMessage = NO_MORE_SUGGESTIONS_PROMPT;
userMessage = NO_MORE_SUGGESTIONS_PROMPT;
} else {
// Detect user intention and map it to the correct prompt
const lastFollowUpQuestion = getLastFollowUpQuestion(chatHistory);
// Only if there is a follow-up question, we don't want to alter the initial debug prompt
if (lastFollowUpQuestion) {
const detectorResult = await getNextUserPrompt(userMessage, lastFollowUpQuestion);
userMessage = detectorResult.prompt;
userMessage = detectorResult.prompt;
increaseHelpCounter(detectorResult.detectedIntent);
}
// First response is also the suggestion, if we want to count that:
@ -241,12 +269,13 @@ export class AIController {
if (error instanceof Error)
console.log('>> ⚠️ <<', `Error: Could not parse LLM output: ${error.toString()}`);
response = error.toString().replace(/Error: Could not parse LLM output: /, '');
response = error.toString().replace(/Error: Could not parse LLM output: /, '');
}
console.log('>> 🤖 <<', response);
addConversationToHistory(userMessage, response);
let debugInfo = '--------------------------------- [DEBUG INFO] -----------------------------------\n';
let debugInfo =
'--------------------------------- [DEBUG INFO] -----------------------------------\n';
debugInfo +=
toolHistory.n8n_documentation.length > 0
? `📄 N8N DOCS DOCUMENTS USED: \n• ${toolHistory.n8n_documentation.join('\n• ')}\n`

4
packages/editor-ui/src/api/ai.ts
View File

@ -114,7 +114,7 @@ export const debugWithAssistant = async (
const headers = {
'Content-Type': 'application/json',
};
const response = await fetch(`${context.baseUrl}/ai/debug-with-assistant`, {
const response = await fetch(`${context.baseUrl}/ai/ai-assistant/error-debug`, {
headers,
method: 'POST',
credentials: 'include',
@ -225,7 +225,7 @@ export const debugChatWithAiErrorHelper = async (
const headers = {
'Content-Type': 'application/json',
};
const response = await fetch(`${context.baseUrl}/ai/debug-chat`, {
const response = await fetch(`${context.baseUrl}/ai/ai-assistant/code-node-debug`, {
headers,
method: 'POST',
credentials: 'include',