We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d126f5d commit faa4c52Copy full SHA for faa4c52
src/memos/llms/qwen.py
@@ -1,7 +1,5 @@
1
-import openai
2
-
3
from memos.configs.llm import QwenLLMConfig
4
-from memos.llms.base import BaseLLM
+from memos.llms.openai import OpenAILLM
5
from memos.llms.utils import remove_thinking_tags
6
from memos.log import get_logger
7
from memos.types import MessageList
@@ -10,12 +8,11 @@
10
8
logger = get_logger(__name__)
11
9
12
13
-class QwenLLM(BaseLLM):
+class QwenLLM(OpenAILLM):
14
"""Qwen (DashScope) LLM class via OpenAI-compatible API."""
15
16
def __init__(self, config: QwenLLMConfig):
17
- self.config = config
18
- self.client = openai.Client(api_key=config.api_key, base_url=config.api_base)
+ super().__init__(config)
19
20
def generate(self, messages: MessageList) -> str:
21
"""Generate a response from Qwen LLM."""
0 commit comments