Kackle commited on
Commit
bb06013
·
verified ·
1 Parent(s): 0427b70

back jto nova

Browse files
Files changed (1) hide show
  1. 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("PixtralAgent initialized.")
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.titan-text-lite-v1"
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