Unverified Commit d26b7363 authored by Yeuoly's avatar Yeuoly

feat: rename api

parent d85ced8a
......@@ -4,4 +4,4 @@ from libs.external_api import ExternalApi
bp = Blueprint('inner_api', __name__, url_prefix='/inner/api')
api = ExternalApi(bp)
from . import authorization, model_runtime, app
\ No newline at end of file
from . import authorization, model_runtime, app, service
\ No newline at end of file
import json
from collections.abc import Generator
from typing import Union
from flask import Response
from flask.helpers import stream_with_context
from flask_restful import Resource, reqparse
from werkzeug.exceptions import InternalServerError
from controllers.console.setup import setup_required
from controllers.inner_api import api
from controllers.inner_api.wraps import inner_api_only
from services.conversation_service import ConversationService
class EnterpriseRenameApi(Resource):
"""Model invoke API for enterprise edition"""
@setup_required
@inner_api_only
def post(self):
request_parser = reqparse.RequestParser()
request_parser.add_argument("tenant_id", type=str, required=True, nullable=False, location="json")
request_parser.add_argument("query", type=str, required=True, nullable=False, location="json")
args = request_parser.parse_args()
return ConversationService.auto_generate_name_string(
tenant_id=args["tenant_id"],
query=args["query"]
)
api.add_resource(EnterpriseRenameApi, "/service/rename")
......@@ -102,6 +102,17 @@ class ConversationService:
db.session.commit()
return conversation
@classmethod
def auto_generate_name_string(cls, tenant_id: str, query: str):
try:
return {
'name': LLMGenerator.generate_conversation_name(tenant_id, query)
}
except:
return {
'name': ''
}
@classmethod
def get_conversation(cls, app_model: App, conversation_id: str, user: Optional[Union[Account, EndUser]]):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment