diff --git a/app/controller/app-center/aiChat.ts b/app/controller/app-center/aiChat.ts
index 350af76..ce9422b 100644
--- a/app/controller/app-center/aiChat.ts
+++ b/app/controller/app-center/aiChat.ts
@@ -10,27 +10,76 @@
  *
  */
 import { Controller } from 'egg';
-import { E_FOUNDATION_MODEL } from '../../lib/enum';
 
 export default class AiChatController extends Controller {
   public async aiChat() {
     const { ctx } = this;
-    const { foundationModel, messages } = ctx.request.body;
-    this.ctx.logger.info('ai接口请求参参数 model选型:', foundationModel);
+    const options = ctx.request.body;
+    this.ctx.logger.info('ai接口请求参数 model选型:', options);
+
+    const messages = options.messages;
     if (!messages || !Array.isArray(messages)) {
       return this.ctx.helper.getResponseData('Not passing the correct message parameter');
     }
-    const model = foundationModel?.model ?? E_FOUNDATION_MODEL.GPT_35_TURBO;
-    const token = foundationModel.token;
-    ctx.body = await ctx.service.appCenter.aiChat.getAnswerFromAi(messages, { model, token });
+    const apiKey = ctx.request.header?.authorization?.replace('Bearer', '');
+    const baseUrl = options?.baseUrl;
+    const model = options?.model;
+    const stream = options?.stream || false;
+    const tools = options?.tools || [];
+
+    if (stream) {
+      ctx.status = 200;
+      ctx.set({
+        'Content-Type': 'text/event-stream',
+        'Cache-Control': 'no-cache',
+        Connection: 'keep-alive'
+      });
+      try {
+        const result = await ctx.service.appCenter.aiChat.getAnswerFromAi(messages, {
+          apiKey,
+          baseUrl,
+          model,
+          stream,
+          tools
+        });
+
+        for await (const chunk of result) {
+          ctx.res.write(`data: ${JSON.stringify(chunk)}\n\n`); // SSE 格式
+        }
+
+        // 添加结束标记
+        ctx.res.write('data: [DONE]');
+      } catch (e: any) {
+        this.ctx.logger.error(`调用AI大模型接口失败: ${(e as Error).message}`);
+      } finally {
+        console.log('end');
+        ctx.res.end(); // 关闭连接
+      }
+
+      return;
+    }
+
+    // 非流式模式
+    ctx.body = await ctx.service.appCenter.aiChat.getAnswerFromAi(messages, {
+      apiKey,
+      baseUrl,
+      model,
+      stream,
+      tools
+    });
   }
 
+  public async search() {
+    const { ctx } = this;
+    const { content } = ctx.request.body;
+
+    ctx.body = await ctx.service.appCenter.aiChat.search(content);
+  }
 
   public async uploadFile() {
     const { ctx } = this;
-    const fileStream = await ctx.getFileStream();
-    const foundationModelObject = JSON.parse(fileStream.fields.foundationModel);
-    const { model, token } = foundationModelObject.foundationModel;
-    ctx.body = await ctx.service.appCenter.aiChat.getFileContentFromAi(fileStream, { model, token });
+    const stream = await ctx.getFileStream();
+
+    ctx.body = await ctx.service.appCenter.aiChat.uploadFile(stream);
   }
 }
diff --git a/app/router/appCenter/base.ts b/app/router/appCenter/base.ts
index 2b7c273..45f694d 100644
--- a/app/router/appCenter/base.ts
+++ b/app/router/appCenter/base.ts
@@ -113,5 +113,6 @@ export default (app: Application) => {
 
   // AI大模型聊天接口
   subRouter.post('/ai/chat', controller.appCenter.aiChat.aiChat);
-  subRouter.post('/ai/files', controller.appCenter.aiChat.uploadFile);
+  subRouter.post('/ai/search', controller.appCenter.aiChat.search);
+  subRouter.post('/ai/uploadFile', controller.appCenter.aiChat.uploadFile);
 };
