专栏名称: 阿里云云原生
发布云原生技术最新资讯、汇集云原生技术最全内容,定期举办云原生活动、直播,阿里产品及用户最佳实践发布。与你并肩探索云原生技术点滴,分享你需要的云原生内容。
目录
相关文章推荐
纪念币预约信息  ·  开始预约!一二五分纸币!10张9.9包邮! ·  4 小时前  
纪念币预约信息  ·  开始预约!国色天香彩色银币,龙头币! ·  3 天前  
纪念币预约信息  ·  下一站!马年纪念币、马年纪念钞! ·  3 天前  
纪念币发行信息  ·  重磅!2025新版100元纸钞发行!开始预约! ·  3 天前  
纪念币发行信息  ·  重磅!2025新版100元纸钞发行!开始预约! ·  3 天前  
51好读  ›  专栏  ›  阿里云云原生

AI 网关基于 IP 地理位置,增强 Prompt 修饰能力

阿里云云原生  · 公众号  ·  · 2024-08-23 16:08

正文

作者:沈鑫糠,来自昆仑数智瑞道云团队,专注于云原生领域产品研发和相关技术。
01

前言

Cloud Native


什么是 Prompt Engineering

提示词工程(Prompt Engineering),也被称为上下文提示(In-Context Prompting),指的是通过结构化文本等方式来完善提示词,引导 LLM 输出我们期望的结果。通过提示词工程可以在不更新模型权重的情况下,让 LLM 完成不同类型的任务。面对同样一个问题,为 LLM 设置不同的提示词,能够得到不同的回答结果。我们可以通过对 Prompt 的优化提升 LLM 的回答精度。

例如:


什么是 AI Gateway
AI Gateway 的定义是 AI Native 的 API Gateway,是基于 API Gateway 的能⼒来满⾜ AI Native 的需求。例如:

  • 将传统的 QPS 限流扩展到 Token 限流。

  • 将传统的负载均衡/重试/fallback 能力延伸,支持对接多个大模型厂商 API,提高整体稳定性。

  • 扩展可观测能力,支持不同模型之间效果的对比 A/B Test,以及对话上下⽂链路 Tracing 等。

