From c6b107ce565ac2016592609cbb4fabb56d0ff53a Mon Sep 17 00:00:00 2001
From: bingchu <152955942+J4ckycjl@users.noreply.github.com>
Date: Wed, 12 Mar 2025 10:13:03 +0800
Subject: [PATCH] fix(knext): set token in request of write_graph (#409)

* fix(knext)set token in request of write_graph

* refine code
---
 knext/builder/client.py                       |  1 +
 .../rest/models/writer_graph_request.py       | 39 ++++++++++++++++---
 knext/graph/client.py                         |  1 +
 .../graph/rest/models/writer_graph_request.py | 39 ++++++++++++++++---
 4 files changed, 68 insertions(+), 12 deletions(-)

diff --git a/knext/builder/client.py b/knext/builder/client.py
index 8666ae53..c448ab5a 100644
--- a/knext/builder/client.py
+++ b/knext/builder/client.py
@@ -31,6 +31,7 @@ class BuilderClient(Client):
             sub_graph=sub_graph,
             operation=operation,
             enable_lead_to=lead_to_builder,
+            token="openspg@8380255d4e49_"
         )
         self._rest_client.builder_job_writer_graph_post(writer_graph_request=request)
 
diff --git a/knext/builder/rest/models/writer_graph_request.py b/knext/builder/rest/models/writer_graph_request.py
index dca97030..a21ce65a 100644
--- a/knext/builder/rest/models/writer_graph_request.py
+++ b/knext/builder/rest/models/writer_graph_request.py
@@ -48,6 +48,7 @@ class WriterGraphRequest(object):
         "operation": "str",
         "sub_graph": "object",
         "enable_lead_to": "bool",
+        "token": "str"
     }
 
     attribute_map = {
@@ -55,15 +56,17 @@ class WriterGraphRequest(object):
         "operation": "operation",
         "sub_graph": "subGraph",
         "enable_lead_to": "enableLeadTo",
+        "token": "token"
     }
 
     def __init__(
-        self,
-        project_id=None,
-        operation=None,
-        sub_graph=None,
-        enable_lead_to=None,
-        local_vars_configuration=None,
+            self,
+            project_id=None,
+            operation=None,
+            sub_graph=None,
+            enable_lead_to=None,
+            token=None,
+            local_vars_configuration=None,
     ):  # noqa: E501
         """WriterGraphRequest - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
@@ -74,6 +77,7 @@ class WriterGraphRequest(object):
         self._operation = None
         self._sub_graph = None
         self._enable_lead_to = None
+        self._token = None
         self.discriminator = None
 
         if project_id is not None:
@@ -84,6 +88,8 @@ class WriterGraphRequest(object):
             self.sub_graph = sub_graph
         if enable_lead_to is not None:
             self.enable_lead_to = enable_lead_to
+        if token is not None:
+            self.token = token
 
     @property
     def project_id(self):
@@ -169,6 +175,27 @@ class WriterGraphRequest(object):
 
         self._enable_lead_to = enable_lead_to
 
+    @property
+    def token(self):
+        """Gets the token of this WriterGraphRequest.  # noqa: E501
+
+
+        :return: The token of this WriterGraphRequest.  # noqa: E501
+        :rtype: str
+        """
+        return self._token
+
+    @token.setter
+    def token(self, token):
+        """Sets the token of this WriterGraphRequest.
+
+
+        :param token: The token of this WriterGraphRequest.  # noqa: E501
+        :type: str
+        """
+
+        self._token = token
+
     def to_dict(self):
         """Returns the model properties as a dict"""
         result = {}
diff --git a/knext/graph/client.py b/knext/graph/client.py
index 08ba7fbc..b7b6e5a5 100644
--- a/knext/graph/client.py
+++ b/knext/graph/client.py
@@ -67,6 +67,7 @@ class GraphClient(Client):
             sub_graph=sub_graph,
             operation=operation,
             enable_lead_to=lead_to_builder,
+            token="openspg@8380255d4e49_"
         )
         self._rest_client.graph_writer_graph_post(writer_graph_request=request)
 
diff --git a/knext/graph/rest/models/writer_graph_request.py b/knext/graph/rest/models/writer_graph_request.py
index dca97030..a21ce65a 100644
--- a/knext/graph/rest/models/writer_graph_request.py
+++ b/knext/graph/rest/models/writer_graph_request.py
@@ -48,6 +48,7 @@ class WriterGraphRequest(object):
         "operation": "str",
         "sub_graph": "object",
         "enable_lead_to": "bool",
+        "token": "str"
     }
 
     attribute_map = {
@@ -55,15 +56,17 @@ class WriterGraphRequest(object):
         "operation": "operation",
         "sub_graph": "subGraph",
         "enable_lead_to": "enableLeadTo",
+        "token": "token"
     }
 
     def __init__(
-        self,
-        project_id=None,
-        operation=None,
-        sub_graph=None,
-        enable_lead_to=None,
-        local_vars_configuration=None,
+            self,
+            project_id=None,
+            operation=None,
+            sub_graph=None,
+            enable_lead_to=None,
+            token=None,
+            local_vars_configuration=None,
     ):  # noqa: E501
         """WriterGraphRequest - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
@@ -74,6 +77,7 @@ class WriterGraphRequest(object):
         self._operation = None
         self._sub_graph = None
         self._enable_lead_to = None
+        self._token = None
         self.discriminator = None
 
         if project_id is not None:
@@ -84,6 +88,8 @@ class WriterGraphRequest(object):
             self.sub_graph = sub_graph
         if enable_lead_to is not None:
             self.enable_lead_to = enable_lead_to
+        if token is not None:
+            self.token = token
 
     @property
     def project_id(self):
@@ -169,6 +175,27 @@ class WriterGraphRequest(object):
 
         self._enable_lead_to = enable_lead_to
 
+    @property
+    def token(self):
+        """Gets the token of this WriterGraphRequest.  # noqa: E501
+
+
+        :return: The token of this WriterGraphRequest.  # noqa: E501
+        :rtype: str
+        """
+        return self._token
+
+    @token.setter
+    def token(self, token):
+        """Sets the token of this WriterGraphRequest.
+
+
+        :param token: The token of this WriterGraphRequest.  # noqa: E501
+        :type: str
+        """
+
+        self._token = token
+
     def to_dict(self):
         """Returns the model properties as a dict"""
         result = {}