Spaces:
Sleeping
Sleeping
Commit ·
79998e9
1
Parent(s): a140d98
Update services/pipeline_generator.py
Browse files
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([
|