23 lines
684 B
Python
23 lines
684 B
Python
|
from botocore.exceptions import ClientError, NoCredentialsError, PartialCredentialsError
|
||
|
|
|
||
|
|
from neo4j_graphrag.embeddings import BedrockEmbeddings
|
||
|
|
|
||
|
|
# AWS credentials are read from environment or ~/.aws/credentials
|
||
|
|
embedder = BedrockEmbeddings(
|
||
|
|
model_id="amazon.titan-embed-text-v2:0",
|
||
|
|
dimensions=1024,
|
||
|
|
region_name="us-east-1",
|
||
|
|
)
|
||
|
|
|
||
|
|
try:
|
||
|
|
res = embedder.embed_query("my question")
|
||
|
|
print(res[:10])
|
||
|
|
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}")
|