Change the role to "user" in format function of ollama chat model (#443)

modelscope · Sep 20, 2024 · fbd0e11 · fbd0e11
1 parent 7c7220a
commit fbd0e11
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/agentscope/models/ollama_model.py b/src/agentscope/models/ollama_model.py
@@ -359,7 +359,7 @@ def format(
         system_content = "\n".join(system_content_template)
 
         system_message = {
-            "role": "system",
+            "role": "user",
             "content": system_content,
         }
 

diff --git a/tests/format_test.py b/tests/format_test.py
@@ -283,7 +283,7 @@ def test_ollama_chat(self) -> None:
         # correct format
         ground_truth = [
             {
-                "role": "system",
+                "role": "user",
                 "content": (
                     "You are a helpful assistant\n"
                     "\n"