kaxil commented on code in PR #62816:
URL: https://github.com/apache/airflow/pull/62816#discussion_r2914866165
##########
providers/common/ai/provider.yaml:
##########
@@ -73,6 +73,176 @@ connection-types:
type:
- string
- 'null'
+ - hook-class-name:
airflow.providers.common.ai.hooks.pydantic_ai.PydanticAIAzureHook
+ connection-type: pydanticai-azure
+ ui-field-behaviour:
+ hidden-fields:
+ - schema
+ - port
+ - login
+ relabeling:
+ password: API Key
+ host: Azure Endpoint
+ placeholders:
+ host: "https://<resource>.openai.azure.com"
+ conn-fields:
+ model:
+ label: Model
+ description: "Azure model identifier (e.g. azure:gpt-4o)"
+ schema:
+ type:
+ - string
+ - 'null'
+ api_version:
+ label: API Version
+ description: "Azure OpenAI API version (e.g. 2024-07-01-preview).
Falls back to OPENAI_API_VERSION."
+ schema:
+ type:
+ - string
+ - 'null'
+ - hook-class-name:
airflow.providers.common.ai.hooks.pydantic_ai.PydanticAIBedrockHook
+ connection-type: pydanticai-bedrock
+ ui-field-behaviour:
+ hidden-fields:
+ - schema
+ - port
+ - login
+ - host
+ - password
+ relabeling: {}
+ placeholders: {}
+ conn-fields:
+ model:
+ label: Model
+ description: "Bedrock model identifier (e.g.
bedrock:us.anthropic.claude-opus-4-5)"
+ schema:
+ type:
+ - string
+ - 'null'
+ region_name:
+ label: AWS Region
+ description: "AWS region (e.g. us-east-1). Falls back to
AWS_DEFAULT_REGION env var."
+ schema:
+ type:
+ - string
+ - 'null'
+ aws_access_key_id:
+ label: AWS Access Key ID
+ description: "IAM access key. Leave empty to use instance role /
environment credential chain."
+ schema:
+ type:
+ - string
+ - 'null'
+ aws_secret_access_key:
+ label: AWS Secret Access Key
+ description: "IAM secret key."
+ schema:
+ type:
+ - string
+ - 'null'
+ aws_session_token:
+ label: AWS Session Token
+ description: "Temporary session token (optional)."
+ schema:
+ type:
+ - string
+ - 'null'
+ profile_name:
+ label: AWS Profile Name
+ description: "Named AWS credentials profile (optional)."
+ schema:
+ type:
+ - string
+ - 'null'
+ api_key:
+ label: Bearer Token
+ description: "AWS bearer token (alt. to IAM key/secret). Falls back to
AWS_BEARER_TOKEN_BEDROCK."
+ schema:
+ type:
+ - string
+ - 'null'
+ base_url:
+ label: Custom Endpoint URL
+ description: "Override the Bedrock runtime endpoint URL (optional)."
+ schema:
+ type:
+ - string
+ - 'null'
+ aws_read_timeout:
+ label: Read Timeout (s)
+ description: "boto3 read timeout in seconds (float, optional)."
+ schema:
+ type:
+ - number
+ - 'null'
+ aws_connect_timeout:
+ label: Connect Timeout (s)
+ description: "boto3 connect timeout in seconds (float, optional)."
+ schema:
+ type:
+ - number
+ - 'null'
+ - hook-class-name:
airflow.providers.common.ai.hooks.pydantic_ai.PydanticAIVertexHook
+ connection-type: pydanticai-vertex
+ ui-field-behaviour:
+ hidden-fields:
+ - schema
+ - port
+ - login
+ - host
+ - password
+ relabeling: {}
+ placeholders: {}
+ conn-fields:
+ model:
+ label: Model
+ description: "Google model identifier (e.g.
google-vertex:gemini-2.0-flash)"
+ schema:
+ type:
+ - string
+ - 'null'
+ project:
+ label: GCP Project
+ description: "Google Cloud project ID. Falls back to
GOOGLE_CLOUD_PROJECT env var."
+ schema:
+ type:
+ - string
+ - 'null'
+ location:
+ label: Location / Region
+ description: "Vertex AI region (e.g. us-central1). Falls back to
GOOGLE_CLOUD_LOCATION env var."
+ schema:
+ type:
+ - string
+ - 'null'
+ vertexai:
+ label: Force Vertex AI Mode
+ description: "Force Vertex AI mode. Auto-detected when
project/location/credentials are set."
+ schema:
+ type:
+ - boolean
+ - 'null'
+ api_key:
+ label: API Key
+ description: "Google API key for Gen Language API or Vertex AI. Falls
back to GOOGLE_API_KEY."
+ schema:
+ type:
+ - string
+ - 'null'
+ service_account_info:
+ label: Service Account Info
+ description: "Service account key as inline dict. Mutually exclusive
with service_account_file."
Review Comment:
Nit: "Mutually exclusive with service_account_file" — we removed
`service_account_file` a few rounds ago, so this reference is stale. Same in
`get_provider_info.py`.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
