@lmliheng/agent 1.0.0 → 1.0.2

package/README.md

@@ -1 +1,33 @@
-26-8-27： 做了prompts里的JSON，表格，拒绝回答等格式的prompt函数
+26-8-27： 做了prompts里的JSON，表格，拒绝回答等格式的prompt函数
+
+用法如下，参考test目录案例：
+```js
+message = new Message(answer, p.refuse('强化学习/华为/许家印'))
+```
+```json
+user: 说说RL
+抱歉，关于强化学习（RL）这个话题，我暂时无法为您提供详细信息。建议您查阅相关学术资料或咨询专业领域人士。如有其他问题，欢迎继续提问。
+user: 还是说说恒大吧
+抱歉，关于恒大集团及许家印相关的话题，我暂时无法为您提供详细信息。建议您通过官方渠道或权威新闻平台了解公开信息。如有其他问题，欢迎继续提问。
+user: 好吧
+感谢您的理解！如果之后有任何其他问题，比如技术、生活、学习或其他我能协助的领域，随时欢迎提问。祝您今天一切顺利！ 😊
+```
+
+json回答:
+```js
+let answer_diff = new Prompts(answer)
+message = new Message(answer_diff.json())
+```
+
+```json
+user: 感谢您的理解！如果之后有任何其他问题，比如技术、生活、学习或其他我能协助的领域，随时欢迎提问。祝您今天一切顺利
+{
+  "name": "友好告别信息",
+  "features": [
+    "表达感谢与理解",
+    "主动提供后续帮助领域",
+    "传递美好祝愿"
+  ],
+  "rating": 9
+}
+```

package/dist/index.mjs

@@ -4,6 +4,8 @@
  * let message=new Message('说说KV cache')
  * let message=new Message('说说KV cache','${prompt.JSON}')
  * 
