Skip to content

Commit d93b330

Browse files
committed
增加qwen3版本支持,增加extra_body参数,默认为不思考模式
1 parent 38b2dd3 commit d93b330

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

biz/llm/client/qwen.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ def __init__(self, api_key: str = None):
1616

1717
self.client = OpenAI(api_key=self.api_key, base_url=self.base_url)
1818
self.default_model = os.getenv("QWEN_API_MODEL", "qwen-coder-plus")
19+
self.extra_body={"enable_thinking": False}
1920

2021
def completions(self,
2122
messages: List[Dict[str, str]],
@@ -25,5 +26,6 @@ def completions(self,
2526
completion = self.client.chat.completions.create(
2627
model=model,
2728
messages=messages,
29+
extra_body=self.extra_body,
2830
)
2931
return completion.choices[0].message.content

0 commit comments

Comments
 (0)