fix: Replace extra_headers and extra_body with extra_params in Coder, ChatSummary, and GitRepo

This commit is contained in:
Paul Gauthier 2024-09-27 13:09:43 -07:00 committed by Paul Gauthier (aider)
parent c24e947b18
commit 810aeccf94
4 changed files with 4 additions and 7 deletions

View file

@ -1409,9 +1409,7 @@ class Coder:
functions, functions,
self.stream, self.stream,
temp, temp,
extra_headers=model.extra_headers, extra_params=model.extra_params,
extra_body=model.extra_body,
max_tokens=model.max_tokens,
) )
self.chat_completion_call_hashes.append(hash_object.hexdigest()) self.chat_completion_call_hashes.append(hash_object.hexdigest())

View file

@ -109,7 +109,7 @@ class ChatSummary:
for model in self.models: for model in self.models:
try: try:
summary = simple_send_with_retries( summary = simple_send_with_retries(
model.name, summarize_messages, extra_headers=model.extra_headers model.name, summarize_messages, extra_params=model.extra_params
) )
if summary is not None: if summary is not None:
summary = prompts.summary_prefix + summary summary = prompts.summary_prefix + summary

View file

@ -193,7 +193,7 @@ class GitRepo:
if max_tokens and num_tokens > max_tokens: if max_tokens and num_tokens > max_tokens:
continue continue
commit_message = simple_send_with_retries( commit_message = simple_send_with_retries(
model.name, messages, extra_headers=model.extra_headers model.name, messages, extra_params=model.extra_params
) )
if commit_message: if commit_message:
break break

View file

@ -96,9 +96,8 @@ def simple_send_with_retries(model_name, messages, extra_params=None):
"messages": messages, "messages": messages,
"functions": None, "functions": None,
"stream": False, "stream": False,
"extra_params": extra_params,
} }
if extra_params is not None:
kwargs["extra_params"] = extra_params
_hash, response = send_completion(**kwargs) _hash, response = send_completion(**kwargs)
return response.choices[0].message.content return response.choices[0].message.content