Skip to content

Commit 27023c9

Browse files
committed
fix(payloads): 使用字典解包
1 parent 645c482 commit 27023c9

3 files changed

Lines changed: 5 additions & 10 deletions

File tree

astrbot/core/provider/sources/anthropic_source.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -515,8 +515,7 @@ async def text_chat(
515515

516516
model = model or self.get_model()
517517

518-
payloads = {"messages": new_messages, "model": model}
519-
payloads.update(kwargs)
518+
payloads = {**kwargs, "messages": new_messages, "model": model}
520519

521520
# Anthropic has a different way of handling system prompts
522521
if system_prompt:
@@ -572,8 +571,7 @@ async def text_chat_stream(
572571

573572
model = model or self.get_model()
574573

575-
payloads = {"messages": new_messages, "model": model}
576-
payloads.update(kwargs)
574+
payloads = {**kwargs, "messages": new_messages, "model": model}
577575

578576
# Anthropic has a different way of handling system prompts
579577
if system_prompt:

astrbot/core/provider/sources/gemini_source.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -757,8 +757,7 @@ async def text_chat(
757757

758758
model = model or self.get_model()
759759

760-
payloads = {"messages": context_query, "model": model}
761-
payloads.update(kwargs)
760+
payloads = {**kwargs, "messages": context_query, "model": model}
762761

763762
retry = 10
764763
keys = self.api_keys.copy()
@@ -813,8 +812,7 @@ async def text_chat_stream(
813812

814813
model = model or self.get_model()
815814

816-
payloads = {"messages": context_query, "model": model}
817-
payloads.update(kwargs)
815+
payloads = {**kwargs, "messages": context_query, "model": model}
818816

819817
retry = 10
820818
keys = self.api_keys.copy()

astrbot/core/provider/sources/openai_source.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -619,8 +619,7 @@ async def _prepare_chat_payload(
619619
context_query.extend(tcr.to_openai_messages())
620620

621621
model = model or self.get_model()
622-
payloads = {"messages": context_query, "model": model}
623-
payloads.update(kwargs)
622+
payloads = {**kwargs, "messages": context_query, "model": model}
624623

625624
self._finally_convert_payload(payloads)
626625

0 commit comments

Comments
 (0)