diff --git a/app/service/app-center/aiChat.ts b/app/service/app-center/aiChat.ts
index 8ed5207..93ee597 100644
--- a/app/service/app-center/aiChat.ts
+++ b/app/service/app-center/aiChat.ts
@@ -10,13 +10,17 @@
  *
  */
 import { Service } from 'egg';
-import { E_FOUNDATION_MODEL } from '../../lib/enum';
-import * as fs from 'fs';
-import * as path from 'path';
-
-const to = require('await-to-js').default;
-const OpenAI = require('openai');
-
+import OpenApi, * as $OpenApi from '@alicloud/openapi-client';
+import OpenApiUtil from '@alicloud/openapi-util';
+import * as $Util from '@alicloud/tea-util';
+import Credential, { Config } from '@alicloud/credentials';
+import OpenAI from 'openai';
+import { ChatCompletionMessageParam } from 'openai/resources/chat/completions';
+import path from 'path';
+import FormData from 'form-data';
+import fs from 'fs';
+import axios from 'axios';
+import pump from 'mz-modules/pump';
 
 export type AiMessage = {
   role: string; // 角色
@@ -25,251 +29,243 @@ export type AiMessage = {
   partial?: boolean;
 };
 
-interface ConfigModel {
-  model: string;
-  token: string;
-}
-
 export default class AiChat extends Service {
   /**
    * 获取ai的答复
    *
-   * 根据后续引进的大模型情况决定，是否通过重构来对不同大模型进行统一的适配
-   *
    * @param messages
    * @param model
    * @return
    */
 
-  async getAnswerFromAi(messages: Array<AiMessage>, chatConfig: any) {
-    let res = await this.requestAnswerFromAi(messages, chatConfig);
-    let answerContent = '';
-    let isFinish = res.choices[0].finish_reason;
+  async getAnswerFromAi(messages: ChatCompletionMessageParam[], chatConfig: any) {
+    let result: any = null;
 
-    if (isFinish !== 'length') {
-      answerContent = res.choices[0]?.message.content;
-    }
-
-    // 若内容过长被截断，继续回复
-    while (isFinish === 'length') {
-      const prefix = res.choices[0].message.content;
-      answerContent += prefix;
-      messages.push({
-        role: 'assistant',
-        content: prefix,
-        partial: true
+    try {
+      const openai = new OpenAI({
+        apiKey: chatConfig.apiKey || process.env.OPEN_AI_API_KEY,
+        baseURL: chatConfig.baseUrl || process.env.OPEN_AI_BASE_URL,
+        defaultHeaders: {
+          'X-DashScope-OssResourceResolve': 'enable'
+        }
       });
 
-      res = await this.requestAnswerFromAi(messages, chatConfig);
-      answerContent += res.choices[0].message.content;
-      isFinish = res.choices[0].finish_reason;
-    }
+      const options: any = {
+        model: chatConfig.model || process.env.OPEN_AI_MODEL,
+        messages,
+        stream: chatConfig.stream
+      };
+      if (chatConfig.tools.length) {
+        options.tools = chatConfig.tools;
+      }
 
-    const code = this.extractCode(answerContent);
-    const schema = this.extractSchemaCode(code);
-    const answer = {
-      role: res.choices[0].message.role,
-      content: answerContent
-    };
-    const replyWithoutCode = this.removeCode(answerContent);
-    return this.ctx.helper.getResponseData({
-      originalResponse: answer,
-      replyWithoutCode,
-      schema
-    });
-  }
+      result = await openai.chat.completions.create(options);
 
-  async requestAnswerFromAi(messages: Array<AiMessage>, chatConfig: any) {
-    const { ctx } = this;
-    this.formatMessage(messages);
-    let res: any = null;
-    try {
-      // 根据大模型的不同匹配不同的配置
-      const aiChatConfig = this.config.aiChat(messages, chatConfig.token);
-      const { httpRequestUrl, httpRequestOption } = aiChatConfig[chatConfig.model];
-      this.ctx.logger.debug(httpRequestOption);
-      res = await ctx.curl(httpRequestUrl, httpRequestOption);
+      return result;
     } catch (e: any) {
-      this.ctx.logger.debug(`调用AI大模型接口失败: ${(e as Error).message}`);
+      this.ctx.logger.error(`调用AI大模型接口失败: ${(e as Error).message}`);
       return this.ctx.helper.getResponseData(`调用AI大模型接口失败: ${(e as Error).message}`);
     }
-
-    if (!res) {
-      return this.ctx.helper.getResponseData(`调用AI大模型接口未返回正确数据.`);
-    }
-
-    // 适配文心一言的响应数据结构，文心的部分异常情况status也是200，需要转为400，以免前端无所适从
-    if (res.data?.error_code) {
-      return this.ctx.helper.getResponseData(res.data?.error_msg);
-    }
-
-    // 适配chatgpt的响应数据结构
-    if (res.status !== 200) {
-      return this.ctx.helper.getResponseData(res.data?.error?.message, res.status);
-    }
-
-    // 适配文心一言的响应数据结构
-    if (chatConfig.model === E_FOUNDATION_MODEL.ERNIE_BOT_TURBO) {
-      return {
-        ...res.data,
-        choices: [
-          {
-            message: {
-              role: 'assistant',
-              content: res.data.result
-            }
-          }
-        ]
-      };
-    }
-
-    return res.data;
   }
 
   /**
-   * 提取回复中的代码
-   *
-   * 暂且只满足回复中只包括一个代码块的场景
+   * 知识库检索
+   * @remarks
+   * 使用凭据初始化账号Client
+   * @returns Client
    *
-   * @param content ai回复的内容
-   * @return 提取的文本
+   * @throws Exception
    */
-  private extractCode(content: string) {
-    const { start, end } = this.getStartAndEnd(content);
-    if (start < 0 || end < 0) {
-      return '';
-    }
-    return content.substring(start, end);
+  private createClient(): OpenApi {
+    const credentialsConfig1 = new Config({
+      type: 'access_key',
+      accessKeyId: process.env.ALIBABA_CLOUD_ACCESS_KEY_ID,
+      accessKeySecret: process.env.ALIBABA_CLOUD_ACCESS_KEY_SECRET
+    });
+    let credential = new Credential(credentialsConfig1);
+    let config = new $OpenApi.Config({
+      credential: credential
+    });
+
+    config.endpoint = `bailian.cn-beijing.aliyuncs.com`;
+    return new OpenApi(config);
   }
 
   /**
-   * 去除回复中的代码
-   *
-   * 暂且只满足回复中只包括一个代码块的场景
+   * @remarks
+   * API 相关
    *
-   * @param content ai回复的内容
-   * @return 去除代码后的回复内容
+   * @param path - string Path parameters
+   * @returns OpenApi.Params
    */
-  private removeCode(content: string) {
-    const { start, end } = this.getStartAndEnd(content);
-    if (start < 0 || end < 0) {
-      return content;
-    }
-    return content.substring(0, start) + '<代码在画布中展示>' + content.substring(end);
+  private createApiInfo(WorkspaceId): $OpenApi.Params {
+    let params = new $OpenApi.Params({
+      // 接口名称
+      action: 'Retrieve',
+      // 接口版本
+      version: '2023-12-29',
+      // 接口协议
+      protocol: 'HTTPS',
+      // 接口 HTTP 方法
+      method: 'POST',
+      authType: 'AK',
+      style: 'ROA',
+      // 接口 PATH
+      pathname: `/${WorkspaceId}/index/retrieve`,
+      // 接口请求体内容格式
+      reqBodyType: 'json',
+      // 接口响应体内容格式
+      bodyType: 'json'
+    });
+    return params;
   }
 
-  private extractSchemaCode(content) {
-    const startMarker = /```json/;
-    const endMarker = /```/;
-
-    const start = content.search(startMarker);
-    const end = content.slice(start + 7).search(endMarker) + start + 7;
-
-    if (start >= 0 && end >= 0) {
-      return JSON.parse(content.substring(start + 7, end).trim());
-    }
-
-    return null;
+  private getSearchList(res) {
+    const list = res?.body?.Data?.Nodes ?? [];
+
+    return {
+      data: list.map((node) => {
+        return {
+          score: node.Score,
+          content: node.Text,
+          doc_name: node.Metadata.doc_name
+        };
+      })
+    };
   }
 
-  private getStartAndEnd(str: string) {
-    const start = str.search(/```|<template>/);
+  async search(content: string[]) {
+    let res: any = null;
+    let client = this.createClient();
+    let params = this.createApiInfo(process.env.ALIBABA_CLOUD_WORKSPACE_ID);
+    // query params
+    let queries: { [key: string]: any } = {};
+    const QUERY = 'Query';
+    const ENABLE_REWRITE = 'EnableRewrite';
+    const INDEX_ID = 'IndexId';
+    queries[QUERY] = content;
+    queries[ENABLE_REWRITE] = true;
+    queries[INDEX_ID] = process.env.ALIBABA_CLOUD_INDEX_ID;
+    // runtime options
+    let runtime = new $Util.RuntimeOptions({});
+    let request = new $OpenApi.OpenApiRequest({
+      query: OpenApiUtil.query(queries)
+    });
 
-    // 匹配对应的结束标记
-    const endMarkerRegex = /```|<\/template>|<\/script>|<\/style>/g;
-    let match;
-    // 如果找不到匹配的结束标记，返回-1
-    let end = -1;
-    while ((match = endMarkerRegex.exec(str)) !== null) {
-      if (match.index > start) {
-        end = match.index + match[0].length;
-      }
+    try {
+      res = await client.callApi(params, request, runtime);
+    } catch (e) {
+      this.ctx.logger.error('Alibaba Cloud search failed', e);
+      return this.ctx.helper.getResponseData(`知识库检索接口调用失败:${e}`);
     }
 
-    return { start, end };
+    return this.getSearchList(res);
   }
 
-  private formatMessage(messages: Array<any>) {
-    const defaultWords: any = {
-      role: 'user',
-      content: `你是一名前端开发专家，编码时遵从以下几条要求:
-      ###
-      1. 只使用 element-ui组件库的el-button 和 el-table组件
-      2. el-table表格组件的使用方式为 <el-table :columns="columnData" :data="tableData"></el-table> columns的columnData表示列数据，其中用title表示列名，field表示表格数据字段； data的tableData表示表格展示的数据。 el-table标签内不得出现子元素
-      3. 使用vue2技术栈
-      4. 回复中只能有一个代码块
-      5. 不要加任何注释
-      6. el-table标签内不得出现el-table-column
-      ###`
-    };
-    const reg = /.*\u7f16\u7801\u65f6\u9075\u4ece\u4ee5\u4e0b\u51e0\u6761\u8981\u6c42.*/;
-    const { role, content } = messages[0];
-    if (role !== 'user') {
-      messages.unshift(defaultWords);
-    } else if (!reg.test(content)) {
-      messages[0].content = `${defaultWords.content}\n${messages[0].content}`;
+  /**
+   * 获取上传凭证
+   * @param {string} modelName - 模型名称
+   * @param {string} apiKey
+   */
+  async getUploadPolicy(modelName, apiKey) {
+    try {
+      const result = await this.ctx.curl('https://dashscope.aliyuncs.com/api/v1/uploads', {
+        method: 'GET',
+        headers: {
+          Authorization: `Bearer ${apiKey}`,
+          'Content-Type': 'application/json'
+        },
+        data: {
+          action: 'getPolicy',
+          model: modelName
+        },
+        dataType: 'json',
+        timeout: 5000
+      });
+
+      return result.data.data;
+    } catch (e) {
+      this.ctx.logger.error('获取上传凭证失败:', e);
+      return this.ctx.helper.getResponseData(`获取上传凭证失败:${e}`);
     }
-    return messages;
   }
 
   /**
-   * 文件上传
-   *
-   * @param model
-   * @return
+   * 上传文件到OSS
+   * @param {Object} policyData - 上传凭证
+   * @param {string} filePath - 本地文件路径
+   * @param {string} filename - 文件名
    */
+  async uploadToOSS(policyData, filePath, filename) {
+    const key = `${policyData.upload_dir}/${filePath}`;
+
+    const formData = new FormData();
+    formData.append('OSSAccessKeyId', policyData.oss_access_key_id);
+    formData.append('Signature', policyData.signature);
+    formData.append('policy', policyData.policy);
+    formData.append('x-oss-object-acl', policyData.x_oss_object_acl);
+    formData.append('x-oss-forbid-overwrite', policyData.x_oss_forbid_overwrite);
+    formData.append('key', key);
+    formData.append('success_action_status', '200');
+    formData.append('file', fs.createReadStream(filePath), filename);
 
-  async getFileContentFromAi(fileStream: any, chatConfig: ConfigModel) {
-    const answer = await this.requestFileContentFromAi(fileStream, chatConfig);
-    return this.ctx.helper.getResponseData({
-      originalResponse: answer
-    });
-  }
+    try {
+      const response = await axios.post(policyData.upload_host, formData, {
+        headers: formData.getHeaders(),
+        timeout: 10000
+      });
 
-  async requestFileContentFromAi(file: any, chatConfig: ConfigModel) {
-    const { ctx } = this;
-    const filename = Date.now() + path.extname(file.filename).toLowerCase();
-    const savePath = path.join(__dirname, filename);
-    const writeStream = fs.createWriteStream(savePath);
-    file.pipe(writeStream);
-    await new Promise(resolve => writeStream.on('close', resolve));
+      if (response.status !== 200) {
+        throw new Error('上传文件到OSS失败');
+      }
 
-    const client = new OpenAI({
-      apiKey: chatConfig.token,
-      baseURL: 'https://api.moonshot.cn/v1'
-    });
+      return {
+        ossUrl: `oss://${key}`,
+        expireTime: new Date(Date.now() + 48 * 60 * 60 * 1000)
+      };
+    } catch (e) {
+      this.ctx.logger.error('上传文件到OSS失败:', e);
+      return this.ctx.helper.getResponseData(`上传文件到OSS失败:${e}`);
+    }
+  }
 
-    // 上传文件
-    const [fileError, fileObject] = await to(client.files.create({
-      file: fs.createReadStream(savePath),
-      purpose: 'file-extract'
-    }));
+  /**
+   * 完整上传流程
+   * @param stream - 文件对象
+   */
+  async uploadFile(stream) {
+    const { modelName, apiKey } = stream.fields;
+    const filename = stream.filename;
+    const tmpDir = path.join(this.config.baseDir, 'tmp');
 
-    if (fileError) {
-      this.ctx.logger.debug(`调用上传图片接口失败: ${fileError.message}`);
-      await fs.promises.unlink(savePath).catch(err => console.error('文件删除失败:', err));
-      return this.ctx.helper.getResponseData(`调用上传图片接口失败: ${fileError.message}`);
+    if (!fs.existsSync(tmpDir)) {
+      fs.mkdirSync(tmpDir, { recursive: true });
     }
+    const tmpFilePath = path.join(tmpDir, filename);
 
-    // 文件解析
-    const imageAnalysisConfig = this.config.parsingFile(fileObject.id, chatConfig.token);
-    const { analysisImageHttpRequestUrl, analysisImageHttpRequestOption } = imageAnalysisConfig[chatConfig.model];
+    try {
+      const writeStream = fs.createWriteStream(tmpFilePath);
+      await pump(stream, writeStream);
+      await new Promise((resolve) => writeStream.on('close', resolve));
 
-    const [analysisError, res] = await to(ctx.curl(analysisImageHttpRequestUrl, analysisImageHttpRequestOption));
+      const policyData = await this.getUploadPolicy(modelName, apiKey);
+      const { ossUrl, expireTime } = await this.uploadToOSS(policyData, tmpFilePath, filename);
 
-    if (analysisError) {
-      this.ctx.logger.debug(`调用解析文件接口失败: ${analysisError.message}`);
-      await fs.promises.unlink(savePath).catch(err => console.error('文件删除失败:', err));
-      return this.ctx.helper.getResponseData(`调用解析文件接口失败: ${analysisError.message}`);
-    }
+      fs.unlinkSync(tmpFilePath);
 
-    // 删除文件
-    await fs.promises.unlink(savePath).catch(err => console.error('文件删除失败:', err));
-    await to(client.files.del(fileObject.id));
-
-    // 返回结果
-    res.data = JSON.parse(res.res.data.toString());
-    return res.data || this.ctx.helper.getResponseData('调用上传图片接口未返回正确数据.');
+      return {
+        success: true,
+        data: {
+          url: ossUrl,
+          expireTime: expireTime.toISOString()
+        }
+      };
+    } catch (e) {
+      if (fs.existsSync(tmpFilePath)) {
+        fs.unlinkSync(tmpFilePath);
+      }
+      this.ctx.logger.error('文件上传失败:', e);
+      return this.ctx.helper.getResponseData(`文件上传失败:${e}`);
+    }
   }
 }
-
diff --git a/config/config.default.ts b/config/config.default.ts
index b0e941f..ef3c9c5 100644
--- a/config/config.default.ts
+++ b/config/config.default.ts
@@ -86,6 +86,19 @@ export default (appInfo) => {
     }
   };
 
+  config.multipart = {
+    mode: 'stream', // 必须设置为 stream 模式
+    autoFields: true, // 自动解析字段
+    fieldNameSize: 100,
+    fieldSize: '100kb',
+    fields: 5,
+    fileSize: '10mb',
+    whitelist: ['.jpg', '.png', '.jpeg'], // 根据需求调整
+    cleanSchedule: {
+      cron: '0 30 4 * * *' // 每天4:30清理临时文件
+    }
+  };
+
   config.previewTemplate = {
     default: {
       vue: '@opentiny/tiny-engine-preview-vue'
@@ -260,7 +273,7 @@ export default (appInfo) => {
             messages
           },
           headers: {
-            Authorization: `Bearer ${process.env.OPENAI_API_KEY}`
+            Authorization: `Bearer ${token || process.env.OPENAI_API_KEY}`
           }
         },
         manufacturer: 'openai'
diff --git a/package.json b/package.json
index c5b205c..d6a53ca 100644
--- a/package.json
+++ b/package.json
@@ -45,6 +45,13 @@
     "branches": 60
   },
   "dependencies": {
+    "@alicloud/credentials": "^2.4.2",
+    "@alicloud/openapi-client": "^0.4.14",
+    "@alicloud/openapi-util": "^0.3.2",
+    "@alicloud/pop-core": "^1.8.0",
+    "@alicloud/tea-console": "^1.0.0",
+    "@alicloud/tea-typescript": "^1.7.1",
+    "@alicloud/tea-util": "^1.4.9",
     "@opentiny/tiny-engine-block-build": "^0.0.1-alpha.0",
     "@opentiny/tiny-engine-dsl-vue": "~1.0.1",
     "@opentiny/tiny-engine-transform": "^1.0.0-alpha.1",
@@ -66,6 +73,7 @@
     "http-status-codes": "^2.2.0",
     "koa-connect": "^2.1.0",
     "moment": "^2.29.4",
+    "mz-modules": "^2.1.0",
     "mysql2": "^3.14.2",
     "openai": "^4.68.4",
     "proxy-agent": "^5.0.0",