Skip to content

Commit ddd747d

Browse files
Fix ModelConfig attribute access in vectorstore factory
- Handle both dict (for tests with mocked config) and ModelConfig (actual config) - Fixes AttributeError: 'ModelConfig' object has no attribute 'get' - Resolves test failures in test_vectorstore_factory_integration.py Co-authored-by: Andrew Brookins <[email protected]>
1 parent b943664 commit ddd747d

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

agent_memory_server/vectorstore_factory.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,15 @@ def create_embeddings() -> Embeddings:
4646
An Embeddings instance
4747
"""
4848
embedding_config = settings.embedding_model_config
49-
provider = embedding_config.get("provider", "openai")
49+
if embedding_config:
50+
# Handle both dict (for tests with mocked config) and ModelConfig (actual config)
51+
provider = (
52+
embedding_config.get("provider", "openai")
53+
if hasattr(embedding_config, 'get')
54+
else embedding_config.provider
55+
)
56+
else:
57+
provider = "openai"
5058

5159
if provider == "openai":
5260
try:

0 commit comments

Comments
 (0)