better message

Author: chenmoneygithub

dspy/predict/predict.py

@@ -68,6 +68,13 @@ def load_state(self, state, use_legacy_loading=False):
         if use_legacy_loading:
             self._load_state_legacy(state)
             return
+        if "signature" not in state and ("signature_instructions" in state or "signature_prefix" in state):
+            # Check if the state is from a version of DSPy prior to v2.5.3.
+            raise ValueError(
+                "The saved state is from a version of DSPy prior to v2.5.3. Please use `use_legacy_loading=True` to "
+                "load the state."
+            )
+
         excluded_keys = ["signature", "extended_signature"]
         for name, value in state.items():
             # `excluded_keys` are fields that go through special handling.

dspy/primitives/module.py

@@ -152,17 +152,17 @@ def dump_state(self, save_verbose):
         print(self.named_parameters())
         return {name: param.dump_state(save_verbose) for name, param in self.named_parameters()}
 
-    def load_state(self, state):
+    def load_state(self, state, use_legacy_loading=False):
         for name, param in self.named_parameters():
-            param.load_state(state[name])
+            param.load_state(state[name], use_legacy_loading=use_legacy_loading)
 
     def save(self, path, save_field_meta=False):
         with open(path, "w") as f:
             f.write(ujson.dumps(self.dump_state(save_field_meta), indent=2))
 
-    def load(self, path):
+    def load(self, path, use_legacy_loading=False):
         with open(path) as f:
-            self.load_state(ujson.loads(f.read()))
+            self.load_state(ujson.loads(f.read()), use_legacy_loading=use_legacy_loading)
 
 
 def postprocess_parameter_name(name, value):