From 6906adfb6a4fd9a0f644981bb20185bca815544b Mon Sep 17 00:00:00 2001
From: liwenrui <liwenrui@easycorp.ltd>
Date: Tue, 25 Jul 2023 13:44:01 +0800
Subject: [PATCH] * fix unable to generate url.

---
 module/ai/control.php | 4 +++-
 module/ai/model.php   | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/module/ai/control.php b/module/ai/control.php
index 18dbbe72b6..6fbf320d4f 100644
--- a/module/ai/control.php
+++ b/module/ai/control.php
@@ -447,8 +447,10 @@ class ai extends control
         $response = $this->ai->executePrompt($prompt, $object);
         if(empty($response)) return $this->send(array('result' => 'fail', 'message' => $this->lang->ai->execute->fail));
 
+        list($objectData, $rawObject) = $object;
+
         $this->ai->setInjectData($prompt->targetForm, $response);
-        $location = $this->ai->getTargetFormLocation($prompt, $object);
+        $location = $this->ai->getTargetFormLocation($prompt, $rawObject);
         if(empty($location)) return $this->send(array('result' => 'fail', 'message' => $this->lang->ai->execute->fail));
 
         $_SESSION['auditPrompt']['prompt'] = $prompt;
diff --git a/module/ai/model.php b/module/ai/model.php
index b1a547937b..91aa85a7c4 100644
--- a/module/ai/model.php
+++ b/module/ai/model.php
@@ -974,7 +974,7 @@ class aiModel extends model
             }
             elseif(!empty($object->$arg))
             {
-                $vars[] = $object->$arg;
+                $vars[] = is_object($object->$arg) && !empty($object->$arg->id) ? $object->$arg->id : $object->$arg;
             }
             else
             {