208b9ade51
- Added support for AI agent services, including `ConversationalAgentService`, to facilitate user interactions through a chat interface. - Implemented `ChatTUI` for a terminal-based chat experience, allowing users to send messages and receive responses from the AI agent. - Updated `EditorManager` to include options for displaying the agent chat widget and performance dashboard. - Enhanced CMake configurations to include new source files for AI services and chat interface components. This commit significantly expands the functionality of the z3ed system, paving the way for a more interactive and user-friendly experience in ROM hacking.
73 lines
2.3 KiB
C++
73 lines
2.3 KiB
C++
#include "cli/service/ai/service_factory.h"
|
|
|
|
#include <iostream>
|
|
|
|
#include "cli/service/ai/ai_service.h"
|
|
#include "cli/service/ai/gemini_ai_service.h"
|
|
#include "cli/service/ai/ollama_ai_service.h"
|
|
|
|
namespace yaze {
|
|
namespace cli {
|
|
|
|
std::unique_ptr<AIService> CreateAIService() {
|
|
// Priority: Ollama (local) > Gemini (remote) > Mock (testing)
|
|
const char* provider_env = std::getenv("YAZE_AI_PROVIDER");
|
|
const char* gemini_key = std::getenv("GEMINI_API_KEY");
|
|
const char* ollama_model = std::getenv("OLLAMA_MODEL");
|
|
const char* gemini_model = std::getenv("GEMINI_MODEL");
|
|
|
|
// Explicit provider selection
|
|
if (provider_env && std::string(provider_env) == "ollama") {
|
|
OllamaConfig config;
|
|
|
|
// Allow model override via env
|
|
if (ollama_model && std::strlen(ollama_model) > 0) {
|
|
config.model = ollama_model;
|
|
}
|
|
|
|
auto service = std::make_unique<OllamaAIService>(config);
|
|
|
|
// Health check
|
|
if (auto status = service->CheckAvailability(); !status.ok()) {
|
|
std::cerr << "⚠️ Ollama unavailable: " << status.message() << std::endl;
|
|
std::cerr << " Falling back to MockAIService" << std::endl;
|
|
return std::make_unique<MockAIService>();
|
|
}
|
|
|
|
std::cout << "🤖 Using Ollama AI with model: " << config.model << std::endl;
|
|
return service;
|
|
}
|
|
|
|
// Gemini if API key provided
|
|
if (gemini_key && std::strlen(gemini_key) > 0) {
|
|
GeminiConfig config(gemini_key);
|
|
|
|
// Allow model override via env
|
|
if (gemini_model && std::strlen(gemini_model) > 0) {
|
|
config.model = gemini_model;
|
|
}
|
|
|
|
auto service = std::make_unique<GeminiAIService>(config);
|
|
|
|
// Health check
|
|
if (auto status = service->CheckAvailability(); !status.ok()) {
|
|
std::cerr << "⚠️ Gemini unavailable: " << status.message() << std::endl;
|
|
std::cerr << " Falling back to MockAIService" << std::endl;
|
|
return std::make_unique<MockAIService>();
|
|
}
|
|
|
|
std::cout << "🤖 Using Gemini AI with model: " << config.model << std::endl;
|
|
return service;
|
|
}
|
|
|
|
// Default: Mock service for testing
|
|
std::cout << "🤖 Using MockAIService (no LLM configured)" << std::endl;
|
|
std::cout
|
|
<< " Tip: Set YAZE_AI_PROVIDER=ollama or GEMINI_API_KEY to enable LLM"
|
|
<< std::endl;
|
|
return std::make_unique<MockAIService>();
|
|
}
|
|
|
|
} // namespace cli
|
|
} // namespace yaze
|