tushare 1.4.10__py3-none-any.whl → 1.4.12__py3-none-any.whl

tushare/__init__.py

@@ -2,7 +2,7 @@
 import codecs
 import os
 
-__version__ = '1.4.10'
+__version__ = '1.4.12'
 __author__ = 'Jimmy Liu'
 
 """

tushare/pro/llm.py

@@ -0,0 +1,124 @@
+import json
+
+import requests
+try:
+    import sseclient
+except:
+    raise ImportError("sseclient not found, please install it using 'pip install sseclient-py'.")
+
+
+from tushare import get_token
+
+BASE_URL = "http://api.waditu.com/dataapi"
+# BASE_URL = "http://127.0.0.1:8000/dataapi"
+API_KEY_PREFIX = "tsgpt-"
+
+
+class LlmClient:
+    def __init__(self, token=None, timetout=120):
+        if not token:
+            token = get_token()
+        self.token = token
+        self.timeout = timetout
+
+    def chat_query(self, model, messages, temperature=None, max_tokens=None, pretty=False):
+        """
+        model string 模型名称， doubao-pro-128k
+        messages list 消息列表
+            [
+                {
+                  "role": "user",
+                  "content": "Hello World"
+                }
+            ]
+        pretty bool 是否只返回回答内容文本
+        """
+        resp = requests.post(
+            f'{BASE_URL}/llm/{model}',
+            json={"params": {
+                "stream": False,
+                "messages": messages,
+                "temperature": temperature,
+                "max_tokens": max_tokens
+            }},
+            headers={"Authorization": f"tstoken-{self.token}"},
+            timeout=self.timeout
+        )
+        if resp.status_code != 200:
+            raise Exception(f"请求出现错误，{resp.content}")
+
+        resp_data = resp.json()
+        if resp_data.get('code') not in (0, None):
+            raise Exception(resp_data['msg'])
+        if pretty:
+            return resp_data['choices'][0]["message"]["content"]
+        else:
+            return resp_data
+
+    def chat_stream(self, model, messages, temperature=None, max_tokens=None, pretty=False):
+        """
+        model string 模型名称， doubao-pro-128k
+        messages list 消息列表
+            [
+                {
+                  "role": "user",
+                  "content": "Hello World"
+                }
+            ]
+        pretty bool 是否只返回回答内容文本
+        """
+        resp = requests.post(
+            f'{BASE_URL}/llm/{model}',
+            json={"params": {
+                "stream": True,
+                "messages": messages,
+                "temperature": temperature,
+                "max_tokens": max_tokens
+            }},
+            headers={"Authorization": f"tstoken-{self.token}"},
+            timeout=self.timeout, stream=True
+        )
+        if resp.status_code != 200:
+            raise Exception(f"请求出现错误，{resp.content}")
+
+        for e in sseclient.SSEClient(resp).events():
+            e_data = json.loads(e.data)
+            if pretty:
+                yield e_data["choices"][0]["delta"]["content"]
+            else:
+                yield e_data
+
+
+def test_chart_query():
+    c = LlmClient()
+    dd = c.chat_query("doubao-pro-128k", [{
+        "role": "user",
+        "content": "你好"
+    }])
+    print(dd)
+    dd = c.chat_query("doubao-pro-128k", [{
+        "role": "user",
+        "content": "你好"
+    }], pretty=True)
+    print(dd)
+
+
+def test_chart_stream():
+    c = LlmClient()
+    dd = c.chat_stream("doubao-pro-128k", [
+        {
+            "role": "user",
+            "content": "你好"
+        }
+    ])
+    for d in dd:
+        print(d)
+
+    dd = c.chat_stream("doubao-pro-128k", [
+        {
+            "role": "user",
+            "content": "你好"
+        }
+    ], pretty=True)
+    for d in dd:
+        print(d)

{tushare-1.4.10.dist-info → tushare-1.4.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tushare
-Version: 1.4.10
+Version: 1.4.12
 Summary: A utility for crawling historical and Real-time Quotes data of China stocks
 Home-page: https://tushare.pro
 Author: Jimmy Liu

{tushare-1.4.10.dist-info → tushare-1.4.12.dist-info}/RECORD

@@ -1,4 +1,4 @@
-tushare/__init__.py,sha256=8yF2REriCYD3iDufFWimHXRrkff2NxiEJqffrWR1EwA,4765
+tushare/__init__.py,sha256=jyET1cC5S1mBEZSSBRhY1j9eacrqXF3pyXTUv5ouuko,4765
 tushare/bond/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tushare/bond/bonds.py,sha256=PJM0xDiWZDpOPwDtbEU9PdP0M_Gu0c599YuB1rbZ3r8,232
 tushare/coins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -19,6 +19,7 @@ tushare/internet/indexes.py,sha256=oteQCv0k2X0pbhXRHwMSviD1AowJWE6xRjKkqr5RNCo,3
 tushare/pro/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tushare/pro/client.py,sha256=w7LfXtlmSd4Aiq7Bz5-tPFfkfjrPS_7jroUiRs7BUWw,1388
 tushare/pro/data_pro.py,sha256=SOHbK-j9CmZguuv0CLIT-Z3hUynEy96mbvvhrGN6y9I,10883
+tushare/pro/llm.py,sha256=gmc8vG7Vzv9JlusiWnSLEmcPX8X-NN-WkLUk4ibB5E8,3561
 tushare/stock/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tushare/stock/billboard.py,sha256=vCrbN7-NGXSpXMc8jq7NcjjHyu9pTcr4uXp559iA1wA,12833
 tushare/stock/classifying.py,sha256=3yWqlLhgj2irk-9ksKWFFS5HY-SnjfEr5erwJxBidG8,10539
@@ -71,8 +72,8 @@ tushare/util/verify_token.py,sha256=cuV3RErWbOC318NANCYL6K1LKZ3wSAL2yMwZHA7tD3s,
 tushare/util/protobuf/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tushare/util/protobuf/funcs.py,sha256=UCdK8FxTyjPZsNzoEeXqYzqrQXUmRMvW5hua6GPA66A,779
 tushare/util/protobuf/response_pb2.py,sha256=vJH9ONkDuJlg6y-q1PvuDZoviKrK7hzNtMieQHK45DI,11347
-tushare-1.4.10.dist-info/LICENSE,sha256=C2j55UI0Ul-1-wA1-rn7OaY6b3vGl4YukiyvYzHsU9o,1503
-tushare-1.4.10.dist-info/METADATA,sha256=i4KoKwZ6XIAWEPTd5rvST6NFXoS7Yu9qfkCBQOFYQqg,2839
-tushare-1.4.10.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-tushare-1.4.10.dist-info/top_level.txt,sha256=HHOxMuqc31KuAIcxpE0t5dAPMKbaiRtjsjTMFd7FlXI,8
-tushare-1.4.10.dist-info/RECORD,,
+tushare-1.4.12.dist-info/LICENSE,sha256=C2j55UI0Ul-1-wA1-rn7OaY6b3vGl4YukiyvYzHsU9o,1503
+tushare-1.4.12.dist-info/METADATA,sha256=qrBJtq0rMmPCKXiSMP4LdbS3AlQjAklBxtLhQvfscgw,2839
+tushare-1.4.12.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+tushare-1.4.12.dist-info/top_level.txt,sha256=HHOxMuqc31KuAIcxpE0t5dAPMKbaiRtjsjTMFd7FlXI,8
+tushare-1.4.12.dist-info/RECORD,,