23 lines
706 B
Python
23 lines
706 B
Python
from botocore.exceptions import ClientError, NoCredentialsError, PartialCredentialsError
|
|
|
|
from neo4j_graphrag.llm import BedrockLLM
|
|
|
|
# AWS credentials are read from environment or ~/.aws/credentials
|
|
llm = BedrockLLM(
|
|
model_name="us.anthropic.claude-sonnet-4-20250514-v1:0",
|
|
model_params={"temperature": 0.7, "maxTokens": 1024},
|
|
region_name="us-east-1",
|
|
)
|
|
|
|
try:
|
|
res = llm.invoke("say something")
|
|
print(res.content)
|
|
except NoCredentialsError:
|
|
print(
|
|
"AWS credentials not found. Run 'aws configure' or set environment variables."
|
|
)
|
|
except PartialCredentialsError as e:
|
|
print(f"Incomplete AWS credentials: {e}")
|
|
except ClientError as e:
|
|
print(f"AWS API error: {e}")
|