Add ollama embeddings (#1634)

This commit is contained in:
Dev Khant
2024-08-03 10:55:40 +05:30
committed by GitHub
parent 1c46fddce1
commit 024089d33e
7 changed files with 57 additions and 16 deletions

View File

@@ -48,7 +48,7 @@ telemetry = AnonymousTelemetry(
def capture_event(event_name, memory_instance, additional_data=None):
event_data = {
"collection": memory_instance.collection_name,
"vector_size": memory_instance.embedding_model.dims,
"vector_size": memory_instance.embedding_model.config.embedding_dims,
"history_store": "sqlite",
"vector_store": f"{memory_instance.vector_store.__class__.__module__}.{memory_instance.vector_store.__class__.__name__}",
"llm": f"{memory_instance.llm.__class__.__module__}.{memory_instance.llm.__class__.__name__}",