edi-core 0.4.0__tar.gz → 0.6.0__tar.gz

{edi_core-0.4.0 → edi_core-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: edi-core
-Version: 0.4.0
+Version: 0.6.0
 Summary: 
 License: MIT
 Author: LtttAZ
@@ -25,4 +25,4 @@ Description-Content-Type: text/markdown
 
 # EDI
 
-
+[PYPI](https://pypi.org/project/edi-core/)

edi_core-0.6.0/README.md

@@ -0,0 +1,3 @@
+# EDI
+
+[PYPI](https://pypi.org/project/edi-core/)

{edi_core-0.4.0 → edi_core-0.6.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "edi-core"
-version = "0.4.0"
+version = "0.6.0"
 description = ""
 authors = ["LtttAZ <louis_zhang_1303@126.com>"]
 license = "MIT"

{edi_core-0.4.0 → edi_core-0.6.0}/src/edi_core/ai/writesonic/__init__.py

@@ -1,8 +1,8 @@
 WRITESONIC_BASE_URL = 'https://api.writesonic.com'
 CHATSONIC_URL_PATH = '/v2/business/content/chatsonic'
 WRITESONIC_ENGINE_PREMIUM = 'premium'
-WRITESONIC_ENGINE_SUPERIUM = 'superium'
-WRITESONIC_ENGINES = (WRITESONIC_ENGINE_PREMIUM, WRITESONIC_ENGINE_SUPERIUM)
+WRITESONIC_ENGINE_SUPERIOR = 'superior'
+WRITESONIC_ENGINES = (WRITESONIC_ENGINE_PREMIUM, WRITESONIC_ENGINE_SUPERIOR)
 WRITESONIC_LANGUAGE_ENGLISH = 'en'
 WRITESONIC_LANGUAGE_CHINESE = 'zh'
 WRITESONIC_LANGUAGES = (WRITESONIC_LANGUAGE_ENGLISH, WRITESONIC_LANGUAGE_CHINESE)

{edi_core-0.4.0 → edi_core-0.6.0}/src/edi_core/ai/writesonic/chatsonic.py

@@ -1,3 +1,4 @@
+import dataclasses
 import os
 from dataclasses import dataclass, field
 from typing import List
@@ -29,7 +30,7 @@ class ChatSonicConfig:
 
 
 @dataclass
-class ChatSonicPayload:
+class ChatSonicRequest:
     @dataclass
     class Message:
         is_sent: bool
@@ -51,6 +52,9 @@ class ChatSonicPayload:
         self.enable_memory = config.enable_memory
         self.enable_google_results = config.enable_google_results
 
+    def to_dict(self):
+        return dataclasses.asdict(self)
+
 
 @dataclass
 class ChatSonicResponse:
@@ -72,14 +76,15 @@ class ChatSonic:
     def __init__(self, config: ChatSonicConfig):
         self._config = config
 
-    def call(self, payload: ChatSonicPayload) -> (ChatSonicResponse, ChatSonicPayload):
+    def call(self, payload: ChatSonicRequest) -> (ChatSonicResponse, ChatSonicRequest):
         url = self._config.get_request_url()
+        payload.set_config(self._config)
 
         raw_response = self._session.post(
             url=url,
             headers=self.get_request_headers(),
             params=self.get_request_params(),
-            json=payload.__dict__,
+            json=payload.to_dict(),
             timeout=60
         )
 
@@ -90,9 +95,9 @@ class ChatSonic:
 
         return response, payload
 
-    def chat(self, message: str) -> (ChatSonicResponse, ChatSonicPayload):
-        payload = ChatSonicPayload(input_text=message);
-        payload.set_config(self._config)
+    def chat(self, message: str, history_data: List[ChatSonicRequest.Message] = []) -> (
+            ChatSonicResponse, ChatSonicRequest):
+        payload = ChatSonicRequest(input_text=message, history_data=history_data)
         return self.call(payload)
 
     def get_request_headers(self):

edi_core-0.4.0/README.md

@@ -1,2 +0,0 @@
-# EDI
-