citrus

？柑橘。

开源（分布式）向量数据库

安装

pip install citrusdb

入门

1。创建索引

 import citrusdb

# Initialize client
citrus = citrusdb . Client ()

# Create index
citrus . create_index (
  name = "example" ,
  max_elements = 1000 ,            # increases dynamically as you insert more vectors
)

2。插入元素

 ids = [ 1 , 2 , 3 ]
documents = [
  "Your time is limited, so don't waste it living someone else's life" ,
  "I'd rather be optimistic and wrong than pessimistic and right." ,
  "Running a start-up is like chewing glass and staring into the abyss."
]

citrus . add ( index = "example" , ids = ids , documents = documents )

您也可以直接通过向量嵌入。如果您像我们在此处完成的字符串列表，请确保您在环境中拥有OPENAI_API_KEY 。默认情况下，我们使用OpenAI来生成嵌入。如果您正在寻求其他提供商的支持，请联系！

3。搜索

 results = citrus . query (
    index = "example" ,
    documents = [ "What is it like to launch a startup" ],
    k = 1 ,
    include = [ "document" , "metadata" ]
)

print ( results )

您可以指定是否要返回关联的文本文档和元数据。默认情况下，仅返回ID。

请启动一个重录，看看在运行查询后获得的结果！ result将包含顶部k搜索命中的ids 。

例子

聊天w/ reving ai播客

Pokedex搜索

面临问题？

随时在此存储库上打开问题！ Discord服务器即将推出！

PS：柑橘尚未完全分发。我们要到达那里;）

---

特别感谢

Devkit-重要的开发人员工具包
DSOC 2023