From 399bc08717518fdc0f7ed5d7c22ef52fd1b42f38 Mon Sep 17 00:00:00 2001
From: Chenjianyu <chenjianyu@@easycorp.ltd>
Date: Thu, 20 Jul 2023 14:19:33 +0800
Subject: [PATCH] * add a fallback for none exist demo data sources.

---
 module/ai/lang/en.php    | 1 +
 module/ai/lang/zh-cn.php | 1 +
 module/ai/model.php      | 2 ++
 3 files changed, 4 insertions(+)

diff --git a/module/ai/lang/en.php b/module/ai/lang/en.php
index c1ff98f79f..c0faac188a 100644
--- a/module/ai/lang/en.php
+++ b/module/ai/lang/en.php
@@ -379,6 +379,7 @@ $lang->ai->dataType->realStarted  = $lang->ai->dataType->datetime;
 $lang->ai->dataType->finishedDate = $lang->ai->dataType->datetime;
 
 $lang->ai->demoData            = new stdclass();
+$lang->ai->demoData->notExist  = 'The demo data does not exist for now.';
 $lang->ai->demoData->story     = array(
     'story' => array
     (
diff --git a/module/ai/lang/zh-cn.php b/module/ai/lang/zh-cn.php
index 61830ad617..6dd44dac17 100644
--- a/module/ai/lang/zh-cn.php
+++ b/module/ai/lang/zh-cn.php
@@ -380,6 +380,7 @@ $lang->ai->dataType->realStarted  = $lang->ai->dataType->datetime;
 $lang->ai->dataType->finishedDate = $lang->ai->dataType->datetime;
 
 $lang->ai->demoData            = new stdclass();
+$lang->ai->demoData->notExist  = '暂无演示数据。';
 $lang->ai->demoData->story     = array(
     'story' => array
     (
diff --git a/module/ai/model.php b/module/ai/model.php
index b7cccd2d12..410c40902e 100644
--- a/module/ai/model.php
+++ b/module/ai/model.php
@@ -575,6 +575,8 @@ class aiModel extends model
      */
     public function generateDemoDataPrompt($module, $source)
     {
+        if(empty($this->lang->ai->demoData->$module)) return $this->lang->ai->demoData->notExist;
+
         $sources = explode(',', $source);
         $sources = array_filter($sources);