BruceSun
/
MaiBot
mirror of https://github.com/Mai-with-u/MaiBot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

请求中增加extra_params功能

pull/1154/head
UnCLAS-Prommer 2025-07-31 01:08:35 +08:00
parent 42a33a406e
commit 483c8fb547
No known key found for this signature in database
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/llm_models/utils_model.py
View File

@ -253,10 +253,15 @@ class LLMRequest:
response_format=response_format, response_format=response_format,
stream_response_handler=stream_response_handler, stream_response_handler=stream_response_handler,
async_response_parser=async_response_parser, async_response_parser=async_response_parser,
extra_params=model_info.extra_params,
) )
elif request_type == RequestType.EMBEDDING: elif request_type == RequestType.EMBEDDING:
assert embedding_input, "embedding_input cannot be empty for embedding requests" assert embedding_input, "embedding_input cannot be empty for embedding requests"
return await client.get_embedding(model_info=model_info, embedding_input=embedding_input) return await client.get_embedding(
model_info=model_info,
embedding_input=embedding_input,
extra_params=model_info.extra_params,
)
except Exception as e: except Exception as e:
logger.debug(f"请求失败: {str(e)}") logger.debug(f"请求失败: {str(e)}")
# 处理异常 # 处理异常