from yandex_cloud_ml_sdk import YCloudML
from yandex_cloud_ml_sdk.search_indexes import (
HybridSearchIndexType,
TextSearchIndexType,
StaticIndexChunkingStrategy,
ReciprocalRankFusionIndexCombinationStrategy,
)
sdk = YCloudML(
folder_id="ID проекта",
auth="КОД АВТОРИЗАЦИИ",
)
instruction = 'ПРОМПТ КАКОЙ_ТО'
# Пример того, как загружаются данные в поисковый индекс
# file = sdk.files.upload("all.txt")
#operation = sdk.search_indexes.create_deferred(
# [file],
# index_type=HybridSearchIndexType(
# chunking_strategy=StaticIndexChunkingStrategy(
# max_chunk_size_tokens=512,
# chunk_overlap_tokens=256,
# ),
# combination_strategy=ReciprocalRankFusionIndexCombinationStrategy(),
# ),
#)
# hybrid_index = operation.wait()
hybrid_index = sdk.search_indexes.get(ИДЕНТИФИКАТОР СОЗДАННОГО ИНДЕКСА)
hybrid_tool = sdk.tools.search_index(hybrid_index)
model = sdk.models.completions("yandexgpt", model_version="rc")
assistant = sdk.assistants.create(model, tools=[hybrid_tool], instruction=instruction)
hybrid_index_thread = sdk.threads.create()
query = "Ваш вопрос по датасету"
hybrid_index_thread.write(query)
run = assistant.run(hybrid_index_thread)
result = run.wait().message
print("\n".join(result.parts))
Post #1635
294
обещал пример RAG на яндекс-облаке
- 🔥 3