amis-rpc-design/libs/amis/scripts/bot/vector_store.py

16 lines
390 B
Python
Raw Permalink Normal View History

2023-10-07 19:42:30 +08:00
import chromadb
import os
from chromadb.config import Settings
client = chromadb.Client(Settings(chroma_db_impl="duckdb+parquet",
persist_directory=os.path.join(
os.path.dirname(__file__), "db")
))
def get_client():
return client
def search(query: str) -> str:
return "d"