优化:统一意图分类逻辑,复用 intent.py,删除冗余的 intent_classifier.py
All checks were successful
构建并部署 AI Agent 服务 / deploy (push) Successful in 5m44s

This commit is contained in:
2026-05-06 18:41:14 +08:00
parent 1dc1ecad62
commit 000af774a3
3 changed files with 64 additions and 143 deletions

View File

@@ -2,12 +2,6 @@
from .formatter import MarkdownFormatter
from .state_base import BaseState
from .intent_classifier import (
IntentType,
IntentResult,
IntentClassifier,
get_intent_classifier
)
from .human_review import (
ReviewManager,
InMemoryReviewStore,
@@ -27,30 +21,9 @@ from .visualization import (
generate_chart
)
# 为了兼容性,添加 classify_intent 函数
def classify_intent(user_input: str, context: str = None):
"""兼容旧代码的 classify_intent 函数"""
from backend.app.core.intent_classifier import get_intent_classifier
import asyncio
classifier = get_intent_classifier()
try:
loop = asyncio.get_event_loop()
if loop.is_running():
task = loop.create_task(classifier.classify(user_input, context))
return asyncio.run_coroutine_threadsafe(task, loop).result()
except RuntimeError:
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
return loop.run_until_complete(classifier.classify(user_input, context))
__all__ = [
"MarkdownFormatter",
"BaseState",
"IntentType",
"IntentResult",
"IntentClassifier",
"classify_intent",
"get_intent_classifier",
"ReviewManager",
"InMemoryReviewStore",
"ReviewStatus",
@@ -62,5 +35,5 @@ __all__ = [
"VisualizationTool",
"ChartData",
"get_visualization_tool",
"generate_chart"
"generate_chart",
]