309 lines
7.6 KiB
Go
309 lines
7.6 KiB
Go
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bytes"
|
|||
|
|
"context"
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"io"
|
|||
|
|
"net/http"
|
|||
|
|
"os"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/Nanqipro/YunQue-Tech-Projects/ai_english_learning/serve/internal/model"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// AIService AI服务接口
|
|||
|
|
type AIService interface {
|
|||
|
|
// 写作批改
|
|||
|
|
CorrectWriting(ctx context.Context, content string, taskType string) (*model.WritingCorrection, error)
|
|||
|
|
// 口语评估
|
|||
|
|
EvaluateSpeaking(ctx context.Context, audioText string, prompt string) (*model.SpeakingEvaluation, error)
|
|||
|
|
// 智能推荐
|
|||
|
|
GetRecommendations(ctx context.Context, userLevel string, learningHistory []string) (*model.AIRecommendation, error)
|
|||
|
|
// 生成练习题
|
|||
|
|
GenerateExercise(ctx context.Context, content string, exerciseType string) (*model.Exercise, error)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type aiService struct {
|
|||
|
|
apiKey string
|
|||
|
|
baseURL string
|
|||
|
|
client *http.Client
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewAIService 创建AI服务实例
|
|||
|
|
func NewAIService() AIService {
|
|||
|
|
apiKey := os.Getenv("OPENAI_API_KEY")
|
|||
|
|
baseURL := os.Getenv("OPENAI_BASE_URL")
|
|||
|
|
if baseURL == "" {
|
|||
|
|
baseURL = "https://api.openai.com/v1"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return &aiService{
|
|||
|
|
apiKey: apiKey,
|
|||
|
|
baseURL: baseURL,
|
|||
|
|
client: &http.Client{
|
|||
|
|
Timeout: 30 * time.Second,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// OpenAI API请求结构
|
|||
|
|
type openAIRequest struct {
|
|||
|
|
Model string `json:"model"`
|
|||
|
|
Messages []message `json:"messages"`
|
|||
|
|
MaxTokens int `json:"max_tokens"`
|
|||
|
|
Temperature float64 `json:"temperature"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type message struct {
|
|||
|
|
Role string `json:"role"`
|
|||
|
|
Content string `json:"content"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type openAIResponse struct {
|
|||
|
|
Choices []struct {
|
|||
|
|
Message message `json:"message"`
|
|||
|
|
} `json:"choices"`
|
|||
|
|
Error *struct {
|
|||
|
|
Message string `json:"message"`
|
|||
|
|
Type string `json:"type"`
|
|||
|
|
} `json:"error,omitempty"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// callOpenAI 调用OpenAI API
|
|||
|
|
func (s *aiService) callOpenAI(ctx context.Context, prompt string) (string, error) {
|
|||
|
|
if s.apiKey == "" {
|
|||
|
|
return "", fmt.Errorf("OpenAI API key not configured")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
reqBody := openAIRequest{
|
|||
|
|
Model: "gpt-3.5-turbo",
|
|||
|
|
Messages: []message{
|
|||
|
|
{
|
|||
|
|
Role: "user",
|
|||
|
|
Content: prompt,
|
|||
|
|
},
|
|||
|
|
},
|
|||
|
|
MaxTokens: 1000,
|
|||
|
|
Temperature: 0.7,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
jsonData, err := json.Marshal(reqBody)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", fmt.Errorf("failed to marshal request: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
req, err := http.NewRequestWithContext(ctx, "POST", s.baseURL+"/chat/completions", bytes.NewBuffer(jsonData))
|
|||
|
|
if err != nil {
|
|||
|
|
return "", fmt.Errorf("failed to create request: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
req.Header.Set("Content-Type", "application/json")
|
|||
|
|
req.Header.Set("Authorization", "Bearer "+s.apiKey)
|
|||
|
|
|
|||
|
|
resp, err := s.client.Do(req)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", fmt.Errorf("failed to send request: %w", err)
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
|
|||
|
|
body, err := io.ReadAll(resp.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", fmt.Errorf("failed to read response: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var openAIResp openAIResponse
|
|||
|
|
if err := json.Unmarshal(body, &openAIResp); err != nil {
|
|||
|
|
return "", fmt.Errorf("failed to unmarshal response: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if openAIResp.Error != nil {
|
|||
|
|
return "", fmt.Errorf("OpenAI API error: %s", openAIResp.Error.Message)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if len(openAIResp.Choices) == 0 {
|
|||
|
|
return "", fmt.Errorf("no response from OpenAI")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return openAIResp.Choices[0].Message.Content, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// CorrectWriting 写作批改
|
|||
|
|
func (s *aiService) CorrectWriting(ctx context.Context, content string, taskType string) (*model.WritingCorrection, error) {
|
|||
|
|
prompt := fmt.Sprintf(`请对以下英语写作进行批改，任务类型：%s
|
|||
|
|
|
|||
|
|
写作内容：
|
|||
|
|
%s
|
|||
|
|
|
|||
|
|
请按照以下JSON格式返回批改结果：
|
|||
|
|
{
|
|||
|
|
"overall_score": 85,
|
|||
|
|
"grammar_score": 80,
|
|||
|
|
"vocabulary_score": 90,
|
|||
|
|
"structure_score": 85,
|
|||
|
|
"content_score": 88,
|
|||
|
|
"corrections": [
|
|||
|
|
{
|
|||
|
|
"original": "错误的句子",
|
|||
|
|
"corrected": "修正后的句子",
|
|||
|
|
"explanation": "修改说明",
|
|||
|
|
"error_type": "grammar"
|
|||
|
|
}
|
|||
|
|
],
|
|||
|
|
"suggestions": [
|
|||
|
|
"建议1",
|
|||
|
|
"建议2"
|
|||
|
|
],
|
|||
|
|
"strengths": [
|
|||
|
|
"优点1",
|
|||
|
|
"优点2"
|
|||
|
|
],
|
|||
|
|
"weaknesses": [
|
|||
|
|
"需要改进的地方1",
|
|||
|
|
"需要改进的地方2"
|
|||
|
|
]
|
|||
|
|
}`, taskType, content)
|
|||
|
|
|
|||
|
|
response, err := s.callOpenAI(ctx, prompt)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析JSON响应
|
|||
|
|
var correction model.WritingCorrection
|
|||
|
|
if err := json.Unmarshal([]byte(response), &correction); err != nil {
|
|||
|
|
// 如果JSON解析失败，返回基本的批改结果
|
|||
|
|
return &model.WritingCorrection{
|
|||
|
|
OverallScore: 75,
|
|||
|
|
Suggestions: []string{"AI批改服务暂时不可用，请稍后重试"},
|
|||
|
|
}, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return &correction, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// EvaluateSpeaking 口语评估
|
|||
|
|
func (s *aiService) EvaluateSpeaking(ctx context.Context, audioText string, prompt string) (*model.SpeakingEvaluation, error) {
|
|||
|
|
evalPrompt := fmt.Sprintf(`请对以下英语口语进行评估，题目：%s
|
|||
|
|
|
|||
|
|
口语内容：
|
|||
|
|
%s
|
|||
|
|
|
|||
|
|
请按照以下JSON格式返回评估结果：
|
|||
|
|
{
|
|||
|
|
"overall_score": 85,
|
|||
|
|
"pronunciation_score": 80,
|
|||
|
|
"fluency_score": 90,
|
|||
|
|
"grammar_score": 85,
|
|||
|
|
"vocabulary_score": 88,
|
|||
|
|
"feedback": "整体评价",
|
|||
|
|
"strengths": [
|
|||
|
|
"优点1",
|
|||
|
|
"优点2"
|
|||
|
|
],
|
|||
|
|
"improvements": [
|
|||
|
|
"需要改进的地方1",
|
|||
|
|
"需要改进的地方2"
|
|||
|
|
]
|
|||
|
|
}`, prompt, audioText)
|
|||
|
|
|
|||
|
|
response, err := s.callOpenAI(ctx, evalPrompt)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析JSON响应
|
|||
|
|
var evaluation model.SpeakingEvaluation
|
|||
|
|
if err := json.Unmarshal([]byte(response), &evaluation); err != nil {
|
|||
|
|
// 如果JSON解析失败，返回基本的评估结果
|
|||
|
|
return &model.SpeakingEvaluation{
|
|||
|
|
OverallScore: 75,
|
|||
|
|
Feedback: "AI评估服务暂时不可用，请稍后重试",
|
|||
|
|
}, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return &evaluation, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetRecommendations 智能推荐
|
|||
|
|
func (s *aiService) GetRecommendations(ctx context.Context, userLevel string, learningHistory []string) (*model.AIRecommendation, error) {
|
|||
|
|
historyStr := strings.Join(learningHistory, ", ")
|
|||
|
|
prompt := fmt.Sprintf(`基于用户的英语水平（%s）和学习历史（%s），请提供个性化的学习推荐。
|
|||
|
|
|
|||
|
|
请按照以下JSON格式返回推荐结果：
|
|||
|
|
{
|
|||
|
|
"recommended_topics": [
|
|||
|
|
"推荐主题1",
|
|||
|
|
"推荐主题2"
|
|||
|
|
],
|
|||
|
|
"difficulty_level": "intermediate",
|
|||
|
|
"study_plan": [
|
|||
|
|
"学习计划步骤1",
|
|||
|
|
"学习计划步骤2"
|
|||
|
|
],
|
|||
|
|
"focus_areas": [
|
|||
|
|
"重点关注领域1",
|
|||
|
|
"重点关注领域2"
|
|||
|
|
]
|
|||
|
|
}`, userLevel, historyStr)
|
|||
|
|
|
|||
|
|
response, err := s.callOpenAI(ctx, prompt)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析JSON响应
|
|||
|
|
var recommendation model.AIRecommendation
|
|||
|
|
if err := json.Unmarshal([]byte(response), &recommendation); err != nil {
|
|||
|
|
// 如果JSON解析失败，返回基本的推荐结果
|
|||
|
|
return &model.AIRecommendation{
|
|||
|
|
RecommendedTopics: []string{"基础语法练习", "日常对话练习"},
|
|||
|
|
DifficultyLevel: "beginner",
|
|||
|
|
StudyPlan: []string{"每天练习30分钟", "重点关注基础词汇"},
|
|||
|
|
}, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return &recommendation, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GenerateExercise 生成练习题
|
|||
|
|
func (s *aiService) GenerateExercise(ctx context.Context, content string, exerciseType string) (*model.Exercise, error) {
|
|||
|
|
prompt := fmt.Sprintf(`基于以下内容生成%s类型的练习题：
|
|||
|
|
|
|||
|
|
内容：
|
|||
|
|
%s
|
|||
|
|
|
|||
|
|
请按照以下JSON格式返回练习题：
|
|||
|
|
{
|
|||
|
|
"title": "练习题标题",
|
|||
|
|
"instructions": "练习说明",
|
|||
|
|
"questions": [
|
|||
|
|
{
|
|||
|
|
"question": "问题1",
|
|||
|
|
"options": ["选项A", "选项B", "选项C", "选项D"],
|
|||
|
|
"correct_answer": "正确答案",
|
|||
|
|
"explanation": "解释"
|
|||
|
|
}
|
|||
|
|
]
|
|||
|
|
}`, exerciseType, content)
|
|||
|
|
|
|||
|
|
response, err := s.callOpenAI(ctx, prompt)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析JSON响应
|
|||
|
|
var exercise model.Exercise
|
|||
|
|
if err := json.Unmarshal([]byte(response), &exercise); err != nil {
|
|||
|
|
// 如果JSON解析失败，返回基本的练习题
|
|||
|
|
return &model.Exercise{
|
|||
|
|
Title: "基础练习",
|
|||
|
|
Instructions: "AI练习生成服务暂时不可用，请稍后重试",
|
|||
|
|
Questions: []model.Question{},
|
|||
|
|
}, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return &exercise, nil
|
|||
|
|
}
|