feat: Add Cohere Command R / R+ model support (#3333)

2024-04-11 01:22:55 +08:00
parent bf63a43bda
commit 826c422ac4
16 changed files with 404 additions and 128 deletions
--- a/api/core/prompt/simple_prompt_transform.py
+++ b/api/core/prompt/simple_prompt_transform.py
@@ -232,8 +232,8 @@ class SimplePromptTransform(PromptTransform):
                    )
                ),
                max_token_limit=rest_tokens,
-                ai_prefix=prompt_rules['human_prefix'] if 'human_prefix' in prompt_rules else 'Human',
-                human_prefix=prompt_rules['assistant_prefix'] if 'assistant_prefix' in prompt_rules else 'Assistant'
+                human_prefix=prompt_rules['human_prefix'] if 'human_prefix' in prompt_rules else 'Human',
+                ai_prefix=prompt_rules['assistant_prefix'] if 'assistant_prefix' in prompt_rules else 'Assistant'
            )

            # get prompt