fix: llm models

This commit is contained in:
namastex888 2024-06-14 23:38:48 +00:00
commit b376bb26be
3 changed files with 4 additions and 4 deletions

View file

@ -48,7 +48,7 @@ class AmazonBedrockComponent(LCModelComponent):
"stability.stable-diffusion-xl-v0",
"stability.stable-diffusion-xl-v1",
],
value="anthropic.claude-instant-v1",
value="anthropic.claude-3-haiku-20240307-v1:0",
),
StrInput(name="credentials_profile_name", display_name="Credentials Profile Name"),
StrInput(name="region_name", display_name="Region Name"),
@ -97,4 +97,4 @@ class AmazonBedrockComponent(LCModelComponent):
)
except Exception as e:
raise ValueError("Could not connect to AmazonBedrock API.") from e
return output
return output

View file

@ -107,4 +107,4 @@ class AnthropicModelComponent(LCModelComponent):
raise ValueError("Could not connect to Anthropic API.") from e
return output

View file

@ -116,4 +116,4 @@ class AzureChatOpenAIComponent(LCModelComponent):
except Exception as e:
raise ValueError("Could not connect to AzureOpenAI API.") from e
return output
return output