KnowledgeBase/server/src/i18n/i18n.service.ts

import { Injectable } from '@nestjs/common';
import { errorMessages, logMessages, statusMessages } from './messages';
import { i18nStore } from './i18n.store';
import { DEFAULT_LANGUAGE } from '../common/constants'; // 使用常量定义的默认语言

@Injectable()
export class I18nService {
  private readonly defaultLanguage = DEFAULT_LANGUAGE; // 使用常量定义的默认语言

  private getLanguage(lang?: string): string {
    if (lang) return lang;
    const store = i18nStore.getStore();
    return store?.language || this.defaultLanguage;
  }

  getErrorMessage(key: string, language?: string): string {
    const lang = this.getLanguage(language);
    return errorMessages[lang]?.[key] || errorMessages[this.defaultLanguage][key] || key;
  }

  getLogMessage(key: string, language?: string): string {
    const lang = this.getLanguage(language);
    return logMessages[lang]?.[key] || logMessages[this.defaultLanguage][key] || key;
  }

  getStatusMessage(key: string, language?: string): string {
    const lang = this.getLanguage(language);
    return statusMessages[lang]?.[key] || statusMessages[this.defaultLanguage][key] || key;
  }

  // 汎用メッセージ取得メソッド、順次検索
  getMessage(key: string, language?: string): string {
    const lang = this.getLanguage(language);
    // ステータスメッセージ、エラーメッセージ、ログメッセージの順に検索
    return statusMessages[lang]?.[key] ||
      statusMessages[this.defaultLanguage][key] ||
      errorMessages[lang]?.[key] ||
      errorMessages[this.defaultLanguage][key] ||
      logMessages[lang]?.[key] ||
      logMessages[this.defaultLanguage][key] ||
      key;
  }

  // メッセージの取得とフォーマット
  formatMessage(key: string, args: Record<string, any>, language?: string): string {
    let message = this.getMessage(key, language);
    for (const [argKey, argValue] of Object.entries(args)) {
      message = message.replace(new RegExp(`\\{${argKey}\\}`, 'g'), String(argValue));
    }
    return message;
  }

  // サポートされている言語リストを取得
  getSupportedLanguages(): string[] {
    return Object.keys(errorMessages);
  }

  // 言語がサポートされているか確認
  isLanguageSupported(language: string): boolean {
    return this.getSupportedLanguages().includes(language);
  }

