Merge pull request Ironclad#334 from wayne-chang/main

Fix display of model selection of Claude 2.1 in ChatAnthropicNode
cddigi · Jan 24, 2024 · c081e2e · c081e2e
2 parents 5a756eb + 4b52786
commit c081e2e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/packages/core/src/plugins/anthropic/nodes/ChatAnthropicNode.ts b/packages/core/src/plugins/anthropic/nodes/ChatAnthropicNode.ts
@@ -181,7 +181,12 @@ export const ChatAnthropicNodeImpl: PluginNodeImpl<ChatAnthropicNode> = {
 
   getBody(data): string {
     return dedent`
-      ${data.model === 'claude-2' ? 'Claude' : 'Claude Instant'}
+      ${
+      data.model === 'claude-2' ? 'Claude 2' :
+      data.model === 'claude-2.1' ? 'Claude 2.1' :
+      data.model.startsWith('claude-instant') ? 'Claude Instant' :
+      'Unknown Model'
+      }
       ${
         data.useTopP
           ? `Top P: ${data.useTopPInput ? '(Using Input)' : data.top_p}`