redhairedshanks1 commited on
Commit
79998e9
·
1 Parent(s): a140d98

Update services/pipeline_generator.py

Browse files
Files changed (1) hide show
  1. services/pipeline_generator.py +1 -2
services/pipeline_generator.py CHANGED
@@ -60,8 +60,7 @@ def generate_pipeline_bedrock(user_input: str, file_path: Optional[str] = None)
60
  try:
61
  llm = ChatBedrock(
62
  model_id="mistral.mistral-large-2402-v1:0",
63
- region_name=os.getenv("AWS_REGION", "us-east-1"),
64
- temperature=0.0,
65
  )
66
 
67
  prompt = ChatPromptTemplate.from_messages([
 
60
  try:
61
  llm = ChatBedrock(
62
  model_id="mistral.mistral-large-2402-v1:0",
63
+ region_name=os.getenv("AWS_REGION", "us-east-1")
 
64
  )
65
 
66
  prompt = ChatPromptTemplate.from_messages([