Skip to content

Commit faa4c52

Browse files
committed
feat: modify qwen llm
1 parent d126f5d commit faa4c52

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

src/memos/llms/qwen.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,5 @@
1-
import openai
2-
31
from memos.configs.llm import QwenLLMConfig
4-
from memos.llms.base import BaseLLM
2+
from memos.llms.openai import OpenAILLM
53
from memos.llms.utils import remove_thinking_tags
64
from memos.log import get_logger
75
from memos.types import MessageList
@@ -10,12 +8,11 @@
108
logger = get_logger(__name__)
119

1210

13-
class QwenLLM(BaseLLM):
11+
class QwenLLM(OpenAILLM):
1412
"""Qwen (DashScope) LLM class via OpenAI-compatible API."""
1513

1614
def __init__(self, config: QwenLLMConfig):
17-
self.config = config
18-
self.client = openai.Client(api_key=config.api_key, base_url=config.api_base)
15+
super().__init__(config)
1916

2017
def generate(self, messages: MessageList) -> str:
2118
"""Generate a response from Qwen LLM."""

0 commit comments

Comments
 (0)