mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-03 01:40:21 +08:00
- add Kiro ListAvailableModels request planning and headers - route Kiro model refresh through upstream fetch - normalize Kiro model payloads and default model metadata - remove profileArn requirement from ListAvailableModels
56 lines
2.3 KiB
Rust
56 lines
2.3 KiB
Rust
mod auth;
|
|
mod converter;
|
|
mod credentials;
|
|
mod headers;
|
|
mod policy;
|
|
mod refresh;
|
|
mod request;
|
|
mod url;
|
|
|
|
use crate::provider_types::{
|
|
ProviderApiFormatInheritance, ProviderLocalEmbeddingSupport, ProviderRuntimePolicy,
|
|
};
|
|
|
|
pub const RUNTIME_POLICY: ProviderRuntimePolicy = ProviderRuntimePolicy {
|
|
fixed_provider: true,
|
|
api_format_inheritance: ProviderApiFormatInheritance::OAuthOrConfiguredBearer,
|
|
enable_format_conversion_by_default: true,
|
|
allow_auth_channel_mismatch_by_default: true,
|
|
oauth_is_bearer_like: true,
|
|
supports_model_fetch: true,
|
|
supports_local_openai_chat_transport: false,
|
|
supports_local_same_format_transport: false,
|
|
local_embedding_support: ProviderLocalEmbeddingSupport::None,
|
|
};
|
|
|
|
pub use auth::{
|
|
build_kiro_request_auth_from_config, is_kiro_claude_messages_transport,
|
|
is_kiro_provider_transport, resolve_local_kiro_bearer_auth, resolve_local_kiro_request_auth,
|
|
supports_local_kiro_auth_prerequisites, supports_local_kiro_request_auth_resolution,
|
|
KiroBearerAuth, KiroRequestAuth, KIRO_AUTH_HEADER, PROVIDER_TYPE,
|
|
};
|
|
pub use converter::convert_claude_messages_to_conversation_state;
|
|
pub use credentials::{generate_machine_id, normalize_machine_id, KiroAuthConfig};
|
|
pub use headers::{
|
|
build_generate_assistant_headers, build_list_available_models_headers, build_mcp_headers,
|
|
AWS_EVENTSTREAM_CONTENT_TYPE, KIRO_EXTERNAL_IDP_TOKEN_TYPE, KIRO_PROFILE_ARN_HEADER,
|
|
KIRO_TOKEN_TYPE_HEADER,
|
|
};
|
|
pub use policy::{
|
|
local_kiro_request_transport_unsupported_reason_with_network,
|
|
supports_local_kiro_request_transport, supports_local_kiro_request_transport_with_network,
|
|
};
|
|
pub use refresh::KiroOAuthRefreshAdapter;
|
|
pub use request::{
|
|
apply_local_body_rules_with_request_headers, apply_local_header_rules_with_request_headers,
|
|
body_rules_are_locally_supported, build_kiro_provider_headers,
|
|
build_kiro_provider_request_body, header_rules_are_locally_supported,
|
|
supports_local_kiro_request_shape, KiroProviderHeadersInput,
|
|
};
|
|
pub use url::{
|
|
build_kiro_generate_assistant_response_url, build_kiro_list_available_models_url,
|
|
build_kiro_mcp_url, build_kiro_mcp_url_from_resolved_url, resolve_kiro_base_url,
|
|
GENERATE_ASSISTANT_RESPONSE_PATH, KIRO_ENVELOPE_NAME, LIST_AVAILABLE_MODELS_PATH, MCP_PATH,
|
|
MCP_STREAM_PATH,
|
|
};
|