소스 검색

system role remove

resphinas 2 년 전
부모
커밋
a5a825e439
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      bot/claude/claude_ai_bot.py

+ 1 - 1
bot/claude/claude_ai_bot.py

@@ -141,7 +141,7 @@ class ClaudeAIBot(Bot, OpenAIImage):
             model = conf().get("model") or "gpt-3.5-turbo"
             # remove system message
             if session.messages[0].get("role") == "system":
-                if model == "wenxin":
+                if model == "wenxin" or model == "claude":
                     session.messages.pop(0)
             logger.info(f"[CLAUDEAI] query={query}")