当前位置：首页 > news >正文

如何使用open ai实现全站语义搜索？

news 来源：原创 2024/9/20 19:32:20

在企业Web应用中使用OpenAI的Function Calling功能来实现全站语义搜索，可以有效提升搜索体验，使用户能够通过自然语言查询来获取更相关的搜索结果。以下是一个实现全站语义搜索的基本步骤：

设置OpenAI的API
首先，你需要确保已经配置了OpenAI的API密钥，并且你的Angular应用能够与该API进行通信。

import { HttpClient, HttpHeaders } from '@angular/common/http';
import { Injectable } from '@angular/core';@Injectable({providedIn: 'root',
})
export class OpenAiService {private apiUrl = 'https://api.openai.com/v1/chat/completions';private apiKey = 'your-openai-api-key'; // 替换为你的API密钥constructor(private http: HttpClient) {}public callFunction(query: string) {const headers = new HttpHeaders({'Content-Type': 'application/json',Authorization: `Bearer ${this.apiKey}`,});const body = {model: 'gpt-4', // 或者其他版本messages: [{ role: 'system', content: 'You are a helpful assistant.' },{ role: 'user', content: query },],functions: [{name: 'search_website',description: 'Search the company website for relevant information',parameters: {type: 'object',properties: {query: { type: 'string', description: 'The search query' },},required: ['query'],},},],function_call: 'auto',};return this.http.post(this.apiUrl, body, { headers });}
}

实现全站搜索逻辑
接下来，你需要在Angular应用中实现搜索功能，并且使用OpenAI的Function Calling来解析用户的自然语言查询。

import { Component } from '@angular/core';
import { OpenAiService } from './open-ai.service';@Component({selector: 'app-search',templateUrl: './search.component.html',styleUrls: ['./search.component.css'],
})
export class SearchComponent {public searchQuery = '';public searchResults: any[] = [];constructor(private openAiService: OpenAiService) {}public performSearch() {this.openAiService.callFunction(this.searchQuery).subscribe((response: any) => {const result = response.choices[0].message.function_call.arguments;this.searchResults = JSON.parse(result).results;});}
}

后端搜索功能集成
由于OpenAI的模型本身无法直接访问你的企业网站数据，你需要在后端实现一个函数来处理模型返回的搜索查询。这个函数会执行实际的搜索操作，并将结果返回给前端。

例如，在你的Node.js后端，你可以这样实现：

const express = require('express');
const app = express();// 模拟数据库或全文搜索引擎
const searchDatabase = (query) => {// 实现你自己的搜索逻辑return [// 返回搜索结果的示例数据{ title: 'Result 1', url: '/result-1', snippet: 'This is a summary of result 1.' },{ title: 'Result 2', url: '/result-2', snippet: 'This is a summary of result 2.' },];
};app.post('/search', (req, res) => {const { query } = req.body;const results = searchDatabase(query);res.json({ results });
});app.listen(3000, () => {console.log('Server running on port 3000');
});

前端与后端集成
最后，前端的Angular应用会通过HTTP请求将用户查询传递给后端，并在接收到后端返回的搜索结果后进行显示。

public performSearch() {this.http.post('/search', { query: this.searchQuery }).subscribe((results: any) => {this.searchResults = results;});
}

显示搜索结果
在模板文件中，显示搜索结果列表：

<div><input type="text" [(ngModel)]="searchQuery" placeholder="Enter your search query" /><button (click)="performSearch()">Search</button>
</div><div *ngIf="searchResults.length > 0"><h3>Search Results:</h3><ul><li *ngFor="let result of searchResults"><a [href]="result.url">{{ result.title }}</a><p>{{ result.snippet }}</p></li></ul>
</div>