From a9cf9b60a47beac1af2a1e6253d9d27b97226345 Mon Sep 17 00:00:00 2001
From: Chenjianyu <chenjianyu@@easycorp.ltd>
Date: Tue, 1 Aug 2023 11:08:10 +0800
Subject: [PATCH] * add get role templates method.

---
 module/ai/control.php |  1 +
 module/ai/model.php   | 13 +++++++++++++
 2 files changed, 14 insertions(+)

diff --git a/module/ai/control.php b/module/ai/control.php
index 262bab3390..c8bb4ea99d 100644
--- a/module/ai/control.php
+++ b/module/ai/control.php
@@ -271,6 +271,7 @@ class ai extends control
         $this->view->promptID       = $promptID;
         $this->view->lastActiveStep = $this->ai->getLastActiveStep($prompt);
         $this->view->title          = "{$this->lang->ai->prompts->common}#{$prompt->id} $prompt->name {$this->lang->colon} " . $this->lang->ai->prompts->assignRole . " {$this->lang->colon} " . $this->lang->ai->prompts->common;
+        $this->view->roleTemplates  = $this->ai->getRoleTemplates();
         $this->display();
     }
 
diff --git a/module/ai/model.php b/module/ai/model.php
index 501d0e5330..c064235c53 100644
--- a/module/ai/model.php
+++ b/module/ai/model.php
@@ -1134,4 +1134,17 @@ class aiModel extends model
 
         $_SESSION['aiInjectData'][$targetForm->m][$targetForm->f] = is_string($data) ? $data : json_encode($data);
     }
+
+    /**
+     * Get role templates from db.
+     *
+     * @access public
+     * @return array
+     */
+    public function getRoleTemplates()
+    {
+        return $this->dao->select('*')->from(TABLE_PROMPTROLE)
+            ->where('deleted')->eq(0)
+            ->fetchAll();
+    }
 }