(event_name, memory_instance, additional_data=None)
| 62 | |
| 63 | |
| 64 | def capture_event(event_name, memory_instance, additional_data=None): |
| 65 | event_data = { |
| 66 | "collection": memory_instance.collection_name, |
| 67 | "vector_size": memory_instance.embedding_model.config.embedding_dims, |
| 68 | "history_store": "sqlite", |
| 69 | "graph_store": f"{memory_instance.graph.__class__.__module__}.{memory_instance.graph.__class__.__name__}" |
| 70 | if memory_instance.config.graph_store.config |
| 71 | else None, |
| 72 | "vector_store": f"{memory_instance.vector_store.__class__.__module__}.{memory_instance.vector_store.__class__.__name__}", |
| 73 | "llm": f"{memory_instance.llm.__class__.__module__}.{memory_instance.llm.__class__.__name__}", |
| 74 | "embedding_model": f"{memory_instance.embedding_model.__class__.__module__}.{memory_instance.embedding_model.__class__.__name__}", |
| 75 | "function": f"{memory_instance.__class__.__module__}.{memory_instance.__class__.__name__}.{memory_instance.api_version}", |
| 76 | } |
| 77 | if additional_data: |
| 78 | event_data.update(additional_data) |
| 79 | |
| 80 | telemetry.capture_event(event_name, event_data) |
| 81 | |
| 82 | |
| 83 | def capture_client_event(event_name, instance, additional_data=None): |
no test coverage detected