I’m trying to get structured output from a model
I’m using model with response model:
class IssueDetails(BaseModel):
error_message: str
issue_details: str
error: bool
model_to_use = "us.amazon.nova-pro-v1:0"
class IssueTypeClassifier:
def __init__(self, model_to_use=model_to_use):
self.session = boto3.Session(profile_name='dev-bedrock-role', region_name='us-east-1')
self.agent = Agent(
model=AwsBedrock(id=model_to_use, session=self.session, temperature=0.0),
tools=[my_tool],
instructions=[
"... removed "
],
**response_model=IssueDetails**
)
def classify_issue(self, issue_key):
response: RunResponse = self.agent.run(f"Get issue details for issue key: {issue_key}")
dict_response = response.to_dict()
return dict_response
but i do not get IssueDetails object fron the ‘run’ method
What do i do wrong?