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

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-0.9.7.dist-info → isrpa-0.9.8.dist-info}/METADATA

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

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

@@ -3,10 +3,10 @@ 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/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.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,,
+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,,