适配新的revtongyi，添加天工ai支持，进行一些细节修改适配maestro项目

free_one_api/entities/channel.py

@@ -80,7 +80,7 @@ def count_tokens(
         num_tokens = 0
         for message in messages:
             for key, value in message.items():
-                num_tokens += len(encoding.encode(value))
+                num_tokens += len(encoding.encode(str(value)))
         num_tokens += 3  # every reply is primed with <|start|>assistant<|message|>
         return num_tokens
 

free_one_api/impls/adapter/qianwen.py

@@ -15,19 +15,33 @@
 
 @adapter.llm_adapter
 class QianWenAdapter(llm.LLMLibAdapter):
-    
+
     @classmethod
     def name(cls) -> str:
         return "xw5xr6/revTongYi"
-    
+
     @classmethod
     def description(self) -> str:
         return "Use leeeduke/revTongYi to access Aliyun TongYi QianWen."
 
     def supported_models(self) -> list[str]:
         return [
             "gpt-3.5-turbo",
-            "gpt-4"
+            "gpt-4",
+            "gpt-4-1106-preview",
+            "gpt-4-vision-preview",
+            "gpt-4",
+            "gpt-4-0314",
+            "gpt-4-0613",
+            "gpt-4-32k",
+            "gpt-4-32k-0314",
+            "gpt-4-32k-0613",
+            "gpt-3.5-turbo-1106",
+            "gpt-3.5-turbo",
+            "gpt-3.5-turbo-16k",
+            "gpt-3.5-turbo-0301",
+            "gpt-3.5-turbo-0613",
+            "gpt-3.5-turbo-16k-0613",
         ]
 
     def function_call_supported(self) -> bool:
@@ -38,84 +52,83 @@ def stream_mode_supported(self) -> bool:
 
     def multi_round_supported(self) -> bool:
         return True
-    
+
     @classmethod
     def config_comment(cls) -> str:
         return \
-"""RevTongYi use cookies that can be extracted from https://qianwen.aliyun.com/
-You should provide cookie string as `cookie` in config:
-{
-    "cookie": "your cookie string"
-}
-
-Method of getting cookie string, please refer to https://github.com/leeeduke/revTongYi
-"""
+            """RevTongYi use cookies that can be extracted from https://qianwen.aliyun.com/
+            You should provide cookie string as `cookie` in config:
+            {
+                "cookie": "your cookie string"
+            }
+            
+            Method of getting cookie string, please refer to https://github.com/leeeduke/revTongYi
+            """
 
     @classmethod
     def supported_path(cls) -> str:
         return "/v1/chat/completions"
-    
+
     chatbot: qwen.Chatbot
-    
+
     def __init__(self, config: dict, eval: evaluation.AbsChannelEvaluation):
         self.config = config
         self.eval = eval
         self.chatbot = qwen.Chatbot(
             cookies_str=config['cookie']
         )
-        
+
     async def test(self) -> typing.Union[bool, str]:
         try:
             # self.chatbot.create_session("Hello, reply 'hi' only.")
             self.chatbot.sessionId = ""
             resp = self.chatbot.ask(
-                "Hello, reply 'hi' only.",
-                sessionId=""
+                "Hello, reply 'hi' only."
             )
-
-            self.chatbot.delete_session(resp.sessionId)
-            
+            print(resp)
+            self.chatbot.delete_session(resp['sessionId'])
+
             return True, ""
         except Exception as e:
             traceback.print_exc()
             return False, str(e)
-        
+
     async def query(self, req: request.Request) -> typing.AsyncGenerator[response.Response, None]:
         prompt = ""
-        
+
         for msg in req.messages:
             prompt += f"{msg['role']}: {msg['content']}\n"
-        
+
         prompt += "assistant: "
-        
+
         random_int = random.randint(0, 1000000000)
-        
+
         prev_text = ""
 
         sessionId = ""
 
         self.chatbot.sessionId = ""
-        
+
         for resp in self.chatbot.ask(
-            prompt=prompt,
-            sessionId="",
-            stream=True,
+                prompt=prompt,
+                # sessionId="",
+                stream=True,
         ):
-            if resp.contents == None or len(resp.contents) == 0:
+            if resp['contents'] == None or len(resp['contents']) == 0:
                 continue
 
-            sessionId = resp.sessionId
-            
+            sessionId = resp['sessionId']
+
             yield response.Response(
                 id=random_int,
                 finish_reason=response.FinishReason.NULL,
-                normal_message=resp.contents[0].content.replace(prev_text, ""),
+                normal_message=resp['contents'][0]['content'].replace(prev_text, ""),
                 function_call=None
             )
-            prev_text = resp.contents[0].content
-        
+            prev_text = resp['contents'][0]['content']
+
         self.chatbot.delete_session(sessionId)
-        
+
         yield response.Response(
             id=random_int,
             finish_reason=response.FinishReason.STOP,