(no commit message)

agent/agent.py

@@ -108,9 +108,13 @@ class PersanaSearcher(PrecompiledAgent):
     # You can optionally take in other variables specific to the environment
     # It is recommended to track any performance-altering variables (model, temperature, etc.) in the config, not the constructor
     def __init__(
-        self, config: PersanaConfig, api_key: Optional[str] = None, train: bool = False
+        self,
+        config: PersanaConfig,
+        api_key: Optional[str] = None,
+        train: bool = False,
+        **kwargs,
     ):
-        super().__init__(config)
+        super().__init__(config, **kwargs)
         self.client = PersanaClient(api_key=api_key)
         self.search_creator = dspy.Predict(ICPTOSearchSignature)
         self.search_creator.set_lm(dspy.LM(config.model))