isrpa 0.9.7__py3-none-any.whl → 0.9.9__py3-none-any.whl

isrpa/credentials.py

@@ -14,9 +14,7 @@ def store_key(key_name,key_value,user_name):
 
     """
     # address = os.environ.get("address", "192.168.12.249")
-    print(address)
     url = f"{address}/console/api/user_credentials"
-    print(url)
     headers = {
         'Content-Type': 'application/json',  # 说明请求体是 JSON 格式
     }
@@ -26,11 +24,8 @@ def store_key(key_name,key_value,user_name):
         'certificate_value': encode(key_value),
         'user_name': user_name
     }
-    print(data)
     response = requests.post(url, headers=headers, json=data)
     if response.status_code != 200:
-        print("请求失败，状态码：", response.status_code)
-        print(response.text)
         return "failure"
     json = response.json()
     return json['certificate_key']
@@ -43,11 +38,8 @@ def get_key(key_name, user_name):
     """
     # address = os.environ.get("address", "192.168.12.249")
     url = f"{address}/console/api/user_credentials?certificate_key={key_name}&user_name={user_name}"
-    print(url)
     response = requests.get(url)
     if response.status_code != 200:
-        print("请求失败，状态码：", response.status_code)
-        print(response.text)
         return "failure"
     json = response.json()
 
@@ -61,7 +53,6 @@ def delete_key(key_name,user_name):
     """
     # address = os.environ.get("address", "192.168.12.249")
     url = f"{address}/console/api/user_credentials"
-    print(url)
     headers = {
         'Content-Type': 'application/json',  # 说明请求体是 JSON 格式
     }
@@ -70,7 +61,6 @@ def delete_key(key_name,user_name):
         'certificate_key': key_name,
         'user_name': user_name
     }
-    print(data)
     requests.delete(url, headers=headers, json=data)
 
 def list(user_name):
@@ -80,11 +70,8 @@ def list(user_name):
     """
     # address = os.environ.get("address", "192.168.12.249")
     url = f"{address}/console/api/user_credentials/list?user_name={user_name}"
-    print(url)
     response = requests.get(url)
     if response.status_code != 200:
-        print("请求失败，状态码：", response.status_code)
-        print(response.text)
         return "failure"
     json = response.json()
 
@@ -98,11 +85,8 @@ def exist(key_name, user_name):
     """
     # address = os.environ.get("address", "192.168.12.249")
     url = f"{address}/console/api/user_credentials?certificate_key={key_name}&user_name={user_name}"
-    print(url)
     response = requests.get(url)
     if response.status_code != 200:
-        print("请求失败，状态码：", response.status_code)
-        print(response.text)
         return False
 
 
@@ -114,7 +98,6 @@ def encode(certificate_value):
         key_name：凭证key
     """
     url = f"{address}/console/api/get_encode_credentials"
-    print(url)
     headers = {
         'Content-Type': 'application/json',  # 说明请求体是 JSON 格式
     }
@@ -122,11 +105,8 @@ def encode(certificate_value):
     data = {
         'certificate_value': certificate_value,
     }
-    print(data)
     response = requests.post(url, headers=headers, json=data)
     if response.status_code != 200:
-        print("请求失败，状态码：", response.status_code)
-        print(response.text)
         return "failure"
     json = response.json()
     return json['certificate_value']

isrpa/llm_chat.py

@@ -1,24 +1,54 @@
+from typing import Optional, Dict, Any, List, Union
 from volcenginesdkarkruntime import Ark
-#暂不统计消耗的token，后续要统计再改，无法传递token没法用dify的大模型，先保证这个功能能用起来
 
-def ai_talk(input_str):
-    """使用方舟SDK调用大模型API"""
+
+def ai_query(
+        query: str,
+        context: Optional[List[Dict[str, str]]] = None,
+        model: str = "deepseek-v3-250324",
+        **kwargs
+) -> Union[str, Dict[str, Any]]:
+    """
+    向AI发送查询，支持对话上下文和函数调用
+
+    参数:
+        query: 用户查询文本
+        context: 之前的对话上下文(role/content格式)
+        model: 使用的AI模型
+        **kwargs: 其他API参数(temperature, max_tokens等)
+
+    返回:
+        普通回答: 返回字符串
+    """
     try:
         # 初始化客户端
         client = Ark(api_key="f0f8b460-c4ea-42a4-a951-5e2e454022ee")
 
+        # 构建消息列表
+        messages = []
+
+        # 添加上下文(如果有)
+        if context:
+            messages.extend(context)
+
+        # 添加当前查询
+        messages.append({"role": "user", "content": query})
+
+        # 设置默认参数
+        default_params = {
+            "model": model,
+            "messages": messages,
+            "max_tokens": kwargs.get("max_tokens", 2048),
+            "temperature": kwargs.get("temperature", 0.5)
+        }
+
         # 调用聊天补全API
-        completion = client.chat.completions.create(
-            model="deepseek-v3-250324",
-            messages=[
-                {"role": "user", "content": input_str}
-            ],
-            max_tokens=512,  # 最大生成token数
-            temperature=0.5  # 温度参数
-        )
-
-        # 返回AI生成的内容
-        return completion.choices[0].message.content
+        completion = client.chat.completions.create(**default_params)
+
+        # 获取响应内容
+        response = completion.choices[0].message.content
+
+        return response
 
     except Exception as e:
-        return f"ai_talk调用时出错: {str(e)}"
+        return f"AI查询时出错: {str(e)}"

isrpa/utils.py

@@ -22,7 +22,6 @@ def getPath(user_name):
     # address = os.environ.get("address", "192.168.12.249")
 
     url = f"{address}/client/getPath"
