Merge pull request #943 from xyloid/fix/ollama_copy

fix(dspy): added `copy` to `OllamaLocal` to propagate `model`, `model_type`, `base_url` and `timeout_s`.

dsp/modules/ollama.py

@@ -180,3 +180,15 @@ def __call__(
         completions = [self._get_choice_text(c) for c in choices]
 
         return completions
+
+    def copy(self, **kwargs):
+        """Returns a copy of the language model with the same parameters."""
+        kwargs = {**self.kwargs, **kwargs}
+
+        return self.__class__(
+            model=self.model_name,
+            model_type=self.model_type,
+            base_url=self.base_url,
+            timeout_s=self.timeout_s,
+            **kwargs,
+        )