mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 11:29:39 +08:00
19 lines
490 B
Python
19 lines
490 B
Python
from __future__ import annotations
|
|
|
|
from ..core import app
|
|
from .vdb import VectorDatabase
|
|
from .vdbs.chroma import ChromaVectorDatabase
|
|
|
|
|
|
class VectorDBManager:
|
|
ap: app.Application
|
|
vector_db: VectorDatabase = None
|
|
|
|
def __init__(self, ap: app.Application):
|
|
self.ap = ap
|
|
|
|
async def initialize(self):
|
|
# 初始化 Chroma 向量数据库（可扩展为多种实现）
|
|
if self.vector_db is None:
|
|
self.vector_db = ChromaVectorDatabase(self.ap)
|