redhairedshanks1 commited on
Commit
e56aeb3
·
1 Parent(s): e2c0c6a

Update services/pipeline_executor.py

Browse files
Files changed (1) hide show
  1. services/pipeline_executor.py +2 -2
services/pipeline_executor.py CHANGED
@@ -38,7 +38,7 @@ def execute_pipeline_bedrock(
38
 
39
  try:
40
  llm = ChatBedrock(
41
- model_id=os.getenv("BEDROCK_MODEL", "anthropic.claude-3-5-sonnet-20241022-v2:0"),
42
  region_name=os.getenv("AWS_REGION", "us-east-1"),
43
  temperature=0.0,
44
  )
@@ -109,7 +109,7 @@ def execute_pipeline_bedrock_streaming(
109
 
110
  try:
111
  llm = ChatBedrock(
112
- model_id=os.getenv("BEDROCK_MODEL", "anthropic.claude-3-5-sonnet-20241022-v2:0"),
113
  region_name=os.getenv("AWS_REGION", "us-east-1"),
114
  temperature=0.0,
115
  )
 
38
 
39
  try:
40
  llm = ChatBedrock(
41
+ model_id="mistral.mistral-large-2402-v1:0",
42
  region_name=os.getenv("AWS_REGION", "us-east-1"),
43
  temperature=0.0,
44
  )
 
109
 
110
  try:
111
  llm = ChatBedrock(
112
+ model_id="mistral.mistral-large-2402-v1:0",
113
  region_name=os.getenv("AWS_REGION", "us-east-1"),
114
  temperature=0.0,
115
  )