Skip to content

Commit f44f65a

Browse files
fix import
1 parent 8e87354 commit f44f65a

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

engine/clients/client_factory.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,10 +25,13 @@ def _import_engine_classes(engine_name: str) -> Dict[str, Type]:
2525
if engine_name in _engine_classes:
2626
return _engine_classes[engine_name]
2727

28-
# Handle special case for vectorsets which uses redis prefix
28+
# Handle special cases for engine naming
2929
if engine_name == "vectorsets":
3030
module_name = f"engine.clients.vectorsets"
3131
class_prefix = "RedisVset"
32+
elif engine_name == "pgvector":
33+
module_name = f"engine.clients.pgvector"
34+
class_prefix = "PgVector"
3235
else:
3336
module_name = f"engine.clients.{engine_name}"
3437
# Convert first letter to uppercase for class name

0 commit comments

Comments
 (0)