isrpa 0.9.6__py3-none-any.whl → 0.9.8__py3-none-any.whl

isrpa/credentials.py

@@ -133,4 +133,4 @@ def encode(certificate_value):
 
 
 if __name__ == '__main__':
-    print(store_key("cc11111c","wqqwqqww","zhouly@i-search.com.cn"))
+    print(exist("ccc","zhouly@i-search.com.cn"))

isrpa/llm_chat.py

@@ -0,0 +1,54 @@
+from typing import Optional, Dict, Any, List, Union
+from volcenginesdkarkruntime import Ark
+
+
+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(**default_params)
+
+        # 获取响应内容
+        response = completion.choices[0].message.content
+
+        return response
+
+    except Exception as e:
+        return f"AI查询时出错: {str(e)}"

{isrpa-0.9.6.dist-info → isrpa-0.9.8.dist-info}/METADATA

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

{isrpa-0.9.6.dist-info → isrpa-0.9.8.dist-info}/RECORD

@@ -2,10 +2,11 @@ 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=LoIf_wxrh8GB4DaJ-sOGuxWeVai-EWlncF0zOmT0BrA,4019
+isrpa/credentials.py,sha256=Qgitrm2pKgA1E_iGPg12Qq2Ka52pgTQCBHnT2U62ZHs,3999
+isrpa/llm_chat.py,sha256=jTuK9wcm3X36V9lXuGuTCdNup_zi2NeG2GuYHzO7gm4,1541
 isrpa/message.py,sha256=N0s6hMaC5S-Pi4EhwcSp-ngfBTBQyW6ymU9sI9dUSeE,11329
 isrpa/utils.py,sha256=uPCb1A1vDB0nTlII2BJz9kS0amk7mvzVA7ilI-UYA8Q,51439
-isrpa-0.9.6.dist-info/METADATA,sha256=3Fr5AxvfiZ85i0qd1WooM0v1rSHd2RFx0PI5kKOCl6A,511
-isrpa-0.9.6.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
-isrpa-0.9.6.dist-info/top_level.txt,sha256=J5HJbtR2WAeKnW1rrpkiuapwnlswv3RgM-riyZD87o0,6
-isrpa-0.9.6.dist-info/RECORD,,
+isrpa-0.9.8.dist-info/METADATA,sha256=LWf7tYP1P5yEyX8DHayHPmQgV8W_Z2fPZzufCbILKAU,511
+isrpa-0.9.8.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
+isrpa-0.9.8.dist-info/top_level.txt,sha256=J5HJbtR2WAeKnW1rrpkiuapwnlswv3RgM-riyZD87o0,6
+isrpa-0.9.8.dist-info/RECORD,,