Procházet zdrojové kódy

feat: make Citations and Attributions display enable default (#5508)

Co-authored-by: StyleZhang <jasonapring2015@outlook.com>
takatost před 10 měsíci
rodič
revize
4a031de0d9

+ 3 - 3
api/constants/model_template.py

@@ -22,7 +22,7 @@ default_app_templates = {
         'model_config': {
             'model': {
                 "provider": "openai",
-                "name": "gpt-4",
+                "name": "gpt-4o",
                 "mode": "chat",
                 "completion_params": {}
             },
@@ -51,7 +51,7 @@ default_app_templates = {
         'model_config': {
             'model': {
                 "provider": "openai",
-                "name": "gpt-4",
+                "name": "gpt-4o",
                 "mode": "chat",
                 "completion_params": {}
             }
@@ -77,7 +77,7 @@ default_app_templates = {
         'model_config': {
             'model': {
                 "provider": "openai",
-                "name": "gpt-4",
+                "name": "gpt-4o",
                 "mode": "chat",
                 "completion_params": {}
             }

+ 4 - 0
api/core/app/apps/advanced_chat/app_generator.py

@@ -84,6 +84,10 @@ class AdvancedChatAppGenerator(MessageBasedAppGenerator):
             workflow=workflow
         )
 
+        if invoke_from == InvokeFrom.DEBUGGER:
+            # always enable retriever resource in debugger mode
+            app_config.additional_features.show_retrieve_source = True
+
         # init application generate entity
         application_generate_entity = AdvancedChatAppGenerateEntity(
             task_id=str(uuid.uuid4()),

+ 5 - 0
api/core/app/apps/agent_chat/app_generator.py

@@ -82,6 +82,11 @@ class AgentChatAppGenerator(MessageBasedAppGenerator):
                 config=args.get('model_config')
             )
 
+            # always enable retriever resource in debugger mode
+            override_model_config_dict["retriever_resource"] = {
+                "enabled": True
+            }
+
         # parse files
         files = args['files'] if args.get('files') else []
         message_file_parser = MessageFileParser(tenant_id=app_model.tenant_id, app_id=app_model.id)

+ 3 - 0
api/core/app/apps/chat/app_config_manager.py

@@ -50,6 +50,9 @@ class ChatAppConfigManager(BaseAppConfigManager):
             app_model_config_dict = app_model_config.to_dict()
             config_dict = app_model_config_dict.copy()
         else:
+            if not override_config_dict:
+                raise Exception('override_config_dict is required when config_from is ARGS')
+
             config_dict = override_config_dict
 
         app_mode = AppMode.value_of(app_model.mode)

+ 5 - 0
api/core/app/apps/chat/app_generator.py

@@ -79,6 +79,11 @@ class ChatAppGenerator(MessageBasedAppGenerator):
                 config=args.get('model_config')
             )
 
+            # always enable retriever resource in debugger mode
+            override_model_config_dict["retriever_resource"] = {
+                "enabled": True
+            }
+
         # parse files
         files = args['files'] if args.get('files') else []
         message_file_parser = MessageFileParser(tenant_id=app_model.tenant_id, app_id=app_model.id)

+ 1 - 1
api/models/model.py

@@ -265,7 +265,7 @@ class AppModelConfig(db.Model):
     @property
     def retriever_resource_dict(self) -> dict:
         return json.loads(self.retriever_resource) if self.retriever_resource \
-            else {"enabled": False}
+            else {"enabled": True}
 
     @property
     def annotation_reply_dict(self) -> dict:

+ 3 - 1
web/app/components/workflow/hooks/use-workflow.ts

@@ -485,7 +485,9 @@ export const useWorkflowInit = () => {
                   nodes: nodesTemplate,
                   edges: edgesTemplate,
                 },
-                features: {},
+                features: {
+                  retriever_resource: { enabled: true },
+                },
               },
             }).then((res) => {
               workflowStore.getState().setDraftUpdatedAt(res.updated_at)