code-repository/maubot-llmplus
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

增加流式功能

Browse Source
This commit is contained in:
taylorxie
2026-03-09 23:51:18 +08:00
parent bf4d2a444c
commit d5d634bf14
2 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
maubot_llmplus/local_paltform.py
View File

@@ -96,7 +96,7 @@ class LmStudio(Platform):
def __init__(self, config: BaseProxyConfig, http: ClientSession) -> None: def __init__(self, config: BaseProxyConfig, http: ClientSession) -> None:
super().__init__(config, http) super().__init__(config, http)
self.temperature = self.config['temperature'] self.temperature = float(self.config['temperature']) if self.config.get('temperature') is not None else None
self.streaming = self.config.get('streaming', False) self.streaming = self.config.get('streaming', False)
def is_streaming_enabled(self) -> bool: def is_streaming_enabled(self) -> bool:

22
maubot_llmplus/thrid_platform.py
View File

@@ -113,12 +113,12 @@ class Deepseek(Platform):
class OpenAi(Platform): class OpenAi(Platform):
max_tokens: int max_tokens: int
temperature: int temperature: float
def __init__(self, config: BaseProxyConfig, http: ClientSession) -> None: def __init__(self, config: BaseProxyConfig, http: ClientSession) -> None:
super().__init__(config, http) super().__init__(config, http)
self.max_tokens = self.config['max_tokens'] self.max_tokens = int(self.config['max_tokens']) if self.config.get('max_tokens') else None
self.temperature = self.config['temperature'] self.temperature = float(self.config['temperature']) if self.config.get('temperature') is not None else None
self.streaming = self.config.get('streaming', False) self.streaming = self.config.get('streaming', False)
def is_streaming_enabled(self) -> bool: def is_streaming_enabled(self) -> bool:
@@ -220,7 +220,7 @@ class Anthropic(Platform):
def __init__(self, config: BaseProxyConfig, http: ClientSession) -> None: def __init__(self, config: BaseProxyConfig, http: ClientSession) -> None:
super().__init__(config, http) super().__init__(config, http)
self.max_tokens = self.config['max_tokens'] self.max_tokens = int(self.config['max_tokens']) if self.config.get('max_tokens') else None
self.streaming = self.config.get('streaming', False) self.streaming = self.config.get('streaming', False)
def is_streaming_enabled(self) -> bool: def is_streaming_enabled(self) -> bool:
@@ -314,8 +314,8 @@ class Gemini(Platform):
def __init__(self, config: BaseProxyConfig, http: ClientSession) -> None: def __init__(self, config: BaseProxyConfig, http: ClientSession) -> None:
super().__init__(config, http) super().__init__(config, http)
self.max_tokens = self.config['max_tokens'] self.max_tokens = int(self.config['max_tokens']) if self.config.get('max_tokens') else None
self.temperature = self.config['temperature'] self.temperature = float(self.config['temperature']) if self.config.get('temperature') is not None else None
self.streaming = self.config.get('streaming', False) self.streaming = self.config.get('streaming', False)
def is_streaming_enabled(self) -> bool: def is_streaming_enabled(self) -> bool:
@@ -431,8 +431,8 @@ class XAi(Platform):
def __init__(self, config: BaseProxyConfig, http: ClientSession) -> None: def __init__(self, config: BaseProxyConfig, http: ClientSession) -> None:
super().__init__(config, http) super().__init__(config, http)
self.temperature = self.config['temperature'] self.temperature = float(self.config['temperature']) if self.config.get('temperature') is not None else None
self.max_tokens = self.config['max_tokens'] self.max_tokens = int(self.config['max_tokens']) if self.config.get('max_tokens') else None
self.streaming = self.config.get('streaming', False) self.streaming = self.config.get('streaming', False)
def is_streaming_enabled(self) -> bool: def is_streaming_enabled(self) -> bool:
@@ -526,9 +526,9 @@ class Qwen(Platform):
def __init__(self, config: BaseProxyConfig, http: ClientSession) -> None: def __init__(self, config: BaseProxyConfig, http: ClientSession) -> None:
super().__init__(config, http) super().__init__(config, http)
self.max_tokens = self.config['max_tokens'] self.max_tokens = int(self.config['max_tokens']) if self.config.get('max_tokens') else None
self.temperature = self.config['temperature'] self.temperature = float(self.config['temperature']) if self.config.get('temperature') is not None else None
self.top_p = self.config['top_p'] self.top_p = float(self.config['top_p']) if self.config.get('top_p') is not None else None
self.enable_thinking = self.config['enable_thinking'] self.enable_thinking = self.config['enable_thinking']
self.streaming = self.config.get('streaming', False) self.streaming = self.config.get('streaming', False)