pip install citrusdb import citrusdb
# Initialize client
citrus = citrusdb . Client ()
# Create index
citrus . create_index (
name = "example" ,
max_elements = 1000 , # increases dynamically as you insert more vectors
) 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 для генерации вторжений. Пожалуйста, обратитесь, если вы ищете поддержку от другого провайдера!
results = citrus . query (
index = "example" ,
documents = [ "What is it like to launch a startup" ],
k = 1 ,
include = [ "document" , "metadata" ]
)
print ( results )Вы можете указать, хотите ли вы возвращать связанный текстовый документ и метаданные. По умолчанию возвращаются только идентификаторы.
Запустите реплику на применении и посмотрите, какой результат вы получите после запуска запроса! result будет содержать ids лучших k поисковых хитов.
чат с подкастом Replit AI
Покедекс поиск
Не стесняйтесь открывать проблемы в этом репозитории! Discord Server скоро появится!
PS: цитрусовые еще не полностью распределены. Мы добираемся туда;)
Особое спасибо
Devkit - Essential Developer Toolkit
DSOC 2023