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-v0",
"stability.stable-diffusion-xl-v1", "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="credentials_profile_name", display_name="Credentials Profile Name"),
StrInput(name="region_name", display_name="Region Name"), StrInput(name="region_name", display_name="Region Name"),
@ -97,4 +97,4 @@ class AmazonBedrockComponent(LCModelComponent):
) )
except Exception as e: except Exception as e:
raise ValueError("Could not connect to AmazonBedrock API.") from 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 raise ValueError("Could not connect to Anthropic API.") from e
return output return output

View file

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