|
| 1 | +<?php |
| 2 | + |
| 3 | +declare(strict_types=1); |
| 4 | + |
| 5 | +namespace WordPress\AiClient\Providers; |
| 6 | + |
| 7 | +use InvalidArgumentException; |
| 8 | +use WordPress\AiClient\Providers\DTO\ProviderMetadata; |
| 9 | +use WordPress\AiClient\Providers\DTO\ProviderModelsMetadata; |
| 10 | +use WordPress\AiClient\Providers\Models\DTO\ModelConfig; |
| 11 | +use WordPress\AiClient\Providers\Models\DTO\ModelMetadata; |
| 12 | +use WordPress\AiClient\Providers\Models\DTO\ModelRequirements; |
| 13 | + |
| 14 | +/** |
| 15 | + * Registry for managing AI providers and their models. |
| 16 | + * |
| 17 | + * This class provides a centralized way to register AI providers, discover |
| 18 | + * their capabilities, and find suitable models based on requirements. |
| 19 | + * |
| 20 | + * @since n.e.x.t |
| 21 | + */ |
| 22 | +class AiProviderRegistry |
| 23 | +{ |
| 24 | + /** |
| 25 | + * @var array<string, string> Mapping of provider IDs to class names. |
| 26 | + */ |
| 27 | + private array $providerClassNames = []; |
| 28 | + |
| 29 | + /** |
| 30 | + * @var array<string, object> Cache of instantiated provider instances. |
| 31 | + */ |
| 32 | + private array $providerInstances = []; |
| 33 | + |
| 34 | + /** |
| 35 | + * Registers a provider class with the registry. |
| 36 | + * |
| 37 | + * @since n.e.x.t |
| 38 | + * |
| 39 | + * @param string $className The fully qualified provider class name. |
| 40 | + * @throws InvalidArgumentException If the class doesn't exist or implement required interface. |
| 41 | + */ |
| 42 | + public function registerProvider(string $className): void |
| 43 | + { |
| 44 | + if (!class_exists($className)) { |
| 45 | + throw new InvalidArgumentException( |
| 46 | + sprintf('Provider class does not exist: %s', $className) |
| 47 | + ); |
| 48 | + } |
| 49 | + |
| 50 | + // TODO: Add interface validation when ProviderInterface is available |
| 51 | + |
| 52 | + // Get provider metadata to extract ID |
| 53 | + $instance = new $className(); |
| 54 | + |
| 55 | + // Check if provider has metadata method |
| 56 | + if (!method_exists($instance, 'metadata')) { |
| 57 | + throw new InvalidArgumentException( |
| 58 | + sprintf('Provider must implement metadata() method: %s', $className) |
| 59 | + ); |
| 60 | + } |
| 61 | + |
| 62 | + $metadata = $instance->metadata(); |
| 63 | + |
| 64 | + if (!$metadata instanceof ProviderMetadata) { |
| 65 | + throw new InvalidArgumentException( |
| 66 | + sprintf('Provider must return ProviderMetadata from metadata() method: %s', $className) |
| 67 | + ); |
| 68 | + } |
| 69 | + |
| 70 | + $this->providerClassNames[$metadata->getId()] = $className; |
| 71 | + } |
| 72 | + |
| 73 | + /** |
| 74 | + * Checks if a provider is registered. |
| 75 | + * |
| 76 | + * @since n.e.x.t |
| 77 | + * |
| 78 | + * @param string $idOrClassName The provider ID or class name to check. |
| 79 | + * @return bool True if the provider is registered. |
| 80 | + */ |
| 81 | + public function hasProvider(string $idOrClassName): bool |
| 82 | + { |
| 83 | + return isset($this->providerClassNames[$idOrClassName]) || |
| 84 | + in_array($idOrClassName, $this->providerClassNames, true); |
| 85 | + } |
| 86 | + |
| 87 | + /** |
| 88 | + * Gets the class name for a registered provider. |
| 89 | + * |
| 90 | + * @since n.e.x.t |
| 91 | + * |
| 92 | + * @param string $id The provider ID. |
| 93 | + * @return string The provider class name. |
| 94 | + * @throws InvalidArgumentException If the provider is not registered. |
| 95 | + */ |
| 96 | + public function getProviderClassName(string $id): string |
| 97 | + { |
| 98 | + if (!isset($this->providerClassNames[$id])) { |
| 99 | + throw new InvalidArgumentException( |
| 100 | + sprintf('Provider not registered: %s', $id) |
| 101 | + ); |
| 102 | + } |
| 103 | + |
| 104 | + return $this->providerClassNames[$id]; |
| 105 | + } |
| 106 | + |
| 107 | + /** |
| 108 | + * Checks if a provider is properly configured. |
| 109 | + * |
| 110 | + * @since n.e.x.t |
| 111 | + * |
| 112 | + * @param string $idOrClassName The provider ID or class name. |
| 113 | + * @return bool True if the provider is configured and ready to use. |
| 114 | + */ |
| 115 | + public function isProviderConfigured(string $idOrClassName): bool |
| 116 | + { |
| 117 | + $instance = $this->getProviderInstance($idOrClassName); |
| 118 | + |
| 119 | + // TODO: Call availability() method when ProviderInterface is available |
| 120 | + // For now, assume configured if we can instantiate |
| 121 | + return $instance !== null; |
| 122 | + } |
| 123 | + |
| 124 | + /** |
| 125 | + * Finds models across all providers that support the given requirements. |
| 126 | + * |
| 127 | + * @since n.e.x.t |
| 128 | + * |
| 129 | + * @param ModelRequirements $modelRequirements The requirements to match against. |
| 130 | + * @return list<ProviderModelsMetadata> List of provider models metadata that match requirements. |
| 131 | + */ |
| 132 | + public function findModelsMetadataForSupport(ModelRequirements $modelRequirements): array |
| 133 | + { |
| 134 | + $results = []; |
| 135 | + |
| 136 | + foreach ($this->providerClassNames as $providerId => $className) { |
| 137 | + $providerResults = $this->findProviderModelsMetadataForSupport($providerId, $modelRequirements); |
| 138 | + if (!empty($providerResults)) { |
| 139 | + $results[] = new ProviderModelsMetadata( |
| 140 | + $this->getProviderInstance($providerId)->metadata(), |
| 141 | + $providerResults |
| 142 | + ); |
| 143 | + } |
| 144 | + } |
| 145 | + |
| 146 | + return $results; |
| 147 | + } |
| 148 | + |
| 149 | + /** |
| 150 | + * Finds models within a specific provider that support the given requirements. |
| 151 | + * |
| 152 | + * @since n.e.x.t |
| 153 | + * |
| 154 | + * @param string $idOrClassName The provider ID or class name. |
| 155 | + * @param ModelRequirements $modelRequirements The requirements to match against. |
| 156 | + * @return list<ModelMetadata> List of model metadata that match requirements. |
| 157 | + */ |
| 158 | + public function findProviderModelsMetadataForSupport( |
| 159 | + string $idOrClassName, |
| 160 | + ModelRequirements $modelRequirements |
| 161 | + ): array { |
| 162 | + $instance = $this->getProviderInstance($idOrClassName); |
| 163 | + |
| 164 | + // TODO: Get model metadata directory when ProviderInterface is available |
| 165 | + // For now, return empty array as placeholder |
| 166 | + return []; |
| 167 | + } |
| 168 | + |
| 169 | + /** |
| 170 | + * Gets a configured model instance from a provider. |
| 171 | + * |
| 172 | + * @since n.e.x.t |
| 173 | + * |
| 174 | + * @param string $idOrClassName The provider ID or class name. |
| 175 | + * @param string $modelId The model identifier. |
| 176 | + * @param ModelConfig|array<string, mixed> $modelConfig The model configuration. |
| 177 | + * @return object The configured model instance. |
| 178 | + * @throws InvalidArgumentException If provider or model is not found. |
| 179 | + */ |
| 180 | + public function getProviderModel(string $idOrClassName, string $modelId, $modelConfig): object |
| 181 | + { |
| 182 | + $instance = $this->getProviderInstance($idOrClassName); |
| 183 | + |
| 184 | + // Normalize config to ModelConfig if needed |
| 185 | + if (is_array($modelConfig)) { |
| 186 | + // TODO: Improve type safety when ModelConfig::fromArray is finalized |
| 187 | + /** @var ModelConfig $modelConfig */ |
| 188 | + $modelConfig = ModelConfig::fromArray($modelConfig); |
| 189 | + } |
| 190 | + |
| 191 | + // TODO: Call model() method when ProviderInterface is available |
| 192 | + throw new InvalidArgumentException('Model instantiation not yet implemented'); |
| 193 | + } |
| 194 | + |
| 195 | + /** |
| 196 | + * Gets or creates a provider instance. |
| 197 | + * |
| 198 | + * @param string $idOrClassName The provider ID or class name. |
| 199 | + * @return object The provider instance. |
| 200 | + * @throws InvalidArgumentException If provider is not registered. |
| 201 | + */ |
| 202 | + private function getProviderInstance(string $idOrClassName): object |
| 203 | + { |
| 204 | + // Handle both ID and class name |
| 205 | + $className = $this->providerClassNames[$idOrClassName] ?? $idOrClassName; |
| 206 | + |
| 207 | + if (!$this->hasProvider($idOrClassName)) { |
| 208 | + throw new InvalidArgumentException( |
| 209 | + sprintf('Provider not registered: %s', $idOrClassName) |
| 210 | + ); |
| 211 | + } |
| 212 | + |
| 213 | + // Use cached instance if available |
| 214 | + if (isset($this->providerInstances[$className])) { |
| 215 | + return $this->providerInstances[$className]; |
| 216 | + } |
| 217 | + |
| 218 | + // Create and cache new instance |
| 219 | + $this->providerInstances[$className] = new $className(); |
| 220 | + |
| 221 | + return $this->providerInstances[$className]; |
| 222 | + } |
| 223 | +} |
0 commit comments