import few_shots from '../prompts/few_shots';

import { Utils } from './utils';

export class AssistantSdk {
  static getAssistantPrompt(params: {
    maxTokens?: number;
    userTimezone?: string;
    userLocation?: string;
  }): string {
    const { maxTokens, userTimezone = 'UTC', userLocation = '' } = params;
    // console.log('[DEBUG_LOG] few_shots:', JSON.stringify(few_shots));
    let selectedFewshots = Utils.selectEquitably?.(few_shots);
    // console.log('[DEBUG_LOG] selectedFewshots after Utils.selectEquitably:', JSON.stringify(selectedFewshots));
    if (!selectedFewshots) {
      selectedFewshots = few_shots;
      // console.log('[DEBUG_LOG] selectedFewshots after fallback:', JSON.stringify(selectedFewshots));
    }
    const sdkDate = new Date().toISOString();
    const [currentDate] = sdkDate.includes('T') ? sdkDate.split('T') : [sdkDate];
    const now = new Date();
    const formattedMinutes = String(now.getMinutes()).padStart(2, '0');
    const currentTime = `${now.getHours()}:${formattedMinutes} ${now.getSeconds()}s`;

    return `# Assistant Knowledge
## Current Context
### Date: ${currentDate} ${currentTime}
### Web Host: open-gsio.seemueller.workers.dev
${maxTokens ? `### Max Response Length: ${maxTokens} tokens (maximum)` : ''}
### Lexicographical Format: Markdown
### User Location: ${userLocation || 'Unknown'}
### Timezone: ${userTimezone}
## Response Framework
1. Use knowledge provided in the current context as the primary source of truth.
2. Format all responses in Markdown.
3. Attribute external sources with footnotes.
## Examples
#### Example 0
**Human**: What is this?
**Assistant**: This is a conversational AI system.
---
${AssistantSdk.useFewshots(selectedFewshots, 5)}
---
## Directive
Continuously monitor the evolving conversation. Dynamically adapt each response.`;
  }

  static useFewshots(fewshots: Record<string, string>, limit = 5): string {
    return Object.entries(fewshots)
      .slice(0, limit)
      .map(([q, a], i) => {
        return `#### Example ${i + 1}\n**Human**: ${q}\n**Assistant**: ${a}`;
      })
      .join('\n---\n');
  }
}