update

chatchat/baidu.py

@@ -1,5 +1,5 @@
 from chatchat.base import Base
-import httpx, time
+import httpx, time, json
 
 class Completion(Base):
     def __init__(self, jfile, model='ERNIE-Speed-8K'):
@@ -85,8 +85,23 @@ def create(self, message):
         return r.json()
 
 class Chat(Completion):
-    def __init__(self, jfile, model='ERNIE-Speed-8K'):
+    def __init__(self, jfile, model='ERNIE-Speed-8K', history=[]):
         super().__init__(jfile, model=model)
+        self.history = history
+
+    def chat(self, message):
+        self.history.append({
+            "role": "user",
+            "content": message,
+        })
+        message = {"messages": self.history}
+        payload = json.dumps(message)
+        url = f'{self.api}?access_token={self.get_access_token()}'
+        r = self.client.post(url, headers=self.headers, data=payload).json()
+        if 'result' in r:
+            self.history.append({
+                "role": "assistant",
+                "content": r['result']
+            })
 
-    def chat(self):
-        ...
+        return r

examples/baidu.py

@@ -22,4 +22,20 @@
 #         'total_tokens': 70
 #     }
 # }
-print(r)
+print(r)
+
+history = [
+    {
+        "role": "user",
+        "content": "简单介绍一下你自己",
+    },
+    {
+        "role": "assistant",
+        "content": "我是人工智能助手，具备智能问答、自然语言处理等多项功能，致力于为用户提供准确、便捷的解答和服务。",
+    }
+]
+chat = cc.baidu.Chat('./data.json', history=history)
+r = chat.chat("说的再详细点！")
+print(r)
+r = chat.chat("给我举个具体的例子。")
+print(r)

setup.py

@@ -3,7 +3,7 @@
 setup(
     name = 'chatchat',
     packages = find_packages(exclude=['examples']),
-    version = '0.0.5',
+    version = '0.0.6',
     license = 'GPL-2.0',
     description = 'Large Language Model API',
     author = 'JiauZhang',