chore: initial commit
This commit is contained in:
你的用户名
51
modules/ai_analyzer.py
Normal file
51
modules/ai_analyzer.py
Normal file
@@ -0,0 +1,51 @@
|
||||
"""AI意图分析模块"""
|
||||
import json
|
||||
import re
|
||||
import logging
|
||||
from typing import Dict
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
class AIAnalyzer:
|
||||
def __init__(self, claude_client):
|
||||
self.claude_client = claude_client
|
||||
self.model = "claude-sonnet-4-20250514"
|
||||
|
||||
async def analyze_intent(self, user_input: str) -> Dict:
|
||||
prompt = f"""分析Telegram群组搜索需求，生成3-5个搜索建议。
|
||||
用户输入："{user_input}"
|
||||
可用命令：/search /text /human /topchat
|
||||
返回JSON：{{"explanation":"说明","suggestions":[{{"command":"/text","keyword":"关键词","description":"描述","icon":"💬"}}]}}"""
|
||||
|
||||
try:
|
||||
response = self.claude_client.messages.create(
|
||||
model=self.model,
|
||||
max_tokens=1000,
|
||||
messages=[{"role": "user", "content": prompt}]
|
||||
)
|
||||
|
||||
ai_response = response.content[0].text.strip()
|
||||
json_match = re.search(r'```json\s*(.*?)\s*```', ai_response, re.DOTALL)
|
||||
if json_match:
|
||||
ai_response = json_match.group(1)
|
||||
|
||||
analysis = json.loads(ai_response)
|
||||
return self._validate(analysis, user_input)
|
||||
except Exception as e:
|
||||
logger.error(f"AI分析失败: {e}")
|
||||
return self._fallback(user_input)
|
||||
|
||||
def _validate(self, analysis, user_input):
|
||||
if 'suggestions' not in analysis:
|
||||
raise ValueError("缺少suggestions")
|
||||
return analysis
|
||||
|
||||
def _fallback(self, user_input):
|
||||
return {
|
||||
"explanation": f"为您搜索「{user_input}」",
|
||||
"suggestions": [
|
||||
{"command": "/search", "keyword": user_input, "description": f"按名称:{user_input}", "icon": "🔍"},
|
||||
{"command": "/text", "keyword": user_input, "description": f"按内容:{user_input}", "icon": "💬"},
|
||||
{"command": "/topchat", "keyword": "", "description": "浏览热门", "icon": "🔥"}
|
||||
]
|
||||
}
|
||||
160
modules/session_manager.py
Normal file
160
modules/session_manager.py
Normal file
@@ -0,0 +1,160 @@
|
||||
"""会话管理模块 - 管理用户交互状态和历史"""
|
||||
from __future__ import annotations
|
||||
|
||||
from datetime import datetime, timedelta
|
||||
from typing import Dict, Optional, Any
|
||||
import logging
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class SessionManager:
|
||||
"""用户会话管理器"""
|
||||
|
||||
def __init__(self, timeout_minutes: int = 30) -> None:
|
||||
self.sessions: Dict[int, Dict[str, Any]] = {}
|
||||
self.session_timeout = timedelta(minutes=timeout_minutes)
|
||||
|
||||
def _now(self) -> datetime:
|
||||
return datetime.now()
|
||||
|
||||
def create_session(self, user_id: int, initial_query: str) -> Dict[str, Any]:
|
||||
"""创建新会话"""
|
||||
session = {
|
||||
"user_id": user_id,
|
||||
"stage": "initial",
|
||||
"initial_query": initial_query,
|
||||
"history": [
|
||||
{
|
||||
"step": "input",
|
||||
"content": initial_query,
|
||||
"timestamp": self._now(),
|
||||
}
|
||||
],
|
||||
"analysis": None,
|
||||
"selected_suggestion": None,
|
||||
"search_results": None,
|
||||
"can_go_back": False,
|
||||
"created_at": self._now(),
|
||||
"last_activity": self._now(),
|
||||
}
|
||||
self.sessions[user_id] = session
|
||||
logger.info("[会话] 创建新会话: user=%s, query=%s", user_id, initial_query)
|
||||
return session
|
||||
|
||||
def get_session(self, user_id: int) -> Optional[Dict[str, Any]]:
|
||||
"""获取会话，包含过期检查"""
|
||||
session = self.sessions.get(user_id)
|
||||
if not session:
|
||||
return None
|
||||
|
||||
if self._now() - session.get("last_activity", self._now()) > self.session_timeout:
|
||||
logger.info("[会话] 会话已过期: user=%s", user_id)
|
||||
self.sessions.pop(user_id, None)
|
||||
return None
|
||||
|
||||
session["last_activity"] = self._now()
|
||||
return session
|
||||
|
||||
def update_stage(self, user_id: int, stage: str, **kwargs: Any) -> Optional[Dict[str, Any]]:
|
||||
"""更新会话阶段并记录历史"""
|
||||
session = self.get_session(user_id)
|
||||
if not session:
|
||||
return None
|
||||
|
||||
session["stage"] = stage
|
||||
session["last_activity"] = self._now()
|
||||
|
||||
history_entry = {
|
||||
"step": stage,
|
||||
"timestamp": self._now(),
|
||||
}
|
||||
history_entry.update(kwargs)
|
||||
session.setdefault("history", []).append(history_entry)
|
||||
|
||||
for key, value in kwargs.items():
|
||||
session[key] = value
|
||||
|
||||
logger.info("[会话] 更新阶段: user=%s, stage=%s", user_id, stage)
|
||||
return session
|
||||
|
||||
def save_analysis(self, user_id: int, analysis: Dict[str, Any]) -> None:
|
||||
"""保存AI分析结果"""
|
||||
session = self.get_session(user_id)
|
||||
if not session:
|
||||
return
|
||||
|
||||
session["analysis"] = analysis
|
||||
session["stage"] = "suggestions"
|
||||
session["can_go_back"] = True
|
||||
|
||||
suggestions = analysis.get("suggestions", [])
|
||||
logger.info("[会话] 保存分析: user=%s, suggestions=%s", user_id, len(suggestions))
|
||||
|
||||
def save_selection(self, user_id: int, suggestion_index: int) -> Optional[Dict[str, Any]]:
|
||||
"""保存用户选择的建议"""
|
||||
session = self.get_session(user_id)
|
||||
if not session:
|
||||
return None
|
||||
|
||||
analysis = session.get("analysis") or {}
|
||||
suggestions = analysis.get("suggestions", [])
|
||||
if 0 <= suggestion_index < len(suggestions):
|
||||
selection = suggestions[suggestion_index]
|
||||
session["selected_suggestion"] = selection
|
||||
session["stage"] = "searching"
|
||||
session.setdefault("history", []).append(
|
||||
{
|
||||
"step": "selection",
|
||||
"timestamp": self._now(),
|
||||
"selection": selection,
|
||||
}
|
||||
)
|
||||
logger.info(
|
||||
"[会话] 保存选择: user=%s, index=%s", user_id, suggestion_index
|
||||
)
|
||||
return selection
|
||||
logger.warning(
|
||||
"[会话] 选择索引无效: user=%s, index=%s, total=%s",
|
||||
user_id,
|
||||
suggestion_index,
|
||||
len(suggestions),
|
||||
)
|
||||
return None
|
||||
|
||||
def can_go_back(self, user_id: int) -> bool:
|
||||
session = self.get_session(user_id)
|
||||
return bool(session and session.get("can_go_back", False))
|
||||
|
||||
def go_back_to_suggestions(self, user_id: int) -> Optional[Dict[str, Any]]:
|
||||
"""返回到建议阶段"""
|
||||
session = self.get_session(user_id)
|
||||
if not session:
|
||||
return None
|
||||
|
||||
analysis = session.get("analysis")
|
||||
if not analysis:
|
||||
return None
|
||||
|
||||
session["stage"] = "suggestions"
|
||||
session["selected_suggestion"] = None
|
||||
logger.info("[会话] 返回建议列表: user=%s", user_id)
|
||||
return analysis
|
||||
|
||||
def clear_session(self, user_id: int) -> None:
|
||||
"""清除会话"""
|
||||
if user_id in self.sessions:
|
||||
self.sessions.pop(user_id, None)
|
||||
logger.info("[会话] 清除会话: user=%s", user_id)
|
||||
|
||||
def get_stats(self) -> Dict[str, Any]:
|
||||
"""获取会话统计信息"""
|
||||
stage_counter: Dict[str, int] = {}
|
||||
for session in self.sessions.values():
|
||||
stage_name = session.get("stage", "unknown")
|
||||
stage_counter[stage_name] = stage_counter.get(stage_name, 0) + 1
|
||||
|
||||
return {
|
||||
"active_sessions": len(self.sessions),
|
||||
"stages": stage_counter,
|
||||
}
|
||||
Reference in New Issue
Block a user