|
| 1 | +import { RESPONSE_FORMAT_TYPE_OPTIONS } from "../../common/constants.mjs"; |
| 2 | +import { parseObject } from "../../common/util.mjs"; |
| 3 | +import deepseek from "../../deepseek.app.mjs"; |
| 4 | + |
| 5 | +export default { |
| 6 | + key: "deepseek-create-chat-completion", |
| 7 | + name: "Create Chat Completion", |
| 8 | + description: "Creates a chat completion using the DeepSeek API. [See the documentation](https://api-docs.deepseek.com/api/create-chat-completion)", |
| 9 | + version: "0.0.1", |
| 10 | + type: "action", |
| 11 | + props: { |
| 12 | + deepseek, |
| 13 | + messages: { |
| 14 | + type: "string[]", |
| 15 | + label: "Messages", |
| 16 | + description: "The messages for the chat conversation as JSON strings. Each message should be a JSON string like '{\"role\": \"user\", \"content\": \"Hello!\"}'. [See the documentation](https://api-docs.deepseek.com/api/create-chat-completion) for further details.", |
| 17 | + }, |
| 18 | + frequencyPenalty: { |
| 19 | + type: "string", |
| 20 | + label: "Frequency Penalty", |
| 21 | + description: "Number between -2.0 and 2.0. Positive values penalize new tokens based on their existing frequency in the text so far, decreasing the model's likelihood to repeat the same line verbatim.", |
| 22 | + optional: true, |
| 23 | + }, |
| 24 | + maxTokens: { |
| 25 | + type: "integer", |
| 26 | + label: "Max Tokens", |
| 27 | + description: "Integer between 1 and 8192. The maximum number of tokens that can be generated in the chat completion. The total length of input tokens and generated tokens is limited by the model's context length. If `max_tokens` is not specified, the default value 4096 is used.", |
| 28 | + optional: true, |
| 29 | + }, |
| 30 | + presencePenalty: { |
| 31 | + type: "string", |
| 32 | + label: "Presence Penalty", |
| 33 | + description: "Number between -2.0 and 2.0. Positive values penalize new tokens based on whether they appear in the text so far, increasing the model's likelihood to talk about new topics.", |
| 34 | + optional: true, |
| 35 | + }, |
| 36 | + responseFormatType: { |
| 37 | + type: "string", |
| 38 | + label: "Response Format Type", |
| 39 | + description: "The format that the model must output. Setting to JSON Object enables JSON Output, which guarantees the message the model generates is valid JSON.", |
| 40 | + options: RESPONSE_FORMAT_TYPE_OPTIONS, |
| 41 | + optional: true, |
| 42 | + }, |
| 43 | + stop: { |
| 44 | + type: "string[]", |
| 45 | + label: "Stop Sequences", |
| 46 | + description: "Up to 16 sequences where the API will stop generating further tokens.", |
| 47 | + optional: true, |
| 48 | + }, |
| 49 | + stream: { |
| 50 | + type: "boolean", |
| 51 | + label: "Stream", |
| 52 | + description: "If set, partial message deltas will be sent. Tokens will be sent as data-only server-sent events (SSE) as they become available, with the stream terminated by a `data: [DONE]` message.", |
| 53 | + optional: true, |
| 54 | + reloadProps: true, |
| 55 | + }, |
| 56 | + streamIncludeUsage: { |
| 57 | + type: "string", |
| 58 | + label: "Stream Include Usage", |
| 59 | + description: "If set, an additional chunk will be streamed before the `data: [DONE]` message. The `usage` field on this chunk shows the token usage statistics for the entire request, and the `choices` field will always be an empty array. All other chunks will also include a `usage` field, but with a null value.", |
| 60 | + optional: true, |
| 61 | + hidden: true, |
| 62 | + }, |
| 63 | + temperature: { |
| 64 | + type: "string", |
| 65 | + label: "Temperature", |
| 66 | + description: "What sampling temperature to use, between 0 and 2. Higher values like 0.8 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. We generally recommend altering this or Top P but not both.", |
| 67 | + optional: true, |
| 68 | + }, |
| 69 | + topP: { |
| 70 | + type: "string", |
| 71 | + label: "Top P", |
| 72 | + description: "An alternative to sampling with temperature, called nucleus sampling, where the model considers the results of the tokens with top_p probability mass. So 0.1 means only the tokens comprising the top 10% probability mass are considered. We generally recommend altering this or Temperature but not both.", |
| 73 | + optional: true, |
| 74 | + }, |
| 75 | + tools: { |
| 76 | + type: "string[]", |
| 77 | + label: "Tools", |
| 78 | + description: "A list of tools the model may call. Currently, only functions are supported as a tool. Use this to provide a list of functions the model may generate JSON inputs for. A max of 128 functions are supported.", |
| 79 | + optional: true, |
| 80 | + }, |
| 81 | + toolChoice: { |
| 82 | + type: "string", |
| 83 | + label: "Tool Choice", |
| 84 | + description: "Controls which (if any) tool is called by the model. [See the documentation](https://api-docs.deepseek.com/api/create-chat-completion) for further details.", |
| 85 | + optional: true, |
| 86 | + }, |
| 87 | + logprobs: { |
| 88 | + type: "boolean", |
| 89 | + label: "Log Probs", |
| 90 | + description: "Whether to return log probabilities of the output tokens or not. If true, returns the log probabilities of each output token returned in the `content` of `message`.", |
| 91 | + optional: true, |
| 92 | + }, |
| 93 | + topLogprobs: { |
| 94 | + type: "string", |
| 95 | + label: "Top Log Probabilities", |
| 96 | + description: "An integer between 0 and 20 specifying the number of most likely tokens to return at each token position, each with an associated log probability. logprobs must be set to true if this parameter is used.", |
| 97 | + optional: true, |
| 98 | + }, |
| 99 | + }, |
| 100 | + async additionalProps(props) { |
| 101 | + props.streamIncludeUsage.hidden = !this.stream; |
| 102 | + return {}; |
| 103 | + }, |
| 104 | + async run({ $ }) { |
| 105 | + const response = await this.deepseek.createModelResponse({ |
| 106 | + $, |
| 107 | + data: { |
| 108 | + messages: parseObject(this.messages), |
| 109 | + model: "deepseek-chat", |
| 110 | + frequency_penalty: this.frequencyPenalty && parseInt(this.frequencyPenalty), |
| 111 | + max_tokens: this.maxTokens, |
| 112 | + presence_penalty: this.presencePenalty && parseInt(this.presencePenalty), |
| 113 | + response_format: this.responseFormatType |
| 114 | + ? { |
| 115 | + type: this.responseFormatType, |
| 116 | + } |
| 117 | + : null, |
| 118 | + stop: parseObject(this.stop), |
| 119 | + stream: this.stream, |
| 120 | + stream_options: this.stream |
| 121 | + ? { |
| 122 | + include_usage: this.streamIncludeUsage, |
| 123 | + } |
| 124 | + : null, |
| 125 | + temperature: this.temperature && parseInt(this.temperature), |
| 126 | + top_p: this.topP && parseInt(this.topP), |
| 127 | + tools: parseObject(this.tools), |
| 128 | + tool_choice: parseObject(this.toolChoice), |
| 129 | + logprobs: this.logprobs, |
| 130 | + top_logprobs: this.topLogprobs && parseInt(this.topLogprobs), |
| 131 | + }, |
| 132 | + }); |
| 133 | + $.export("$summary", "Chat completion created"); |
| 134 | + return response; |
| 135 | + }, |
| 136 | +}; |
0 commit comments