优化:统一意图分类逻辑,复用 intent.py,删除冗余的 intent_classifier.py
All checks were successful
构建并部署 AI Agent 服务 / deploy (push) Successful in 5m44s
All checks were successful
构建并部署 AI Agent 服务 / deploy (push) Successful in 5m44s
This commit is contained in:
@@ -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",
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user