57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
from django.shortcuts import render
|
|
|
|
from agent_core.orchestrator import run_agent
|
|
from agent_core.results import AgentResult
|
|
from apps.audit.services import create_audit_log
|
|
from apps.documents.models import UploadedDocument
|
|
from apps.scenarios.services import ScenarioNotFound, get_scenario
|
|
|
|
from .forms import ChatForm
|
|
|
|
|
|
def index(request, scenario_id: str):
|
|
try:
|
|
scenario = get_scenario(scenario_id)
|
|
except ScenarioNotFound:
|
|
return render(
|
|
request,
|
|
"chat/index.html",
|
|
{
|
|
"scenario": None,
|
|
"form": ChatForm(),
|
|
"error": "场景不存在，请返回首页检查配置。",
|
|
},
|
|
status=404,
|
|
)
|
|
|
|
result = None
|
|
audit_log = None
|
|
documents = UploadedDocument.objects.filter(
|
|
scenario_id=scenario["id"],
|
|
status=UploadedDocument.STATUS_INDEXED,
|
|
)
|
|
form = ChatForm(request.POST or None, documents=documents)
|
|
if request.method == "POST" and form.is_valid():
|
|
message = form.cleaned_data["message"]
|
|
try:
|
|
result = run_agent(
|
|
scenario,
|
|
message,
|
|
options={"document_ids": form.cleaned_data["document_ids"]},
|
|
)
|
|
except Exception as exc:
|
|
result = AgentResult(status="failed", error=str(exc), answer="")
|
|
audit_log = create_audit_log(scenario["id"], scenario["name"], message, result)
|
|
|
|
return render(
|
|
request,
|
|
"chat/index.html",
|
|
{
|
|
"scenario": scenario,
|
|
"form": form,
|
|
"documents": documents,
|
|
"result": result,
|
|
"audit_log": audit_log,
|
|
},
|
|
)
|