# Initialize ErnieBot Deep Search
search_engine = ErnieBotDeepSearch()

# Example search
query = "深度学习的应用"
documents = [
    "机器学习是人工智能的一个分支",
    "深度学习在图像识别领域取得了突破性进展",
    "自然语言处理是深度学习的重要应用领域"
]

# Perform deep search
results = search_engine.deep_search(query, documents)
ranked_results = torch.argsort(results, descending=True)

# Training example
training_batch = (
    query,
    [documents[1], documents[2]],  # positive examples
    [documents[0]]                 # negative examples
)
loss = search_engine.train_step(training_batch)