Higress (https://github.com/alibaba/higress) 是阿⾥云开源的⼀款 AI Gateway,为开发者提供了一站式的 AI 插件集和增强后端模型调度处理能力,使得 AI 与网关的集成更加便捷和高效。官方提供了丰富的插件库,涵盖 AI、流量管理、安全防护等常用功能,满足 90% 以上的业务场景需求。此外还支持 Wasm 插件扩展,支持多语言编写 Wasm 插件,插件更新采用热插拔机制对流量无损。

本文主要基于 Higress 的 Wasm 插件机制,使用 Go 进行插件的整体开发和能力增强。具体可以参考: https://higress.io/docs/latest/plugins/custom/

02

Geo-IP 插件与 AI 提示词

修饰插件的结合与使用

Cloud Native


插件参数配置

配置 Geo-IP 插件

启用 Geo-IP 插件(目前仅支持 IPv4),并配置:
BashipProtocal: "ipv4"
配置 AI 提示词修饰插件

启用 AI 提示词修饰插件,在配置的 content 中通过变量,注入地理位置信息:
Bashprepend:- content: "提问用户当前的地理位置信息是,国家:${geo-country},省份:${geo-province}, 城市:${geo-city}"  role: "system"append:- content: "每次回答完问题,尝试进行反问"  role: "user"

插件效果
示例请求⼀:
Bashcurl 'http://<这里换成网关公网IP>/api/openai/v1/chat/completions' \-H 'Accept: application/json, text/event-stream' \-H 'Content-Type: application/json' \--data-raw '{  "model":"qwen-long",  "frequency_penalty":0,  "max_tokens":800,  "stream":false,  "presence_penalty":0,  "temperature":0.7,  "top_p":0.95,  "messages":[{"role":"user","content":"附近有什么旅游景点?"}] }'

示例响应一:

Bash{..."content": "附近热门的旅游景点有故宫、天安门广场、恭王府等。您对哪个景点更感兴趣呢?"...}

示例请求二:

Bashcurl 'http://<这里换成网关公网IP>/api/openai/v1/chat/completions' \-H 'Accept: application/json, text/event-stream' \-H 'Content-Type: application/json' \--data-raw '{  "model":"qwen-long",  "frequency_penalty":0,  "max_tokens":800,  "stream":false,  "presence_penalty":0,  "temperature":0.7,  "top_p":0.95,  "messages":[{"role":"user","content":"怎么去香港?"}] }'
示例响应二:
Bash{..."content": "从您所在的位置(北京市)前往香港,您可以选择乘坐飞机、火车或者长途汽车等多种交通方式。\n\n1. **飞机**:从北京首都国际机场出发,有多家航空公司提供直飞香港的服务。\n2. **高铁**:北京西站有直达香港西九龙站的高速列车。\n3. **长途汽车**:虽然不是主流方式,但也有部分长途汽车线路连接两地。\n\n您计划什么时候出发呢?不同的出行日期可能会有不同的票务情况和价格。"...}
示例请求三:

Bashcurl 'http://<这里换成网关公网IP>/api/openai/v1/chat/completions' \-H 'Accept: application/json, text/event-stream' \-H 'Content-Type: application/json' \--data-raw '{  "model":"qwen-long",  "frequency_penalty":0,  "max_tokens":800,  "stream":false,  "presence_penalty":0,  "temperature":0.7,  "top_p":0.95,  "messages":[{"role":"user","content":"特色小吃有哪些?"}] }'

示例响应三:

Bash{..."content": "我可以帮您大概介绍一下北京的一些特色小吃:\n\n1. **豆汁儿配焦圈**:北京的传统小吃之一,豆汁儿口感独特,搭配焦圈食用。\n2. **炸酱面**:北京非常有名的面条吃法,面条通常拌上特制的炸酱以及各种蔬菜丝。\n3. **卤煮火烧**:以猪内脏和豆腐干为主要原料,炖煮而成,汤味醇厚,是老北京的传统食品。\n4. **炒肝**:一种类似卤煮的小吃,但是主要用猪肝和大肠,加上蒜泥等调味品。\n5. **艾窝窝**:糯米制成的甜食,外皮软糯,里面包裹着红豆沙或其他甜馅。\n6. **糖葫芦**:串起来的山楂果,表面裹上一层糖浆,酸甜可口,很受欢迎。\n\n您对哪一种小吃比较感兴趣呢?或者您想了解更多有关这些小吃的信息吗?"...}
03

实现原理

Cloud Native

1. Geo-IP 插件基于用户 IP 获取用户相关的地理位置信息,如:国家、省份、城市等。
2. 基于 Higress 的 Property 机制实现跨插件间的数据传递,将 Geo-IP 插件获取的地理位置信息传递给 AI 提示词修饰插件。

3. AI 提示词插件基于用户配置的提示词模板、用户输入,以及地理位置信息,生成增强后的用户请求发送给 LLM。


Geo-IP 插件实现

用户 IP 的获取:

当请求进入 onHttpRequestHeader( ) 函数时,如果用户是直接连接 Higress,就取 remote_address 作为用户 IP。如果用户通过代理服务访问 Higress,请求里包含请求头 X-forwarded-for,就解析 XFF 头取第一个 IP 地址作为原始用户 IP。

离线 IP 地址定位库内存优化:

Geo IP 插件内置离线 IP 地址定位库,数据来源: https://gitee.com/lionsoul/ip2region
由于默认提供的文件格式为:起始 IP 地址|结束 IP 地址|国家|区域|省份|城市|ISP,直接对数据进行存储存在较大的内存开销。因此首先对原始数据进行处理,将原始数据拆分为多个:IP 地址/子网掩码|国家|区域|省份|城市|ISP 的格式。然后基于加工后的数据格式,可以通过基数树(Radix Tree)的数据结构在内存中进行存储,最终在降低内存占用的同时,保证了 IP 地址的查找效率。

地理位置信息的传递:

起初希望通过 setContext 进行信息的传递。后来在和 Higress Maintainer 的沟通中了解到 Context 没有办法跨插件传递信息。需要通过 setProperty 设置到 metadata 中,其他插件通过 getProperty 获取。在此感谢 johnlanni 的热心解答和帮助。


AI 提示词修饰插件增强实现
用户配置:
prepend:






请到「今天看啥」查看全文