AI APIs

پیاده‌سازی API رتبه‌بندی مجدد Cohere برای بازیابی با دقت بالا در خطوط لوله RAG سازمانی

تولید تقویت‌شده با بازیابی (RAG) به استاندارد طلایی برای مستندسازی مدل‌های زبانی بزرگ (LLM) در داده‌های واقعی و خاص سازمان تبدیل شده است. با این حال، یک گلوگاه رایج در سیستم‌های RAG تولیدی، «شکاف کیفیت بازیابی» است. جستجوی برداری استاندارد به شباهت کسینوس برای دریافت اسناد برتر (top-k) متکی است. اگرچه این روش کارآمد است، اما این معیار اغلب در درک مرتبطی معنایی ظریف مورد نیاز برای پرسش‌های پیچیده شکست می‌خورد و منجر به آلودگی زمینه و توهم‌سازی می‌شود. اینجاست که API رتبه‌بندی مجدد Cohere درخشش خود را نشان می‌دهد.

محدودیت جستجوی برداری اولیه

وقتی یک پرسش کاربر و مجموعه اسناد شما را در یک فضای برداری ابعاد بالا جاسازی می‌کنید، همسایگان نزدیک معمولاً مرتبط هستند. اما در داده‌های سازمانی متراکم (فکر کنید به قراردادها حقوقی، سوابق پزشکی یا دستورالعمل‌های فنی)، بردار «نزدیک‌ترین» ممکن است سندی باشد که اصطلاحات مشترک دارد اما پاسخ منطقی خاص را ندارد. بازیابی اولیه برای یادآوری (recall) طراحی شده است؛ یک شبکه گسترده پرتاب می‌کند. رتبه‌بندی مجدد برای دقت (precision) طراحی شده است؛ ماهی را انتخاب می‌کند.

نحوه عملکرد مدل رتبه‌بندی مجدد Cohere

مدل رتبه‌بندی مجدد Cohere (به طور خاص rerank-english-v3.0) از معماری‌های کراس-انکودر استفاده می‌کند. برخلاف دوانکودرها (که در جستجوی اولیه استفاده می‌شوند) که پرسش و سند را به صورت مستقل جاسازی می‌کنند، کراس-انکودرها جفت پرسش و سند را به صورت مشترک پردازش می‌کنند. این امر به مدل اجازه می‌دهد تا تعاملات و وابستگی‌های سطح توکن را درک کند و امتیاز مرتبطی بسیار دقیق‌تری ارائه دهد.

استراتژی پیاده‌سازی: خط لوله دو مرحله‌ای

برای پیاده‌سازی این روش، نباید پایگاه داده برداری خود را جایگزین کنید. به جای آن، Rerank را به عنوان یک مرحله پس‌پردازش در نظر بگیرید. جریان کاری استاندارد به شرح زیر است:

  1. بازیابی: با استفاده از جستجوی شباهت استاندارد، ۵۰ تا ۱۰۰ کاندیدای برتر را از پایگاه داده برداری خود (مثلاً Pinecone، Weaviate، Milvus) دریافت کنید.
  2. رتبه‌بندی مجدد: این کاندیداها را همراه با پرسش اصلی کاربر به API رتبه‌بندی مجدد Cohere ارسال کنید.
  3. انتخاب: ۳ تا ۵ پاراگراف مرتبط برتر را بر اساس امتیازهای جدید رتبه‌بندی مجدد انتخاب کنید.
  4. تولید: زمینه انتخاب شده را برای پاسخ نهایی به LLM خود منتقل کنید.

نمونه کد: یکپارچه‌سازی Rerank با پایتون

در اینجا یک مثال عملی با استفاده از SDK پایتون cohere آورده شده است. توجه داشته باشید که باید فقط تعداد محدودی از کاندیداها (مثلاً ۲۰ تا ۵۰) را ارسال کنید تا تعادلی بین تأخیر (latency) و هزینه برقرار شود.

import cohere

client = cohere.ClientClient(api_key="YOUR_API_KEY")

# پرسش فرضی کاربر
query = "What are the SLA penalties for late data delivery?"

# ۵۰ سند برتر بازیابی شده از پایگاه داده برداری شما
candidates = [
    "Section 4.1: Delivery times must be within 24 hours...",
    "Section 4.2: Penalties of 5% per day apply for late delivery...",
    "Section 2.0: Definitions of 'Delivery' and 'Data'...",
    # ... ۴۷ سند دیگر مرتبط اما کمتر خاص
]

# فراخوانی API رتبه‌بندی مجدد
results = client.rerank(
    model="rerank-english-v3.0",
    query=query,
    documents=candidates,
    top_n=3  # ما فقط به ۳ مورد برتر برای زمینه LLM نیاز داریم
)

# آماده‌سازی زمینه برای LLM
top_relevant_docs = [result['document'] for result in results['results']]
print("Top Documents Selected:")
for doc in top_relevant_docs:
    print(f"- {doc[:100]}...")

بهترین روش‌ها برای استقرار سازمانی

  • بهینه‌سازی اندازه دسته (Batch Size): اگرچه می‌توانید صدها سند را رتبه‌بندی مجدد کنید، اما تأخیر به صورت خطی افزایش می‌یابد. برای چت‌بات‌های بلادرنگ، استخر کاندیداها را زیر ۵۰ نگه دارید.
  • کش کردن نتایج رتبه‌بندی مجدد: اگر پرسش‌های مشابه مکرر هستند، شناسه‌های اسناد رتبه‌بندی شده را در کش ذخیره کنید. این کار می‌تواند هزینه‌های API را به طور قابل توجهی کاهش دهد.
  • آستانه‌بندی (Thresholding): اگر امتیاز رتبه‌بندی مجدد برتر زیر یک آستانه خاص (مثلاً ۰.۵) باشد، به جای مجبور کردن یک زمینه با کیفیت پایین، در نظر بگیرید که از پاسخ دادن خودداری کنید یا اعلام کنید که اطلاعات یافت نشد.

نتیجه‌گیری

افزودن API رتبه‌بندی مجدد Cohere به خط لوله RAG شما یک اصلاح کم‌تلاش و پراثر است. این روش شکاف بین «مرتبط» و «مناسب» را پر می‌کند و اطمینان می‌دهد که LLM شما زمینه‌ای با وفاداری بالا دریافت می‌کند. برای کاربردهای سازمانی که دقت غیرقابل مذاکره است، رتبه‌بندی مجدد کراس-انکودر دیگر اختیاری نیست—این یک بهترین روش است.

Share: