Coverage for examples/CustomMarketPlugin/agentlib_plugin/custom_market.py: 100%

1from agentlib_flexquant import flexibility_market 

2from agentlib.core.datamodels import AgentVariable 

3from pydantic import Field 

4 

5class CustomMarketConfig(flexibility_market.FlexibilityMarketModuleConfig): 

6 custom_string: str = Field( 

7 title="Custom String", 

8 description="Just an example of a custom String.", 

9 ) 

10 

11class CustomMarketModule(flexibility_market.FlexibilityMarketModule): 

12 config: CustomMarketConfig 

13 

14 def custom_flexibility_callback(self, inp: AgentVariable, name: str): 

15 """Placeholder for a custom flexibility callback.""" 

16 offer = inp.value 

17 offer.status = "Custom status" 

18 self.write_results(offer=offer) 

19 

20 print("My custom string is: " + self.config.custom_string) 

21 print("Recieved offer: " + str(offer)) 

22 pass