From 08714af5ed8438c27c121526e58764dd1b1d4537 Mon Sep 17 00:00:00 2001
From: zhaokexin <zhaokexin@kylinos.cn>
Date: Mon, 22 Apr 2024 09:39:17 +0800
Subject: [PATCH] =?UTF-8?q?nlp=E6=94=AF=E6=8C=81=E5=88=9B=E5=BB=BA?=
 =?UTF-8?q?=E5=A4=9A=E4=B8=AA=E9=80=9A=E4=BF=A1=E8=BF=9E=E6=8E=A5,?=
 =?UTF-8?q?=E6=AF=8F=E6=AC=A1=E9=9A=8F=E6=9C=BA=E4=BA=A7=E7=94=9F=E4=B8=80?=
 =?UTF-8?q?=E4=B8=AAsessionId=E7=94=A8=E4=BA=8E=E5=8C=BA=E5=88=86=E4=B8=8D?=
 =?UTF-8?q?=E5=90=8C=E7=9A=84=E8=BF=9E=E6=8E=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/services/textprocessor.cpp      | 123 +++++++++++-----
 src/services/textprocessor.h        |  38 +++--
 src/services/textprocessor.xml      |  13 +-
 src/services/textprocessorglue.c    | 218 +++++++++++++++++++++++-----
 src/services/textprocessorglue.h    |  49 ++++---
 src/services/textprocessorproxy.cpp | 106 ++++++++++----
 src/services/textprocessorproxy.h   |  26 +++-
 7 files changed, 426 insertions(+), 147 deletions(-)

diff --git a/src/services/textprocessor.cpp b/src/services/textprocessor.cpp
index 5abf13e..3f43f49 100644
--- a/src/services/textprocessor.cpp
+++ b/src/services/textprocessor.cpp
@@ -24,6 +24,8 @@
 const std::string TextProcessor::objectPath_ =
     "/org/openkylin/aisdk/textprocessor";
 
+std::set<int> TextProcessor::sessionIdSet;
+
 TextProcessor::TextProcessor(GDBusConnection &connection,
     ai_engine::AiEnginePluginManager &aiEnginePluginManager)
     : connection_(connection)
