Skip to content

Commit 9c1fff9

Browse files
Merge pull request stanfordnlp#1012 from omkar-sh/bug/vllm_extra_param_fix
bug(dspy): removed extra params from kwargs sent to vllm pydantic model
2 parents 17d76b7 + ec96941 commit 9c1fff9

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

dsp/modules/hf_client.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -154,6 +154,10 @@ def _generate(self, prompt, **kwargs):
154154
messages = [{"role": "user", "content": prompt}]
155155
if system_prompt:
156156
messages.insert(0, {"role": "system", "content": system_prompt})
157+
158+
kwargs.pop("port", None)
159+
kwargs.pop("url", None)
160+
157161
payload = {
158162
"model": self.kwargs["model"],
159163
"messages": messages,
@@ -180,6 +184,9 @@ def _generate(self, prompt, **kwargs):
180184
print("Failed to parse JSON response:", response.text)
181185
raise Exception("Received invalid JSON response from server")
182186
else:
187+
kwargs.pop("port", None)
188+
kwargs.pop("url", None)
189+
183190
payload = {
184191
"model": self.kwargs["model"],
185192
"prompt": prompt,

0 commit comments

Comments
 (0)