From 92359fb6b9d7fcea92a1f25e57061989585b001d Mon Sep 17 00:00:00 2001
From: liunux4odoo <liunux@qq.com>
Date: Wed, 20 Sep 2023 23:02:43 +0800
Subject: [PATCH] remove depends on volcengine

---
 server/model_workers/fangzhou.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/server/model_workers/fangzhou.py b/server/model_workers/fangzhou.py
index f2865b7..33a6b7d 100644
--- a/server/model_workers/fangzhou.py
+++ b/server/model_workers/fangzhou.py
@@ -74,12 +74,11 @@ class FangZhouWorker(ApiModelWorker):
         self.api_key = config.get("api_key")
         self.secret_key = config.get("secret_key")
 
-        from volcengine.maas import ChatRole
         self.conv = conv.Conversation(
             name=self.model_names[0],
             system_message="你是一个聪明、对人类有帮助的人工智能，你可以对人类提出的问题给出有用、详细、礼貌的回答。",
             messages=[],
-            roles=[ChatRole.USER, ChatRole.ASSISTANT, ChatRole.SYSTEM],
+            roles=["user", "assistant", "system"],
             sep="\n### ",
             stop_str="###",
         )