(self, messages, user_id, timestamp)
| 24 | self.client = Zep(api_key=api_key) |
| 25 | |
| 26 | def add(self, messages, user_id, timestamp): |
| 27 | iso_date = datetime.fromtimestamp(timestamp).isoformat() |
| 28 | for msg in messages: |
| 29 | self.client.graph.add( |
| 30 | data=msg.get("role") + ": " + msg.get("content"), |
| 31 | type="message", |
| 32 | created_at=iso_date, |
| 33 | group_id=user_id, |
| 34 | ) |
| 35 | |
| 36 | def search(self, query, user_id, top_k): |
| 37 | search_results = ( |
no test coverage detected