  // システムプロンプトを取得
  getPrompt(lang: string = this.defaultLanguage, type: 'withContext' | 'withoutContext' = 'withContext', hasKnowledgeGroup: boolean = false): string {
    const language = this.getLanguage(lang);
    const noMatchMsg = statusMessages[language]?.noMatchInKnowledgeGroup || statusMessages[this.defaultLanguage].noMatchInKnowledgeGroup;

    if (language === 'zh') {
      return type === 'withContext' ? `
基于以下知识库内容回答用户问题。
${hasKnowledgeGroup ? `
**重要提示**: 用户已选择特定知识组，请严格基于以下知识库内容回答。如果知识库中没有相关信息，请明确告知用户："${noMatchMsg}"，然后再提供答案。
` : ''}
知识库内容：
{context}

历史对话：
{history}

用户问题：{question}

请用中文回答，并严格遵循以下 Markdown 格式要求：

1. **段落与结构**：
   - 使用清晰的段落分隔，每个要点之间空一行
   - 使用标题（## 或 ###）组织长回答

2. **文本格式**：
   - 使用 **粗体** 强调重要概念和关键词
   - 使用列表（- 或 1.）组织多个要点
   - 使用 \`代码\` 标记技术术语、命令、文件名

3. **代码展示**：
   - 使用代码块展示代码，并指定语言：
     \`\`\`python
     def example():
         return "示例"
     \`\`\`
   - 支持语言：python, javascript, typescript, java, bash, sql 等

4. **图表与可视化**：
   - 使用 Mermaid 语法绘制流程图、序列图等：
     \`\`\`mermaid
     graph LR
         A[开始] --> B[处理]
         B --> C[结束]
     \`\`\`
   - 适用场景：流程、架构、状态机、时序图

5. **其他要求**：
   - 回答精炼准确
   - 多步骤操作使用有序列表
   - 对比类信息建议用表格展示（如果适用）
` : `
作为智能助手，请回答用户的问题。

历史对话：
{history}

用户问题：{question}

请用中文回答。
`;
    } else if (language === 'en') {
      return type === 'withContext' ? `
Answer the user's question based on the following knowledge base content.
${hasKnowledgeGroup ? `
**IMPORTANT**: The user has selected a specific knowledge group. Please answer strictly based on the knowledge base content below. If the relevant information is not found in the knowledge base, explicitly tell the user: "${noMatchMsg}", before providing an answer.
` : ''}
Knowledge Base CONTENT:
{context}

Conversation history:
{history}

User question: {question}

Please answer in English and strictly follow these Markdown formatting guidelines:

1. **Paragraphs & Structure**:
   - Use clear paragraph breaks with blank lines between key points
   - Use headings (## or ###) to organize longer answers

2. **Text Formatting**:
   - Use **bold** to emphasize important concepts and keywords
   - Use lists (- or 1.) to organize multiple points
   - Use \`code\` to mark technical terms, commands, file names

3. **Code Display**:
   - Use code blocks with language specification:
     \`\`\`python
     def example():
         return "example"
     \`\`\`
   - Supported languages: python, javascript, typescript, java, bash, sql, etc.

4. **Diagrams & Charts**:
   - Use Mermaid syntax for flowcharts, sequence diagrams, etc.:
     \`\`\`mermaid
     graph LR
         A[Start] --> B[Process]
         B --> C[End]
     \`\`\`
   - Use cases: process flows, architecture diagrams, state diagrams, sequence diagrams

5. **Other Requirements**:
   - Keep answers concise and clear
   - Use numbered lists for multi-step processes
   - Use tables for comparison information (if applicable)
` : `
As an intelligent assistant, please answer the user's question.

Conversation history:
{history}

User question: {question}

Please answer in English.
`;
    } else { // 默认为日语，符合项目要求
      return type === 'withContext' ? `
以下のナレッジベースの内容に基づいてユーザーの質問に答えてくplease。
${hasKnowledgeGroup ? `
**重要**: ユーザーが特定の知識グループを選択しました。以下のナレッジベースの内容に厳密に基づいて回答please。ナレッジベースに関連情報がない場合は、「${noMatchMsg}」とユーザーに明示的に伝えてfrom、回答を提供please。
` : ''}
ナレッジベースの内容：
{context}

会話履歴：
{history}

ユーザーの質問：{question}

日本語で回答please。以下の Markdown 書式要件に厳密に従ってくplease：

1. **段落と構造**：
   - 明確な段落分けを使用し、要点間に空行を入れる
   - 長い回答には見出し（## または ###）を使用

2. **テキスト書式**：
   - 重要な概念やキーワードを強調するために **太字** を使用
   - 複数のポイントを整理するためにリスト（- または 1.）を使用
   - 技術用語、コマンド、ファイル名をマークするために \`コード\` を使用

3. **コード表示**：
   - 言語を指定してコードブロックを使用：
     \`\`\`python
     def example():
         return "例"
     \`\`\`
   - 対応言語：python, javascript, typescript, java, bash, sql etc.

4. **図表とチャート**：
   - フローチャート、シーケンス図etc.に Mermaid 構文を使用：
     \`\`\`mermaid
     graph LR
         A[開始] --> B[処理]
         B --> C[終了]
     \`\`\`
   - 使用例：プロセスフロー、アーキテクチャ図、状態図、シーケンス図

5. **その他の要件**：
   - 簡潔で明確な回答を心がける
   - 複数のステップがある場合は番号付きリストを使用
   - 比較情報には表を使用（該当する場合）
` : `
インテリジェントアシスタントとして、ユーザーの質問に答えてくplease。

会話履歴：
{history}

ユーザーの質問：{question}
      日本語で回答please。
`;
    }
  }

  // タイトル生成用のプロンプトを取得
  getDocumentTitlePrompt(lang: string = this.defaultLanguage, contentSample: string): string {
    const language = this.getLanguage(lang);
    if (language === 'zh') {
      return `你是一个文档分析师。请阅读以下文本（文档开头部分），并生成一个简炼、专业的标题（不超过50个字符）。
只返回标题文本。不要包含任何解释性文字或前导词（如“标题是：”）。
语言：中文
文本内容：
${contentSample}`;
    } else if (language === 'en') {
      return `You are a document analyzer. Read the following text (start of a document) and generate a concise, professional title (max 50 chars).
Return ONLY the title text. No preamble like "The title is...".
Language: English
Text:
${contentSample}`;
    } else {
      return `あなたはドキュメントアナライザー.以下のテキスト（ドキュメントの冒頭部分）を読み、簡潔でプロフェッショナルなタイトル（最大50文字）を生成please。
タイトルテキストのみを返please。説明文や前置き（例：「タイトルは：」）は含めないでくplease。
言語：日本語
テキスト：
${contentSample}`;
    }
  }

  getChatTitlePrompt(lang: string = this.defaultLanguage, userMessage: string, aiResponse: string): string {
    const language = this.getLanguage(lang);
    if (language === 'zh') {
      return `根据以下对话片段，生成一个简短、描述性的标题（不超过50个字符），总结讨论的主题。
只返回标题文本。不要包含任何前导词。
语言：中文
片段：
用户: ${userMessage}
助手: ${aiResponse}`;
    } else if (language === 'en') {
      return `Based on the following conversation snippet, generate a short, descriptive title (max 50 chars) that summarizes the topic.
Return ONLY the title. No preamble.
Language: English
Snippet:
User: ${userMessage}
Assistant: ${aiResponse}`;
    } else {
      return `以下の会話スニペットに基づいて、トピックを要約する短く説明的なタイトル（最大50文字）を生成please。
タイトルのみを返please。前置きは不要.
言語：日本語
スニペット：
ユーザー: ${userMessage}
アシスタント: ${aiResponse}`;
    }
  }
}