Get agent info including MemRL specific details.
(self)
| 974 | self._init_memrl() |
| 975 | |
| 976 | def get_info(self) -> Dict[str, Any]: |
| 977 | """Get agent info including MemRL specific details.""" |
| 978 | info = super().get_info() |
| 979 | |
| 980 | # Add MemRL specific info |
| 981 | info.update({ |
| 982 | "memrl_config": { |
| 983 | "retrieve_num": self.retrieve_num, |
| 984 | "candidate_top_k": self.candidate_top_k, |
| 985 | "similarity_threshold": self.similarity_threshold, |
| 986 | "epsilon": self.epsilon, |
| 987 | "learning_rate": self.learning_rate, |
| 988 | "initial_q": self.initial_q, |
| 989 | "weight_sim": self.weight_sim, |
| 990 | "weight_q": self.weight_q, |
| 991 | }, |
| 992 | "strategy": { |
| 993 | "build": self.build_strategy, |
| 994 | "retrieve": self.retrieve_strategy, |
| 995 | "update": self.update_strategy, |
| 996 | }, |
| 997 | "embedding": { |
| 998 | "model": self.embedding_model, |
| 999 | "provider": self.embedding_provider, |
| 1000 | }, |
| 1001 | }) |
| 1002 | |
| 1003 | # Add LLM stats if available |
| 1004 | if self._tracked_llm: |
| 1005 | info["memrl_llm_stats"] = self._tracked_llm.get_stats() |
| 1006 | |
| 1007 | return info |