feat: [STG-1756] forward Vertex model config

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 8
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/browserbase/stagehand-80502d74c1be605e77d45ff2b54297fe34ce85dbad1e8f2dfa30ba6d09601219.yml
-openapi_spec_hash: fd62f768756a400c3ecd695bfcf3845a
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/browserbase/stagehand-c7910965e66e73ad8b65b6cc391d431094b2a6c6577c3e9d82feaa8138e74cff.yml
+openapi_spec_hash: 37748bb69c22a9ce721d9b5a5861f964
 config_hash: 1fb12ae9b478488bc1e56bfbdc210b01

src/Sessions/ModelConfig.php

@@ -4,229 +4,28 @@
 
 namespace Stagehand\Sessions;
 
-use Stagehand\Core\Attributes\Optional;
-use Stagehand\Core\Attributes\Required;
-use Stagehand\Core\Concerns\SdkModel;
-use Stagehand\Core\Contracts\BaseModel;
-use Stagehand\Sessions\ModelConfig\GoogleAuthOptions;
-use Stagehand\Sessions\ModelConfig\Provider;
+use Stagehand\Core\Concerns\SdkUnion;
+use Stagehand\Core\Conversion\Contracts\Converter;
+use Stagehand\Core\Conversion\Contracts\ConverterSource;
+use Stagehand\Sessions\ModelConfig\GenericModelConfigObject;
+use Stagehand\Sessions\ModelConfig\VertexModelConfigObject;
 
 /**
- * @phpstan-import-type GoogleAuthOptionsShape from \Stagehand\Sessions\ModelConfig\GoogleAuthOptions
+ * @phpstan-import-type VertexModelConfigObjectShape from \Stagehand\Sessions\ModelConfig\VertexModelConfigObject
+ * @phpstan-import-type GenericModelConfigObjectShape from \Stagehand\Sessions\ModelConfig\GenericModelConfigObject
  *
- * @phpstan-type ModelConfigShape = array{
- *   modelName: string,
- *   apiKey?: string|null,
- *   baseURL?: string|null,
- *   googleAuthOptions?: null|GoogleAuthOptions|GoogleAuthOptionsShape,
- *   headers?: array<string,string>|null,
- *   location?: string|null,
- *   project?: string|null,
- *   provider?: null|Provider|value-of<Provider>,
- * }
+ * @phpstan-type ModelConfigVariants = VertexModelConfigObject|GenericModelConfigObject
+ * @phpstan-type ModelConfigShape = ModelConfigVariants|VertexModelConfigObjectShape|GenericModelConfigObjectShape
  */
