swagginty/persana-lead-gen
1
1
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

(no commit message)

Browse Source
This commit is contained in:
Ty Todd
2025-10-06 21:57:28 -07:00
parent d0dbad868e
commit c47ea4f3b3
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
agent/agent.py
View File

@@ -108,9 +108,13 @@ class PersanaSearcher(PrecompiledAgent):
# You can optionally take in other variables specific to the environment # 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 # It is recommended to track any performance-altering variables (model, temperature, etc.) in the config, not the constructor
def __init__( 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.client = PersanaClient(api_key=api_key)
self.search_creator = dspy.Predict(ICPTOSearchSignature) self.search_creator = dspy.Predict(ICPTOSearchSignature)
self.search_creator.set_lm(dspy.LM(config.model)) self.search_creator.set_lm(dspy.LM(config.model))