feat:(chat-api): add ChatService APIs. (#23)

add ChatService APIs:
1.chat_again
2.get_conversation_list
3.get_conversation_inputs
4.update_conversation
5.delete_conversation
6.stop_message
7.clear_message
8.get_conversation_messages
9.get_message_info
10.delete_message
11.feedback
12.set_message_answer_used
13.get_suggested_questions
14.run_app_workflow
15.sync_run_app_workflow
16.query_run_app_process
17.list_oauth2_token
18.event_trigger_webhook
19.chat_continue
20.list_long_memory
21.update_long_memory
22.delete_long_memory
23.clear_long_memory
24.async_resume_app_workflow
25.set_conversation_top
26.cancel_conversation_top
27.query_skill_async_task
28.sync_resume_app_workflow_blocking
29.sync_resume_app_workflow_streaming
30.get_app_user_variables
31.set_app_user_variables
32.query_trigger_run_records

Author: hannengfang

libs/api/hiagent_api/base.py

@@ -67,21 +67,21 @@ def __call__(self, r):
 
 class Service(object):
     def __init__(
-        self,
-        service_info: ServiceInfo,
-        api_info: dict[str, ApiInfo],
-        http_client: Optional[httpx.Client] = None,
-        async_http_client: Optional[httpx.AsyncClient] = None,
+            self,
+            service_info: ServiceInfo,
+            api_info: dict[str, ApiInfo],
+            http_client: Optional[httpx.Client] = None,
+            async_http_client: Optional[httpx.AsyncClient] = None,
     ):
         self.service_info = service_info
         self.api_info = api_info
         self.init()
         self.init_http_client(http_client, async_http_client)
 
     def init_http_client(
-        self,
-        http_client: Optional[httpx.Client] = None,
-        async_http_client: Optional[httpx.AsyncClient] = None,
+            self,
+            http_client: Optional[httpx.Client] = None,
+            async_http_client: Optional[httpx.AsyncClient] = None,
     ):
         if http_client:
             self.http_client = http_client
@@ -361,9 +361,9 @@ async def _arequest(self, action, params):
     def prepare_request(self, api_info, params, doseq=0):
         for key in params:
             if (
-                type(params[key]) == int
-                or type(params[key]) == float
-                or type(params[key]) == bool
+                    type(params[key]) == int
+                    or type(params[key]) == float
+                    or type(params[key]) == bool
             ):
                 params[key] = str(params[key])
             elif sys.version_info[0] != 3:
@@ -444,12 +444,12 @@ def sign_sts2(self, policy, expire):
         inner_token.signature = Util.to_hex(Util.hmac_sha256(key, sign_str))
 
         sts.session_token = (
-            "STS2"
-            + base64.b64encode(
-                json.dumps(inner_token, cls=ComplexEncoder, sort_keys=True)
-                .replace(" ", "")
-                .encode("utf-8")
-            ).decode()
+                "STS2"
+                + base64.b64encode(
+            json.dumps(inner_token, cls=ComplexEncoder, sort_keys=True)
+            .replace(" ", "")
+            .encode("utf-8")
+        ).decode()
         )
         return sts
 
@@ -459,14 +459,14 @@ def to_rfc3339(t):
         pos = format_time.find("+")
         if pos == -1:
             pos = format_time.find("-")
-        return format_time[: pos + 3] + ":" + format_time[pos + 3 : pos + 5]
+        return format_time[: pos + 3] + ":" + format_time[pos + 3: pos + 5]
 
 
 class AppAPIMixin:
     def __init__(
-        self,
-        http_client: httpx.Client,
-        async_http_client: httpx.AsyncClient,
+            self,
+            http_client: httpx.Client,
+            async_http_client: httpx.AsyncClient,
     ) -> None:
         self.base_url = ""
         self.http_client = http_client
@@ -475,7 +475,7 @@ def __init__(
     def set_app_base_url(self, base_url: str):
         self.base_url = base_url
 
-    async def _apost(self, app_key: str, action: str, params: dict) -> str:
+    async def _apost(self, app_key: str, action: str, params: dict, _headers: Optional[dict]) -> str:
         if self.base_url == "":
             raise Exception(
                 "base_url not set, you should call set_app_base_url() first"
@@ -484,7 +484,8 @@ async def _apost(self, app_key: str, action: str, params: dict) -> str:
         app_url = f"{self.base_url}/{action}"
 
         headers = {"Apikey": f"{app_key}", "Content-Type": "application/json"}
-
+        if _headers is not None:
+            headers.update(_headers)
         response = await self.async_http_client.post(
             app_url, json=params, headers=headers
         )
@@ -497,15 +498,17 @@ async def _apost(self, app_key: str, action: str, params: dict) -> str:
             raise Exception("empty response")
         return res_text
 
-    def _post(self, app_key: str, action: str, params: dict) -> str:
+    def _post(self, app_key: str, action: str, params: dict, _headers: Optional[dict]) -> str:
         if self.base_url == "":
             raise Exception(
                 "base_url not set, you should call set_app_base_url() first"
             )
 
         app_url = f"{self.base_url}/{action}"
-
         headers = {"Apikey": f"{app_key}", "Content-Type": "application/json"}
+        if _headers is not None:
+            headers.update(_headers)
+
         response = self.http_client.post(app_url, json=params, headers=headers)
         try:
             response.raise_for_status()  # Raise an exception for bad status codes
@@ -517,7 +520,7 @@ def _post(self, app_key: str, action: str, params: dict) -> str:
         return res_text
 
     def _sse_post(
-        self, app_key: str, action: str, params: dict
+            self, app_key: str, action: str, params: dict
     ) -> Generator[ServerSentEvent, None, None]:
         if self.base_url == "":
             raise Exception(
@@ -529,17 +532,17 @@ def _sse_post(
         headers = {"Apikey": f"{app_key}", "Content-Type": "application/json"}
 
         with connect_sse(
-            self.http_client,
-            method="POST",
-            url=app_url,
-            json=params,
-            headers=headers,
+                self.http_client,
+                method="POST",
+                url=app_url,
+                json=params,
+                headers=headers,
         ) as event_source:
             for sse in event_source.iter_sse():
                 yield sse
 
     async def _asse_post(
-        self, app_key: str, action: str, params: dict
+            self, app_key: str, action: str, params: dict
     ) -> AsyncGenerator[ServerSentEvent, None]:
         if self.base_url == "":
             raise Exception(
@@ -551,11 +554,11 @@ async def _asse_post(
         headers = {"Apikey": f"{app_key}", "Content-Type": "application/json"}
 
         async with aconnect_sse(
-            self.async_http_client,
-            method="POST",
-            url=app_url,
-            json=params,
-            headers=headers,
+                self.async_http_client,
+                method="POST",
+                url=app_url,
+                json=params,
+                headers=headers,
         ) as event_source:
             async for sse in event_source.aiter_sse():
                 yield sse