-    print(url)
     headers = {
         'Content-Type': 'application/json',  # 说明请求体是 JSON 格式
     }
@@ -30,10 +29,8 @@ def getPath(user_name):
     data = {
         'user_name': user_name
     }
-    print(data)
     response = requests.post(url, headers=headers, json=data)
     if response.status_code != 200:
-        print("请求失败，状态码：", response.status_code)
         return "failure"
     return response.json()
 
@@ -49,7 +46,6 @@ def upload_file(file_path, dest_file, user_name):
 
     # address = os.environ.get("address", "192.168.12.249")
     url = f"{address}/client/noticeUpload"
-    print(url)
     headers = {
         'Content-Type': 'application/json',  # 说明请求体是 JSON 格式
     }
@@ -59,7 +55,6 @@ def upload_file(file_path, dest_file, user_name):
         'dest_file': dest_file,
         'user_name': user_name
     }
-    print(data)
     requests.post(url, headers=headers, json=data)
 
 
@@ -73,7 +68,6 @@ def vCode(image: str, code_type, apiKey, secretKey):
     :return:
     """
     url = "https://ai.i-search.com.cn/ocr/v2/vCode"
-    print(url)
     headers = {
         'Content-Type': 'application/json',  # 说明请求体是 JSON 格式
     }
@@ -85,11 +79,9 @@ def vCode(image: str, code_type, apiKey, secretKey):
         'apiKey': apiKey,
         'secretKey': secretKey
     }
-    print(data)
     response = requests.post(url, headers=headers, json=data)
     status_code = response.status_code
     if status_code != 200:
-        print("请求失败，状态码：", status_code)
         return {"error_msg": "failure", "error_code": status_code}
     return response.json()
 

{isrpa-0.9.7.dist-info → isrpa-0.9.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: isrpa
-Version: 0.9.7
+Version: 0.9.9
 Summary: isrpa package
 Home-page: https://github.com/yourusername/mypackage
 Author: ysq

isrpa-0.9.9.dist-info/RECORD

@@ -0,0 +1,12 @@
+isrpa/OCR.py,sha256=nSDmVLo_z9GV7GUY57xBh2kumERG5_xtpUPxeF6ziQE,2509
+isrpa/Template.py,sha256=UP1uovLvJzXEPGo_hzLrIYOu3vZP8Cf6m3G2hGzQ6XA,1958
+isrpa/Template_Exception.py,sha256=an6eQ1NMB8a9R-J0yyMdwHIZcZeDtdV5f7fWKZQsZ3s,819
+isrpa/__init__.py,sha256=pG-YPVG0gJIJ6s4xcAz9S_CnUxpUcz33wl9eNUSgxGk,20
+isrpa/credentials.py,sha256=B9CKBgvik7U9YwwXIgtDb1jUfZuNcditLzyaL_-0arw,3342
+isrpa/llm_chat.py,sha256=jTuK9wcm3X36V9lXuGuTCdNup_zi2NeG2GuYHzO7gm4,1541
+isrpa/message.py,sha256=N0s6hMaC5S-Pi4EhwcSp-ngfBTBQyW6ymU9sI9dUSeE,11329
+isrpa/utils.py,sha256=Du2_yQJlB-JBm6Y6a3kT74mOGCtM3XffyX_5_cFudRI,51213
+isrpa-0.9.9.dist-info/METADATA,sha256=BV0HEf3Ixu5wLaOjR4GlbopVrgc5ZmGz_Vtmj1CvE3Q,511
+isrpa-0.9.9.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
+isrpa-0.9.9.dist-info/top_level.txt,sha256=J5HJbtR2WAeKnW1rrpkiuapwnlswv3RgM-riyZD87o0,6
+isrpa-0.9.9.dist-info/RECORD,,

isrpa-0.9.7.dist-info/RECORD

@@ -1,12 +0,0 @@
-isrpa/OCR.py,sha256=nSDmVLo_z9GV7GUY57xBh2kumERG5_xtpUPxeF6ziQE,2509
-isrpa/Template.py,sha256=UP1uovLvJzXEPGo_hzLrIYOu3vZP8Cf6m3G2hGzQ6XA,1958
-isrpa/Template_Exception.py,sha256=an6eQ1NMB8a9R-J0yyMdwHIZcZeDtdV5f7fWKZQsZ3s,819
-isrpa/__init__.py,sha256=pG-YPVG0gJIJ6s4xcAz9S_CnUxpUcz33wl9eNUSgxGk,20
-isrpa/credentials.py,sha256=Qgitrm2pKgA1E_iGPg12Qq2Ka52pgTQCBHnT2U62ZHs,3999
-isrpa/llm_chat.py,sha256=lrufXl2JnnezXPF3FOMmmQFbqMTLjhCy7ujJTWeu460,863
-isrpa/message.py,sha256=N0s6hMaC5S-Pi4EhwcSp-ngfBTBQyW6ymU9sI9dUSeE,11329
-isrpa/utils.py,sha256=uPCb1A1vDB0nTlII2BJz9kS0amk7mvzVA7ilI-UYA8Q,51439
-isrpa-0.9.7.dist-info/METADATA,sha256=Y_yaHUXMpR5uicuMfDZ1Mh88aQF_ol7-yiDl-lOdHn0,511
-isrpa-0.9.7.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
-isrpa-0.9.7.dist-info/top_level.txt,sha256=J5HJbtR2WAeKnW1rrpkiuapwnlswv3RgM-riyZD87o0,6
-isrpa-0.9.7.dist-info/RECORD,,