Initial commit: Telegram bot with Cloudflare Workers
- OpenAI GPT-4o-mini with Function Calling - Cloudflare D1 for user profiles and message buffer - Sliding window (3 summaries max) for infinite context - Tools: weather, search, time, calculator - Workers AI fallback support - Webhook security with rate limiting Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
155
src/n8n-service.ts
Normal file
155
src/n8n-service.ts
Normal file
@@ -0,0 +1,155 @@
|
||||
import { Env, IntentAnalysis, N8nResponse } from './types';
|
||||
|
||||
// n8n으로 처리할 기능 목록
|
||||
const N8N_CAPABILITIES = [
|
||||
'weather', // 날씨
|
||||
'search', // 검색
|
||||
'image', // 이미지 생성
|
||||
'translate', // 번역
|
||||
'schedule', // 일정
|
||||
'reminder', // 알림
|
||||
'news', // 뉴스
|
||||
'calculate', // 계산
|
||||
'summarize_url', // URL 요약
|
||||
];
|
||||
|
||||
// AI가 의도를 분석하여 n8n 호출 여부 결정
|
||||
export async function analyzeIntent(
|
||||
ai: Ai,
|
||||
userMessage: string
|
||||
): Promise<IntentAnalysis> {
|
||||
const prompt = `사용자 메시지를 분석하여 어떤 처리가 필요한지 JSON으로 응답하세요.
|
||||
|
||||
## 외부 기능이 필요한 경우 (action: "n8n")
|
||||
- 날씨 정보: type = "weather"
|
||||
- 웹 검색: type = "search"
|
||||
- 이미지 생성: type = "image"
|
||||
- 번역: type = "translate"
|
||||
- 일정/캘린더: type = "schedule"
|
||||
- 알림 설정: type = "reminder"
|
||||
- 뉴스: type = "news"
|
||||
- 복잡한 계산: type = "calculate"
|
||||
- URL/링크 요약: type = "summarize_url"
|
||||
|
||||
## 일반 대화인 경우 (action: "chat")
|
||||
- 인사, 잡담, 질문, 조언 요청 등
|
||||
|
||||
## 응답 형식 (JSON만 출력)
|
||||
{"action": "n8n", "type": "weather", "confidence": 0.9}
|
||||
또는
|
||||
{"action": "chat", "confidence": 0.95}
|
||||
|
||||
## 사용자 메시지
|
||||
${userMessage}
|
||||
|
||||
JSON:`;
|
||||
|
||||
try {
|
||||
const response = await ai.run('@cf/meta/llama-3.1-8b-instruct', {
|
||||
messages: [{ role: 'user', content: prompt }],
|
||||
max_tokens: 100,
|
||||
});
|
||||
|
||||
const text = response.response || '';
|
||||
|
||||
// JSON 추출
|
||||
const jsonMatch = text.match(/\{[^}]+\}/);
|
||||
if (jsonMatch) {
|
||||
const parsed = JSON.parse(jsonMatch[0]);
|
||||
return {
|
||||
action: parsed.action || 'chat',
|
||||
type: parsed.type,
|
||||
confidence: parsed.confidence || 0.5,
|
||||
};
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('Intent analysis error:', error);
|
||||
}
|
||||
|
||||
// 기본값: 일반 대화
|
||||
return { action: 'chat', confidence: 0.5 };
|
||||
}
|
||||
|
||||
// n8n Webhook 호출
|
||||
export async function callN8n(
|
||||
env: Env,
|
||||
type: string,
|
||||
userMessage: string,
|
||||
userId: number,
|
||||
chatId: string,
|
||||
userProfile?: string | null
|
||||
): Promise<N8nResponse> {
|
||||
if (!env.N8N_WEBHOOK_URL) {
|
||||
return { error: 'n8n이 설정되지 않았습니다.' };
|
||||
}
|
||||
|
||||
const webhookUrl = `${env.N8N_WEBHOOK_URL}/webhook/telegram-bot`;
|
||||
|
||||
try {
|
||||
const response = await fetch(webhookUrl, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
},
|
||||
body: JSON.stringify({
|
||||
type,
|
||||
message: userMessage,
|
||||
user_id: userId,
|
||||
chat_id: chatId,
|
||||
profile: userProfile,
|
||||
timestamp: new Date().toISOString(),
|
||||
}),
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
console.error('n8n error:', response.status, await response.text());
|
||||
return { error: `n8n 호출 실패 (${response.status})` };
|
||||
}
|
||||
|
||||
const data = await response.json() as N8nResponse;
|
||||
return data;
|
||||
} catch (error) {
|
||||
console.error('n8n fetch error:', error);
|
||||
return { error: 'n8n 연결 실패' };
|
||||
}
|
||||
}
|
||||
|
||||
// 스마트 라우팅: AI 판단 → n8n 또는 로컬 AI
|
||||
export async function smartRoute(
|
||||
env: Env,
|
||||
userMessage: string,
|
||||
userId: number,
|
||||
chatId: string,
|
||||
userProfile?: string | null
|
||||
): Promise<{ useN8n: boolean; n8nType?: string; n8nResponse?: N8nResponse }> {
|
||||
// n8n URL이 없으면 항상 로컬 AI 사용
|
||||
if (!env.N8N_WEBHOOK_URL) {
|
||||
return { useN8n: false };
|
||||
}
|
||||
|
||||
// 의도 분석
|
||||
const intent = await analyzeIntent(env.AI, userMessage);
|
||||
|
||||
// n8n 호출이 필요하고 신뢰도가 높은 경우
|
||||
if (intent.action === 'n8n' && intent.confidence >= 0.7 && intent.type) {
|
||||
const n8nResponse = await callN8n(
|
||||
env,
|
||||
intent.type,
|
||||
userMessage,
|
||||
userId,
|
||||
chatId,
|
||||
userProfile
|
||||
);
|
||||
|
||||
// n8n 응답이 성공적인 경우
|
||||
if (n8nResponse.reply && !n8nResponse.error) {
|
||||
return {
|
||||
useN8n: true,
|
||||
n8nType: intent.type,
|
||||
n8nResponse,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
return { useN8n: false };
|
||||
}
|
||||
Reference in New Issue
Block a user