Files

2.4 KiB
Raw Permalink Blame History

DeepSeek 使用

1. 创建相关文件

创建一个yaml文件例如 user_deepseek_llm.yaml 将以下内容粘贴到您的user_deepseek_llm.yaml文件当中

name: 'user_deepseek_llm'
description: 'default default_deepseek_llm llm with spi'
model_name: 'deepseek-chat'
max_tokens: 1000
temperature: 0.5
api_key: '${DEEPSEEK_API_KEY}'
api_base: 'https://api.deepseek.com/v1'
organization: '${DEEPSEEK_ORGANIZATION}'
proxy: '${DEEPSEEK_PROXY}'
streaming: True
metadata:
  type: 'LLM'
  module: 'agentuniverse.llm.default.deep_seek_openai_style_llm'
  class: 'DefaultDeepSeekLLM'

note: api_key/api_base/organization/proxy等模型参数有三种配置方法

  1. 直接字符串值直接在配置文件中输入API密钥字符串。

    api_key: 'sk-***'
    
  2. 环境变量占位符:使用${VARIABLE_NAME}语法从环境变量中加载。当agentUniverse启动时会自动从环境变量读取相应的值。

    api_key: '${DEEPSEEK_API_KEY}'
    
  3. 自定义函数加载:使用@FUNC注解在运行时通过自定义函数动态加载API密钥。

    api_key: '@FUNC(load_api_key(model_name="deepseek"))'
    

    该函数需要在yaml_func_extension.py文件的YamlFuncExtension类中定义可参考样例工程中的YamlFuncExtension当agentUniverse加载此配置时

    • 解析@FUNC注解
    • 执行load_api_key函数并传入相应参数
    • 用函数返回值替换注解内容

2. 环境设置

示例yaml中模型密钥等参数使用环境变量占位符下面将介绍环境变量设置方法。

必须配置DEEPSEEK_API_KEY
可选配置DEEPSEEK_API_BASE, DEEPSEEK_PROXY, DEEPSEEK_ORGANIZATION

2.1 通过python代码配置

import os
os.environ['DEEPSEEK_API_KEY'] = 'sk-***'
os.environ['DEEPSEEK_API_BASE'] = 'https://xxxxxx'

2.2 通过配置文件配置

在项目的config目录下的custom_key.toml当中添加配置

DEEPSEEK_API_KEY="sk-******"
DEEPSEEK_API_BASE="https://xxxxxx"
DEEPSEEK_ORGANIZATION = ''
DEEPSEEK_PROXY = ''

3. DEEPSEEK API KEY 获取

参考 DEEPSEEK 官方文档:https://platform.deepseek.com/api_keys

4. Tips

在agentuniverse中我们已经创建了一个name为default_deepseek_llm的llm用户在配置DEEPSEEK_API_KEY之后可以直接使用。