Unverified Commit ba019a26 authored by Yeuoly's avatar Yeuoly

fix: unknown error

parent 08bf23a5
......@@ -45,6 +45,8 @@ class EnterpriseAppInvokeApi(Resource):
args = request_parser.parse_args()
try:
app_id = args['app_id']
app_model: App = db.session.query(App).filter(App.id == app_id).first()
if app_model is None:
......@@ -53,7 +55,6 @@ class EnterpriseAppInvokeApi(Resource):
# disable auto generate name
args['auto_generate_name'] = False
try:
response = CompletionService.completion(
app_model=app_model,
user=current_user,
......
......@@ -523,7 +523,7 @@ class GenerateTaskPipeline:
# show_retrieve_source
if 'retriever_resources' in self._task_state.metadata:
if self._application_generate_entity.invoke_from in [InvokeFrom.DEBUGGER, InvokeFrom.SERVICE_API]:
if self._application_generate_entity.invoke_from in [InvokeFrom.DEBUGGER, InvokeFrom.SERVICE_API, InvokeFrom.INNER_API]:
metadata['retriever_resources'] = self._task_state.metadata['retriever_resources']
else:
metadata['retriever_resources'] = []
......@@ -537,11 +537,11 @@ class GenerateTaskPipeline:
})
# show annotation reply
if 'annotation_reply' in self._task_state.metadata:
if self._application_generate_entity.invoke_from in [InvokeFrom.DEBUGGER, InvokeFrom.SERVICE_API]:
if self._application_generate_entity.invoke_from in [InvokeFrom.DEBUGGER, InvokeFrom.SERVICE_API, InvokeFrom.INNER_API]:
metadata['annotation_reply'] = self._task_state.metadata['annotation_reply']
# show usage
if self._application_generate_entity.invoke_from in [InvokeFrom.DEBUGGER, InvokeFrom.SERVICE_API]:
if self._application_generate_entity.invoke_from in [InvokeFrom.DEBUGGER, InvokeFrom.SERVICE_API, InvokeFrom.INNER_API]:
metadata['usage'] = self._task_state.metadata['usage']
return metadata
......
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