我想要一天分享一點「LLM從底層堆疊的技術」,並且每篇文章長度控制在三分鐘以內,讓大家不會壓力太大,但是又能夠每天成長一點。
延續 AI說書 - 從0開始 - 218 | OpenAI GPT 4 & RAG 安裝完相關依賴,今天來撰寫一個函數,此函數透過使用使用者請求中的關鍵字來選擇 URL,以下程式碼根據在使用者請求中偵測到的關鍵字選擇知識庫中的 URL:
def select_urls_based_on_query(user_query):
climate_urls = ["https://en.wikipedia.org/wiki/Climate_change",
"https://en.wikipedia.org/wiki/Effects_of_climate_change"]
rag_urls = ["https://en.wikipedia.org/wiki/Large_language_model",
"https://huggingface.co/blog/ray-rag"]
if "climate" in user_query.lower():
return climate_urls
elif "RAG" in user_query:
return rag_urls
return []