✨ feat: 支持查询天气

src/pages/api/openai.api.ts

@@ -1,8 +1,13 @@
 import { OpenAIStream, StreamingTextResponse } from 'ai';
 import { Configuration, OpenAIApi } from 'openai-edge';
+import { ChatCompletionFunctions, ChatCompletionRequestMessage } from 'openai-edge/types/api';
 
 import { OpenAIStreamPayload } from '@/types/openai';
 
+import { plugins } from './plugins';
+
+export const runtime = 'edge';
+
 const isDev = process.env.NODE_ENV === 'development';
 const OPENAI_PROXY_URL = process.env.OPENAI_PROXY_URL;
 
@@ -13,19 +18,38 @@ const config = new Configuration({
 
 const openai = new OpenAIApi(config, isDev && OPENAI_PROXY_URL ? OPENAI_PROXY_URL : undefined);
 
-export const runtime = 'edge';
+const functions: ChatCompletionFunctions[] = plugins.map((f) => f.schema);
 
 export default async function handler(req: Request) {
   // Extract the `messages` from the body of the request
   const { messages, ...params } = (await req.json()) as OpenAIStreamPayload;
 
-  console.log(params);
+  const formatMessages = messages.map((m) => ({ content: m.content, role: m.role }));
+
   const response = await openai.createChatCompletion({
+    functions,
+    messages: formatMessages,
     stream: true,
     ...params,
-    messages: messages.map((m) => ({ content: m.content, role: m.role })),
   });
 
-  const stream = OpenAIStream(response);
+  const stream = OpenAIStream(response, {
+    experimental_onFunctionCall: async ({ name, arguments: args }, createFunctionCallMessages) => {
+      const func = plugins.find((f) => f.name === name);
+
+      if (func) {
+        const result = await func.runner(args as any);
+
+        const newMessages = createFunctionCallMessages(result) as ChatCompletionRequestMessage[];
+
+        return openai.createChatCompletion({
+          functions,
+          messages: [...formatMessages, ...newMessages],
+          stream: true,
+          ...params,
+        });
+      }
+    },
+  });
   return new StreamingTextResponse(stream);
 }

src/pages/api/plugins/index.ts

@@ -0,0 +1,3 @@
+import getWeather from './weather';
+
+export const plugins = [getWeather];

src/pages/api/plugins/weather/index.ts

@@ -0,0 +1,20 @@
+import runner from './runner';
+
+const schema = {
+  description: '获取当前天气情况',
+  name: 'fetchWeather',
+  parameters: {
+    properties: {
+      city: {
+        description: '城市名称',
+        type: 'string',
+      },
+    },
+    required: ['city'],
+    type: 'object',
+  },
+};
+
+const getWeather = { name: 'fetchWeather', runner, schema };
+
+export default getWeather;

src/pages/api/plugins/weather/runner.ts

@@ -0,0 +1,30 @@
+const weatherBaseURL = 'https://restapi.amap.com/v3/weather/weatherInfo';
+
+const citySearchURL = 'https://restapi.amap.com/v3/config/district';
+
+const KEY = process.env.GAODE_WEATHER_KEY;
+
+interface WeatherParams {
+  city: string;
+  extensions?: 'base' | 'all';
+}
+
+const fetchCityCode = async (keywords: string): Promise<string> => {
+  const URL = `${citySearchURL}?keywords=${keywords}&subdistrict=0&extensions=base&key=${KEY}`;
+  const res = await fetch(URL);
+
+  const data = await res.json();
+
+  return data.districts[0].adcode;
+};
+
+const fetchWeather = async ({ city, extensions = 'all' }: WeatherParams) => {
+  const cityCode = await fetchCityCode(city);
+
+  const URL = `${weatherBaseURL}?city=${cityCode}&extensions=${extensions}&key=${KEY}`;
+  const res = await fetch(URL);
+
+  return await res.json();
+};
+
+export default fetchWeather;