chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,21 @@
|
||||
# Copyright (c) Microsoft. All rights reserved.
|
||||
|
||||
from pydantic import ConfigDict, Field
|
||||
|
||||
from semantic_kernel.kernel_pydantic import KernelBaseModel
|
||||
from semantic_kernel.utils.feature_stage_decorator import experimental
|
||||
|
||||
|
||||
@experimental
|
||||
class BedrockActionGroupModel(KernelBaseModel):
|
||||
"""Bedrock Action Group Model.
|
||||
|
||||
Model field definitions for the Amazon Bedrock Action Group Service:
|
||||
https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/bedrock-agent/client/create_agent_action_group.html
|
||||
"""
|
||||
|
||||
# This model_config will merge with the KernelBaseModel.model_config
|
||||
model_config = ConfigDict(extra="allow")
|
||||
|
||||
action_group_id: str = Field(..., alias="actionGroupId", description="The unique identifier of the action group.")
|
||||
action_group_name: str = Field(..., alias="actionGroupName", description="The name of the action group.")
|
||||
@@ -0,0 +1,19 @@
|
||||
# Copyright (c) Microsoft. All rights reserved.
|
||||
|
||||
from enum import Enum
|
||||
|
||||
from semantic_kernel.utils.feature_stage_decorator import experimental
|
||||
|
||||
|
||||
@experimental
|
||||
class BedrockAgentEventType(str, Enum):
|
||||
"""Bedrock Agent Event Type."""
|
||||
|
||||
# Contains the text response from the agent.
|
||||
CHUNK = "chunk"
|
||||
# Contains the trace information (reasoning process) from the agent.
|
||||
TRACE = "trace"
|
||||
# Contains the function call requests from the agent.
|
||||
RETURN_CONTROL = "returnControl"
|
||||
# Contains the files generated by the agent using the code interpreter.
|
||||
FILES = "files"
|
||||
@@ -0,0 +1,24 @@
|
||||
# Copyright (c) Microsoft. All rights reserved.
|
||||
|
||||
from pydantic import ConfigDict, Field
|
||||
|
||||
from semantic_kernel.kernel_pydantic import KernelBaseModel
|
||||
from semantic_kernel.utils.feature_stage_decorator import experimental
|
||||
|
||||
|
||||
@experimental
|
||||
class BedrockAgentModel(KernelBaseModel):
|
||||
"""Bedrock Agent Model.
|
||||
|
||||
Model field definitions for the Amazon Bedrock Agent Service:
|
||||
https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/bedrock-agent/client/create_agent.html
|
||||
"""
|
||||
|
||||
# This model_config will merge with the KernelBaseModel.model_config
|
||||
model_config = ConfigDict(extra="allow")
|
||||
|
||||
agent_id: str | None = Field(default=None, alias="agentId", description="The unique identifier of the agent.")
|
||||
agent_name: str | None = Field(default=None, alias="agentName", description="The name of the agent.")
|
||||
agent_version: str | None = Field(default=None, alias="agentVersion", description="The version of the agent.")
|
||||
foundation_model: str | None = Field(default=None, alias="foundationModel", description="The foundation model.")
|
||||
agent_status: str | None = Field(default=None, alias="agentStatus", description="The status of the agent.")
|
||||
@@ -0,0 +1,23 @@
|
||||
# Copyright (c) Microsoft. All rights reserved.
|
||||
|
||||
|
||||
from enum import Enum
|
||||
|
||||
from semantic_kernel.utils.feature_stage_decorator import experimental
|
||||
|
||||
|
||||
@experimental
|
||||
class BedrockAgentStatus(str, Enum):
|
||||
"""Bedrock Agent Status.
|
||||
|
||||
https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_PrepareAgent.html#API_agent_PrepareAgent_ResponseElements
|
||||
"""
|
||||
|
||||
CREATING = "CREATING"
|
||||
PREPARING = "PREPARING"
|
||||
PREPARED = "PREPARED"
|
||||
NOT_PREPARED = "NOT_PREPARED"
|
||||
DELETING = "DELETING"
|
||||
FAILED = "FAILED"
|
||||
VERSIONING = "VERSIONING"
|
||||
UPDATING = "UPDATING"
|
||||
Reference in New Issue
Block a user