Merge pull request #522 from insop/insop/issue_521

BootstrapFewShot failing due to lm.copy for AzureOpenAI, #521
stanfordnlp · Mar 2, 2024 · 5ecc042 · 5ecc042
2 parents cc9f4f3 + 5cdd7eb
commit 5ecc042
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/dsp/modules/azure_openai.py b/dsp/modules/azure_openai.py
@@ -107,6 +107,9 @@ def __init__(
             kwargs["model"] = model
 
         self.kwargs = {
+            "api_base": api_base,
+            "api_version": api_version,
+            "api_key": api_key,
             "temperature": 0.0,
             "max_tokens": 150,
             "top_p": 1,

diff --git a/dsp/modules/lm.py b/dsp/modules/lm.py
@@ -101,4 +101,4 @@ def copy(self, **kwargs):
         kwargs = {**self.kwargs, **kwargs}
         model = kwargs.pop('model')
 
-        return self.__class__(model, **kwargs)
+        return self.__class__(model=model, **kwargs)