@@ -37,16 +39,20 @@ TextProcessor::~TextProcessor() {
 
 void TextProcessor::stopProcess()
 {
-    if (nlpEngine_) {
-        nlpEngine_->stopChat();
+    gintAndNlpEngineMap::iterator iter = nlpEngineMap_.begin();
+    while (iter != nlpEngineMap_.end()) {
+        iter->second->stopChat();
+        ++iter;
     }
 
     std::lock_guard<std::mutex> locker(mutex_);
     stopped = true;
 }
 
-void TextProcessor::setChatCallback() {
-    nlpEngine_->setChatResultCallback([this](const std::string &result) {
+void TextProcessor::setSessionChatCallback(gint sessionId) {
+    nlpEngineMap_.at(sessionId)->setChatResultCallback(
+                [this, sessionId](const std::string &result) {
+        std::cout << "setChatCallback sessionId " << sessionId << std::endl;
         std::cout << "aisdk_text_processor_complete_chat: \n " << result << std::endl;
         {
             std::lock_guard<std::mutex> locker(mutex_);
@@ -55,12 +61,15 @@ void TextProcessor::setChatCallback() {
             }
         }
 
+        std::string interfaceName = "org.openkylin.aisdk.textprocessor"
+                + std::to_string((int)sessionId);
+
         std::shared_ptr<TextProcessor> textProcessor =
                 static_cast<TextProcessor *>(this)->getSharedObject();
         g_dbus_connection_emit_signal(&(textProcessor->connection_),
                                       "org.openkylin.aisdk.textprocessor",
                                       "/org/openkylin/aisdk/textprocessor",
-                                      "org.openkylin.aisdk.textprocessor",
+                                      interfaceName.c_str(),
                                       "ChatResult",
                                       g_variant_new("(s)", result.c_str()), nullptr);
     });
@@ -72,6 +81,7 @@ void TextProcessor::onHandleInitEngine(AisdkTextProcessor *delegate,
 {
     std::cout << "On init engine..." <<  engineName << appId << apiKey
               << secretKey << invocation << std::endl;
+
     TextProcessor *textProcessor = static_cast<TextProcessor *>(userData);
     if (!textProcessor) {
         return;
@@ -80,22 +90,33 @@ void TextProcessor::onHandleInitEngine(AisdkTextProcessor *delegate,
         new config::PromptTemplate(config::promptTemplateConfigFile(engineName),
                                    config::promptFilePath(engineName))
     );
-    textProcessor->initEngine(engineName, appId, apiKey, secretKey);
-    textProcessor->setChatCallback();
+
+    int sessionId = textProcessor->generateRandomSessionId();
+    bool initResult = textProcessor->initEngine(
+                engineName, appId, apiKey, secretKey, sessionId);
+
+    if (initResult) {
+        textProcessor->setSessionChatCallback(sessionId);
+    }
+
+    aisdk_text_processor_complete_init(delegate, invocation, sessionId, initResult);
+    std::cout << "On init engine..." << sessionId << std::endl;
 }
 
-bool TextProcessor::onHandleChat(AisdkTextProcessor *delegate,
-                  GDBusMethodInvocation *invocation,
-                  gchar *question, gpointer userData) {
+bool TextProcessor::onHandleChat(
+        AisdkTextProcessor *delegate, GDBusMethodInvocation *invocation,
+        gchar *question, gint sessionId, gpointer userData) {
 
     std::cout << "On handle chat..." << invocation << std::endl;
+
     std::shared_ptr<TextProcessor> textProcessor = static_cast<TextProcessor *>(userData)->getSharedObject();
-    if (!textProcessor) {
+    if (!textProcessor ||
+            textProcessor->nlpEngineMap_.find(sessionId) == textProcessor->nlpEngineMap_.end()) {
         return false;
     }
 
     std::string userQuestion = question;
-    cpr::async([delegate, userQuestion, textProcessor]() {
+    cpr::async([delegate, userQuestion, textProcessor, sessionId]() {
         std::string message;
         {
             std::lock_guard<std::mutex> locker(textProcessor->mutex_);
@@ -104,15 +125,15 @@ bool TextProcessor::onHandleChat(AisdkTextProcessor *delegate,
             textProcessor->actorChanged = false;
         }
 
-        textProcessor->nlpEngine_->chat(message);
+        textProcessor->nlpEngineMap_.at(sessionId)->chat(message);
     });
 
     return true;
 }
 
-void TextProcessor::onSetActor(AisdkTextProcessor *delegate,
-           GDBusMethodInvocation *invocation, gint actor,
-                gpointer userData) {
+void TextProcessor::onSetActor(
+        AisdkTextProcessor *delegate, GDBusMethodInvocation *invocation,
+        gint actor, gint sessionId, gpointer userData) {
     std::cout << "On Set Actor..." << std::endl;
     TextProcessor *textProcessor = static_cast<TextProcessor *>(userData);
     if (!textProcessor) {
@@ -132,41 +153,44 @@ void TextProcessor::onSetActor(AisdkTextProcessor *delegate,
     }
 }
 
-void TextProcessor::onStopChat(AisdkTextProcessor *delegate,
-                GDBusMethodInvocation *invocation,
-                gpointer userData) {
+void TextProcessor::onStopChat(
+        AisdkTextProcessor *delegate, GDBusMethodInvocation *invocation,
+        gint sessionId, gpointer userData) {
     std::cout << "On Stop Chat..." << std::endl;
     TextProcessor *textProcessor = static_cast<TextProcessor *>(userData);
-    if (!textProcessor) {
+    if (!textProcessor ||
+            textProcessor->nlpEngineMap_.find(sessionId) == textProcessor->nlpEngineMap_.end()) {
         return;
     }
 
-    textProcessor->nlpEngine_->stopChat();
+    textProcessor->nlpEngineMap_.at(sessionId)->stopChat();
 }
 
-void TextProcessor::onSetContextSize(AisdkTextProcessor *delegate,
-                GDBusMethodInvocation *invocation, gint size,
-                gpointer userData) {
-    std::cout << "On Set Context Size..." << std::endl;
+void TextProcessor::onSetContextSize(
+        AisdkTextProcessor *delegate, GDBusMethodInvocation *invocation,
+        gint size, gint sessionId, gpointer userData) {
+    std::cout << "On Set Context Size...sessionId " << sessionId << std::endl;
     TextProcessor *textProcessor = static_cast<TextProcessor *>(userData);
-    if (!textProcessor) {
+    if (!textProcessor ||
+            textProcessor->nlpEngineMap_.find(sessionId) == textProcessor->nlpEngineMap_.end()) {
         return;
     }
 
-    textProcessor->nlpEngine_->setContextSize(size);
+    textProcessor->nlpEngineMap_.at(sessionId)->setContextSize(size);
 }
 
-void TextProcessor::onClearContext(AisdkTextProcessor *delegate,
-                      GDBusMethodInvocation *invocation,
-                      gpointer userData) {
+void TextProcessor::onClearContext(
+        AisdkTextProcessor *delegate, GDBusMethodInvocation *invocation,
+        gint sessionId, gpointer userData) {
     std::cout << "On Clear Context..." << std::endl;
 
     TextProcessor *textProcessor = static_cast<TextProcessor *>(userData);
-    if (!textProcessor) {
+    if (!textProcessor ||
+            textProcessor->nlpEngineMap_.find(sessionId) == textProcessor->nlpEngineMap_.end()) {
         return;
     }
 
-    textProcessor->nlpEngine_->clearContext();
+    textProcessor->nlpEngineMap_.at(sessionId)->clearContext();
 }
 
 void TextProcessor::exportSkeleton() {
@@ -211,17 +235,36 @@ void TextProcessor::unexportSkeleton() {
         g_dbus_interface_skeleton_unexport_from_connection(
             G_DBUS_INTERFACE_SKELETON(delegate_), &connection_);
     }
-
+    sessionIdSet.clear();
     g_object_unref(delegate_);
 }
 
-void TextProcessor::initEngine(const std::string &engineName,
-    const std::string &appId, const std::string &apiKey, const std::string &secretKey)
+bool TextProcessor::initEngine(const std::string &engineName,
+    const std::string &appId, const std::string &apiKey, const std::string &secretKey, gint sessionId)
 {
-    nlpEngine_ = aiEnginePluginManager_.createNlpEngine(engineName);
-    if (nlpEngine_->isCloud()) {
-        static_cast<ai_engine::lm::nlp::AbstractCloudNlpEngine *>(nlpEngine_.get())->setAppId(appId);
-        static_cast<ai_engine::lm::nlp::AbstractCloudNlpEngine *>(nlpEngine_.get())->setApiKey(apiKey);
-        static_cast<ai_engine::lm::nlp::AbstractCloudNlpEngine *>(nlpEngine_.get())->setSecretKey(secretKey);
+    auto nlpEngine = aiEnginePluginManager_.createNlpEngine(engineName);
+    if (!nlpEngine) {
+        std::cout << "initEngine nlpEngine is nullptr !" << std::endl;
+        return false;
+    }
+    nlpEngineMap_[sessionId] = std::move(nlpEngine);
+
+    if (nlpEngineMap_.at(sessionId)->isCloud()) {
+        static_cast<ai_engine::lm::nlp::AbstractCloudNlpEngine *>(
+                    nlpEngineMap_.at(sessionId).get())->setAppId(appId);
+        static_cast<ai_engine::lm::nlp::AbstractCloudNlpEngine *>(
+                    nlpEngineMap_.at(sessionId).get())->setApiKey(apiKey);
+        static_cast<ai_engine::lm::nlp::AbstractCloudNlpEngine *>(
+                    nlpEngineMap_.at(sessionId).get())->setSecretKey(secretKey);
+    }
+    return true;
+}
+
+int TextProcessor::generateRandomSessionId() {
+    int sessionId = std::rand();
+    while (sessionIdSet.find(sessionId) != sessionIdSet.end()) {
+        sessionId = std::rand();
     }
+    sessionIdSet.insert(sessionId);
+    return sessionId;
 }
diff --git a/src/services/textprocessor.h b/src/services/textprocessor.h
index 5fc6738..031a586 100644
--- a/src/services/textprocessor.h
+++ b/src/services/textprocessor.h
@@ -21,6 +21,7 @@
 #include <string>
 #include <memory>
 #include <mutex>
+#include <set>
 #include <kylin-ai/ai-engine/large-model/nlpengine.h>
 #include "engine/aienginepluginmanager.h"
 #include "config/prompttemplate.h"
@@ -39,29 +40,35 @@ public:
 private:
     void exportSkeleton();
     void unexportSkeleton();
+    int generateRandomSessionId();
 
-    void initEngine(const std::string &engineName, const std::string &appId,
-        const std::string &apiKey, const std::string &secretKey);
-    void setChatCallback();
+    bool initEngine(const std::string &engineName, const std::string &appId,
+        const std::string &apiKey, const std::string &secretKey, gint sessionId);
+    void setSessionChatCallback(gint sessionId);
     
     static void onHandleInitEngine(AisdkTextProcessor *delegate,
         GDBusMethodInvocation *invocation, gchar *engineName, gchar *appId,
         gchar *apiKey, gchar *secretKey, gpointer userData);
 
-    static bool onHandleChat(AisdkTextProcessor *delegate,
-        GDBusMethodInvocation *invocation, gchar *question, gpointer userData);
+    static bool onHandleChat(
+            AisdkTextProcessor *delegate, GDBusMethodInvocation *invocation,
+            gchar *question, gint sessionId, gpointer userData);
 
-    static void onSetActor(AisdkTextProcessor *delegate,
-        GDBusMethodInvocation *invocation, gint actor, gpointer userData);
+    static void onSetActor(
+            AisdkTextProcessor *delegate, GDBusMethodInvocation *invocation,
+            gint actor, gint sessionId, gpointer userData);
 
-    static void onStopChat(AisdkTextProcessor *delegate,
-        GDBusMethodInvocation *invocation, gpointer userData);
+    static void onStopChat(
+            AisdkTextProcessor *delegate, GDBusMethodInvocation *invocation,
+            gint sessionId, gpointer userData);
 
-    static void onSetContextSize(AisdkTextProcessor *delegate,
-         GDBusMethodInvocation *invocation, gint size, gpointer userData);
+    static void onSetContextSize(
+            AisdkTextProcessor *delegate, GDBusMethodInvocation *invocation,
+            gint size, gint sessionId, gpointer userData);
 
-    static void onClearContext(AisdkTextProcessor *delegate,
-        GDBusMethodInvocation *invocation, gpointer userData);
+    static void onClearContext(
+            AisdkTextProcessor *delegate, GDBusMethodInvocation *invocation,
+            gint sessionId, gpointer userData);
 
 private:
     std::shared_ptr<TextProcessor> getSharedObject() {
@@ -73,7 +80,6 @@ private:
     AisdkTextProcessor *delegate_ = nullptr;
     GDBusConnection &connection_;
     ai_engine::AiEnginePluginManager &aiEnginePluginManager_;
-    std::unique_ptr<ai_engine::lm::nlp::AbstractNlpEngine> nlpEngine_ { nullptr };
     std::unique_ptr<config::PromptTemplate> promptTemplate_;
     std::string currentPrompt_;
     int actorId_ { -1 };
@@ -82,6 +88,10 @@ private:
     std::mutex mutex_;
 
     static const std::string objectPath_;
+
+    static std::set<int> sessionIdSet;
+    typedef std::map<gint, std::unique_ptr<ai_engine::lm::nlp::AbstractNlpEngine>> gintAndNlpEngineMap;
+    gintAndNlpEngineMap nlpEngineMap_;
 };
 
 #endif
diff --git a/src/services/textprocessor.xml b/src/services/textprocessor.xml
index ce3f1d3..6f80d58 100644
--- a/src/services/textprocessor.xml
+++ b/src/services/textprocessor.xml
@@ -2,12 +2,12 @@
 
 <!--
  Copyright 2024 KylinSoft Co., Ltd.
- 
+
  This program is free software: you can redistribute it and/or modify it under
  the terms of the GNU General Public License as published by the Free Software
  Foundation, either version 3 of the License, or (at your option) any later
  version.
- 
+
  This program is distributed in the hope that it will be useful, but WITHOUT
  ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
@@ -24,24 +24,31 @@
             <arg name="app_id" type="s" direction="in" />
             <arg name="api_key" type="s" direction="in" />
             <arg name="secret_key" type="s" direction="in" />
+            <arg name="session_Id" type="i" direction="out" />
+            <arg name="initResult" type="b" direction="out" />
         </method>
         <method name="chat">
             <arg name="question" type="s" direction="in" />
             <arg name="answer" type="s" direction="out" />
+            <arg name="session_Id" type="i" direction="in" />
         </method>
 
         <method name="stop_chat">
+        <arg name="session_Id" type="i" direction="in" />
         </method>
-        
+
         <method name="set_actor">
             <arg name="actor" type="i" direction="in" />
+            <arg name="session_Id" type="i" direction="in" />
         </method>
 
         <method name="set_context">
             <arg name="size" type="i" direction="in" />
+            <arg name="session_Id" type="i" direction="in" />
         </method>
 
         <method name="clear_context">
+            <arg name="session_Id" type="i" direction="in" />
         </method>
 
     </interface>
diff --git a/src/services/textprocessorglue.c b/src/services/textprocessorglue.c
index 0057bac..324eb44 100644
--- a/src/services/textprocessorglue.c
+++ b/src/services/textprocessorglue.c
@@ -219,13 +219,42 @@ static const GDBusArgInfo * const _aisdk_text_processor_method_info_init_IN_ARG_
   NULL
 };
 
+static const _ExtendedGDBusArgInfo _aisdk_text_processor_method_info_init_OUT_ARG_session_Id =
+{
+  {
+    -1,
+    (gchar *) "session_Id",
+    (gchar *) "i",
+    NULL
+  },
+  FALSE
+};
+
+static const _ExtendedGDBusArgInfo _aisdk_text_processor_method_info_init_OUT_ARG_initResult =
+{
+  {
+    -1,
+    (gchar *) "initResult",
+    (gchar *) "b",
+    NULL
+  },
+  FALSE
+};
+
+static const GDBusArgInfo * const _aisdk_text_processor_method_info_init_OUT_ARG_pointers[] =
+{
+  &_aisdk_text_processor_method_info_init_OUT_ARG_session_Id.parent_struct,
+  &_aisdk_text_processor_method_info_init_OUT_ARG_initResult.parent_struct,
+  NULL
+};
+
 static const _ExtendedGDBusMethodInfo _aisdk_text_processor_method_info_init =
 {
   {
     -1,
     (gchar *) "init",
     (GDBusArgInfo **) &_aisdk_text_processor_method_info_init_IN_ARG_pointers,
-    NULL,
+    (GDBusArgInfo **) &_aisdk_text_processor_method_info_init_OUT_ARG_pointers,
     NULL
   },
   "handle-init",
@@ -243,9 +272,21 @@ static const _ExtendedGDBusArgInfo _aisdk_text_processor_method_info_chat_IN_ARG
   FALSE
 };
 
+static const _ExtendedGDBusArgInfo _aisdk_text_processor_method_info_chat_IN_ARG_session_Id =
+{
+  {
+    -1,
+    (gchar *) "session_Id",
+    (gchar *) "i",
+    NULL
+  },
+  FALSE
+};
+
 static const GDBusArgInfo * const _aisdk_text_processor_method_info_chat_IN_ARG_pointers[] =
 {
   &_aisdk_text_processor_method_info_chat_IN_ARG_question.parent_struct,
+  &_aisdk_text_processor_method_info_chat_IN_ARG_session_Id.parent_struct,
   NULL
 };
 
@@ -279,12 +320,29 @@ static const _ExtendedGDBusMethodInfo _aisdk_text_processor_method_info_chat =
   FALSE
 };
 
+static const _ExtendedGDBusArgInfo _aisdk_text_processor_method_info_stop_chat_IN_ARG_session_Id =
+{
+  {
+    -1,
+    (gchar *) "session_Id",
+    (gchar *) "i",
+    NULL
+  },
+  FALSE
+};
+
+static const GDBusArgInfo * const _aisdk_text_processor_method_info_stop_chat_IN_ARG_pointers[] =
+{
+  &_aisdk_text_processor_method_info_stop_chat_IN_ARG_session_Id.parent_struct,
+  NULL
+};
+
 static const _ExtendedGDBusMethodInfo _aisdk_text_processor_method_info_stop_chat =
 {
   {
     -1,
     (gchar *) "stop_chat",
-    NULL,
+    (GDBusArgInfo **) &_aisdk_text_processor_method_info_stop_chat_IN_ARG_pointers,
     NULL,
     NULL
   },
@@ -303,9 +361,21 @@ static const _ExtendedGDBusArgInfo _aisdk_text_processor_method_info_set_actor_I
   FALSE
 };
 
+static const _ExtendedGDBusArgInfo _aisdk_text_processor_method_info_set_actor_IN_ARG_session_Id =
+{
+  {
+    -1,
+    (gchar *) "session_Id",
+    (gchar *) "i",
+    NULL
+  },
+  FALSE
+};
+
 static const GDBusArgInfo * const _aisdk_text_processor_method_info_set_actor_IN_ARG_pointers[] =
 {
   &_aisdk_text_processor_method_info_set_actor_IN_ARG_actor.parent_struct,
+  &_aisdk_text_processor_method_info_set_actor_IN_ARG_session_Id.parent_struct,
   NULL
 };
 
@@ -333,9 +403,21 @@ static const _ExtendedGDBusArgInfo _aisdk_text_processor_method_info_set_context
   FALSE
 };
 
+static const _ExtendedGDBusArgInfo _aisdk_text_processor_method_info_set_context_IN_ARG_session_Id =
+{
+  {
+    -1,
+    (gchar *) "session_Id",
+    (gchar *) "i",
+    NULL
+  },
+  FALSE
+};
+
 static const GDBusArgInfo * const _aisdk_text_processor_method_info_set_context_IN_ARG_pointers[] =
 {
   &_aisdk_text_processor_method_info_set_context_IN_ARG_size.parent_struct,
+  &_aisdk_text_processor_method_info_set_context_IN_ARG_session_Id.parent_struct,
   NULL
 };
 
@@ -352,12 +434,29 @@ static const _ExtendedGDBusMethodInfo _aisdk_text_processor_method_info_set_cont
   FALSE
 };
 
+static const _ExtendedGDBusArgInfo _aisdk_text_processor_method_info_clear_context_IN_ARG_session_Id =
+{
+  {
+    -1,
+    (gchar *) "session_Id",
+    (gchar *) "i",
+    NULL
+  },
+  FALSE
+};
+
+static const GDBusArgInfo * const _aisdk_text_processor_method_info_clear_context_IN_ARG_pointers[] =
+{
+  &_aisdk_text_processor_method_info_clear_context_IN_ARG_session_Id.parent_struct,
+  NULL
+};
+
 static const _ExtendedGDBusMethodInfo _aisdk_text_processor_method_info_clear_context =
 {
   {
     -1,
     (gchar *) "clear_context",
-    NULL,
+    (GDBusArgInfo **) &_aisdk_text_processor_method_info_clear_context_IN_ARG_pointers,
     NULL,
     NULL
   },
@@ -478,6 +577,7 @@ aisdk_text_processor_default_init (AisdkTextProcessorIface *iface)
    * @object: A #AisdkTextProcessor.
    * @invocation: A #GDBusMethodInvocation.
    * @arg_question: Argument passed by remote caller.
+   * @arg_session_Id: Argument passed by remote caller.
    *
    * Signal emitted when a remote caller is invoking the <link linkend="gdbus-method-org-openkylin-aisdk-TextProcessor.chat">chat()</link> D-Bus method.
    *
@@ -493,13 +593,14 @@ aisdk_text_processor_default_init (AisdkTextProcessorIface *iface)
     NULL,
     g_cclosure_marshal_generic,
     G_TYPE_BOOLEAN,
-    2,
-    G_TYPE_DBUS_METHOD_INVOCATION, G_TYPE_STRING);
+    3,
+    G_TYPE_DBUS_METHOD_INVOCATION, G_TYPE_STRING, G_TYPE_INT);
 
   /**
    * AisdkTextProcessor::handle-stop-chat:
    * @object: A #AisdkTextProcessor.
    * @invocation: A #GDBusMethodInvocation.
+   * @arg_session_Id: Argument passed by remote caller.
    *
    * Signal emitted when a remote caller is invoking the <link linkend="gdbus-method-org-openkylin-aisdk-TextProcessor.stop_chat">stop_chat()</link> D-Bus method.
    *
@@ -515,14 +616,15 @@ aisdk_text_processor_default_init (AisdkTextProcessorIface *iface)
     NULL,
     g_cclosure_marshal_generic,
     G_TYPE_BOOLEAN,
-    1,
-    G_TYPE_DBUS_METHOD_INVOCATION);
+    2,
+    G_TYPE_DBUS_METHOD_INVOCATION, G_TYPE_INT);
 
   /**
    * AisdkTextProcessor::handle-set-actor:
    * @object: A #AisdkTextProcessor.
    * @invocation: A #GDBusMethodInvocation.
    * @arg_actor: Argument passed by remote caller.
+   * @arg_session_Id: Argument passed by remote caller.
    *
    * Signal emitted when a remote caller is invoking the <link linkend="gdbus-method-org-openkylin-aisdk-TextProcessor.set_actor">set_actor()</link> D-Bus method.
    *
@@ -538,14 +640,15 @@ aisdk_text_processor_default_init (AisdkTextProcessorIface *iface)
     NULL,
     g_cclosure_marshal_generic,
     G_TYPE_BOOLEAN,
-    2,
-    G_TYPE_DBUS_METHOD_INVOCATION, G_TYPE_INT);
+    3,
+    G_TYPE_DBUS_METHOD_INVOCATION, G_TYPE_INT, G_TYPE_INT);
 
   /**
    * AisdkTextProcessor::handle-set-context:
    * @object: A #AisdkTextProcessor.
    * @invocation: A #GDBusMethodInvocation.
    * @arg_size: Argument passed by remote caller.
+   * @arg_session_Id: Argument passed by remote caller.
    *
    * Signal emitted when a remote caller is invoking the <link linkend="gdbus-method-org-openkylin-aisdk-TextProcessor.set_context">set_context()</link> D-Bus method.
    *
@@ -561,13 +664,14 @@ aisdk_text_processor_default_init (AisdkTextProcessorIface *iface)
     NULL,
     g_cclosure_marshal_generic,
     G_TYPE_BOOLEAN,
-    2,
-    G_TYPE_DBUS_METHOD_INVOCATION, G_TYPE_INT);
+    3,
+    G_TYPE_DBUS_METHOD_INVOCATION, G_TYPE_INT, G_TYPE_INT);
 
   /**
    * AisdkTextProcessor::handle-clear-context:
    * @object: A #AisdkTextProcessor.
    * @invocation: A #GDBusMethodInvocation.
+   * @arg_session_Id: Argument passed by remote caller.
    *
    * Signal emitted when a remote caller is invoking the <link linkend="gdbus-method-org-openkylin-aisdk-TextProcessor.clear_context">clear_context()</link> D-Bus method.
    *
@@ -583,8 +687,8 @@ aisdk_text_processor_default_init (AisdkTextProcessorIface *iface)
     NULL,
     g_cclosure_marshal_generic,
     G_TYPE_BOOLEAN,
-    1,
-    G_TYPE_DBUS_METHOD_INVOCATION);
+    2,
+    G_TYPE_DBUS_METHOD_INVOCATION, G_TYPE_INT);
 
 }
 
@@ -633,6 +737,8 @@ aisdk_text_processor_call_init (
 /**
  * aisdk_text_processor_call_init_finish:
  * @proxy: A #AisdkTextProcessorProxy.
+ * @out_session_Id: (out) (optional): Return location for return parameter or %NULL to ignore.
+ * @out_initResult: (out) (optional): Return location for return parameter or %NULL to ignore.
  * @res: The #GAsyncResult obtained from the #GAsyncReadyCallback passed to aisdk_text_processor_call_init().
  * @error: Return location for error or %NULL.
  *
@@ -643,6 +749,8 @@ aisdk_text_processor_call_init (
 gboolean
 aisdk_text_processor_call_init_finish (
     AisdkTextProcessor *proxy,
+    gint *out_session_Id,
+    gboolean *out_initResult,
     GAsyncResult *res,
     GError **error)
 {
@@ -651,7 +759,9 @@ aisdk_text_processor_call_init_finish (
   if (_ret == NULL)
     goto _out;
   g_variant_get (_ret,
-                 "()");
+                 "(ib)",
+                 out_session_Id,
+                 out_initResult);
   g_variant_unref (_ret);
 _out:
   return _ret != NULL;
@@ -664,6 +774,8 @@ _out:
  * @arg_app_id: Argument to pass with the method invocation.
  * @arg_api_key: Argument to pass with the method invocation.
  * @arg_secret_key: Argument to pass with the method invocation.
+ * @out_session_Id: (out) (optional): Return location for return parameter or %NULL to ignore.
+ * @out_initResult: (out) (optional): Return location for return parameter or %NULL to ignore.
  * @cancellable: (nullable): A #GCancellable or %NULL.
  * @error: Return location for error or %NULL.
  *
@@ -680,6 +792,8 @@ aisdk_text_processor_call_init_sync (
     const gchar *arg_app_id,
     const gchar *arg_api_key,
     const gchar *arg_secret_key,
+    gint *out_session_Id,
+    gboolean *out_initResult,
     GCancellable *cancellable,
     GError **error)
 {
@@ -698,7 +812,9 @@ aisdk_text_processor_call_init_sync (
   if (_ret == NULL)
     goto _out;
   g_variant_get (_ret,
-                 "()");
+                 "(ib)",
+                 out_session_Id,
+                 out_initResult);
   g_variant_unref (_ret);
 _out:
   return _ret != NULL;
@@ -708,6 +824,7 @@ _out:
  * aisdk_text_processor_call_chat:
  * @proxy: A #AisdkTextProcessorProxy.
  * @arg_question: Argument to pass with the method invocation.
+ * @arg_session_Id: Argument to pass with the method invocation.
  * @cancellable: (nullable): A #GCancellable or %NULL.
  * @callback: A #GAsyncReadyCallback to call when the request is satisfied or %NULL.
  * @user_data: User data to pass to @callback.
@@ -722,14 +839,16 @@ void
 aisdk_text_processor_call_chat (
     AisdkTextProcessor *proxy,
     const gchar *arg_question,
+    gint arg_session_Id,
     GCancellable *cancellable,
     GAsyncReadyCallback callback,
     gpointer user_data)
 {
   g_dbus_proxy_call (G_DBUS_PROXY (proxy),
     "chat",
-    g_variant_new ("(s)",
-                   arg_question),
+    g_variant_new ("(si)",
+                   arg_question,
+                   arg_session_Id),
     G_DBUS_CALL_FLAGS_NONE,
     -1,
     cancellable,
@@ -771,6 +890,7 @@ _out:
  * aisdk_text_processor_call_chat_sync:
  * @proxy: A #AisdkTextProcessorProxy.
  * @arg_question: Argument to pass with the method invocation.
+ * @arg_session_Id: Argument to pass with the method invocation.
  * @out_answer: (out) (optional): Return location for return parameter or %NULL to ignore.
  * @cancellable: (nullable): A #GCancellable or %NULL.
  * @error: Return location for error or %NULL.
@@ -785,6 +905,7 @@ gboolean
 aisdk_text_processor_call_chat_sync (
     AisdkTextProcessor *proxy,
     const gchar *arg_question,
+    gint arg_session_Id,
     gchar **out_answer,
     GCancellable *cancellable,
     GError **error)
@@ -792,8 +913,9 @@ aisdk_text_processor_call_chat_sync (
   GVariant *_ret;
   _ret = g_dbus_proxy_call_sync (G_DBUS_PROXY (proxy),
     "chat",
-    g_variant_new ("(s)",
-                   arg_question),
+    g_variant_new ("(si)",
+                   arg_question,
+                   arg_session_Id),
     G_DBUS_CALL_FLAGS_NONE,
     -1,
     cancellable,
@@ -811,6 +933,7 @@ _out:
 /**
  * aisdk_text_processor_call_stop_chat:
  * @proxy: A #AisdkTextProcessorProxy.
+ * @arg_session_Id: Argument to pass with the method invocation.
  * @cancellable: (nullable): A #GCancellable or %NULL.
  * @callback: A #GAsyncReadyCallback to call when the request is satisfied or %NULL.
  * @user_data: User data to pass to @callback.
@@ -824,13 +947,15 @@ _out:
 void
 aisdk_text_processor_call_stop_chat (
     AisdkTextProcessor *proxy,
+    gint arg_session_Id,
     GCancellable *cancellable,
     GAsyncReadyCallback callback,
     gpointer user_data)
 {
   g_dbus_proxy_call (G_DBUS_PROXY (proxy),
     "stop_chat",
-    g_variant_new ("()"),
+    g_variant_new ("(i)",
+                   arg_session_Id),
     G_DBUS_CALL_FLAGS_NONE,
     -1,
     cancellable,
@@ -868,6 +993,7 @@ _out:
 /**
  * aisdk_text_processor_call_stop_chat_sync:
  * @proxy: A #AisdkTextProcessorProxy.
+ * @arg_session_Id: Argument to pass with the method invocation.
  * @cancellable: (nullable): A #GCancellable or %NULL.
  * @error: Return location for error or %NULL.
  *
@@ -880,13 +1006,15 @@ _out:
 gboolean
 aisdk_text_processor_call_stop_chat_sync (
     AisdkTextProcessor *proxy,
+    gint arg_session_Id,
     GCancellable *cancellable,
     GError **error)
 {
   GVariant *_ret;
   _ret = g_dbus_proxy_call_sync (G_DBUS_PROXY (proxy),
     "stop_chat",
-    g_variant_new ("()"),
+    g_variant_new ("(i)",
+                   arg_session_Id),
     G_DBUS_CALL_FLAGS_NONE,
     -1,
     cancellable,
@@ -904,6 +1032,7 @@ _out:
  * aisdk_text_processor_call_set_actor:
  * @proxy: A #AisdkTextProcessorProxy.
  * @arg_actor: Argument to pass with the method invocation.
+ * @arg_session_Id: Argument to pass with the method invocation.
  * @cancellable: (nullable): A #GCancellable or %NULL.
  * @callback: A #GAsyncReadyCallback to call when the request is satisfied or %NULL.
  * @user_data: User data to pass to @callback.
@@ -918,14 +1047,16 @@ void
 aisdk_text_processor_call_set_actor (
     AisdkTextProcessor *proxy,
     gint arg_actor,
+    gint arg_session_Id,
     GCancellable *cancellable,
     GAsyncReadyCallback callback,
     gpointer user_data)
 {
   g_dbus_proxy_call (G_DBUS_PROXY (proxy),
     "set_actor",
-    g_variant_new ("(i)",
-                   arg_actor),
+    g_variant_new ("(ii)",
+                   arg_actor,
+                   arg_session_Id),
     G_DBUS_CALL_FLAGS_NONE,
     -1,
     cancellable,
@@ -964,6 +1095,7 @@ _out:
  * aisdk_text_processor_call_set_actor_sync:
  * @proxy: A #AisdkTextProcessorProxy.
  * @arg_actor: Argument to pass with the method invocation.
+ * @arg_session_Id: Argument to pass with the method invocation.
  * @cancellable: (nullable): A #GCancellable or %NULL.
  * @error: Return location for error or %NULL.
  *
@@ -977,14 +1109,16 @@ gboolean
 aisdk_text_processor_call_set_actor_sync (
     AisdkTextProcessor *proxy,
     gint arg_actor,
+    gint arg_session_Id,
     GCancellable *cancellable,
     GError **error)
 {
   GVariant *_ret;
   _ret = g_dbus_proxy_call_sync (G_DBUS_PROXY (proxy),
     "set_actor",
-    g_variant_new ("(i)",
-                   arg_actor),
+    g_variant_new ("(ii)",
+                   arg_actor,
+                   arg_session_Id),
     G_DBUS_CALL_FLAGS_NONE,
     -1,
     cancellable,
@@ -1002,6 +1136,7 @@ _out:
  * aisdk_text_processor_call_set_context:
  * @proxy: A #AisdkTextProcessorProxy.
  * @arg_size: Argument to pass with the method invocation.
+ * @arg_session_Id: Argument to pass with the method invocation.
  * @cancellable: (nullable): A #GCancellable or %NULL.
  * @callback: A #GAsyncReadyCallback to call when the request is satisfied or %NULL.
  * @user_data: User data to pass to @callback.
@@ -1016,14 +1151,16 @@ void
 aisdk_text_processor_call_set_context (
     AisdkTextProcessor *proxy,
     gint arg_size,
+    gint arg_session_Id,
     GCancellable *cancellable,
     GAsyncReadyCallback callback,
     gpointer user_data)
 {
   g_dbus_proxy_call (G_DBUS_PROXY (proxy),
     "set_context",
-    g_variant_new ("(i)",
-                   arg_size),
+    g_variant_new ("(ii)",
+                   arg_size,
+                   arg_session_Id),
     G_DBUS_CALL_FLAGS_NONE,
     -1,
     cancellable,
@@ -1062,6 +1199,7 @@ _out:
  * aisdk_text_processor_call_set_context_sync:
  * @proxy: A #AisdkTextProcessorProxy.
  * @arg_size: Argument to pass with the method invocation.
+ * @arg_session_Id: Argument to pass with the method invocation.
  * @cancellable: (nullable): A #GCancellable or %NULL.
  * @error: Return location for error or %NULL.
  *
@@ -1075,14 +1213,16 @@ gboolean
 aisdk_text_processor_call_set_context_sync (
     AisdkTextProcessor *proxy,
     gint arg_size,
+    gint arg_session_Id,
     GCancellable *cancellable,
     GError **error)
 {
   GVariant *_ret;
   _ret = g_dbus_proxy_call_sync (G_DBUS_PROXY (proxy),
     "set_context",
-    g_variant_new ("(i)",
-                   arg_size),
+    g_variant_new ("(ii)",
+                   arg_size,
+                   arg_session_Id),
     G_DBUS_CALL_FLAGS_NONE,
     -1,
     cancellable,
@@ -1099,6 +1239,7 @@ _out:
 /**
  * aisdk_text_processor_call_clear_context:
  * @proxy: A #AisdkTextProcessorProxy.
+ * @arg_session_Id: Argument to pass with the method invocation.
  * @cancellable: (nullable): A #GCancellable or %NULL.
  * @callback: A #GAsyncReadyCallback to call when the request is satisfied or %NULL.
  * @user_data: User data to pass to @callback.
@@ -1112,13 +1253,15 @@ _out:
 void
 aisdk_text_processor_call_clear_context (
     AisdkTextProcessor *proxy,
+    gint arg_session_Id,
     GCancellable *cancellable,
     GAsyncReadyCallback callback,
     gpointer user_data)
 {
   g_dbus_proxy_call (G_DBUS_PROXY (proxy),
     "clear_context",
-    g_variant_new ("()"),
+    g_variant_new ("(i)",
+                   arg_session_Id),
     G_DBUS_CALL_FLAGS_NONE,
     -1,
     cancellable,
@@ -1156,6 +1299,7 @@ _out:
 /**
  * aisdk_text_processor_call_clear_context_sync:
  * @proxy: A #AisdkTextProcessorProxy.
+ * @arg_session_Id: Argument to pass with the method invocation.
  * @cancellable: (nullable): A #GCancellable or %NULL.
  * @error: Return location for error or %NULL.
  *
@@ -1168,13 +1312,15 @@ _out:
 gboolean
 aisdk_text_processor_call_clear_context_sync (
     AisdkTextProcessor *proxy,
+    gint arg_session_Id,
     GCancellable *cancellable,
     GError **error)
 {
   GVariant *_ret;
   _ret = g_dbus_proxy_call_sync (G_DBUS_PROXY (proxy),
     "clear_context",
-    g_variant_new ("()"),
+    g_variant_new ("(i)",
+                   arg_session_Id),
     G_DBUS_CALL_FLAGS_NONE,
     -1,
     cancellable,
@@ -1192,6 +1338,8 @@ _out:
  * aisdk_text_processor_complete_init:
  * @object: A #AisdkTextProcessor.
  * @invocation: (transfer full): A #GDBusMethodInvocation.
+ * @session_Id: Parameter to return.
+ * @initResult: Parameter to return.
  *
  * Helper function used in service implementations to finish handling invocations of the <link linkend="gdbus-method-org-openkylin-aisdk-TextProcessor.init">init()</link> D-Bus method. If you instead want to finish handling an invocation by returning an error, use g_dbus_method_invocation_return_error() or similar.
  *
@@ -1200,10 +1348,14 @@ _out:
 void
 aisdk_text_processor_complete_init (
     AisdkTextProcessor *object,
-    GDBusMethodInvocation *invocation)
+    GDBusMethodInvocation *invocation,
+    gint session_Id,
+    gboolean initResult)
 {
   g_dbus_method_invocation_return_value (invocation,
-    g_variant_new ("()"));
+    g_variant_new ("(ib)",
+                   session_Id,
+                   initResult));
 }
 
 /**
diff --git a/src/services/textprocessorglue.h b/src/services/textprocessorglue.h
index c6d00b1..df4c880 100644
--- a/src/services/textprocessorglue.h
+++ b/src/services/textprocessorglue.h
@@ -1,19 +1,3 @@
-/*
- * Copyright 2024 KylinSoft Co., Ltd.
- *
- * This program is free software: you can redistribute it and/or modify it under
- * the terms of the GNU General Public License as published by the Free Software
- * Foundation, either version 3 of the License, or (at your option) any later
- * version.
- *
- * This program is distributed in the hope that it will be useful, but WITHOUT
- * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
- * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License along with
- * this program. If not, see <https://www.gnu.org/licenses/>.
- */
-
 /*
  * This file is generated by gdbus-codegen, do not modify it.
  *
@@ -49,11 +33,13 @@ struct _AisdkTextProcessorIface
   gboolean (*handle_chat) (
     AisdkTextProcessor *object,
     GDBusMethodInvocation *invocation,
-    const gchar *arg_question);
+    const gchar *arg_question,
+    gint arg_session_Id);
 
   gboolean (*handle_clear_context) (
     AisdkTextProcessor *object,
-    GDBusMethodInvocation *invocation);
+    GDBusMethodInvocation *invocation,
+    gint arg_session_Id);
 
   gboolean (*handle_init) (
     AisdkTextProcessor *object,
@@ -66,16 +52,19 @@ struct _AisdkTextProcessorIface
   gboolean (*handle_set_actor) (
     AisdkTextProcessor *object,
     GDBusMethodInvocation *invocation,
-    gint arg_actor);
+    gint arg_actor,
+    gint arg_session_Id);
 
   gboolean (*handle_set_context) (
     AisdkTextProcessor *object,
     GDBusMethodInvocation *invocation,
-    gint arg_size);
+    gint arg_size,
+    gint arg_session_Id);
 
   gboolean (*handle_stop_chat) (
     AisdkTextProcessor *object,
-    GDBusMethodInvocation *invocation);
+    GDBusMethodInvocation *invocation,
+    gint arg_session_Id);
 
 };
 
@@ -88,7 +77,9 @@ guint aisdk_text_processor_override_properties (GObjectClass *klass, guint prope
 /* D-Bus method call completion functions: */
 void aisdk_text_processor_complete_init (
     AisdkTextProcessor *object,
-    GDBusMethodInvocation *invocation);
+    GDBusMethodInvocation *invocation,
+    gint session_Id,
+    gboolean initResult);
 
 void aisdk_text_processor_complete_chat (
     AisdkTextProcessor *object,
@@ -126,6 +117,8 @@ void aisdk_text_processor_call_init (
 
 gboolean aisdk_text_processor_call_init_finish (
     AisdkTextProcessor *proxy,
+    gint *out_session_Id,
+    gboolean *out_initResult,
     GAsyncResult *res,
     GError **error);
 
@@ -135,12 +128,15 @@ gboolean aisdk_text_processor_call_init_sync (
     const gchar *arg_app_id,
     const gchar *arg_api_key,
     const gchar *arg_secret_key,
+    gint *out_session_Id,
+    gboolean *out_initResult,
     GCancellable *cancellable,
     GError **error);
 
 void aisdk_text_processor_call_chat (
     AisdkTextProcessor *proxy,
     const gchar *arg_question,
+    gint arg_session_Id,
     GCancellable *cancellable,
     GAsyncReadyCallback callback,
     gpointer user_data);
@@ -154,12 +150,14 @@ gboolean aisdk_text_processor_call_chat_finish (
 gboolean aisdk_text_processor_call_chat_sync (
     AisdkTextProcessor *proxy,
     const gchar *arg_question,
+    gint arg_session_Id,
     gchar **out_answer,
     GCancellable *cancellable,
     GError **error);
 
 void aisdk_text_processor_call_stop_chat (
     AisdkTextProcessor *proxy,
+    gint arg_session_Id,
     GCancellable *cancellable,
     GAsyncReadyCallback callback,
     gpointer user_data);
@@ -171,12 +169,14 @@ gboolean aisdk_text_processor_call_stop_chat_finish (
 
 gboolean aisdk_text_processor_call_stop_chat_sync (
     AisdkTextProcessor *proxy,
+    gint arg_session_Id,
     GCancellable *cancellable,
     GError **error);
 
 void aisdk_text_processor_call_set_actor (
     AisdkTextProcessor *proxy,
     gint arg_actor,
+    gint arg_session_Id,
     GCancellable *cancellable,
     GAsyncReadyCallback callback,
     gpointer user_data);
@@ -189,12 +189,14 @@ gboolean aisdk_text_processor_call_set_actor_finish (
 gboolean aisdk_text_processor_call_set_actor_sync (
     AisdkTextProcessor *proxy,
     gint arg_actor,
+    gint arg_session_Id,
     GCancellable *cancellable,
     GError **error);
 
 void aisdk_text_processor_call_set_context (
     AisdkTextProcessor *proxy,
     gint arg_size,
+    gint arg_session_Id,
     GCancellable *cancellable,
     GAsyncReadyCallback callback,
     gpointer user_data);
@@ -207,11 +209,13 @@ gboolean aisdk_text_processor_call_set_context_finish (
 gboolean aisdk_text_processor_call_set_context_sync (
     AisdkTextProcessor *proxy,
     gint arg_size,
+    gint arg_session_Id,
     GCancellable *cancellable,
     GError **error);
 
 void aisdk_text_processor_call_clear_context (
     AisdkTextProcessor *proxy,
+    gint arg_session_Id,
     GCancellable *cancellable,
     GAsyncReadyCallback callback,
     gpointer user_data);
@@ -223,6 +227,7 @@ gboolean aisdk_text_processor_call_clear_context_finish (
 
 gboolean aisdk_text_processor_call_clear_context_sync (
     AisdkTextProcessor *proxy,
+    gint arg_session_Id,
     GCancellable *cancellable,
     GError **error);
 
diff --git a/src/services/textprocessorproxy.cpp b/src/services/textprocessorproxy.cpp
index 68aae6b..1bc58cf 100644
--- a/src/services/textprocessorproxy.cpp
+++ b/src/services/textprocessorproxy.cpp
@@ -15,19 +15,79 @@
  */
 
 #include "services/textprocessorproxy.h"
-
 #include "services/serverproxy.h"
-#include "services/textprocessor.h"
+#include <iostream>
+
+namespace {
+
+const char *textProcessorObjectPath = "/org/openkylin/aisdk/textprocessor";
+const char *textProcessorInterfaceName = "org.openkylin.aisdk.textprocessor";
+
+}
+
+void onChatResult(GDBusConnection *connection,
+                  const gchar      *sender_name,
+                  const gchar      *object_path,
+                  const gchar      *interface_name,
+                  const gchar      *signal_name,
+                  GVariant         *parameters,
+                  gpointer          user_data) {
+    auto *proxy = static_cast<TextProcessorProxy *>(user_data);
+    const gchar *cstr = nullptr;
+    g_variant_get(parameters, "(s)", &cstr);
+    std::string result(cstr);
+    proxy->chatCallback(result, user_data);
+}
 
 TextProcessorProxy::TextProcessorProxy() { init(); }
 
 TextProcessorProxy::~TextProcessorProxy() { destroy(); }
 
+bool TextProcessorProxy::initEngine(const std::string &engineName,
+    const std::string &apiKey, const std::string &secretKey, const std::string &appId)
+{
+    GError *error = nullptr;
+    gboolean initResult;
+    aisdk_text_processor_call_init_sync(
+                delegate_, engineName.c_str(), appId.c_str(), apiKey.c_str(),
+                secretKey.c_str(), &sessionId_, &initResult, nullptr, &error);
+
+    if (!initResult) {
+        return false;
+    }
+
+    std::cout << "proxy initEngine sessionId " << sessionId_ << std::endl;
+    connectSignal();
+    return true;
+}
+
 void TextProcessorProxy::chat(const std::string &question) {
-    aisdk_text_processor_call_chat(delegate_, question.c_str(), nullptr,
-                                   reinterpret_cast<GAsyncReadyCallback>(
-                                       TextProcessorProxy::onChatResultReady),
-                                   this);
+    aisdk_text_processor_call_chat(
+        delegate_, question.c_str(), sessionId_, nullptr, nullptr, this);
+}
+
+void TextProcessorProxy::stopChat()
+{
+    aisdk_text_processor_call_stop_chat(delegate_, sessionId_,
+                                        nullptr, nullptr, nullptr);
+}
+
+void TextProcessorProxy::setActor(int actor)
+{
+    aisdk_text_processor_call_set_actor(
+        delegate_, actor, sessionId_, nullptr, nullptr, nullptr);
+}
+
+void TextProcessorProxy::setContext(int size)
+{
+    aisdk_text_processor_call_set_context(delegate_, size, sessionId_,
+                                          nullptr, nullptr, nullptr);
+}
+
+void TextProcessorProxy::clearContext()
+{
+    aisdk_text_processor_call_clear_context(delegate_, sessionId_,
+                                            nullptr, nullptr, nullptr);
 }
 
 void TextProcessorProxy::init() {
@@ -40,13 +100,12 @@ void TextProcessorProxy::init() {
     }
 
     auto *connection = ServerProxy::getInstance().getConnection();
-    const auto &objectPath = TextProcessor::getObjectPath();
     delegate_ = aisdk_text_processor_proxy_new_sync(
-        connection, G_DBUS_PROXY_FLAGS_NONE, nullptr, objectPath.c_str(),
+        connection, G_DBUS_PROXY_FLAGS_NONE, nullptr, textProcessorObjectPath,
         nullptr, &error);
     if (delegate_ == nullptr) {
         g_printerr("Error creating text processor proxy %s: %s\n",
-                   objectPath.c_str(), error->message);
+                   textProcessorObjectPath, error->message);
         g_error_free(error);
     }
 }
@@ -57,26 +116,11 @@ void TextProcessorProxy::destroy() {
     }
 }
 
-void TextProcessorProxy::onChatResultReady(AisdkTextProcessor *delegate,
-                                           GAsyncResult *result,
-                                           gpointer userData) {
-    auto *proxy = static_cast<TextProcessorProxy *>(userData);
-    if (proxy->chatCallback == nullptr) {
-        return;
-    }
-
-    gchar *answer = nullptr;
-    GError *error = nullptr;
-    bool success = aisdk_text_processor_call_chat_finish(delegate, &answer,
-                                                         result, &error);
-    if (success) {
-        proxy->chatCallback(answer, proxy->userData_);
-        g_free(answer);
-    } else {
-        proxy->chatCallback("", proxy->userData_);
-        g_printerr("Error get text processor proxy chat result: %s\n",
-                   error->message);
+void TextProcessorProxy::connectSignal() {
+    auto *connection = ServerProxy::getInstance().getConnection();
+    std::string interfaceName = textProcessorInterfaceName + std::to_string(sessionId_);
 
-        g_error_free(error);
-    }
-}
\ No newline at end of file
+    g_dbus_connection_signal_subscribe(connection, nullptr,
+        interfaceName.c_str(), "ChatResult", textProcessorObjectPath, NULL,
+        G_DBUS_SIGNAL_FLAGS_NONE, onChatResult, this, nullptr);
+}
diff --git a/src/services/textprocessorproxy.h b/src/services/textprocessorproxy.h
index 8721c8d..aaf45cf 100644
--- a/src/services/textprocessorproxy.h
+++ b/src/services/textprocessorproxy.h
@@ -18,9 +18,9 @@
 #define SERVICES_TEXTPROCESSORPROXY_H
 
 #include <functional>
-
-#include "services/textprocessorglue.h"
 #include <string>
+#include "services/textprocessorglue.h"
+
 
 class TextProcessorProxy {
 public:
@@ -30,6 +30,10 @@ public:
     TextProcessorProxy();
     ~TextProcessorProxy();
 
+    bool initEngine(const std::string &engineName,
+        const std::string &apiKey = {}, const std::string &secretKey = {},
+        const std::string &appId = {});
+
     void setChatCallback(ChatCallback callback) {
         chatCallback = std::move(callback);
     }
@@ -37,18 +41,32 @@ public:
     void setUserData(void *userData) { userData_ = userData; }
 
     void chat(const std::string &question);
+    void stopChat();
+    void setActor(int actor);
+    void setContext(int size);
+    void clearContext();
 
 private:
     void init();
     void destroy();
+    void connectSignal();
+    void generateRandomSessionId();
+
+    friend void onChatResult(GDBusConnection *connection,
+                        const gchar      *sender_name,
+                        const gchar      *object_path,
+                        const gchar      *interface_name,
+                        const gchar      *signal_name,
+                        GVariant         *parameters,
+                        gpointer          user_data);
 
-    static void onChatResultReady(AisdkTextProcessor *delegate,
-                                  GAsyncResult *result, gpointer userData);
 
 private:
     AisdkTextProcessor *delegate_ = nullptr;
 
     ChatCallback chatCallback = nullptr;
     void *userData_ = nullptr;
+
+    int sessionId_;
 };
 #endif
-- 
Gitee