AI說書 - 從0開始 - 219 | GPT 4 & RAG 關鍵字偵測函數

我想要一天分享一點「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 []