# examples\nstrings, relatednesses = strings_ranked_by_relatedness(\"curling gold medal\", df, top_n=5)\nfor string, relatedness in zip(strings, relatednesses):\n print(f\"{relatedness=:.3f}\")\n display(string)<\/pre><\/div>\n\n\n\n\u8fd0\u884c\u7ed3\u679c\uff1a<\/p>\n\n\n\n![\"\"](\"https:\/\/www.aqwu.net\/wp\/wp-content\/uploads\/2024\/04\/\u56fe\u7247-14-1024x504.png\")
<\/figure>\n\n\n\n7. \u8be2\u95ee<\/strong><\/h2>\n\n\n\n\u901a\u8fc7\u4e0a\u9762\u7684\u641c\u7d22\u529f\u80fd\uff0c\u6211\u4eec\u73b0\u5728\u53ef\u4ee5\u81ea\u52a8\u68c0\u7d22\u76f8\u5173\u77e5\u8bc6\u5e76\u5c06\u5176\u63d2\u5165\u5230 GPT \u7684\u6d88\u606f\u4e2d\u3002<\/p>\n\n\n\n
\u4e0b\u9762\uff0c\u6211\u4eec\u5b9a\u4e49\u4e00\u4e2a\u51fd\u6570\uff1aask<\/code><\/p>\n\n\n\n\n- \u63a5\u53d7\u7528\u6237\u67e5\u8be2<\/li>\n\n\n\n
- \u641c\u7d22\u4e0e\u67e5\u8be2\u76f8\u5173\u7684\u6587\u672c<\/li>\n\n\n\n
- \u5c06\u8be5\u6587\u672c\u586b\u5145\u5230 GPT \u7684\u6d88\u606f\u4e2d<\/li>\n\n\n\n
- \u5c06\u6d88\u606f\u53d1\u9001\u7ed9 GPT<\/li>\n\n\n\n
- \u8fd4\u56de GPT \u7684\u7b54\u6848<\/li>\n<\/ul>\n\n\n\n
def num_tokens(text: str, model: str = GPT_MODEL) -> int:\n \"\"\"Return the number of tokens in a string.\"\"\"\n encoding = tiktoken.encoding_for_model(model)\n return len(encoding.encode(text))\n\n\ndef query_message(\n query: str,\n df: pd.DataFrame,\n model: str,\n token_budget: int\n) -> str:\n \"\"\"Return a message for GPT, with relevant source texts pulled from a dataframe.\"\"\"\n strings, relatednesses = strings_ranked_by_relatedness(query, df)\n introduction = 'Use the below articles on the 2026 Winter Olympics to answer the subsequent question. If the answer cannot be found in the articles, write \"I could not find an answer.\"'\n question = f\"\\n\\nQuestion: {query}\"\n message = introduction\n for string in strings:\n next_article = f'\\n\\nWikipedia article section:\\n\"\"\"\\n{string}\\n\"\"\"'\n if (\n num_tokens(message + next_article + question, model=model)\n > token_budget\n ):\n break\n else:\n message += next_article\n return message + question\n\n\ndef ask(\n query: str,\n df: pd.DataFrame = df,\n model: str = GPT_MODEL,\n token_budget: int = 4096 - 500,\n print_message: bool = False,\n) -> str:\n \"\"\"Answers a query using GPT and a dataframe of relevant texts and embeddings.\"\"\"\n message = query_message(query, df, model=model, token_budget=token_budget)\n if print_message:\n print(message)\n messages = [\n {\"role\": \"system\", \"content\": \"You answer questions about the 2026 Winter Olympics.\"},\n {\"role\": \"user\", \"content\": message},\n ]\n response = client.chat.completions.create(\n model=model,\n messages=messages,\n temperature=0\n )\n response_message = response.choices[0].message.content\n return response_message\n\n<\/pre><\/div>\n\n\n\n7.1 \u793a\u4f8b\u95ee\u9898<\/strong><\/h3>\n\n\n\n\u6700\u540e\uff0c\u8ba9\u6211\u4eec\u95ee\u6211\u4eec\u7684\u7cfb\u7edf\u5173\u4e8e\u91d1\u724c\u51b0\u58f6\u8fd0\u52a8\u5458\u7684\u539f\u59cb\u95ee\u9898\uff1a<\/p>\n\n\n\n
# 2026\u5e74\u51ac\u5b63\u5965\u8fd0\u4f1a\u4e0a\u54ea\u4e9b\u8fd0\u52a8\u5458\u8d62\u5f97\u4e86\u51b0\u58f6\u9879\u76ee\u7684\u91d1\u724c\uff1f\nask('Which athletes won the gold medal in curling at the 2026 Winter Olympics?')<\/pre><\/div>\n\n\n\n\"The athletes who won the gold medal in curling at the 2026 Winter Olympics were from Italy (Stefania Constantini and Amos Mosaner) in the mixed doubles tournament and from Great Britain (Bruce Mouat, Grant Hardie, Bobby Lammie, Hammy McMillan Jr., and Ross Whyte) in the men's tournament.\"<\/pre><\/div>\n\n\n\n\u7ffb\u8bd1\u4e3a\u4e2d\u6587\uff1a<\/p>\n\n\n\n
\u57282026\u5e74\u51ac\u5b63\u5965\u8fd0\u4f1a\u4e0a\uff0c\u8d62\u5f97\u6df7\u5408\u53cc\u6253\u51b0\u58f6\u91d1\u724c\u7684\u8fd0\u52a8\u5458\u6765\u81ea\u610f\u5927\u5229\uff08Stefania Constantini \u548c Amos Mosaner\uff09\uff0c\u800c\u8d62\u5f97\u7537\u5b50\u6bd4\u8d5b\u91d1\u724c\u7684\u8fd0\u52a8\u5458\u6765\u81ea\u82f1\u56fd\uff08Bruce Mouat, Grant Hardie, Bobby Lammie, Hammy McMillan Jr., \u4ee5\u53ca Ross Whyte\uff09\u3002<\/pre><\/div>\n\n\n\ngpt-3.5-turbo<\/code> \u5c3d\u7ba1\u5bf9 2026 \u5e74\u51ac\u5965\u4f1a\u4e00\u65e0\u6240\u77e5\uff0c\u4f46\u6211\u4eec\u7684\u641c\u7d22\u7cfb\u7edf\u80fd\u591f\u68c0\u7d22\u5230\u53c2\u8003\u6587\u672c\u4f9b\u6a21\u578b\u9605\u8bfb\uff0c\u4f7f\u5176\u80fd\u591f\u6b63\u786e\u5217\u51fa\u7537\u5b50\u548c\u5973\u5b50\u9526\u6807\u8d5b\u7684\u91d1\u724c\u83b7\u5f97\u8005\u3002<\/p>\n\n\n\n\u7136\u800c\uff0c\u5b83\u4ecd\u7136\u4e0d\u662f\u5f88\u5b8c\u7f8e\u2014\u2014\u8be5\u6a21\u578b\u672a\u80fd\u5217\u51fa\u6df7\u5408\u53cc\u6253\u9879\u76ee\u7684\u91d1\u724c\u5f97\u4e3b\u3002<\/p>\n\n\n\n
7.2 \u9519\u8bef\u7b54\u6848\u7591\u96be\u89e3\u7b54<\/strong><\/h3>\n\n\n\n\u8981\u67e5\u770b\u9519\u8bef\u662f\u7531\u4e8e\u7f3a\u5c11\u76f8\u5173\u6e90\u6587\u672c\uff08\u5373\u641c\u7d22\u6b65\u9aa4\u5931\u8d25\uff09\u8fd8\u662f\u7f3a\u4e4f\u63a8\u7406\u53ef\u9760\u6027\uff08\u5373\u8be2\u95ee\u6b65\u9aa4\u5931\u8d25\uff09\uff0c\u60a8\u53ef\u4ee5\u67e5\u770b\u6587\u672c GPT \u662f\u901a\u8fc7\u8bbe\u7f6e\u7ed9\u51fa\u7684\u3002print_message=True<\/code><\/p>\n\n\n\n\u5728\u8fd9\u79cd\u7279\u6b8a\u60c5\u51b5\u4e0b\uff0c\u67e5\u770b\u4e0b\u9762\u7684\u6587\u5b57\uff0c\u770b\u8d77\u6765\u7ed9\u6a21\u578b\u7684 #1 \u6587\u7ae0\u786e\u5b9e\u5305\u542b\u6240\u6709\u4e09\u4e2a\u9879\u76ee\u7684\u5956\u724c\u83b7\u5f97\u8005\uff0c\u4f46\u540e\u6765\u7684\u7ed3\u679c\u5f3a\u8c03\u4e86\u7537\u5b50\u548c\u5973\u5b50\u9526\u6807\u8d5b\uff0c\u8fd9\u53ef\u80fd\u4f1a\u5206\u6563\u6a21\u578b\u7684\u6ce8\u610f\u529b\uff0c\u65e0\u6cd5\u7ed9\u51fa\u66f4\u5b8c\u6574\u7684\u7b54\u6848\u3002<\/p>\n\n\n\n
# set print_message=True to see the source text GPT was working off of\n# 2026\u5e74\u51ac\u5b63\u5965\u8fd0\u4f1a\u4e0a\u54ea\u4e9b\u8fd0\u52a8\u5458\u8d62\u5f97\u4e86\u51b0\u58f6\u9879\u76ee\u7684\u91d1\u724c\uff1f\nask('Which athletes won the gold medal in curling at the 2026 Winter Olympics?', print_message=True)<\/pre><\/div>\n\n\n\n![\"\"](\"https:\/\/www.aqwu.net\/wp\/wp-content\/uploads\/2024\/04\/\u56fe\u7247-15-1024x287.png\")
<\/figure>\n\n\n\n\u77e5\u9053\u8fd9\u4e2a\u9519\u8bef\u662f\u7531\u4e8e\u8be2\u95ee\u6b65\u9aa4\u4e2d\u7684\u63a8\u7406\u4e0d\u5b8c\u5584\uff0c\u800c\u4e0d\u662f\u641c\u7d22\u6b65\u9aa4\u4e2d\u7684\u68c0\u7d22\u4e0d\u5b8c\u7f8e\u9020\u6210\u7684\uff0c\u8ba9\u6211\u4eec\u4e13\u6ce8\u4e8e\u6539\u8fdb\u8be2\u95ee\u6b65\u9aa4\u3002<\/p>\n\n\n\n
\u6539\u5584\u7ed3\u679c\u7684\u6700\u7b80\u5355\u65b9\u6cd5\u662f\u4f7f\u7528\u529f\u80fd\u66f4\u5f3a\u5927\u7684\u6a21\u578b\uff0c\u4f8b\u5982 .\u8ba9\u6211\u4eec\u8bd5\u8bd5 GPT-4<\/code> \u5427\u3002<\/p>\n\n\n\n# 2026\u5e74\u51ac\u5b63\u5965\u8fd0\u4f1a\u4e0a\u54ea\u4e9b\u8fd0\u52a8\u5458\u8d62\u5f97\u4e86\u51b0\u58f6\u9879\u76ee\u7684\u91d1\u724c\uff1f\nask('Which athletes won the gold medal in curling at the 2026 Winter Olympics?', model=\"gpt-4\")<\/pre><\/div>\n\n\n\n\"The athletes who won the gold medal in curling at the 2026 Winter Olympics are:\nMen's tournament: Niklas Edin, Oskar Eriksson, Rasmus Wran\u00e5, Christoffer Sundgren, and Daniel Magnusson from Sweden.\nWomen's tournament: Eve Muirhead, Vicky Wright, Jennifer Dodds, Hailey Duff, and Mili Smith from Great Britain.\nMixed doubles tournament: Stefania Constantini and Amos Mosaner from Italy.\"<\/pre><\/div>\n\n\n\n\u7ffb\u8bd1\u4e3a\u4e2d\u6587\uff1a<\/p>\n\n\n\n
\u57282026\u5e74\u51ac\u5b63\u5965\u8fd0\u4f1a\u4e0a\u8d62\u5f97\u51b0\u58f6\u91d1\u724c\u7684\u8fd0\u52a8\u5458\u662f\uff1a\n\n\u7537\u5b50\u6bd4\u8d5b\uff1a\u6765\u81ea\u745e\u5178\u7684Niklas Edin\u3001Oskar Eriksson\u3001Rasmus Wran\u00e5\u3001Christoffer Sundgren\u548cDaniel Magnusson\u3002\n\n\u5973\u5b50\u6bd4\u8d5b\uff1a\u6765\u81ea\u82f1\u56fd\u7684Eve Muirhead\u3001Vicky Wright\u3001Jennifer Dodds\u3001Hailey Duff\u548cMili Smith\u3002\n\n\u6df7\u5408\u53cc\u6253\u6bd4\u8d5b\uff1a\u6765\u81ea\u610f\u5927\u5229\u7684Stefania Constantini\u548cAmos Mosaner\u3002<\/pre><\/div>\n\n\n\n
![\"\"](\"https:\/\/www.aqwu.net\/wp\/wp-content\/uploads\/2024\/04\/\u56fe\u7247-13-1024x421.png\")
<\/figure>\n\n\n\n