We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent afad96d commit 2e1969dCopy full SHA for 2e1969d
pyproject.toml
@@ -27,6 +27,7 @@ dependencies = [
27
"numba>=0.60.0",
28
"nanoid>=2.0.0",
29
"mcp>=1.6.0",
30
+ "sentence-transformers>=2.5.0",
31
]
32
33
[project.optional-dependencies]
redis_memory_server/extraction.py
@@ -28,7 +28,10 @@ def get_topic_model() -> BERTopic:
"""
global _topic_model
if _topic_model is None:
- _topic_model = BERTopic.load(settings.topic_model)
+ # TODO: Expose this as a config option
+ _topic_model = BERTopic.load(
+ settings.topic_model, embedding_model="all-MiniLM-L6-v2"
34
+ )
35
return _topic_model # type: ignore
36
37
0 commit comments