Update predict.py

stanfordnlp · Jan 9, 2025 · e35a9cc · e35a9cc
1 parent 238e312
commit e35a9cc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dspy/predict/predict.py b/dspy/predict/predict.py
@@ -108,6 +108,8 @@ def forward(self, **kwargs):
 
         import dspy
         adapter = dspy.settings.adapter or dspy.ChatAdapter()
+        tools = kwawrgs.pop("tools", None)
+        config["tools"] = tools
         completions = adapter(lm, lm_kwargs=config, signature=signature, demos=demos, inputs=kwargs)
 
         pred = Prediction.from_completions(completions, signature=signature)