clean up mercury

This commit is contained in:
MaxRobinsonTheGreat 2025-08-23 15:40:16 -05:00
parent 7d3fb22dfe
commit f41f03cdb3
3 changed files with 6 additions and 9 deletions

View file

@ -65,7 +65,7 @@ You can configure the agent's name, model, and prompts in their profile like `an
| `hyperbolic` | `HYPERBOLIC_API_KEY` | `hyperbolic/deepseek-ai/DeepSeek-V3` | [docs](https://docs.hyperbolic.xyz/docs/getting-started) | | `hyperbolic` | `HYPERBOLIC_API_KEY` | `hyperbolic/deepseek-ai/DeepSeek-V3` | [docs](https://docs.hyperbolic.xyz/docs/getting-started) |
| `vllm` | n/a | `vllm/llama3` | n/a | | `vllm` | n/a | `vllm/llama3` | n/a |
| `cerebras` | `CEREBRAS_API_KEY` | `cerebras/llama-3.3-70b` | [docs](https://inference-docs.cerebras.ai/introduction) | | `cerebras` | `CEREBRAS_API_KEY` | `cerebras/llama-3.3-70b` | [docs](https://inference-docs.cerebras.ai/introduction) |
| `mercury(EA)` | `MERCURY_API_KEY` | `mercury-coder-small` | [docs](https://www.inceptionlabs.ai/) | | `mercury` | `MERCURY_API_KEY` | `mercury-coder-small` | [docs](https://www.inceptionlabs.ai/) |
If you use Ollama, to install the models used by default (generation and embedding), execute the following terminal command: If you use Ollama, to install the models used by default (generation and embedding), execute the following terminal command:
`ollama pull llama3.1 && ollama pull nomic-embed-text` `ollama pull llama3.1 && ollama pull nomic-embed-text`

View file

@ -3,13 +3,7 @@
"cooldown": 5000, "cooldown": 5000,
"model": { "model": "mercury/mercury-coder-small",
"api": "mercury",
"url": "https://api.inceptionlabs.ai/v1",
"model": "mercury-coder-small"
},
"embedding": "openai", "embedding": "openai"
"description":"Official Website Introduction:The world’s first diffusion large language models"
} }

View file

@ -3,12 +3,15 @@ import { getKey, hasKey } from '../utils/keys.js';
import { strictFormat } from '../utils/text.js'; import { strictFormat } from '../utils/text.js';
export class Mercury { export class Mercury {
static prefix = 'mercury';
constructor(model_name, url, params) { constructor(model_name, url, params) {
this.model_name = model_name; this.model_name = model_name;
this.params = params; this.params = params;
let config = {}; let config = {};
if (url) if (url)
config.baseURL = url; config.baseURL = url;
else
config.baseURL = "https://api.inceptionlabs.ai/v1";
config.apiKey = getKey('MERCURY_API_KEY'); config.apiKey = getKey('MERCURY_API_KEY');