Spaces:
Sleeping
Sleeping
back jto nova
Browse files- nova_agent.py +2 -2
nova_agent.py
CHANGED
|
@@ -9,7 +9,7 @@ load_dotenv()
|
|
| 9 |
|
| 10 |
class NovaProAgent:
|
| 11 |
def __init__(self):
|
| 12 |
-
print("
|
| 13 |
|
| 14 |
# Get AWS credentials from environment variables
|
| 15 |
aws_access_key_id = os.getenv('AWS_ACCESS_KEY_ID')
|
|
@@ -28,7 +28,7 @@ class NovaProAgent:
|
|
| 28 |
region_name='us-east-1'
|
| 29 |
)
|
| 30 |
|
| 31 |
-
self.model_id = "amazon.
|
| 32 |
self.content_type = "application/json"
|
| 33 |
self.accept = "application/json"
|
| 34 |
|
|
|
|
| 9 |
|
| 10 |
class NovaProAgent:
|
| 11 |
def __init__(self):
|
| 12 |
+
print("NovaProAgent initialized.")
|
| 13 |
|
| 14 |
# Get AWS credentials from environment variables
|
| 15 |
aws_access_key_id = os.getenv('AWS_ACCESS_KEY_ID')
|
|
|
|
| 28 |
region_name='us-east-1'
|
| 29 |
)
|
| 30 |
|
| 31 |
+
self.model_id = "amazon.nova-pro-v1:0"
|
| 32 |
self.content_type = "application/json"
|
| 33 |
self.accept = "application/json"
|
| 34 |
|