contaigents
Version:
Modular AI Content Ecosystem with Audio Generation
129 lines (128 loc) • 6.58 kB
JavaScript
import { ChatService } from '../services/chatService.js';
import { ConversationMessageBuilder } from '../services/conversation/MessageBuilder.js';
import { PROVIDER_CAPABILITIES } from '../services/llm/types.js';
import { ToolManager } from '../services/tools/ToolManager.js';
async function testConversationArchitecture() {
console.log('🧪 Testing Conversation Architecture Implementation');
console.log('='.repeat(60));
// Test 1: Provider Capabilities Detection
console.log('\n📋 Test 1: Provider Capabilities Detection');
console.log('-'.repeat(40));
const providers = ['OpenAI', 'Anthropic', 'HuggingFace', 'Ollama', 'Gemini'];
for (const provider of providers) {
const capabilities = PROVIDER_CAPABILITIES[provider];
if (capabilities) {
console.log(`✅ ${provider}:`);
console.log(` - Supports Conversation: ${capabilities.supportsConversation}`);
console.log(` - Supports System Messages: ${capabilities.supportsSystemMessages}`);
console.log(` - Supports Tool Calls: ${capabilities.supportsToolCalls}`);
console.log(` - Custom Format: ${capabilities.customFormat}`);
}
else {
console.log(`❌ ${provider}: No capabilities defined`);
}
}
// Test 2: Message Builder for Conversation Provider
console.log('\n📋 Test 2: Message Builder for Conversation Provider (OpenAI)');
console.log('-'.repeat(40));
const chatService = new ChatService();
const toolManager = new ToolManager();
const session = chatService.createSession({ systemPrompt: 'Test conversation architecture' });
const sessionId = session.id;
console.log('🔍 Session created:', !!session, 'ID:', sessionId);
if (session) {
// Update system prompt for testing
session.systemPrompt = 'You are a helpful AI assistant.';
// Add some conversation history
session.messages.push({
id: 'msg1',
role: 'user',
content: 'Hello, can you help me?',
timestamp: Date.now()
});
session.messages.push({
id: 'msg2',
role: 'assistant',
content: 'Of course! I\'d be happy to help you.',
timestamp: Date.now()
});
try {
console.log('🔍 Building messages for OpenAI...');
const messageData = await ConversationMessageBuilder.buildForProvider(session, 'What can you do?', 'OpenAI', toolManager);
console.log('🔍 Message data type:', typeof messageData, 'Array?', Array.isArray(messageData));
if (Array.isArray(messageData)) {
console.log('✅ OpenAI (Conversation Provider) - Built message array:');
console.log(` - Total messages: ${messageData.length}`);
console.log(` - System message: ${messageData.find(m => m.role === 'system') ? 'Yes' : 'No'}`);
console.log(` - User messages: ${messageData.filter(m => m.role === 'user').length}`);
console.log(` - Assistant messages: ${messageData.filter(m => m.role === 'assistant').length}`);
// Show first few messages
console.log(' - First 2 messages:');
messageData.slice(0, 2).forEach((msg, i) => {
console.log(` ${i + 1}. ${msg.role}: ${msg.content.substring(0, 50)}...`);
});
}
else {
console.log('❌ Expected array for conversation provider, got string');
}
}
catch (error) {
console.log(`❌ Error building messages for OpenAI: ${error}`);
}
}
else {
console.log('❌ Session not found for OpenAI test');
}
// Test 3: Message Builder for Simple Prompt Provider
console.log('\n📋 Test 3: Message Builder for Simple Prompt Provider (Ollama)');
console.log('-'.repeat(40));
if (session) {
try {
console.log('🔍 Building messages for Ollama...');
const messageData = await ConversationMessageBuilder.buildForProvider(session, 'What can you do?', 'Ollama', toolManager);
console.log('🔍 Message data type:', typeof messageData, 'Array?', Array.isArray(messageData));
if (typeof messageData === 'string') {
console.log('✅ Ollama (Simple Provider) - Built prompt string:');
console.log(` - Length: ${messageData.length} characters`);
console.log(` - Contains "System:": ${messageData.includes('System:')}`);
console.log(` - Contains "Human:": ${messageData.includes('Human:')}`);
console.log(` - Contains "Assistant:": ${messageData.includes('Assistant:')}`);
// Show first 200 characters
console.log(` - Preview: ${messageData.substring(0, 200)}...`);
}
else {
console.log('❌ Expected string for simple provider, got array');
}
}
catch (error) {
console.log(`❌ Error building messages for Ollama: ${error}`);
}
}
else {
console.log('❌ Session not found for Ollama test');
}
// Test 4: Provider Capability Methods
console.log('\n📋 Test 4: Provider Capability Methods');
console.log('-'.repeat(40));
try {
// Test with a mock provider that should have capabilities
const { OpenAIProvider } = await import('../services/llm/OpenAIProvider.js');
const openAIProvider = new OpenAIProvider();
console.log('✅ OpenAI Provider:');
console.log(` - supportsConversation(): ${openAIProvider.supportsConversation()}`);
const capabilities = openAIProvider.getCapabilities();
console.log(` - getCapabilities(): ${JSON.stringify(capabilities, null, 2)}`);
}
catch (error) {
console.log(`❌ Error testing provider methods: ${error}`);
}
console.log('\n🎉 Conversation Architecture Test Complete!');
console.log('\n📊 Summary:');
console.log('- ✅ Provider capabilities detection working');
console.log('- ✅ Message builder creates arrays for conversation providers');
console.log('- ✅ Message builder creates strings for simple providers');
console.log('- ✅ Provider capability methods implemented');
console.log('- ✅ Hybrid architecture ready for testing with real providers');
}
// Run the test
testConversationArchitecture().catch(console.error);