tushare 1.4.10__tar.gz → 1.4.11__tar.gz

{tushare-1.4.10 → tushare-1.4.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tushare
-Version: 1.4.10
+Version: 1.4.11
 Summary: A utility for crawling historical and Real-time Quotes data of China stocks
 Home-page: https://tushare.pro
 Author: Jimmy Liu
@@ -16,6 +16,13 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: License :: OSI Approved :: BSD License
 Description-Content-Type: text/plain
 License-File: LICENSE
+Requires-Dist: pandas
+Requires-Dist: requests
+Requires-Dist: lxml
+Requires-Dist: simplejson
+Requires-Dist: bs4
+Requires-Dist: websocket-client>=0.57.0
+Requires-Dist: tqdm
 
 
 TuShare

{tushare-1.4.10 → tushare-1.4.11}/setup.py

@@ -97,7 +97,7 @@ def read_install_requires():
 
 setup(
     name='tushare',
-    version='1.4.10',
+    version='1.4.11',
     description='A utility for crawling historical and Real-time Quotes data of China stocks',
     # long_description=read("README.rst"),
     long_description=long_desc,

{tushare-1.4.10 → tushare-1.4.11}/tushare/__init__.py

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

tushare-1.4.11/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 LlmClent:
+    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 = LlmClent()
+    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 = LlmClent()
+    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 → tushare-1.4.11}/tushare.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tushare
-Version: 1.4.10
+Version: 1.4.11
 Summary: A utility for crawling historical and Real-time Quotes data of China stocks
 Home-page: https://tushare.pro
 Author: Jimmy Liu
@@ -16,6 +16,13 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: License :: OSI Approved :: BSD License
 Description-Content-Type: text/plain
 License-File: LICENSE
+Requires-Dist: pandas
+Requires-Dist: requests
+Requires-Dist: lxml
+Requires-Dist: simplejson
+Requires-Dist: bs4
+Requires-Dist: websocket-client>=0.57.0
+Requires-Dist: tqdm
 
 
 TuShare

{tushare-1.4.10 → tushare-1.4.11}/tushare.egg-info/SOURCES.txt

@@ -27,6 +27,7 @@ tushare/internet/indexes.py
 tushare/pro/__init__.py
 tushare/pro/client.py
 tushare/pro/data_pro.py
+tushare/pro/llm.py
 tushare/stock/__init__.py
 tushare/stock/billboard.py
 tushare/stock/classifying.py