fix: unknown error

api/controllers/inner_api/app.py

@@ -45,15 +45,16 @@ class EnterpriseAppInvokeApi(Resource):
 
         args = request_parser.parse_args()
 
-        app_id = args['app_id']
-        app_model: App = db.session.query(App).filter(App.id == app_id).first()
-        if app_model is None:
-            raise NotFound("App Not Exists.")
-        
-        # disable auto generate name
-        args['auto_generate_name'] = False
         
         try:
+            app_id = args['app_id']
+            app_model: App = db.session.query(App).filter(App.id == app_id).first()
+            if app_model is None:
+                raise NotFound("App Not Exists.")
+            
+            # disable auto generate name
+            args['auto_generate_name'] = False
+            
             response = CompletionService.completion(
                 app_model=app_model,
                 user=current_user,

api/core/app_runner/generate_task_pipeline.py

@@ -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