fix signature input type (#8413)

Author: TomeHirata

dspy/predict/chain_of_thought.py

@@ -11,7 +11,7 @@
 class ChainOfThought(Module):
     def __init__(
         self,
-        signature: Type[Signature],
+        signature: Union[str, Type[Signature]],
         rationale_field: Optional[Union[OutputField, FieldInfo]] = None,
         rationale_field_type: Type = str,
         **config: dict[str, Any],

dspy/predict/code_act.py

@@ -1,14 +1,13 @@
 import inspect
 import logging
-from inspect import Signature
 from typing import Callable, Optional, Type, Union
 
 import dspy
 from dspy.adapters.types.tool import Tool
 from dspy.predict.program_of_thought import ProgramOfThought
 from dspy.predict.react import ReAct
 from dspy.primitives.python_interpreter import PythonInterpreter
-from dspy.signatures.signature import ensure_signature
+from dspy.signatures.signature import Signature, ensure_signature
 
 logger = logging.getLogger(__name__)
 

dspy/predict/predict.py

@@ -1,5 +1,6 @@
 import logging
 import random
+from typing import Optional, Type, Union
 
 from pydantic import BaseModel
 
@@ -10,13 +11,14 @@
 from dspy.predict.parameter import Parameter
 from dspy.primitives.module import Module
 from dspy.primitives.prediction import Prediction
-from dspy.signatures.signature import ensure_signature
+from dspy.signatures.signature import Signature, ensure_signature
+from dspy.utils.callback import BaseCallback
 
 logger = logging.getLogger(__name__)
 
 
 class Predict(Module, Parameter):
-    def __init__(self, signature, callbacks=None, **config):
+    def __init__(self, signature: Union[str, Type[Signature]], callbacks: Optional[list[BaseCallback]] = None, **config):
         super().__init__(callbacks=callbacks)
         self.stage = random.randbytes(8).hex()
         self.signature = ensure_signature(signature)