-final class ModelConfig implements BaseModel
+final class ModelConfig implements ConverterSource
 {
-    /** @use SdkModel<ModelConfigShape> */
-    use SdkModel;
+    use SdkUnion;
 
     /**
-     * Model name string with provider prefix (e.g., 'openai/gpt-5-nano').
+     * @return list<string|Converter|ConverterSource>|array<string,string|Converter|ConverterSource>
      */
-    #[Required]
-    public string $modelName;
-
-    /**
-     * API key for the model provider.
-     */
-    #[Optional]
-    public ?string $apiKey;
-
-    /**
-     * Base URL for the model provider.
-     */
-    #[Optional]
-    public ?string $baseURL;
-
-    /**
-     * google-auth-library options used to authenticate Vertex AI models.
-     */
-    #[Optional]
-    public ?GoogleAuthOptions $googleAuthOptions;
-
-    /**
-     * Custom headers sent with every request to the model provider.
-     *
-     * @var array<string,string>|null $headers
-     */
-    #[Optional(map: 'string')]
-    public ?array $headers;
-
-    /**
-     * Google Cloud location for Vertex AI models.
-     */
-    #[Optional]
-    public ?string $location;
-
-    /**
-     * Google Cloud project ID for Vertex AI models.
-     */
-    #[Optional]
-    public ?string $project;
-
-    /**
-     * AI provider for the model (or provide a baseURL endpoint instead).
-     *
-     * @var value-of<Provider>|null $provider
-     */
-    #[Optional(enum: Provider::class)]
-    public ?string $provider;
-
-    /**
-     * `new ModelConfig()` is missing required properties by the API.
-     *
-     * To enforce required parameters use
-     * ```
-     * ModelConfig::with(modelName: ...)
-     * ```
-     *
-     * Otherwise ensure the following setters are called
-     *
-     * ```
-     * (new ModelConfig)->withModelName(...)
-     * ```
-     */
-    public function __construct()
-    {
-        $this->initialize();
-    }
-
-    /**
-     * Construct an instance from the required parameters.
-     *
-     * You must use named parameters to construct any parameters with a default value.
-     *
-     * @param GoogleAuthOptions|GoogleAuthOptionsShape|null $googleAuthOptions
-     * @param array<string,string>|null $headers
-     * @param Provider|value-of<Provider>|null $provider
-     */
-    public static function with(
-        string $modelName,
-        ?string $apiKey = null,
-        ?string $baseURL = null,
-        GoogleAuthOptions|array|null $googleAuthOptions = null,
-        ?array $headers = null,
-        ?string $location = null,
-        ?string $project = null,
-        Provider|string|null $provider = null,
-    ): self {
-        $self = new self;
-
-        $self['modelName'] = $modelName;
-
-        null !== $apiKey && $self['apiKey'] = $apiKey;
-        null !== $baseURL && $self['baseURL'] = $baseURL;
-        null !== $googleAuthOptions && $self['googleAuthOptions'] = $googleAuthOptions;
-        null !== $headers && $self['headers'] = $headers;
-        null !== $location && $self['location'] = $location;
-        null !== $project && $self['project'] = $project;
-        null !== $provider && $self['provider'] = $provider;
-
-        return $self;
-    }
-
-    /**
-     * Model name string with provider prefix (e.g., 'openai/gpt-5-nano').
-     */
-    public function withModelName(string $modelName): self
-    {
-        $self = clone $this;
-        $self['modelName'] = $modelName;
-
-        return $self;
-    }
-
-    /**
-     * API key for the model provider.
-     */
-    public function withAPIKey(string $apiKey): self
-    {
-        $self = clone $this;
-        $self['apiKey'] = $apiKey;
-
-        return $self;
-    }
-
-    /**
-     * Base URL for the model provider.
-     */
-    public function withBaseURL(string $baseURL): self
-    {
-        $self = clone $this;
-        $self['baseURL'] = $baseURL;
-
-        return $self;
-    }
-
-    /**
-     * google-auth-library options used to authenticate Vertex AI models.
-     *
-     * @param GoogleAuthOptions|GoogleAuthOptionsShape $googleAuthOptions
-     */
-    public function withGoogleAuthOptions(
-        GoogleAuthOptions|array $googleAuthOptions
-    ): self {
-        $self = clone $this;
-        $self['googleAuthOptions'] = $googleAuthOptions;
-
-        return $self;
-    }
-
-    /**
-     * Custom headers sent with every request to the model provider.
-     *
-     * @param array<string,string> $headers
-     */
-    public function withHeaders(array $headers): self
+    public static function variants(): array
     {
-        $self = clone $this;
-        $self['headers'] = $headers;
-
-        return $self;
-    }
-
-    /**
-     * Google Cloud location for Vertex AI models.
-     */
-    public function withLocation(string $location): self
-    {
-        $self = clone $this;
-        $self['location'] = $location;
-
-        return $self;
-    }
-
-    /**
-     * Google Cloud project ID for Vertex AI models.
-     */
-    public function withProject(string $project): self
-    {
-        $self = clone $this;
-        $self['project'] = $project;
-
-        return $self;
-    }
-
-    /**
-     * AI provider for the model (or provide a baseURL endpoint instead).
-     *
-     * @param Provider|value-of<Provider> $provider
-     */
-    public function withProvider(Provider|string $provider): self
-    {
-        $self = clone $this;
-        $self['provider'] = $provider;
-
-        return $self;
+        return [VertexModelConfigObject::class, GenericModelConfigObject::class];
     }
 }