+ * 增加对话：
+ * MessageAdd(message,role)
  */
 class Message {
 
@@ -13,7 +15,7 @@ class Message {
      * SystemMessage默认为空
      */
     constructor(text, SystemMessage = '') {
-        this.message = [
+        this.content = [
             { "role": "system", "content": `${SystemMessage}` },
             { "role": "user", "content": `${text}` }
         ];
@@ -23,10 +25,16 @@ class Message {
      * role可以选择system,assistant,tool，user
      */
     MessageAdd(message, role) {
-        this.message.push({ "role": role, "content": message });
+        this.content.push({ "role": role, "content": message });
     }
 }
 
+/**
+ * @propmts对象
+ * 
+ * 
+ */
+
 class Prompts {
 
     constructor(text) {
@@ -123,4 +131,55 @@ class Prompts {
     }
 }
 
-export { Message, Prompts };
+/**
+ * @ds单多轮对话调用
+ */
+async function callDeepSeek(messages, options = {}) {
+    const API_URL = process.env.DEEPSEEK_BASE_URL || 'https://api.deepseek.com/chat/completions';
+    const MODEL = process.env.DEEPSEEK_MODEL || 'deepseek-v4-flash';
+    // 调用模型前先检查 API Key，避免发送无效请求。
+    if (!process.env.DEEPSEEK_API_KEY) {
+        throw new Error('缺少 DEEPSEEK_API_KEY，请先在 .env 中完成配置。')
+    }
+    const startedAt = Date.now();
+    const response = await fetch(API_URL, {
+        method: 'POST',
+        headers: {
+            Authorization: `Bearer ${process.env.DEEPSEEK_API_KEY}`,
+            'Content-Type': 'application/json'
+        },
+        body: JSON.stringify({
+            model: MODEL,
+            messages,
+            ...options,
+        })
+    });
+
+    /**
+     * 无论 HTTP 状态码是否成功，都先解析响应体。
+     *
+     * 这样接口失败时，也可以将服务端返回的错误信息
+     * 一起放入异常中，方便定位问题。
+     */
+    const data = await response.json();
+
+    if (!response.ok) {
+        throw new Error(
+            `DeepSeek 调用失败：${response.status} ${JSON.stringify(data)}`
+        )
+    }
+
+    const choice = data.choices?.[0];
+    if (!choice?.message) {
+        throw new Error(`DeepSeek 没有返回有效消息：${JSON.stringify(data)}`)
+    }
+
+    return {
+        message: choice.message,
+        finishReason: choice.finish_reason,
+        latencyMs: Date.now() - startedAt,
+        usage: data.usage
+    }
+}
+
+export { Message, Prompts, callDeepSeek };

package/index.js

@@ -1,2 +1,5 @@
 export { default as Message } from './message/index.js'
-export { default as Prompts } from './prompts/index.js'
+export { default as Prompts } from './prompts/index.js'
+
+
+export { callDeepSeek } from './llm/deepseek_client.js'

package/llm/deepseek_client.js

@@ -0,0 +1,80 @@
+/**
+ * @ds单多轮对话调用
+ */
+export async function callDeepSeek(messages, options = {}) {
+    const API_URL = process.env.DEEPSEEK_BASE_URL || 'https://api.deepseek.com/chat/completions'
+    const MODEL = process.env.DEEPSEEK_MODEL || 'deepseek-v4-flash'
+    // 调用模型前先检查 API Key，避免发送无效请求。
+    if (!process.env.DEEPSEEK_API_KEY) {
+        throw new Error('缺少 DEEPSEEK_API_KEY，请先在 .env 中完成配置。')
+    }
+    const startedAt = Date.now()
+    const response = await fetch(API_URL, {
+        method: 'POST',
+        headers: {
+            Authorization: `Bearer ${process.env.DEEPSEEK_API_KEY}`,
+            'Content-Type': 'application/json'
+        },
+        body: JSON.stringify({
+            model: MODEL,
+            messages,
+            ...options,
+        })
+    })
+
+    /**
+     * 无论 HTTP 状态码是否成功，都先解析响应体。
+     *
+     * 这样接口失败时，也可以将服务端返回的错误信息
+     * 一起放入异常中，方便定位问题。
+     */
+    const data = await response.json()
+
+    if (!response.ok) {
+        throw new Error(
+            `DeepSeek 调用失败：${response.status} ${JSON.stringify(data)}`
+        )
+    }
+
+    const choice = data.choices?.[0]
+    if (!choice?.message) {
+        throw new Error(`DeepSeek 没有返回有效消息：${JSON.stringify(data)}`)
+    }
+
+    return {
+        message: choice.message,
+        finishReason: choice.finish_reason,
+        latencyMs: Date.now() - startedAt,
+        usage: data.usage
+    }
+}
+
+/**
+ * @余额查询
+ */
+export async function getDeepseekBalance() {
+    const token = process.env.DEEPSEEK_API_KEY
+    if (!token) {
+        throw new Error('DEEPSEEK_API_KEY 未设置，检查 .env 和 dotenv 加载顺序')
+    }
+    try {
+        const response = await fetch('https://api.deepseek.com/user/balance', {
+            method: 'GET',
+            headers: {
+                'Accept': 'application/json',
+                'Authorization': `Bearer ${process.env.DEEPSEEK_API_KEY}`
+            }
+        })
+        if (!response.ok) {
+            throw new Error(`HTTP error! status: ${response.status}`)
+        }
+        // console.log(response)
+        const data = await response.json()
+        return data
+
+    } catch (error) {
+        console.error('请求失败:', error.message)
+    }
+}
+
+

package/message/index.js

@@ -6,6 +6,8 @@
  * let message=new Message('说说KV cache')
  * let message=new Message('说说KV cache','${prompt.JSON}')
  * 
+ * 增加对话：
+ * MessageAdd(message,role)
  */
 export default class Message {
 
@@ -15,7 +17,7 @@ export default class Message {
      * SystemMessage默认为空
      */
     constructor(text, SystemMessage = '') {
-        this.message = [
+        this.content = [
             { "role": "system", "content": `${SystemMessage}` },
             { "role": "user", "content": `${text}` }
         ]
@@ -25,7 +27,7 @@ export default class Message {
      * role可以选择system,assistant,tool，user
      */
     MessageAdd(message, role) {
-        this.message.push({ "role": role, "content": message })
+        this.content.push({ "role": role, "content": message })
     }
 }
 

package/package.json

@@ -4,7 +4,7 @@
     "access": "public"
   },
   "name": "@lmliheng/agent",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "agent包，目前有message，prompts模块，最近一次更新是26-8-27",
   "keywords": [
     "prompt",
@@ -14,9 +14,7 @@
   "license": "MIT",
   "main": "index.js",
   "scripts": {
-    "test": "node --test",
-    "mcp": "node index.mjs --server",
-    "disk": "node src/Disk/disk.js",
+    "llm": "node --env-file=.env ./test/llm.test.js",
     "build": "rollup -c",
     "p": "npm version patch && npm publish --registry=https://registry.npmjs.org/"
   },
@@ -27,4 +25,4 @@
   "devDependencies": {
     "rollup": "^4.62.4"
   }
-}
+}

package/prompts/index.js

@@ -1,3 +1,10 @@
+
+/**
+ * @propmts对象
+ * 
+ * 
+ */
+
 export default class Prompts {
 
     constructor(text) {

package/test/.env

@@ -0,0 +1 @@
+DEEPSEEK_API_KEY='sk-67cc769fcef146819f6aaedcc0716f74'

package/test/llm.test.js

@@ -0,0 +1,42 @@
+import { callDeepSeek, Message, Prompts } from '../index.js'
+import * as readline from 'readline'
+
+
+// node --env-file=.env llm.test.js chat
+
+if (process.argv[2] === 'chat') {
+    const rl = readline.createInterface({
+        input: process.stdin,
+        output: process.stdout,
+    })
+
+    let p = new Prompts()
+    let message
+    function ask() {
+        rl.question('user: ', async (answer) => {
+            if (answer.toLowerCase() === 'stop') {
+                rl.close()
+                return
+            }
+            let answer_diff = new Prompts(answer)
+            if (message === undefined) {
+                message = new Message(answer_diff.json(), p.refuse('强化学习/华为/许家印'))
+            } else {
+                message.MessageAdd(answer, 'user')
+            }
+
+            let res = await callDeepSeek(message.content, {
+                temperature: 0.9
+            })
+            console.log(res.message.content)
+            message.MessageAdd(res.message.content, 'assistant')
+            ask()
+        })
+    }
+    ask()
+}
+
+if (process.argv[2] === 'balance') {
+    let balance = await getDeepseekBalance()
+    console.table(balance.balance_infos)
+}