fz-mirrors/AstrBot
1
0
Fork 0
mirror of https://github.com/AstrBotDevs/AstrBot synced 2026-07-15 17:30:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: cannot use tools in siliconflow provider (#6829)

Browse Source 
* fix: cannot use tools in siliconflow provider

* fix: handle empty choices in ChatCompletionStreamState
This commit is contained in:
Soulter
2026-03-23 12:53:37 +08:00
committed by GitHub
parent ade42227e4
commit b7e8b335a7
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
astrbot/core/provider/sources/openai_source.py
View File

@@ -329,14 +329,20 @@ class ProviderOpenAIOfficial(Provider):
state = ChatCompletionStreamState()
async for chunk in stream:
try:
state.handle_chunk(chunk)
except Exception as e:
logger.warning("Saving chunk state error: " + str(e))
if not chunk.choices:
continue
choice = chunk.choices[0]
delta = choice.delta
# siliconflow workaround
if dtcs := delta.tool_calls:
for tc in dtcs:
if tc.function and tc.function.arguments:
tc.type = "function"
try:
state.handle_chunk(chunk)
except Exception as e:
logger.error("Saving chunk state error: " + str(e))
# logger.debug(f"chunk delta: {delta}")
# handle the content delta
reasoning = self._extract_reasoning_content(chunk)