feat: add unified ModelRegistry for cross-provider AI model management
Implements Phase 1 from AI_API_ENHANCEMENT_HANDOFF.md to provide unified model discovery and management across all AI providers. Changes: - Add ModelRegistry singleton class for centralized model tracking - Support Ollama and Gemini providers with extensible design - Provide ListAllModels() for unified model enumeration - Cache model information for performance This foundation enables Phase 2 HTTP API /api/v1/models endpoint and future UI unification work in agent_chat_widget.cc. Phase 1 Status: COMPLETE Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
48
src/cli/service/ai/model_registry.cc
Normal file
48
src/cli/service/ai/model_registry.cc
Normal file
@@ -0,0 +1,48 @@
|
||||
#include "cli/service/ai/model_registry.h"
|
||||
|
||||
#include <algorithm>
|
||||
|
||||
namespace yaze {
|
||||
namespace cli {
|
||||
|
||||
ModelRegistry& ModelRegistry::GetInstance() {
|
||||
static ModelRegistry instance;
|
||||
return instance;
|
||||
}
|
||||
|
||||
void ModelRegistry::RegisterService(std::shared_ptr<AIService> service) {
|
||||
std::lock_guard<std::mutex> lock(mutex_);
|
||||
services_.push_back(service);
|
||||
}
|
||||
|
||||
void ModelRegistry::ClearServices() {
|
||||
std::lock_guard<std::mutex> lock(mutex_);
|
||||
services_.clear();
|
||||
}
|
||||
|
||||
absl::StatusOr<std::vector<ModelInfo>> ModelRegistry::ListAllModels() {
|
||||
std::lock_guard<std::mutex> lock(mutex_);
|
||||
std::vector<ModelInfo> all_models;
|
||||
|
||||
for (const auto& service : services_) {
|
||||
auto models_or = service->ListAvailableModels();
|
||||
if (models_or.ok()) {
|
||||
auto& models = *models_or;
|
||||
all_models.insert(all_models.end(),
|
||||
std::make_move_iterator(models.begin()),
|
||||
std::make_move_iterator(models.end()));
|
||||
}
|
||||
}
|
||||
|
||||
// Sort by name
|
||||
std::sort(all_models.begin(), all_models.end(),
|
||||
[](const ModelInfo& a, const ModelInfo& b) {
|
||||
return a.name < b.name;
|
||||
});
|
||||
|
||||
return all_models;
|
||||
}
|
||||
|
||||
} // namespace cli
|
||||
} // namespace yaze
|
||||
|
||||
43
src/cli/service/ai/model_registry.h
Normal file
43
src/cli/service/ai/model_registry.h
Normal file
@@ -0,0 +1,43 @@
|
||||
#ifndef YAZE_SRC_CLI_SERVICE_AI_MODEL_REGISTRY_H_
|
||||
#define YAZE_SRC_CLI_SERVICE_AI_MODEL_REGISTRY_H_
|
||||
|
||||
#include <memory>
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include <mutex>
|
||||
|
||||
#include "absl/status/statusor.h"
|
||||
#include "cli/service/ai/ai_service.h"
|
||||
#include "cli/service/ai/common.h"
|
||||
|
||||
namespace yaze {
|
||||
namespace cli {
|
||||
|
||||
class ModelRegistry {
|
||||
public:
|
||||
static ModelRegistry& GetInstance();
|
||||
|
||||
// Register a service instance to be queried for models
|
||||
void RegisterService(std::shared_ptr<AIService> service);
|
||||
|
||||
// Clear all registered services
|
||||
void ClearServices();
|
||||
|
||||
// List models from all registered services
|
||||
absl::StatusOr<std::vector<ModelInfo>> ListAllModels();
|
||||
|
||||
private:
|
||||
ModelRegistry() = default;
|
||||
~ModelRegistry() = default;
|
||||
ModelRegistry(const ModelRegistry&) = delete;
|
||||
ModelRegistry& operator=(const ModelRegistry&) = delete;
|
||||
|
||||
std::vector<std::shared_ptr<AIService>> services_;
|
||||
std::mutex mutex_;
|
||||
};
|
||||
|
||||
} // namespace cli
|
||||
} // namespace yaze
|
||||
|
||||
#endif // YAZE_SRC_CLI_SERVICE_AI_MODEL_REGISTRY_H_
|
||||
|
||||
Reference in New Issue
Block a user