diff --git a/.env.example b/.env.example
index ee896bf1ea2..03f3b3bbcfb 100644
--- a/.env.example
+++ b/.env.example
@@ -100,32 +100,32 @@ MEDIUM_HYPERBOLIC_MODEL= # Default: meta-llama/Meta-Llama-3.1-70B-Instruc
LARGE_HYPERBOLIC_MODEL= # Default: meta-llama/Meta-Llama-3.1-405-Instruct
# Infera Configuration
-INFERA_API_KEY= # visit api.infera.org/docs to obtain an API key under /signup_user
-INFERA_MODEL= # Default: llama3.2:latest
-INFERA_SERVER_URL= # Default: https://api.infera.org/
-SMALL_INFERA_MODEL= #Recommended: llama3.2:latest
-MEDIUM_INFERA_MODEL= #Recommended: mistral-nemo:latest
-LARGE_INFERA_MODEL= #Recommended: mistral-small:latest
-
-# Venice Configuration
-VENICE_API_KEY= # generate from venice settings
-SMALL_VENICE_MODEL= # Default: llama-3.3-70b
-MEDIUM_VENICE_MODEL= # Default: llama-3.3-70b
-LARGE_VENICE_MODEL= # Default: llama-3.1-405b
-IMAGE_VENICE_MODEL= # Default: fluently-xl
-
-# Nineteen.ai Configuration
-NINETEEN_AI_API_KEY= # Get a free api key from https://nineteen.ai/app/api
-SMALL_NINETEEN_AI_MODEL= # Default: unsloth/Llama-3.2-3B-Instruct
-MEDIUM_NINETEEN_AI_MODEL= # Default: unsloth/Meta-Llama-3.1-8B-Instruct
-LARGE_NINETEEN_AI_MODEL= # Default: hugging-quants/Meta-Llama-3.1-70B-Instruct-AWQ-INT4
-IMAGE_NINETEEN_AI_MODE= # Default: dataautogpt3/ProteusV0.4-Lightning
-
-# Akash Chat API Configuration docs: https://chatapi.akash.network/documentation
-AKASH_CHAT_API_KEY= # Get from https://chatapi.akash.network/
-SMALL_AKASH_CHAT_API_MODEL= # Default: Meta-Llama-3-2-3B-Instruct
-MEDIUM_AKASH_CHAT_API_MODEL= # Default: Meta-Llama-3-3-70B-Instruct
-LARGE_AKASH_CHAT_API_MODEL= # Default: Meta-Llama-3-1-405B-Instruct-FP8
+ INFERA_API_KEY= # visit api.infera.org/docs to obtain an API key under /signup_user
+ INFERA_MODEL= # Default: llama3.2:latest
+ INFERA_SERVER_URL= # Default: https://api.infera.org/
+ SMALL_INFERA_MODEL= #Recommended: llama3.2:latest
+ MEDIUM_INFERA_MODEL= #Recommended: mistral-nemo:latest
+ LARGE_INFERA_MODEL= #Recommended: mistral-small:latest
+
+ # Venice Configuration
+ VENICE_API_KEY= # generate from venice settings
+ SMALL_VENICE_MODEL= # Default: llama-3.3-70b
+ MEDIUM_VENICE_MODEL= # Default: llama-3.3-70b
+ LARGE_VENICE_MODEL= # Default: llama-3.1-405b
+ IMAGE_VENICE_MODEL= # Default: fluently-xl
+
+ # Nineteen.ai Configuration
+ NINETEEN_AI_API_KEY= # Get a free api key from https://nineteen.ai/app/api
+ SMALL_NINETEEN_AI_MODEL= # Default: unsloth/Llama-3.2-3B-Instruct
+ MEDIUM_NINETEEN_AI_MODEL= # Default: unsloth/Meta-Llama-3.1-8B-Instruct
+ LARGE_NINETEEN_AI_MODEL= # Default: hugging-quants/Meta-Llama-3.1-70B-Instruct-AWQ-INT4
+ IMAGE_NINETEEN_AI_MODE= # Default: dataautogpt3/ProteusV0.4-Lightning
+
+ # Akash Chat API Configuration docs: https://chatapi.akash.network/documentation
+ AKASH_CHAT_API_KEY= # Get from https://chatapi.akash.network/
+ SMALL_AKASH_CHAT_API_MODEL= # Default: Meta-Llama-3-2-3B-Instruct
+ MEDIUM_AKASH_CHAT_API_MODEL= # Default: Meta-Llama-3-3-70B-Instruct
+ LARGE_AKASH_CHAT_API_MODEL= # Default: Meta-Llama-3-1-405B-Instruct-FP8
# Livepeer configuration
LIVEPEER_GATEWAY_URL= # Free inference gateways and docs: https://livepeer-eliza.com/
@@ -184,6 +184,12 @@ MEDIUM_GOOGLE_MODEL= # Default: gemini-1.5-flash-latest
LARGE_GOOGLE_MODEL= # Default: gemini-1.5-pro-latest
EMBEDDING_GOOGLE_MODEL= # Default: text-embedding-004
+# Mistral Configuration
+MISTRAL_MODEL=
+SMALL_MISTRAL_MODEL= # Default: mistral-small-latest
+MEDIUM_MISTRAL_MODEL= # Default: mistral-large-latest
+LARGE_MISTRAL_MODEL= # Default: mistral-large-latest
+
# Groq Configuration
GROQ_API_KEY= # Starts with gsk_
SMALL_GROQ_MODEL= # Default: llama-3.1-8b-instant
@@ -232,6 +238,14 @@ MEDIUM_VOLENGINE_MODEL= # Default: doubao-pro-128k
LARGE_VOLENGINE_MODEL= # Default: doubao-pro-256k
VOLENGINE_EMBEDDING_MODEL= # Default: doubao-embedding
+# DeepSeek Configuration
+DEEPSEEK_API_KEY= #Your DeepSeek API key
+DEEPSEEK_API_URL= # Default: https://api.deepseek.com
+SMALL_DEEPSEEK_MODEL= # Default: deepseek-chat
+MEDIUM_DEEPSEEK_MODEL= # Default: deepseek-chat
+LARGE_DEEPSEEK_MODEL= # Default: deepseek-chat
+
+
# fal.ai Configuration
FAL_API_KEY=
FAL_AI_LORA_PATH=
@@ -257,6 +271,13 @@ TOGETHER_API_KEY= # Together API Key
#### Crypto Plugin Configurations ####
######################################
+# CoinMarketCap / CMC
+COINMARKETCAP_API_KEY=
+
+# CoinGecko
+COINGECKO_API_KEY=
+COINGECKO_PRO_API_KEY=
+
# EVM
EVM_PRIVATE_KEY=
EVM_PROVIDER_URL=
@@ -299,6 +320,10 @@ STARKNET_ADDRESS=
STARKNET_PRIVATE_KEY=
STARKNET_RPC_URL=
+# Lens Network Configuration
+LENS_ADDRESS=
+LENS_PRIVATE_KEY=
+
# Coinbase
COINBASE_COMMERCE_KEY= # From Coinbase developer portal
COINBASE_API_KEY= # From Coinbase developer portal
@@ -385,6 +410,12 @@ FLOW_ENDPOINT_URL= # Default: https://mainnet.onflow.org
INTERNET_COMPUTER_PRIVATE_KEY=
INTERNET_COMPUTER_ADDRESS=
+
+#Cloudflare AI Gateway
+CLOUDFLARE_GW_ENABLED= # Set to true to enable Cloudflare AI Gateway
+CLOUDFLARE_AI_ACCOUNT_ID= # Cloudflare AI Account ID - found in the Cloudflare Dashboard under AI Gateway
+CLOUDFLARE_AI_GATEWAY_ID= # Cloudflare AI Gateway ID - found in the Cloudflare Dashboard under AI Gateway
+
# Aptos
APTOS_PRIVATE_KEY= # Aptos private key
APTOS_NETWORK= # Must be one of mainnet, testnet
@@ -467,6 +498,8 @@ GIPHY_API_KEY=
# OpenWeather
OPEN_WEATHER_API_KEY= # OpenWeather API key
+
+
# EchoChambers Configuration
ECHOCHAMBERS_API_URL=http://127.0.0.1:3333
ECHOCHAMBERS_API_KEY=testingkey0011
@@ -500,3 +533,57 @@ TAVILY_API_KEY=
# Verifiable Inference Configuration
VERIFIABLE_INFERENCE_ENABLED=false # Set to false to disable verifiable inference
VERIFIABLE_INFERENCE_PROVIDER=opacity # Options: opacity
+
+
+# Autonome Configuration
+AUTONOME_JWT_TOKEN=
+AUTONOME_RPC=https://wizard-bff-rpc.alt.technology/v1/bff/aaa/apps
+
+####################################
+#### Akash Network Configuration ####
+####################################
+AKASH_ENV=mainnet
+AKASH_NET=https://raw.githubusercontent.com/ovrclk/net/master/mainnet
+RPC_ENDPOINT=https://rpc.akashnet.net:443
+AKASH_GAS_PRICES=0.025uakt
+AKASH_GAS_ADJUSTMENT=1.5
+AKASH_KEYRING_BACKEND=os
+AKASH_FROM=default
+AKASH_FEES=20000uakt
+AKASH_DEPOSIT=500000uakt
+AKASH_MNEMONIC=
+AKASH_WALLET_ADDRESS=
+# Akash Pricing API
+AKASH_PRICING_API_URL=https://console-api.akash.network/v1/pricing
+# Default values # 1 CPU = 1000 1GB = 1000000000 1GB = 1000000000
+AKASH_DEFAULT_CPU=1000
+AKASH_DEFAULT_MEMORY=1000000000
+AKASH_DEFAULT_STORAGE=1000000000
+AKASH_SDL=example.sdl.yml
+# Close deployment
+# Close all deployments = closeAll
+# Close a single deployment = dseq and add the value in AKASH_CLOSE_DSEQ
+AKASH_CLOSE_DEP=closeAll
+AKASH_CLOSE_DSEQ=19729929
+# Provider Info we added one to check you will have to pass this into the action
+AKASH_PROVIDER_INFO=akash1ccktptfkvdc67msasmesuy5m7gpc76z75kukpz
+# Deployment Status
+# AKASH_DEP_STATUS = dseq or param_passed when you are building you wil pass the dseq dinamically to test you
+# you can pass the dseq using AKASH_DEP_DSEQ 19729929 is an example of a dseq we test while build.
+AKASH_DEP_STATUS=dseq
+AKASH_DEP_DSEQ=19729929
+# Gas Estimation Options: close, create, or update
+# qseq is required when operation is "close" 19729929 is an example of a dseq we test while build.
+AKASH_GAS_OPERATION=close
+AKASH_GAS_DSEQ=19729929
+# Manifest
+# Values: "auto" | "manual" | "validate_only" Default: "auto"
+AKASH_MANIFEST_MODE=auto
+# Default: Will use the SDL directory
+AKASH_MANIFEST_PATH=
+# Values: "strict" | "lenient" | "none" - Default: "strict"
+AKASH_MANIFEST_VALIDATION_LEVEL=strict
+
+# Quai Network Ecosystem
+QUAI_PRIVATE_KEY=
+QUAI_RPC_URL=https://rpc.quai.network
diff --git a/.github/workflows/greetings.yml b/.github/workflows/greetings.yml
index 750e5ce458b..4e39d95ffff 100644
--- a/.github/workflows/greetings.yml
+++ b/.github/workflows/greetings.yml
@@ -12,5 +12,5 @@ jobs:
- uses: actions/first-interaction@v1
with:
repo-token: ${{ secrets.GITHUB_TOKEN }}
- issue-message: "Hello @${{ github.actor }}! Welcome to the ai16z community. Thank you for opening your first issue; we appreciate your contribution. You are now a ai16z contributor!"
- pr-message: "Hi @${{ github.actor }}! Welcome to the ai16z community. Thanks for submitting your first pull request; your efforts are helping us accelerate towards AGI. We'll review it shortly. You are now a ai16z contributor!"
+ issue-message: "Hello @${{ github.actor }}! Welcome to the elizaOS community. Thank you for opening your first issue; we appreciate your contribution. You are now an elizaOS contributor!"
+ pr-message: "Hi @${{ github.actor }}! Welcome to the elizaOS community. Thanks for submitting your first pull request; your efforts are helping us accelerate towards AGI. We'll review it shortly. You are now an elizaOS contributor!"
diff --git a/.github/workflows/integrationTests.yaml b/.github/workflows/integrationTests.yaml
index 0dcef61c065..b21aac7b558 100644
--- a/.github/workflows/integrationTests.yaml
+++ b/.github/workflows/integrationTests.yaml
@@ -3,7 +3,7 @@ on:
push:
branches:
- "*"
- pull_request_target:
+ pull_request:
branches:
- "*"
@@ -33,12 +33,9 @@ jobs:
- name: Build packages
run: pnpm build
- - name: Check for API key
- run: |
- if [ -z "$OPENAI_API_KEY" ]; then
- echo "Error: OPENAI_API_KEY is not set."
- exit 1
- fi
-
- name: Run integration tests
- run: pnpm run integrationTests
+ env:
+ OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+ COINBASE_COMMERCE_KEY: ${{ secrets.COINBASE_COMMERCE_KEY }}
+ run: |
+ pnpm run integrationTests
diff --git a/.github/workflows/pnpm-lockfile-check.yml b/.github/workflows/pnpm-lockfile-check.yml
index a048b3703f2..3b303f8809e 100644
--- a/.github/workflows/pnpm-lockfile-check.yml
+++ b/.github/workflows/pnpm-lockfile-check.yml
@@ -2,7 +2,7 @@ name: Pnpm Lockfile Check
on:
pull_request:
- branches: ["*"]
+ branches: [main]
jobs:
check-lockfile:
@@ -38,4 +38,4 @@ jobs:
owner: context.repo.owner,
repo: context.repo.repo,
body: '❌ The pnpm-lockfile is out of date. Please run `pnpm install --no-frozen-lockfile` and commit the updated pnpm-lock.yaml file.'
- })
\ No newline at end of file
+ })
diff --git a/.gitignore b/.gitignore
index 86be41efaf2..7c6c92eb7b9 100644
--- a/.gitignore
+++ b/.gitignore
@@ -60,4 +60,4 @@ agent/content
eliza.manifest
eliza.manifest.sgx
-eliza.sig
+eliza.sig
\ No newline at end of file
diff --git a/.vscode/settings.json b/.vscode/settings.json
index 8bd129fed74..7d430c55039 100644
--- a/.vscode/settings.json
+++ b/.vscode/settings.json
@@ -43,5 +43,16 @@
},
"[shellscript]": {
"editor.defaultFormatter": "foxundermoon.shell-format"
+ },
+ "explorer.fileNesting.enabled": true,
+ "explorer.fileNesting.patterns": {
+ "*.ts": "${capture}.js",
+ "*.js": "${capture}.js.map, ${capture}.min.js, ${capture}.d.ts",
+ "*.jsx": "${capture}.js",
+ "*.tsx": "${capture}.ts",
+ "tsconfig.json": "tsconfig.*.json",
+ "package.json": "package-lock.json, yarn.lock, pnpm-lock.yaml, bun.lockb,pnpm-workspace.yaml",
+ "README.md": "*.md",
+ "Dockerfile": "docker-compose-docs.yaml,docker-compose.yaml,Dockerfile.docs"
}
}
\ No newline at end of file
diff --git a/README.md b/README.md
index 8a6db86caf6..e32978e6d95 100644
--- a/README.md
+++ b/README.md
@@ -12,7 +12,7 @@
## 🌍 README Translations
-[中文说明](./README_CN.md) | [日本語の説明](./README_JA.md) | [한국어 설명](./README_KOR.md) | [Français](./README_FR.md) | [Português](./README_PTBR.md) | [Türkçe](./README_TR.md) | [Русский](./README_RU.md) | [Español](./README_ES.md) | [Italiano](./README_IT.md) | [ไทย](./README_TH.md) | [Deutsch](./README_DE.md) | [Tiếng Việt](./README_VI.md) | [עִברִית](https://github.com/elizaos/Elisa/blob/main/README_HE.md) | [Tagalog](./README_TG.md) | [Polski](./README_PL.md) | [Arabic](./README_AR.md) | [Hungarian](./README_HU.md) | [Srpski](./README_RS.md) | [Română](./README_RO.md) | [Nederlands](./README_NL.md)
+[中文说明](./README_CN.md) | [日本語の説明](./README_JA.md) | [한국어 설명](./README_KOR.md) | [Persian](./README_FA.md) | [Français](./README_FR.md) | [Português](./README_PTBR.md) | [Türkçe](./README_TR.md) | [Русский](./README_RU.md) | [Español](./README_ES.md) | [Italiano](./README_IT.md) | [ไทย](./README_TH.md) | [Deutsch](./README_DE.md) | [Tiếng Việt](./README_VI.md) | [עִברִית](https://github.com/elizaos/Elisa/blob/main/README_HE.md) | [Tagalog](./README_TG.md) | [Polski](./README_PL.md) | [Arabic](./README_AR.md) | [Hungarian](./README_HU.md) | [Srpski](./README_RS.md) | [Română](./README_RO.md) | [Nederlands](./README_NL.md) | [Ελληνικά](./README_GR.md)
## 🚩 Overview
@@ -80,6 +80,8 @@ git clone https://github.com/elizaos/eliza.git
# Checkout the latest release
# This project iterates fast, so we recommend checking out the latest release
git checkout $(git describe --tags --abbrev=0)
+# If the above doesn't checkout the latest release, this should work:
+# git checkout $(git describe --tags `git rev-list --tags --max-count=1`)
```
### Start Eliza with Gitpod
diff --git a/README_FA.md b/README_FA.md
new file mode 100644
index 00000000000..cfc386f4cd3
--- /dev/null
+++ b/README_FA.md
@@ -0,0 +1,147 @@
+# الیزا 🤖
+
+
+
+
+
+📖 [مستندات](https://elizaos.github.io/eliza/) | 🎯 [نمونهها](https://github.com/thejoven/awesome-eliza)
+
+
+
+## 🚩 معرفی کلی
+
+
+
+
+
+
+## تاریخچه ستارهها
+
+[](https://star-history.com/#elizaos/eliza&Date)
+
diff --git a/README_GR.md b/README_GR.md
new file mode 100644
index 00000000000..6898498b812
--- /dev/null
+++ b/README_GR.md
@@ -0,0 +1,148 @@
+# Eliza 🤖
+
+
+
+
+
+📖 [Τεκμηρίωση](https://elizaos.github.io/eliza/) | 🎯 [Παραδείγματα](https://github.com/thejoven/awesome-eliza)
+
+
+
+## 🌍 Μεταφράσεις README
+[中文说明](./README_CN.md) | [日本語の説明](./README_JA.md) | [한국어 설명](./README_KOR.md) | [Persian](./README_FA.md) | [Français](./README_FR.md) | [Português](./README_PTBR.md) | [Türkçe](./README_TR.md) | [Русский](./README_RU.md) | [Español](./README_ES.md) | [Italiano](./README_IT.md) | [ไทย](./README_TH.md) | [Deutsch](./README_DE.md) | [Tiếng Việt](./README_VI.md) | [עִברִית](https://github.com/elizaos/Elisa/blob/main/README_HE.md) | [Tagalog](./README_TG.md) | [Polski](./README_PL.md) | [Arabic](./README_AR.md) | [Hungarian](./README_HU.md) | [Srpski](./README_RS.md) | [Română](./README_RO.md) | [Nederlands](./README_NL.md) | [Ελληνικά](./README_GR.md)
+
+## 🚩 Επισκόπηση
+
+
+
+
+_Como visto dando funcionamento em [@DegenSpartanAI](https://x.com/degenspartanai) e [@MarcAIndreessen](https://x.com/pmairca)_
+
+- Framework Multi-agente de simulação
+- Adicione quantos personagens únicos quiser com o [characterfile](https://github.com/lalalune/characterfile/)
+- Conectores completos para Discord e Twitter, com suporte para canais de voz no Discord
+- Memória RAG completa para conversas e documentos
+- Pode ler links e PDFs, transcrever áudios e vídeos, resumir conversas e muito mais
+- Altamente extensível - crie suas próprias ações e clientes para ampliar as capacidades do Eliza
+- Suporte para modelos de código aberto e locais (configuração padrão com Nous Hermes Llama 3.1B)
+- Suporte ao OpenAI para inferência em nuvem em dispositivos com configurações leves
+- Modo "Perguntar ao Claude" para chamadas a Claude em consultas mais complexas
+- 100% Typescript
+
+# Iniciando
+
+**Pré-requisitos (OBRIGATÓRIO):**
+
+- [Node.js 23+](https://docs.npmjs.com/downloading-and-installing-node-js-and-npm)
+- [pnpm](https://pnpm.io/installation)
+
+### Edite o arquivo .env
+
+- Copie .env.example para .env e preencha com valores apropriados
+- Edite as variáveis de ambiente do TWITTER para adicionar o nome de usuário e a senha do seu bot
+
+### Edite o arquivo de personagem (character file)
+
+- Verifique o arquivo `src/core/defaultCharacter.ts` - você pode modificá-lo
+- Você também pode carregar personagens com o comando `pnpm start --characters="path/to/your/character.json"` e executar vários bots ao mesmo tempo.
+
+Após configurar o arquivo .env e o arquivo de personagem (character file), você pode iniciar o bot com o seguinte comando:
+
+```
+pnpm i
+pnpm start
+```
+
+# Personalizando Eliza
+
+### Adicionando ações personalizadas
+
+Para evitar conflitos no diretório principal, recomendamos adicionar ações personalizadas a um diretório chamado `custom_actions` e, em seguida, incluí-las no arquivo `elizaConfig.yaml`. Consulte o arquivo `elizaConfig.example.yaml` para um exemplo.
+
+## Rodando com diferentes modelos
+
+### Rode com Llama
+
+Você pode executar modelos Llama 70B ou 405B configurando a variável de ambiente `XAI_MODEL` para `meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo` ou `meta-llama/Meta-Llama-3.1-405B-Instruct`
+
+### Rode com Grok
+
+Você pode executar modelos Grok configurando a variável de ambiente `XAI_MODEL` para `grok-beta`.
+
+### Rode com OpenAI
+
+Você pode executar modelos OpenAI configurando a variável de ambiente para `gpt-4-mini` or `gpt-4o`
+
+## Requisitos Adicionais
+
+Você pode precisar instalar o Sharp. Se aparecer um erro ao iniciar, tente instalá-lo com o seguinte comando:
+
+```
+pnpm install --include=optional sharp
+```
+
+# Configuração do Ambiente
+
+Você precisará adicionar variáveis de ambiente ao seu arquivo .env para conectar a diversas plataformas:
+
+```
+# Variaveis de ambiente obrigatorias
+DISCORD_APPLICATION_ID=
+DISCORD_API_TOKEN= # Bot token
+OPENAI_API_KEY=sk-* # OpenAI API key, começando com sk-
+ELEVENLABS_XI_API_KEY= # API key da elevenlabs
+
+# Configuracoes ELEVENLABS
+ELEVENLABS_MODEL_ID=eleven_multilingual_v2
+ELEVENLABS_VOICE_ID=21m00Tcm4TlvDq8ikWAM
+ELEVENLABS_VOICE_STABILITY=0.5
+ELEVENLABS_VOICE_SIMILARITY_BOOST=0.9
+ELEVENLABS_VOICE_STYLE=0.66
+ELEVENLABS_VOICE_USE_SPEAKER_BOOST=false
+ELEVENLABS_OPTIMIZE_STREAMING_LATENCY=4
+ELEVENLABS_OUTPUT_FORMAT=pcm_16000
+
+TWITTER_DRY_RUN=false
+TWITTER_USERNAME= # Usuário da conta
+TWITTER_PASSWORD= # Senha da conta
+TWITTER_EMAIL= # Email da conta
+
+X_SERVER_URL=
+XAI_API_KEY=
+XAI_MODEL=
+
+
+# Para perguntas ao Claude
+ANTHROPIC_API_KEY=
+
+WALLET_SECRET_KEY=EXAMPLE_WALLET_SECRET_KEY
+WALLET_PUBLIC_KEY=EXAMPLE_WALLET_PUBLIC_KEY
+
+BIRDEYE_API_KEY=
+
+SOL_ADDRESS=So11111111111111111111111111111111111111112
+SLIPPAGE=1
+RPC_URL=https://api.mainnet-beta.solana.com
+HELIUS_API_KEY=
+
+
+## Telegram
+TELEGRAM_BOT_TOKEN=
+
+TOGETHER_API_KEY=
+```
+
+# Configuração de Inferência Local
+
+### Configuração CUDA
+
+Se você tiver uma GPU NVIDIA, pode instalar o CUDA para acelerar significativamente a inferência local.
+
+```
+pnpm install
+npx --no node-llama-cpp source download --gpu cuda
+```
+
+Certifique-se de que você instalou o CUDA Toolkit, incluindo o cuDNN e cuBLAS.
+
+### Rodando localmente
+
+Add XAI_MODEL e defina-o para uma das opções mencionadas em [Run with
+Llama](#run-with-llama) - você pode deixar X_SERVER_URL e XAI_API_KEY em branco,
+pois o modelo será baixado do Hugging Face e consultado localmente.
+
+# Clientes
+
+## Discord Bot
+
+Para ajuda com a configuração do seu bot no Discord, consulte aqui: https://discordjs.guide/preparations/setting-up-a-bot-application.html
+
+# Desenvolvimento
+
+## Testando
+
+Para executar a suíte de testes:
+
+```bash
+pnpm test # Executar os testes uma vez
+pnpm test:watch # Executar os testes no modo de observação/monitoramento (watch mode)
+```
+
+Para testes específicos de banco de dados:
+
+```bash
+pnpm test:sqlite # Rode testes com SQLite
+pnpm test:sqljs # Rode testes com SQL.js
+```
+
+Os testes são escritos usando o Jest e podem ser encontrados nos arquivos. O ambiente de teste está configurado para:
+
+- Carregar variáveis de ambiente do arquivo `.env.test`
+- Usar um tempo limite de 2 minutos para testes de longa duração
+- Suportar módulos ESM
+- Executar os testes em sequência (--runInBand)
+
+Para criar novos testes, adicione um arquivo `.test.ts` ao lado do código que você está testando.
+
+## Atualizações da Documentação
+
+Por favor, verifique se a documentação fornecida está correta. Para fazer isso, execute o serviço de documentação (docs) abaixo.
+
+```console
+docker compose -f docker-compose-docs.yaml up --build
+```
+
+O servidor do Docusaurus será iniciado e você poderá verificar a documentação localmente em https://localhost:3000/eliza.
diff --git a/docs/api/classes/DatabaseAdapter.md b/docs/api/classes/DatabaseAdapter.md
index 65f8186e762..46cf221d279 100644
--- a/docs/api/classes/DatabaseAdapter.md
+++ b/docs/api/classes/DatabaseAdapter.md
@@ -194,7 +194,7 @@ Retrieves memories based on the specified parameters.
• **params**
-An object containing parameters for the memory retrieval.
+An object containing parameters for memory retrieval.
• **params.agentId**: \`$\{string\}-$\{string\}-$\{string\}-$\{string\}-$\{string\}\`
@@ -300,7 +300,7 @@ An object containing parameters for the embedding retrieval.
`Promise`\<`object`[]\>
-A Promise that resolves to an array of objects containing embeddings and levenshtein scores.
+A Promise that resolves to an array of objects containing embeddings and Levenshtein scores.
#### Implementation of
diff --git a/docs/api/functions/composeContext.md b/docs/api/functions/composeContext.md
index 86ed7bb61ab..055bdb28c07 100644
--- a/docs/api/functions/composeContext.md
+++ b/docs/api/functions/composeContext.md
@@ -20,7 +20,7 @@ The parameters for composing the context.
• **params.state**: [`State`](../interfaces/State.md)
-The state object containing values to replace the placeholders in the template.
+The state object contains values to replace the placeholders in the template.
• **params.template**: `string` | `Function`
diff --git a/docs/api/functions/configureSettings.md b/docs/api/functions/configureSettings.md
index 97e013f435b..876d1dc54cf 100644
--- a/docs/api/functions/configureSettings.md
+++ b/docs/api/functions/configureSettings.md
@@ -10,7 +10,7 @@ Configures environment settings for browser usage
• **settings**: `Settings`
-Object containing environment variables
+The object containing environment variables
## Returns
diff --git a/docs/api/functions/splitChunks.md b/docs/api/functions/splitChunks.md
index b066d0777fc..4e9f5745dd1 100644
--- a/docs/api/functions/splitChunks.md
+++ b/docs/api/functions/splitChunks.md
@@ -24,7 +24,7 @@ Number of characters to overlap between chunks (default: 100)
`Promise`\<`string`[]\>
-Promise resolving to array of text chunks with bleed sections
+Promise resolving to an array of text chunks with bleed sections
## Defined in
diff --git a/docs/api/interfaces/ActionExample.md b/docs/api/interfaces/ActionExample.md
index f084d8ab5d9..40f694afa13 100644
--- a/docs/api/interfaces/ActionExample.md
+++ b/docs/api/interfaces/ActionExample.md
@@ -2,7 +2,7 @@
# Interface: ActionExample
-Example content with associated user for demonstration purposes
+Example content with the associated user for demonstration purposes
## Properties
diff --git a/docs/api/interfaces/ConversationExample.md b/docs/api/interfaces/ConversationExample.md
index c7d21073075..30435854682 100644
--- a/docs/api/interfaces/ConversationExample.md
+++ b/docs/api/interfaces/ConversationExample.md
@@ -10,7 +10,7 @@ Example conversation content with user ID
> **userId**: \`$\{string\}-$\{string\}-$\{string\}-$\{string\}-$\{string\}\`
-UUID of user in conversation
+UUID of the user in conversation
#### Defined in
diff --git a/docs/docs/advanced/fine-tuning.md b/docs/docs/advanced/fine-tuning.md
index 7822e9010ff..2a3220ddac6 100644
--- a/docs/docs/advanced/fine-tuning.md
+++ b/docs/docs/advanced/fine-tuning.md
@@ -22,6 +22,7 @@ enum ModelProviderName {
LLAMACLOUD,
LLAMALOCAL,
GOOGLE,
+ MISTRAL,
REDPILL,
OPENROUTER,
HEURIST,
diff --git a/docs/docs/advanced/verified-inference.md b/docs/docs/advanced/verified-inference.md
new file mode 100644
index 00000000000..2b8692bebbc
--- /dev/null
+++ b/docs/docs/advanced/verified-inference.md
@@ -0,0 +1,83 @@
+---
+sidebar_position: 18
+---
+
+# 🪪 Verified Inference
+
+## Overview
+
+With verified inference, you can turn your Eliza agent fully verifiable on-chain on Solana with an OpenAI compatible TEE API. This proves that your agent’s thoughts and outputs are free from human control thus increasing the trust of the agent.
+
+Compared to [fully deploying the agent in a TEE](https://elizaos.github.io/eliza/docs/advanced/eliza-in-tee/), this is a more light-weight solution which only verifies the inference calls and only needs a single line of code change.
+
+The API supports all OpenAI models out of the box, including your fine-tuned models. The following guide will walk you through how to use verified inference API with Eliza.
+
+## Background
+
+The API is built on top of [Sentience Stack](https://github.com/galadriel-ai/Sentience), which cryptographically verifies agent's LLM inferences inside TEEs, posts those proofs on-chain on Solana, and makes the verified inference logs available to read and display to users.
+
+Here’s how it works:
+
+
+1. The agent sends a request containing a message with the desired LLM model to the TEE.
+2. The TEE securely processes the request by calling the LLM API.
+3. The TEE sends back the `{Message, Proof}` to the agent.
+4. The TEE submits the attestation with `{Message, Proof}` to Solana.
+5. The Proof of Sentience SDK is used to read the attestation from Solana and verify it with `{Message, Proof}`. The proof log can be added to the agent website/app.
+
+To verify the code running inside the TEE, use instructions [from here](https://github.com/galadriel-ai/sentience/tree/main/verified-inference/verify).
+
+## Tutorial
+
+1. **Create a free API key on [Galadriel dashboard](https://dashboard.galadriel.com/login)**
+2. **Configure the environment variables**
+ ```bash
+ GALADRIEL_API_KEY=gal-* # Get from https://dashboard.galadriel.com/
+ # Use any model supported by OpenAI
+ SMALL_GALADRIEL_MODEL= # Default: gpt-4o-mini
+ MEDIUM_GALADRIEL_MODEL= # Default: gpt-4o
+ LARGE_GALADRIEL_MODEL= # Default: gpt-4o
+ # If you wish to use a fine-tuned model you will need to provide your own OpenAI API key
+ GALADRIEL_FINE_TUNE_API_KEY= # starting with sk-
+ ```
+3. **Configure your character to use `galadriel`**
+
+ In your character file set the `modelProvider` as `galadriel`.
+ ```
+ "modelProvider": "galadriel"
+ ```
+4. **Run your agent.**
+
+ Reminder how to run an agent is [here](https://elizaos.github.io/eliza/docs/quickstart/#create-your-first-agent).
+ ```bash
+ pnpm start --character="characters/.json"
+ pnpm start:client
+ ```
+5. **Get the history of all of your verified inference calls**
+ ```javascript
+ const url = 'https://api.galadriel.com/v1/verified/chat/completions?limit=100&filter=mine';
+ const headers = {
+ 'accept': 'application/json',
+ 'Authorization': 'Bearer '// Replace with your Galadriel API key
+ };
+
+ const response = await fetch(url, { method: 'GET', headers });
+ const data = await response.json();
+ console.log(data);
+ ```
+
+ Use this to build a verified logs terminal to your agent front end, for example:
+
+
+6. **Check your inferences in the explorer.**
+
+ You can also see your inferences with proofs in the [Galadriel explorer](https://explorer.galadriel.com/). For specific inference responses use `https://explorer.galadriel.com/details/`
+
+ The `hash` param is returned with every inference request.
+ 
+
+7. **Check proofs posted on Solana.**
+
+ You can also see your inferences with proofs on Solana. For specific inference responses: `https://explorer.solana.com/tx/<>tx_hash?cluster=devnet`
+
+ The `tx_hash` param is returned with every inference request.
diff --git a/docs/docs/core/actions.md b/docs/docs/core/actions.md
index 529ff18ea3d..0f710e0c90d 100644
--- a/docs/docs/core/actions.md
+++ b/docs/docs/core/actions.md
@@ -179,7 +179,7 @@ const continueAction: Action = {
name: "CONTINUE",
similes: ["ELABORATE", "KEEP_TALKING"],
description:
- "Used when the message requires a follow-up. Don't use when conversation is finished.",
+ "Used when the message requires a follow-up. Don't use when the conversation is finished.",
validate: async (runtime, message) => {
// Validation logic
return true;
diff --git a/docs/docs/core/characterfile.md b/docs/docs/core/characterfile.md
index f9ca0648fc3..8b5a278f459 100644
--- a/docs/docs/core/characterfile.md
+++ b/docs/docs/core/characterfile.md
@@ -140,7 +140,7 @@ Array used for Retrieval Augmented Generation (RAG), containing facts or referen
#### `messageExamples`
-Sample conversations for establishing interaction patterns, helps establish the character's conversational style.
+Sample conversations for establishing interaction patterns, help establish the character's conversational style.
```json
"messageExamples": [
@@ -191,7 +191,7 @@ The `style` object defines behavior patterns across contexts:
### Adjectives Array
- Words that describe the character's traits and personality
-- Used for generating responses with consistent tone
+- Used for generating responses with a consistent tone
- Can be used in "Mad Libs" style content generation
### Settings Configuration
diff --git a/docs/docs/core/evaluators.md b/docs/docs/core/evaluators.md
index 7811e9d9531..43cfb96caa7 100644
--- a/docs/docs/core/evaluators.md
+++ b/docs/docs/core/evaluators.md
@@ -119,7 +119,7 @@ interface Objective {
### Handler Implementation
- Use runtime services appropriately
-- Store results in correct memory manager
+- Store results in the correct memory manager
- Handle errors gracefully
- Maintain state consistency
diff --git a/docs/docs/faq.md b/docs/docs/faq.md
index 5e0baea5660..0f26446fe2b 100644
--- a/docs/docs/faq.md
+++ b/docs/docs/faq.md
@@ -68,4 +68,4 @@ There are several ways to contribute to the Eliza project:
- **Participate in community discussions**: Share your memecoin insights, propose new ideas, and engage with other community members.
- **Contribute to the development of the Eliza platform**: https://github.com/orgs/elizaos/projects/1/views/3
-- **Help build the Eliza ecosystem**: Create applicatoins / tools, resources, and memes. Give feedback, and spread the word
+- **Help build the Eliza ecosystem**: Create applications / tools, resources, and memes. Give feedback, and spread the word
diff --git a/docs/docs/guides/configuration.md b/docs/docs/guides/configuration.md
index b260a4d8079..a87d61046ca 100644
--- a/docs/docs/guides/configuration.md
+++ b/docs/docs/guides/configuration.md
@@ -71,6 +71,59 @@ HEURIST_API_KEY=
# Livepeer Settings
LIVEPEER_GATEWAY_URL=
```
+
+### Cloudflare AI Gateway Integration
+
+Eliza supports routing API calls through [Cloudflare AI Gateway](https://developers.cloudflare.com/ai-gateway/), which provides several benefits:
+
+- Detailed analytics and monitoring of message traffic and response times
+- Cost optimization through request caching and usage tracking across providers
+- Improved latency through Cloudflare's global network
+- Comprehensive visibility into message content and token usage
+- Cost analysis and comparison between different AI providers
+- Usage patterns and trends visualization
+- Request/response logging for debugging and optimization
+
+To enable Cloudflare AI Gateway:
+
+```bash
+# Cloudflare AI Gateway Settings
+CLOUDFLARE_GW_ENABLED=true
+CLOUDFLARE_AI_ACCOUNT_ID=your-account-id
+CLOUDFLARE_AI_GATEWAY_ID=your-gateway-id
+```
+
+Supported providers through Cloudflare AI Gateway:
+- OpenAI
+- Anthropic
+- Groq
+
+When enabled, Eliza will automatically route requests through your Cloudflare AI Gateway endpoint. The gateway URL is constructed in the format:
+```
+https://gateway.ai.cloudflare.com/v1/${accountId}/${gatewayId}/${provider}
+```
+
+If the gateway configuration is incomplete or disabled, Eliza will fall back to direct API calls.
+
+```bash
+# Cloudflare AI Gateway Settings
+CLOUDFLARE_GW_ENABLED=true
+CLOUDFLARE_AI_ACCOUNT_ID=your-account-id
+CLOUDFLARE_AI_GATEWAY_ID=your-gateway-id
+```
+
+Supported providers through Cloudflare AI Gateway:
+- OpenAI
+- Anthropic
+- Groq
+
+When enabled, Eliza will automatically route requests through your Cloudflare AI Gateway endpoint. The gateway URL is constructed in the format:
+```
+https://gateway.ai.cloudflare.com/v1/${accountId}/${gatewayId}/${provider}
+```
+
+If the gateway configuration is incomplete or disabled, Eliza will fall back to direct API calls.
+
### Image Generation
Configure image generation in your character file:
diff --git a/docs/docs/packages/clients.md b/docs/docs/packages/clients.md
index ad4d173d9e7..24fa4bfb289 100644
--- a/docs/docs/packages/clients.md
+++ b/docs/docs/packages/clients.md
@@ -35,11 +35,11 @@ graph TD
## Available Clients
-- **Discord** (`@eliza/client-discord`) - Full Discord bot integration
-- **Twitter** (`@eliza/client-twitter`) - Twitter bot and interaction handling
-- **Telegram** (`@eliza/client-telegram`) - Telegram bot integration
-- **Direct** (`@eliza/client-direct`) - Direct API interface for custom integrations
-- **Auto** (`@eliza/client-auto`) - Automated trading and interaction client
+- **Discord** (`@elizaos/client-discord`) - Full Discord bot integration
+- **Twitter** (`@elizaos/client-twitter`) - Twitter bot and interaction handling
+- **Telegram** (`@elizaos/client-telegram`) - Telegram bot integration
+- **Direct** (`@elizaos/client-direct`) - Direct API interface for custom integrations
+- **Auto** (`@elizaos/client-auto`) - Automated trading and interaction client
---
@@ -47,19 +47,19 @@ graph TD
```bash
# Discord
-pnpm add @eliza/client-discord
+pnpm add @elizaos/client-discord
# Twitter
-pnpm add @eliza/client-twitter
+pnpm add @elizaos/client-twitter
# Telegram
-pnpm add @eliza/client-telegram
+pnpm add @elizaos/client-telegram
# Direct API
-pnpm add @eliza/client-direct
+pnpm add @elizaos/client-direct
# Auto Client
-pnpm add @eliza/client-auto
+pnpm add @elizaos/client-auto
```
---
@@ -71,7 +71,7 @@ The Discord client provides full integration with Discord's features including v
### Basic Setup
```typescript
-import { DiscordClientInterface } from "@eliza/client-discord";
+import { DiscordClientInterface } from "@elizaos/client-discord";
// Initialize client
const client = await DiscordClientInterface.start(runtime);
@@ -133,7 +133,7 @@ The Twitter client enables posting, searching, and interacting with Twitter user
### Basic Setup
```typescript
-import { TwitterClientInterface } from "@eliza/client-twitter";
+import { TwitterClientInterface } from "@elizaos/client-twitter";
// Initialize client
const client = await TwitterClientInterface.start(runtime);
@@ -192,7 +192,7 @@ The Telegram client provides messaging and bot functionality for Telegram.
### Basic Setup
```typescript
-import { TelegramClientInterface } from "@eliza/client-telegram";
+import { TelegramClientInterface } from "@elizaos/client-telegram";
// Initialize client
const client = await TelegramClientInterface.start(runtime);
@@ -225,7 +225,7 @@ The Direct client provides a REST API interface for custom integrations.
### Basic Setup
```typescript
-import { DirectClientInterface } from "@eliza/client-direct";
+import { DirectClientInterface } from "@elizaos/client-direct";
// Initialize client
const client = await DirectClientInterface.start(runtime);
@@ -258,7 +258,7 @@ The Auto client enables automated interactions and trading.
### Basic Setup
```typescript
-import { AutoClientInterface } from "@eliza/client-auto";
+import { AutoClientInterface } from "@elizaos/client-auto";
// Initialize client
const client = await AutoClientInterface.start(runtime);
diff --git a/docs/package.json b/docs/package.json
index 4b5d443ce69..07a265458d5 100644
--- a/docs/package.json
+++ b/docs/package.json
@@ -1,6 +1,6 @@
{
"name": "eliza-docs",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"private": true,
"packageManager": "pnpm@9.4.0",
"scripts": {
diff --git a/docs/sidebars.js b/docs/sidebars.js
index e2f74c6e87b..93cc9719f9a 100644
--- a/docs/sidebars.js
+++ b/docs/sidebars.js
@@ -117,6 +117,11 @@ const sidebars = {
id: "advanced/eliza-in-tee",
label: "Eliza in TEE",
},
+ {
+ type: "doc",
+ id: "advanced/verified-inference",
+ label: "Verified Inference",
+ },
],
},
{
diff --git a/lerna.json b/lerna.json
index b03a6a059cf..c772c6adb2a 100644
--- a/lerna.json
+++ b/lerna.json
@@ -1,5 +1,5 @@
{
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"packages": [
"packages/*",
"docs",
diff --git a/package.json b/package.json
index f550c3cfe54..af32159bf96 100644
--- a/package.json
+++ b/package.json
@@ -27,6 +27,7 @@
"devDependencies": {
"@commitlint/cli": "18.6.1",
"@commitlint/config-conventional": "18.6.3",
+ "@types/jest": "^29.5.11",
"@typescript-eslint/eslint-plugin": "8.16.0",
"@typescript-eslint/parser": "8.16.0",
"@vitest/eslint-plugin": "1.1.13",
@@ -35,18 +36,17 @@
"eslint": "9.16.0",
"eslint-config-prettier": "9.1.0",
"husky": "9.1.7",
+ "jest": "^29.7.0",
"lerna": "8.1.5",
"only-allow": "1.2.1",
"prettier": "3.4.1",
+ "ts-jest": "^29.1.1",
"turbo": "2.3.3",
"typedoc": "0.26.11",
"typescript": "5.6.3",
- "vite": "5.4.11",
- "vitest": "2.1.5",
"viem": "2.21.58",
- "ts-jest": "^29.1.1",
- "@types/jest": "^29.5.11",
- "jest": "^29.7.0"
+ "vite": "5.4.11",
+ "vitest": "2.1.5"
},
"pnpm": {
"overrides": {
@@ -64,6 +64,7 @@
"@vitest/eslint-plugin": "1.0.1",
"amqplib": "0.10.5",
"csv-parse": "5.6.0",
+ "langdetect": "^0.2.1",
"ollama-ai-provider": "0.16.1",
"optional": "0.1.4",
"pnpm": "9.14.4",
@@ -74,4 +75,4 @@
"workspaces": [
"packages/*"
]
-}
+}
\ No newline at end of file
diff --git a/packages/adapter-pglite/package.json b/packages/adapter-pglite/package.json
index 7f7167333e1..6bd9cff0112 100644
--- a/packages/adapter-pglite/package.json
+++ b/packages/adapter-pglite/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/adapter-pglite",
- "version": "0.1.7-alpha.2",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
diff --git a/packages/adapter-postgres/migrations/20240318103238_remote_schema.sql b/packages/adapter-postgres/migrations/20240318103238_remote_schema.sql
index 2867a12aea6..30b0854ce3d 100644
--- a/packages/adapter-postgres/migrations/20240318103238_remote_schema.sql
+++ b/packages/adapter-postgres/migrations/20240318103238_remote_schema.sql
@@ -21,7 +21,7 @@ BEGIN
FROM pg_extension
WHERE extname = 'vector'
) THEN
- CREATE EXTENSION vector
+ CREATE EXTENSION vector IF NOT EXISTS
SCHEMA extensions;
END IF;
END $$;
@@ -33,7 +33,7 @@ BEGIN
FROM pg_extension
WHERE extname = 'fuzzystrmatch'
) THEN
- CREATE EXTENSION fuzzystrmatch
+ CREATE EXTENSION fuzzystrmatch IF NOT EXISTS
SCHEMA extensions;
END IF;
END $$;
diff --git a/packages/adapter-postgres/package.json b/packages/adapter-postgres/package.json
index 4f58661a795..37240661bfd 100644
--- a/packages/adapter-postgres/package.json
+++ b/packages/adapter-postgres/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/adapter-postgres",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
diff --git a/packages/adapter-postgres/src/__tests__/vector-extension.test.ts b/packages/adapter-postgres/src/__tests__/vector-extension.test.ts
index 7ced5873718..a22c51c79f6 100644
--- a/packages/adapter-postgres/src/__tests__/vector-extension.test.ts
+++ b/packages/adapter-postgres/src/__tests__/vector-extension.test.ts
@@ -3,7 +3,7 @@ import pg from 'pg';
import fs from 'fs';
import path from 'path';
import { describe, test, expect, beforeEach, afterEach, vi, beforeAll } from 'vitest';
-import { DatabaseAdapter, elizaLogger, type Memory, type Content, EmbeddingProvider } from '@elizaos/core';
+import { elizaLogger, type Memory, type Content } from '@elizaos/core';
// Increase test timeout
vi.setConfig({ testTimeout: 15000 });
@@ -41,7 +41,7 @@ vi.mock('@elizaos/core', () => ({
const parseVectorString = (vectorStr: string): number[] => {
if (!vectorStr) return [];
// Remove brackets and split by comma
- return vectorStr.replace(/[\[\]]/g, '').split(',').map(Number);
+ return vectorStr.replace(/[[\]]/g, '').split(',').map(Number);
};
describe('PostgresDatabaseAdapter - Vector Extension Validation', () => {
@@ -111,7 +111,7 @@ describe('PostgresDatabaseAdapter - Vector Extension Validation', () => {
user: 'postgres',
password: 'postgres'
});
-
+
const setupClient = await setupPool.connect();
try {
await cleanDatabase(setupClient);
@@ -133,13 +133,13 @@ describe('PostgresDatabaseAdapter - Vector Extension Validation', () => {
user: 'postgres',
password: 'postgres'
});
-
+
testClient = await testPool.connect();
elizaLogger.debug('Database connection established');
-
+
await cleanDatabase(testClient);
elizaLogger.debug('Database cleaned');
-
+
adapter = new PostgresDatabaseAdapter({
host: 'localhost',
port: 5433,
@@ -254,7 +254,7 @@ describe('PostgresDatabaseAdapter - Vector Extension Validation', () => {
elizaLogger.debug('Attempting initialization with error...');
await expect(adapter.init()).rejects.toThrow('Schema read error');
elizaLogger.success('Error thrown as expected');
-
+
// Verify no tables were created
elizaLogger.debug('Verifying rollback...');
const { rows } = await testClient.query(`
@@ -277,19 +277,19 @@ describe('PostgresDatabaseAdapter - Vector Extension Validation', () => {
describe('Memory Operations with Vector', () => {
const TEST_UUID = 'aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee';
const TEST_TABLE = 'test_memories';
-
+
beforeEach(async () => {
elizaLogger.info('Setting up memory operations test...');
try {
// Ensure clean state and proper initialization
await adapter.init();
-
+
// Verify vector extension and search path
await testClient.query(`
SET search_path TO public, extensions;
SELECT set_config('app.use_openai_embedding', 'true', false);
`);
-
+
// Create necessary account and room first
await testClient.query('BEGIN');
try {
@@ -298,19 +298,19 @@ describe('PostgresDatabaseAdapter - Vector Extension Validation', () => {
VALUES ($1, 'test@test.com')
ON CONFLICT (id) DO NOTHING
`, [TEST_UUID]);
-
+
await testClient.query(`
INSERT INTO rooms (id)
VALUES ($1)
ON CONFLICT (id) DO NOTHING
`, [TEST_UUID]);
-
+
await testClient.query('COMMIT');
} catch (error) {
await testClient.query('ROLLBACK');
throw error;
}
-
+
} catch (error) {
elizaLogger.error('Memory operations setup failed:', {
error: error instanceof Error ? error.message : String(error)
@@ -324,7 +324,7 @@ describe('PostgresDatabaseAdapter - Vector Extension Validation', () => {
const content: Content = {
text: 'test content'
};
-
+
const memory: Memory = {
id: TEST_UUID,
content,
@@ -383,7 +383,7 @@ describe('PostgresDatabaseAdapter - Vector Extension Validation', () => {
await testClient.query('ROLLBACK');
throw error;
}
-
+
// Act
const results = await adapter.searchMemoriesByEmbedding(embedding, {
tableName: TEST_TABLE,
@@ -405,7 +405,7 @@ describe('PostgresDatabaseAdapter - Vector Extension Validation', () => {
const content: Content = {
text: 'test content'
};
-
+
const memory: Memory = {
id: TEST_UUID,
content,
@@ -430,4 +430,4 @@ describe('PostgresDatabaseAdapter - Vector Extension Validation', () => {
}
}, { timeout: 30000 }); // Increased timeout for retry attempts
});
-});
\ No newline at end of file
+});
\ No newline at end of file
diff --git a/packages/adapter-postgres/src/index.ts b/packages/adapter-postgres/src/index.ts
index efbca97a91d..5f257bb7190 100644
--- a/packages/adapter-postgres/src/index.ts
+++ b/packages/adapter-postgres/src/index.ts
@@ -4,31 +4,31 @@ import { v4 } from "uuid";
import pg from "pg";
type Pool = pg.Pool;
-import {
- QueryConfig,
- QueryConfigValues,
- QueryResult,
- QueryResultRow,
-} from "pg";
import {
Account,
Actor,
+ DatabaseAdapter,
+ EmbeddingProvider,
GoalStatus,
+ Participant,
+ RAGKnowledgeItem,
+ elizaLogger,
+ getEmbeddingConfig,
type Goal,
+ type IDatabaseCacheAdapter,
type Memory,
type Relationship,
type UUID,
- type IDatabaseCacheAdapter,
- Participant,
- elizaLogger,
- getEmbeddingConfig,
- DatabaseAdapter,
- EmbeddingProvider,
- RAGKnowledgeItem
} from "@elizaos/core";
import fs from "fs";
-import { fileURLToPath } from "url";
import path from "path";
+import {
+ QueryConfig,
+ QueryConfigValues,
+ QueryResult,
+ QueryResultRow,
+} from "pg";
+import { fileURLToPath } from "url";
const __filename = fileURLToPath(import.meta.url); // get the resolved path to the file
const __dirname = path.dirname(__filename); // get the name of the directory
@@ -199,7 +199,7 @@ export class PostgresDatabaseAdapter
return true;
} catch (error) {
elizaLogger.error("Failed to validate vector extension:", {
- error: error instanceof Error ? error.message : String(error)
+ error: error instanceof Error ? error.message : String(error),
});
return false;
}
@@ -239,8 +239,10 @@ export class PostgresDatabaseAdapter
);
`);
- if (!rows[0].exists || !await this.validateVectorSetup()) {
- elizaLogger.info("Applying database schema - tables or vector extension missing");
+ if (!rows[0].exists || !(await this.validateVectorSetup())) {
+ elizaLogger.info(
+ "Applying database schema - tables or vector extension missing"
+ );
const schema = fs.readFileSync(
path.resolve(__dirname, "../schema.sql"),
"utf8"
@@ -329,6 +331,7 @@ export class PostgresDatabaseAdapter
roomIds: UUID[];
agentId?: UUID;
tableName: string;
+ limit?: number;
}): Promise {
return this.withDatabase(async () => {
if (params.roomIds.length === 0) return [];
@@ -344,6 +347,13 @@ export class PostgresDatabaseAdapter
queryParams = [...queryParams, params.agentId];
}
+ // Add sorting, and conditionally add LIMIT if provided
+ query += ` ORDER BY "createdAt" DESC`;
+ if (params.limit) {
+ query += ` LIMIT $${queryParams.length + 1}`;
+ queryParams.push(params.limit.toString());
+ }
+
const { rows } = await this.pool.query(query, queryParams);
return rows.map((row) => ({
...row,
@@ -1515,12 +1525,17 @@ export class PostgresDatabaseAdapter
const { rows } = await this.pool.query(sql, queryParams);
- return rows.map(row => ({
+ return rows.map((row) => ({
id: row.id,
agentId: row.agentId,
- content: typeof row.content === 'string' ? JSON.parse(row.content) : row.content,
- embedding: row.embedding ? new Float32Array(row.embedding) : undefined,
- createdAt: row.createdAt.getTime()
+ content:
+ typeof row.content === "string"
+ ? JSON.parse(row.content)
+ : row.content,
+ embedding: row.embedding
+ ? new Float32Array(row.embedding)
+ : undefined,
+ createdAt: row.createdAt.getTime(),
}));
}, "getKnowledge");
}
@@ -1536,7 +1551,7 @@ export class PostgresDatabaseAdapter
const cacheKey = `embedding_${params.agentId}_${params.searchText}`;
const cachedResult = await this.getCache({
key: cacheKey,
- agentId: params.agentId
+ agentId: params.agentId,
});
if (cachedResult) {
@@ -1586,24 +1601,29 @@ export class PostgresDatabaseAdapter
const { rows } = await this.pool.query(sql, [
vectorStr,
params.agentId,
- `%${params.searchText || ''}%`,
+ `%${params.searchText || ""}%`,
params.match_threshold,
- params.match_count
+ params.match_count,
]);
- const results = rows.map(row => ({
+ const results = rows.map((row) => ({
id: row.id,
agentId: row.agentId,
- content: typeof row.content === 'string' ? JSON.parse(row.content) : row.content,
- embedding: row.embedding ? new Float32Array(row.embedding) : undefined,
+ content:
+ typeof row.content === "string"
+ ? JSON.parse(row.content)
+ : row.content,
+ embedding: row.embedding
+ ? new Float32Array(row.embedding)
+ : undefined,
createdAt: row.createdAt.getTime(),
- similarity: row.combined_score
+ similarity: row.combined_score,
}));
await this.setCache({
key: cacheKey,
agentId: params.agentId,
- value: JSON.stringify(results)
+ value: JSON.stringify(results),
});
return results;
@@ -1614,35 +1634,52 @@ export class PostgresDatabaseAdapter
return this.withDatabase(async () => {
const client = await this.pool.connect();
try {
- await client.query('BEGIN');
-
- const sql = `
- INSERT INTO knowledge (
- id, "agentId", content, embedding, "createdAt",
- "isMain", "originalId", "chunkIndex", "isShared"
- ) VALUES ($1, $2, $3, $4, to_timestamp($5/1000.0), $6, $7, $8, $9)
- ON CONFLICT (id) DO NOTHING
- `;
+ await client.query("BEGIN");
const metadata = knowledge.content.metadata || {};
- const vectorStr = knowledge.embedding ?
- `[${Array.from(knowledge.embedding).join(",")}]` : null;
-
- await client.query(sql, [
- knowledge.id,
- metadata.isShared ? null : knowledge.agentId,
- knowledge.content,
- vectorStr,
- knowledge.createdAt || Date.now(),
- metadata.isMain || false,
- metadata.originalId || null,
- metadata.chunkIndex || null,
- metadata.isShared || false
- ]);
+ const vectorStr = knowledge.embedding
+ ? `[${Array.from(knowledge.embedding).join(",")}]`
+ : null;
+
+ // If this is a chunk, use createKnowledgeChunk
+ if (metadata.isChunk && metadata.originalId) {
+ await this.createKnowledgeChunk({
+ id: knowledge.id,
+ originalId: metadata.originalId,
+ agentId: metadata.isShared ? null : knowledge.agentId,
+ content: knowledge.content,
+ embedding: knowledge.embedding,
+ chunkIndex: metadata.chunkIndex || 0,
+ isShared: metadata.isShared || false,
+ createdAt: knowledge.createdAt || Date.now(),
+ });
+ } else {
+ // This is a main knowledge item
+ await client.query(
+ `
+ INSERT INTO knowledge (
+ id, "agentId", content, embedding, "createdAt",
+ "isMain", "originalId", "chunkIndex", "isShared"
+ ) VALUES ($1, $2, $3, $4, to_timestamp($5/1000.0), $6, $7, $8, $9)
+ ON CONFLICT (id) DO NOTHING
+ `,
+ [
+ knowledge.id,
+ metadata.isShared ? null : knowledge.agentId,
+ knowledge.content,
+ vectorStr,
+ knowledge.createdAt || Date.now(),
+ true,
+ null,
+ null,
+ metadata.isShared || false,
+ ]
+ );
+ }
- await client.query('COMMIT');
+ await client.query("COMMIT");
} catch (error) {
- await client.query('ROLLBACK');
+ await client.query("ROLLBACK");
throw error;
} finally {
client.release();
@@ -1652,19 +1689,100 @@ export class PostgresDatabaseAdapter
async removeKnowledge(id: UUID): Promise {
return this.withDatabase(async () => {
- await this.pool.query('DELETE FROM knowledge WHERE id = $1', [id]);
+ const client = await this.pool.connect();
+ try {
+ await client.query("BEGIN");
+
+ // Check if this is a pattern-based chunk deletion (e.g., "id-chunk-*")
+ if (typeof id === "string" && id.includes("-chunk-*")) {
+ const mainId = id.split("-chunk-")[0];
+ // Delete chunks for this main ID
+ await client.query(
+ 'DELETE FROM knowledge WHERE "originalId" = $1',
+ [mainId]
+ );
+ } else {
+ // First delete all chunks associated with this knowledge item
+ await client.query(
+ 'DELETE FROM knowledge WHERE "originalId" = $1',
+ [id]
+ );
+ // Then delete the main knowledge item
+ await client.query("DELETE FROM knowledge WHERE id = $1", [
+ id,
+ ]);
+ }
+
+ await client.query("COMMIT");
+ } catch (error) {
+ await client.query("ROLLBACK");
+ elizaLogger.error("Error removing knowledge", {
+ error:
+ error instanceof Error ? error.message : String(error),
+ id,
+ });
+ throw error;
+ } finally {
+ client.release();
+ }
}, "removeKnowledge");
}
async clearKnowledge(agentId: UUID, shared?: boolean): Promise {
return this.withDatabase(async () => {
- const sql = shared ?
- 'DELETE FROM knowledge WHERE ("agentId" = $1 OR "isShared" = true)' :
- 'DELETE FROM knowledge WHERE "agentId" = $1';
+ const sql = shared
+ ? 'DELETE FROM knowledge WHERE ("agentId" = $1 OR "isShared" = true)'
+ : 'DELETE FROM knowledge WHERE "agentId" = $1';
await this.pool.query(sql, [agentId]);
}, "clearKnowledge");
}
+
+ private async createKnowledgeChunk(params: {
+ id: UUID;
+ originalId: UUID;
+ agentId: UUID | null;
+ content: any;
+ embedding: Float32Array | undefined | null;
+ chunkIndex: number;
+ isShared: boolean;
+ createdAt: number;
+ }): Promise {
+ const vectorStr = params.embedding
+ ? `[${Array.from(params.embedding).join(",")}]`
+ : null;
+
+ // Store the pattern-based ID in the content metadata for compatibility
+ const patternId = `${params.originalId}-chunk-${params.chunkIndex}`;
+ const contentWithPatternId = {
+ ...params.content,
+ metadata: {
+ ...params.content.metadata,
+ patternId,
+ },
+ };
+
+ await this.pool.query(
+ `
+ INSERT INTO knowledge (
+ id, "agentId", content, embedding, "createdAt",
+ "isMain", "originalId", "chunkIndex", "isShared"
+ ) VALUES ($1, $2, $3, $4, to_timestamp($5/1000.0), $6, $7, $8, $9)
+ ON CONFLICT (id) DO NOTHING
+ `,
+ [
+ v4(), // Generate a proper UUID for PostgreSQL
+ params.agentId,
+ contentWithPatternId, // Store the pattern ID in metadata
+ vectorStr,
+ params.createdAt,
+ false,
+ params.originalId,
+ params.chunkIndex,
+ params.isShared,
+ ]
+ );
+ }
}
export default PostgresDatabaseAdapter;
diff --git a/packages/adapter-redis/package.json b/packages/adapter-redis/package.json
index 055460a270a..fdd3b2d18ad 100644
--- a/packages/adapter-redis/package.json
+++ b/packages/adapter-redis/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/adapter-redis",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
diff --git a/packages/adapter-sqlite/package.json b/packages/adapter-sqlite/package.json
index 74642dee834..8b45f36b5b2 100644
--- a/packages/adapter-sqlite/package.json
+++ b/packages/adapter-sqlite/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/adapter-sqlite",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
diff --git a/packages/adapter-sqljs/package.json b/packages/adapter-sqljs/package.json
index 967c00a44cb..3c2661a1682 100644
--- a/packages/adapter-sqljs/package.json
+++ b/packages/adapter-sqljs/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/adapter-sqljs",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
diff --git a/packages/adapter-sqljs/src/index.ts b/packages/adapter-sqljs/src/index.ts
index db27215e100..6df3c93f03c 100644
--- a/packages/adapter-sqljs/src/index.ts
+++ b/packages/adapter-sqljs/src/index.ts
@@ -859,7 +859,7 @@ export class SqlJsDatabaseAdapter
return JSON.parse(cachedResult);
}
- let sql = `
+ const sql = `
WITH vector_scores AS (
SELECT id,
1 / (1 + vec_distance_L2(embedding, ?)) as vector_score
diff --git a/packages/adapter-supabase/package.json b/packages/adapter-supabase/package.json
index 9c267b86a4b..f1785ef9746 100644
--- a/packages/adapter-supabase/package.json
+++ b/packages/adapter-supabase/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/adapter-supabase",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
diff --git a/packages/adapter-supabase/seed.sql b/packages/adapter-supabase/seed.sql
index 063c5fbe532..4385fa47864 100644
--- a/packages/adapter-supabase/seed.sql
+++ b/packages/adapter-supabase/seed.sql
@@ -1,3 +1,3 @@
-INSERT INTO public.accounts (id, name, email, avatarUrl, details) VALUES ('00000000-0000-0000-0000-000000000000', 'Default Agent', 'default@agent.com', '', '{}');
-INSERT INTO public.rooms (id) VALUES ('00000000-0000-0000-0000-000000000000');
-INSERT INTO public.participants (userId, roomId) VALUES ('00000000-0000-0000-0000-000000000000', '00000000-0000-0000-0000-000000000000');
+INSERT INTO public.accounts (id, name, email, "avatarUrl", details) VALUES ('00000000-0000-0000-0000-000000000000', 'Default Agent', 'default@agent.com', '', '{}');
+INSERT INTO public.rooms (id, "createdAt") VALUES ('00000000-0000-0000-0000-000000000000', NOW());
+INSERT INTO public.participants (id, "createdAt", "userId", "roomId", "userState", last_messsage_read) VALUES ('00000000-0000-0000-0000-000000000000', NOW(), 'Default Agent', '00000000-0000-0000-0000-000000000000', NULL, NULL);
diff --git a/packages/adapter-supabase/src/index.ts b/packages/adapter-supabase/src/index.ts
index 9c8d643f61a..3800b845c98 100644
--- a/packages/adapter-supabase/src/index.ts
+++ b/packages/adapter-supabase/src/index.ts
@@ -20,12 +20,12 @@ export class SupabaseDatabaseAdapter extends DatabaseAdapter {
.from("rooms")
.select("id")
.eq("id", roomId)
- .single();
+ .maybeSingle();
if (error) {
- throw new Error(`Error getting room: ${error.message}`);
+ elizaLogger.error(`Error getting room: ${error.message}`);
+ return null;
}
-
return data ? (data.id as UUID) : null;
}
@@ -56,7 +56,7 @@ export class SupabaseDatabaseAdapter extends DatabaseAdapter {
.single();
if (error) {
- console.error("Error getting participant user state:", error);
+ elizaLogger.error("Error getting participant user state:", error);
return null;
}
@@ -75,7 +75,7 @@ export class SupabaseDatabaseAdapter extends DatabaseAdapter {
.eq("userId", userId);
if (error) {
- console.error("Error setting participant user state:", error);
+ elizaLogger.error("Error setting participant user state:", error);
throw new Error("Failed to set participant user state");
}
}
@@ -127,7 +127,7 @@ export class SupabaseDatabaseAdapter extends DatabaseAdapter {
const { data, error } = await query;
if (error) {
- console.error("Error retrieving memories by room IDs:", error);
+ elizaLogger.error("Error retrieving memories by room IDs:", error);
return [];
}
@@ -155,7 +155,7 @@ export class SupabaseDatabaseAdapter extends DatabaseAdapter {
.from("accounts")
.upsert([account]);
if (error) {
- console.error(error.message);
+ elizaLogger.error(error.message);
return false;
}
return true;
@@ -175,7 +175,7 @@ export class SupabaseDatabaseAdapter extends DatabaseAdapter {
.eq("id", params.roomId);
if (response.error) {
- console.error("Error!" + response.error);
+ elizaLogger.error("Error!" + response.error);
return [];
}
const { data } = response;
@@ -194,7 +194,7 @@ export class SupabaseDatabaseAdapter extends DatabaseAdapter {
)
.flat();
} catch (error) {
- console.error("error", error);
+ elizaLogger.error("error", error);
throw error;
}
}
@@ -267,7 +267,7 @@ export class SupabaseDatabaseAdapter extends DatabaseAdapter {
});
if (error) {
- console.error("Error inserting log:", error);
+ elizaLogger.error("Error inserting log:", error);
throw new Error(error.message);
}
}
@@ -357,7 +357,7 @@ export class SupabaseDatabaseAdapter extends DatabaseAdapter {
.single();
if (error) {
- console.error("Error retrieving memory by ID:", error);
+ elizaLogger.error("Error retrieving memory by ID:", error);
return null;
}
@@ -571,7 +571,7 @@ export class SupabaseDatabaseAdapter extends DatabaseAdapter {
.insert({ userId: userId, roomId: roomId });
if (error) {
- console.error(`Error adding participant: ${error.message}`);
+ elizaLogger.error(`Error adding participant: ${error.message}`);
return false;
}
return true;
@@ -585,7 +585,7 @@ export class SupabaseDatabaseAdapter extends DatabaseAdapter {
.eq("roomId", roomId);
if (error) {
- console.error(`Error removing participant: ${error.message}`);
+ elizaLogger.error(`Error removing participant: ${error.message}`);
return false;
}
return true;
@@ -695,7 +695,7 @@ export class SupabaseDatabaseAdapter extends DatabaseAdapter {
.single();
if (error) {
- console.error('Error fetching cache:', error);
+ elizaLogger.error('Error fetching cache:', error);
return undefined;
}
@@ -717,7 +717,7 @@ export class SupabaseDatabaseAdapter extends DatabaseAdapter {
});
if (error) {
- console.error('Error setting cache:', error);
+ elizaLogger.error('Error setting cache:', error);
return false;
}
diff --git a/packages/client-auto/package.json b/packages/client-auto/package.json
index dc0fd9b22b3..6ee732d0851 100644
--- a/packages/client-auto/package.json
+++ b/packages/client-auto/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/client-auto",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
diff --git a/packages/client-direct/package.json b/packages/client-direct/package.json
index d0d81ddeeb7..f8a407b58be 100644
--- a/packages/client-direct/package.json
+++ b/packages/client-direct/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/client-direct",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"main": "dist/index.js",
"module": "dist/index.js",
"type": "module",
diff --git a/packages/client-direct/src/api.ts b/packages/client-direct/src/api.ts
index 2780831c24e..c19ac5279c1 100644
--- a/packages/client-direct/src/api.ts
+++ b/packages/client-direct/src/api.ts
@@ -6,6 +6,7 @@ import {
AgentRuntime,
elizaLogger,
getEnvVariable,
+ UUID,
validateCharacterConfig,
ServiceType,
} from "@elizaos/core";
@@ -13,7 +14,38 @@ import {
import { TeeLogQuery, TeeLogService } from "@elizaos/plugin-tee-log";
import { REST, Routes } from "discord.js";
import { DirectClient } from ".";
-import { stringToUuid } from "@elizaos/core";
+import { validateUuid } from "@elizaos/core";
+
+interface UUIDParams {
+ agentId: UUID;
+ roomId?: UUID;
+}
+
+function validateUUIDParams(
+ params: { agentId: string; roomId?: string },
+ res: express.Response
+): UUIDParams | null {
+ const agentId = validateUuid(params.agentId);
+ if (!agentId) {
+ res.status(400).json({
+ error: "Invalid AgentId format. Expected to be a UUID: xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx",
+ });
+ return null;
+ }
+
+ if (params.roomId) {
+ const roomId = validateUuid(params.roomId);
+ if (!roomId) {
+ res.status(400).json({
+ error: "Invalid RoomId format. Expected to be a UUID: xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx",
+ });
+ return null;
+ }
+ return { agentId, roomId };
+ }
+
+ return { agentId };
+}
export function createApiRouter(
agents: Map,
@@ -48,7 +80,11 @@ export function createApiRouter(
});
router.get("/agents/:agentId", (req, res) => {
- const agentId = req.params.agentId;
+ const { agentId } = validateUUIDParams(req.params, res) ?? {
+ agentId: null,
+ };
+ if (!agentId) return;
+
const agent = agents.get(agentId);
if (!agent) {
@@ -56,7 +92,7 @@ export function createApiRouter(
return;
}
- let character = agent?.character;
+ const character = agent?.character;
if (character?.settings?.secrets) {
delete character.settings.secrets;
}
@@ -68,8 +104,11 @@ export function createApiRouter(
});
router.post("/agents/:agentId/set", async (req, res) => {
- const agentId = req.params.agentId;
- console.log("agentId", agentId);
+ const { agentId } = validateUUIDParams(req.params, res) ?? {
+ agentId: null,
+ };
+ if (!agentId) return;
+
let agent: AgentRuntime = agents.get(agentId);
// update character
@@ -104,7 +143,11 @@ export function createApiRouter(
});
router.get("/agents/:agentId/channels", async (req, res) => {
- const agentId = req.params.agentId;
+ const { agentId } = validateUUIDParams(req.params, res) ?? {
+ agentId: null,
+ };
+ if (!agentId) return;
+
const runtime = agents.get(agentId);
if (!runtime) {
@@ -130,8 +173,12 @@ export function createApiRouter(
});
router.get("/agents/:agentId/:roomId/memories", async (req, res) => {
- const agentId = req.params.agentId;
- const roomId = stringToUuid(req.params.roomId);
+ const { agentId, roomId } = validateUUIDParams(req.params, res) ?? {
+ agentId: null,
+ roomId: null,
+ };
+ if (!agentId || !roomId) return;
+
let runtime = agents.get(agentId);
// if runtime is null, look for runtime with the same name
diff --git a/packages/client-direct/src/index.ts b/packages/client-direct/src/index.ts
index 138994bbee7..15d80181088 100644
--- a/packages/client-direct/src/index.ts
+++ b/packages/client-direct/src/index.ts
@@ -378,14 +378,12 @@ export class DirectClient {
// hyperfi specific parameters
let nearby = [];
- let messages = [];
let availableEmotes = [];
if (body.nearby) {
nearby = body.nearby;
}
if (body.messages) {
- messages = body.messages;
// loop on the messages and record the memories
// might want to do this in parallel
for (const msg of body.messages) {
@@ -507,10 +505,17 @@ export class DirectClient {
schema: hyperfiOutSchema,
});
+ if (!response) {
+ res.status(500).send(
+ "No response from generateMessageResponse"
+ );
+ return;
+ }
+
let hfOut;
try {
hfOut = hyperfiOutSchema.parse(response.object);
- } catch (e) {
+ } catch {
elizaLogger.error(
"cant serialize response",
response.object
@@ -520,7 +525,7 @@ export class DirectClient {
}
// do this in the background
- const rememberThis = new Promise(async (resolve) => {
+ new Promise((resolve) => {
const contentObj: Content = {
text: hfOut.say,
};
@@ -550,45 +555,38 @@ export class DirectClient {
content: contentObj,
};
- await runtime.messageManager.createMemory(responseMessage); // 18.2ms
-
- if (!response) {
- res.status(500).send(
- "No response from generateMessageResponse"
- );
- return;
- }
-
- let message = null as Content | null;
-
- const messageId = stringToUuid(Date.now().toString());
- const memory: Memory = {
- id: messageId,
- agentId: runtime.agentId,
- userId,
- roomId,
- content,
- createdAt: Date.now(),
- };
-
- // run evaluators (generally can be done in parallel with processActions)
- // can an evaluator modify memory? it could but currently doesn't
- await runtime.evaluate(memory, state); // 0.5s
-
- // only need to call if responseMessage.content.action is set
- if (contentObj.action) {
- // pass memory (query) to any actions to call
- const _result = await runtime.processActions(
- memory,
- [responseMessage],
- state,
- async (newMessages) => {
- message = newMessages;
- return [memory];
+ runtime.messageManager.createMemory(responseMessage).then(() => {
+ const messageId = stringToUuid(Date.now().toString());
+ const memory: Memory = {
+ id: messageId,
+ agentId: runtime.agentId,
+ userId,
+ roomId,
+ content,
+ createdAt: Date.now(),
+ };
+
+ // run evaluators (generally can be done in parallel with processActions)
+ // can an evaluator modify memory? it could but currently doesn't
+ runtime.evaluate(memory, state).then(() => {
+ // only need to call if responseMessage.content.action is set
+ if (contentObj.action) {
+ // pass memory (query) to any actions to call
+ runtime.processActions(
+ memory,
+ [responseMessage],
+ state,
+ async (_newMessages) => {
+ // FIXME: this is supposed override what the LLM said/decided
+ // but the promise doesn't make this possible
+ //message = newMessages;
+ return [memory];
+ }
+ ); // 0.674s
}
- ); // 0.674s
- }
- resolve(true);
+ resolve(true);
+ });
+ });
});
res.json({ response: hfOut });
}
diff --git a/packages/client-discord/package.json b/packages/client-discord/package.json
index 49f7ac89e53..9414c99c7b9 100644
--- a/packages/client-discord/package.json
+++ b/packages/client-discord/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/client-discord",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
diff --git a/packages/client-discord/src/actions/joinvoice.ts b/packages/client-discord/src/actions/joinvoice.ts
index dbfa556482e..71c879712af 100644
--- a/packages/client-discord/src/actions/joinvoice.ts
+++ b/packages/client-discord/src/actions/joinvoice.ts
@@ -8,6 +8,8 @@ import {
IAgentRuntime,
Memory,
State,
+ generateText,
+ ModelClass,
} from "@elizaos/core";
import {
Channel,
@@ -17,6 +19,7 @@ import {
Guild,
GuildMember,
} from "discord.js";
+import { joinVoiceChannel } from "@discordjs/voice";
export default {
name: "JOIN_VOICE",
@@ -66,12 +69,7 @@ export default {
return false;
}
- const client = state.discordClient as Client;
-
- // Check if the client is connected to any voice channel
- const isConnectedToVoice = client.voice.adapters.size === 0;
-
- return isConnectedToVoice;
+ return true;
},
description: "Join a voice channel to participate in voice chat.",
handler: async (
@@ -115,31 +113,30 @@ export default {
);
});
- if (!state.voiceManager) {
- state.voiceManager = new VoiceManager({
- client: state.discordClient,
- runtime: runtime,
- });
- }
-
if (targetChannel) {
- state.voiceManager.joinVoiceChannel({
+ joinVoiceChannel({
channelId: targetChannel.id,
guildId: (discordMessage as DiscordMessage).guild?.id as string,
adapterCreator: (client.guilds.cache.get(id) as Guild)
.voiceAdapterCreator,
+ selfDeaf: false,
+ selfMute: false,
+ group: client.user.id,
});
return true;
} else {
const member = (discordMessage as DiscordMessage)
.member as GuildMember;
if (member?.voice?.channel) {
- state.voiceManager.joinVoiceChannel({
+ joinVoiceChannel({
channelId: member.voice.channel.id,
guildId: (discordMessage as DiscordMessage).guild
?.id as string,
adapterCreator: (client.guilds.cache.get(id) as Guild)
.voiceAdapterCreator,
+ selfDeaf: false,
+ selfMute: false,
+ group: client.user.id,
});
return true;
}
@@ -204,12 +201,15 @@ You should only respond with the name of the voice channel or none, no commentar
});
if (targetChannel) {
- state.voiceManager.joinVoiceChannel({
+ joinVoiceChannel({
channelId: targetChannel.id,
guildId: (discordMessage as DiscordMessage).guild
?.id as string,
adapterCreator: (client.guilds.cache.get(id) as Guild)
.voiceAdapterCreator,
+ selfDeaf: false,
+ selfMute: false,
+ group: client.user.id,
});
return true;
}
diff --git a/packages/client-farcaster/package.json b/packages/client-farcaster/package.json
index ceb30f634db..5da998b55c1 100644
--- a/packages/client-farcaster/package.json
+++ b/packages/client-farcaster/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/client-farcaster",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
diff --git a/packages/client-github/package.json b/packages/client-github/package.json
index 9859b5708ec..27c19428132 100644
--- a/packages/client-github/package.json
+++ b/packages/client-github/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/client-github",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
diff --git a/packages/client-lens/package.json b/packages/client-lens/package.json
index 186e45cc745..5c51296bf6c 100644
--- a/packages/client-lens/package.json
+++ b/packages/client-lens/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/client-lens",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
diff --git a/packages/client-slack/package.json b/packages/client-slack/package.json
index 98bb8c05ddd..bdeccf37b5d 100644
--- a/packages/client-slack/package.json
+++ b/packages/client-slack/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/client-slack",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"description": "Slack client plugin for Eliza framework",
"type": "module",
"main": "dist/index.js",
diff --git a/packages/client-slack/src/actions/chat_with_attachments.ts b/packages/client-slack/src/actions/chat_with_attachments.ts
index e059cc47b4b..b40353020d7 100644
--- a/packages/client-slack/src/actions/chat_with_attachments.ts
+++ b/packages/client-slack/src/actions/chat_with_attachments.ts
@@ -5,7 +5,6 @@ import {
parseJSONObjectFromText,
getModelSettings,
} from "@elizaos/core";
-import { models } from "@elizaos/core";
import {
Action,
ActionExample,
diff --git a/packages/client-slack/src/actions/summarize_conversation.ts b/packages/client-slack/src/actions/summarize_conversation.ts
index b487757cadd..14649521222 100644
--- a/packages/client-slack/src/actions/summarize_conversation.ts
+++ b/packages/client-slack/src/actions/summarize_conversation.ts
@@ -6,7 +6,6 @@ import {
parseJSONObjectFromText,
getModelSettings,
} from "@elizaos/core";
-import { models } from "@elizaos/core";
import { getActorDetails } from "@elizaos/core";
import {
Action,
diff --git a/packages/client-telegram/package.json b/packages/client-telegram/package.json
index 83277b76f3e..622c2d2ba6a 100644
--- a/packages/client-telegram/package.json
+++ b/packages/client-telegram/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/client-telegram",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
diff --git a/packages/client-telegram/src/messageManager.ts b/packages/client-telegram/src/messageManager.ts
index 73240efa013..3daf8f42eb9 100644
--- a/packages/client-telegram/src/messageManager.ts
+++ b/packages/client-telegram/src/messageManager.ts
@@ -507,7 +507,7 @@ export class MessageManager {
// Check if team member has direct interest first
if (
- this.runtime.character.clientConfig?.discord?.isPartOfTeam &&
+ this.runtime.character.clientConfig?.telegram?.isPartOfTeam &&
!this._isTeamLeader() &&
this._isRelevantToTeamMember(messageText, chatId)
) {
diff --git a/packages/client-twitter/package.json b/packages/client-twitter/package.json
index 88e51d6d09a..566cb86d7d0 100644
--- a/packages/client-twitter/package.json
+++ b/packages/client-twitter/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/client-twitter",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
diff --git a/packages/client-twitter/src/interactions.ts b/packages/client-twitter/src/interactions.ts
index ebc9ca9e64a..e95a2f211e6 100644
--- a/packages/client-twitter/src/interactions.ts
+++ b/packages/client-twitter/src/interactions.ts
@@ -14,6 +14,8 @@ import {
stringToUuid,
elizaLogger,
getEmbeddingZeroVector,
+ IImageDescriptionService,
+ ServiceType
} from "@elizaos/core";
import { ClientBase } from "./base";
import { buildConversationThread, sendTweet, wait } from "./utils.ts";
@@ -43,6 +45,8 @@ Recent interactions between {{agentName}} and other users:
Current Post:
{{currentPost}}
+Here is the descriptions of images in the Current post.
+{{imageDescriptions}}
Thread of Tweets You Are Replying To:
{{formattedConversation}}
@@ -53,6 +57,8 @@ Thread of Tweets You Are Replying To:
Here is the current post text again. Remember to include an action if the current post text includes a prompt that asks for one of the available actions mentioned above (does not need to be exact)
{{currentPost}}
+Here is the descriptions of images in the Current post.
+{{imageDescriptions}}
` + messageCompletionFooter;
export const twitterShouldRespondTemplate = (targetUsersStr: string) =>
@@ -342,11 +348,34 @@ export class TwitterInteractionClient {
elizaLogger.debug("formattedConversation: ", formattedConversation);
+ const imageDescriptionsArray = [];
+ try{
+ elizaLogger.debug('Getting images');
+ for (const photo of tweet.photos) {
+ elizaLogger.debug(photo.url);
+ const description = await this.runtime
+ .getService(
+ ServiceType.IMAGE_DESCRIPTION
+ )
+ .describeImage(photo.url);
+ imageDescriptionsArray.push(description);
+ }
+ } catch (error) {
+ // Handle the error
+ elizaLogger.error("Error Occured during describing image: ", error);
+}
+
+
+
+
let state = await this.runtime.composeState(message, {
twitterClient: this.client.twitterClient,
twitterUserName: this.client.twitterConfig.TWITTER_USERNAME,
currentPost,
formattedConversation,
+ imageDescriptions: imageDescriptionsArray.length > 0
+ ? `\nImages in Tweet:\n${imageDescriptionsArray.map((desc, i) =>
+ `Image ${i + 1}: Title: ${desc.title}\nDescription: ${desc.description}`).join("\n\n")}`:""
});
// check if the tweet exists, save if it doesn't
@@ -413,7 +442,6 @@ export class TwitterInteractionClient {
this.runtime.character?.templates?.messageHandlerTemplate ||
twitterMessageHandlerTemplate,
});
-
elizaLogger.debug("Interactions prompt:\n" + context);
const response = await generateMessageResponse({
@@ -624,4 +652,4 @@ export class TwitterInteractionClient {
return thread;
}
-}
+}
\ No newline at end of file
diff --git a/packages/client-twitter/src/plugins/SttTtsSpacesPlugin.ts b/packages/client-twitter/src/plugins/SttTtsSpacesPlugin.ts
index fd306f65a4b..b23a25d79c2 100644
--- a/packages/client-twitter/src/plugins/SttTtsSpacesPlugin.ts
+++ b/packages/client-twitter/src/plugins/SttTtsSpacesPlugin.ts
@@ -28,6 +28,9 @@ interface PluginConfig {
* - On speaker mute -> flush STT -> GPT -> TTS -> push to Janus
*/
export class SttTtsPlugin implements Plugin {
+ name = "SttTtsPlugin";
+ description = "Speech-to-text (OpenAI) + conversation + TTS (ElevenLabs)";
+
private space?: Space;
private janus?: JanusClient;
@@ -64,7 +67,7 @@ export class SttTtsPlugin implements Plugin {
private ttsQueue: string[] = [];
private isSpeaking = false;
- onAttach(space: Space) {
+ onAttach(_space: Space) {
elizaLogger.log("[SttTtsPlugin] onAttach => space was attached");
}
diff --git a/packages/client-twitter/src/post.ts b/packages/client-twitter/src/post.ts
index e0aff4b3a61..93d89930259 100644
--- a/packages/client-twitter/src/post.ts
+++ b/packages/client-twitter/src/post.ts
@@ -8,6 +8,7 @@ import {
stringToUuid,
TemplateType,
UUID,
+ truncateToCompleteSentence,
} from "@elizaos/core";
import { elizaLogger } from "@elizaos/core";
import { ClientBase } from "./base.ts";
@@ -77,40 +78,6 @@ Tweet:
# Respond with qualifying action tags only. Default to NO action unless extremely confident of relevance.` +
postActionResponseFooter;
-/**
- * Truncate text to fit within the Twitter character limit, ensuring it ends at a complete sentence.
- */
-function truncateToCompleteSentence(
- text: string,
- maxTweetLength: number
-): string {
- if (text.length <= maxTweetLength) {
- return text;
- }
-
- // Attempt to truncate at the last period within the limit
- const lastPeriodIndex = text.lastIndexOf(".", maxTweetLength - 1);
- if (lastPeriodIndex !== -1) {
- const truncatedAtPeriod = text.slice(0, lastPeriodIndex + 1).trim();
- if (truncatedAtPeriod.length > 0) {
- return truncatedAtPeriod;
- }
- }
-
- // If no period, truncate to the nearest whitespace within the limit
- const lastSpaceIndex = text.lastIndexOf(" ", maxTweetLength - 1);
- if (lastSpaceIndex !== -1) {
- const truncatedAtSpace = text.slice(0, lastSpaceIndex).trim();
- if (truncatedAtSpace.length > 0) {
- return truncatedAtSpace + "...";
- }
- }
-
- // Fallback: Hard truncate and add ellipsis
- const hardTruncated = text.slice(0, maxTweetLength - 3).trim();
- return hardTruncated + "...";
-}
-
interface PendingTweet {
cleanedContent: string;
roomId: UUID;
@@ -399,7 +366,6 @@ export class TwitterPostClient {
async handleNoteTweet(
client: ClientBase,
- runtime: IAgentRuntime,
content: string,
tweetId?: string
) {
@@ -465,11 +431,7 @@ export class TwitterPostClient {
let result;
if (cleanedContent.length > DEFAULT_MAX_TWEET_LENGTH) {
- result = await this.handleNoteTweet(
- client,
- runtime,
- cleanedContent
- );
+ result = await this.handleNoteTweet(client, cleanedContent);
} else {
result = await this.sendStandardTweet(client, cleanedContent);
}
@@ -1204,7 +1166,6 @@ export class TwitterPostClient {
if (replyText.length > DEFAULT_MAX_TWEET_LENGTH) {
result = await this.handleNoteTweet(
this.client,
- this.runtime,
replyText,
tweet.id
);
diff --git a/packages/client-twitter/src/utils.ts b/packages/client-twitter/src/utils.ts
index d11ed5b534f..0f2c125ed12 100644
--- a/packages/client-twitter/src/utils.ts
+++ b/packages/client-twitter/src/utils.ts
@@ -212,15 +212,18 @@ export async function sendTweet(
})
);
}
+
+ const cleanChunk = deduplicateMentions(chunk.trim())
+
const result = await client.requestQueue.add(async () =>
isLongTweet
? client.twitterClient.sendLongTweet(
- chunk.trim(),
+ cleanChunk,
previousTweetId,
mediaData
)
: client.twitterClient.sendTweet(
- chunk.trim(),
+ cleanChunk,
previousTweetId,
mediaData
)
@@ -345,7 +348,7 @@ function extractUrls(paragraph: string): {
function splitSentencesAndWords(text: string, maxLength: number): string[] {
// Split by periods, question marks and exclamation marks
// Note that URLs in text have been replaced with `<>` and won't be split by dots
- const sentences = text.match(/[^\.!\?]+[\.!\?]+|[^\.!\?]+$/g) || [text];
+ const sentences = text.match(/[^.!?]+[.!?]+|[^.!?]+$/g) || [text];
const chunks: string[] = [];
let currentChunk = "";
@@ -397,6 +400,33 @@ function splitSentencesAndWords(text: string, maxLength: number): string[] {
return chunks;
}
+function deduplicateMentions(paragraph: string) {
+ // Regex to match mentions at the beginning of the string
+ const mentionRegex = /^@(\w+)(?:\s+@(\w+))*(\s+|$)/;
+
+ // Find all matches
+ const matches = paragraph.match(mentionRegex);
+
+ if (!matches) {
+ return paragraph; // If no matches, return the original string
+ }
+
+ // Extract mentions from the match groups
+ let mentions = matches.slice(0, 1)[0].trim().split(' ')
+
+ // Deduplicate mentions
+ mentions = [...new Set(mentions)];
+
+ // Reconstruct the string with deduplicated mentions
+ const uniqueMentionsString = mentions.join(' ');
+
+ // Find where the mentions end in the original string
+ const endOfMentions = paragraph.indexOf(matches[0]) + matches[0].length;
+
+ // Construct the result by combining unique mentions with the rest of the string
+ return uniqueMentionsString + ' ' + paragraph.slice(endOfMentions);
+}
+
function restoreUrls(
chunks: string[],
placeholderMap: Map
@@ -424,4 +454,4 @@ function splitParagraph(paragraph: string, maxLength: number): string[] {
const restoredChunks = restoreUrls(splittedChunks, placeholderMap);
return restoredChunks;
-}
\ No newline at end of file
+}
diff --git a/packages/core/__tests__/uuid.test.ts b/packages/core/__tests__/uuid.test.ts
new file mode 100644
index 00000000000..f737acea0b0
--- /dev/null
+++ b/packages/core/__tests__/uuid.test.ts
@@ -0,0 +1,108 @@
+import { beforeEach, describe, expect, it } from "vitest";
+import { stringToUuid } from "../src/uuid";
+import type { UUID } from "../src/types";
+
+describe("UUID Module", () => {
+ // Helper function to generate test strings
+ const generateTestString = (): string =>
+ Math.random().toString(36).substring(7);
+
+ // Test data setup
+ let testString: string;
+ let testNumber: number;
+
+ beforeEach(() => {
+ testString = generateTestString();
+ testNumber = Math.floor(Math.random() * 1000);
+ });
+
+ describe("stringToUuid", () => {
+ it("should generate a valid UUID matching the standard format", () => {
+ const result = stringToUuid(testString) as UUID;
+ expect(result).toMatch(
+ /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i
+ );
+ });
+
+ it("should generate consistent UUIDs for identical inputs", () => {
+ const input = testString;
+ const uuid1 = stringToUuid(input) as UUID;
+ const uuid2 = stringToUuid(input) as UUID;
+ expect(uuid1).toBe(uuid2);
+ });
+
+ it("should generate unique UUIDs for different inputs", () => {
+ const input1 = testString;
+ const input2 = generateTestString();
+ const uuid1 = stringToUuid(input1) as UUID;
+ const uuid2 = stringToUuid(input2) as UUID;
+ expect(uuid1).not.toBe(uuid2);
+ });
+
+ describe("input handling", () => {
+ it("should convert number inputs to strings correctly", () => {
+ const numberUuid = stringToUuid(testNumber) as UUID;
+ const stringUuid = stringToUuid(testNumber.toString()) as UUID;
+ expect(numberUuid).toBe(stringUuid);
+ });
+
+ it("should throw TypeError for invalid input types", () => {
+ expect(() => stringToUuid(undefined as any)).toThrow(TypeError);
+ expect(() => stringToUuid(null as any)).toThrow(TypeError);
+ expect(() => stringToUuid({} as any)).toThrow(TypeError);
+ });
+
+ it("should handle empty string input", () => {
+ const result = stringToUuid("") as UUID;
+ expect(result).toMatch(
+ /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i
+ );
+ });
+
+ it("should handle Unicode characters and emojis consistently", () => {
+ const unicodeInput = "Hello 世界! 🌍";
+ const result1 = stringToUuid(unicodeInput) as UUID;
+ const result2 = stringToUuid(unicodeInput) as UUID;
+ expect(result1).toBe(result2);
+ expect(result1).toMatch(
+ /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i
+ );
+ });
+ });
+
+ describe("UUID version and variant bits", () => {
+ it("should set correct version bits (version 5)", () => {
+ const uuid = stringToUuid(testString) as UUID;
+ const versionChar = uuid.split("-")[2][0];
+ expect(versionChar).toBe("0");
+ });
+
+ it("should set correct variant bits (RFC4122)", () => {
+ const uuid = stringToUuid(testString) as UUID;
+ const variantByte = parseInt(
+ uuid.split("-")[3].slice(0, 2),
+ 16
+ );
+ expect(variantByte >= 0x80 && variantByte <= 0xbf).toBe(true);
+ });
+ });
+
+ describe("encoding handling", () => {
+ it("should handle URL-unsafe characters", () => {
+ const urlUnsafeInput = "test?query=value¶m=123";
+ const result = stringToUuid(urlUnsafeInput) as UUID;
+ expect(result).toMatch(
+ /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i
+ );
+ });
+
+ it("should handle very long inputs", () => {
+ const longInput = "a".repeat(1000);
+ const result = stringToUuid(longInput) as UUID;
+ expect(result).toMatch(
+ /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i
+ );
+ });
+ });
+ });
+});
diff --git a/packages/core/generation.ts b/packages/core/generation.ts
deleted file mode 100644
index 74d41237738..00000000000
--- a/packages/core/generation.ts
+++ /dev/null
@@ -1,1970 +0,0 @@
-import { createAnthropic } from "@ai-sdk/anthropic";
-import { createGoogleGenerativeAI } from "@ai-sdk/google";
-import { createGroq } from "@ai-sdk/groq";
-import { createOpenAI } from "@ai-sdk/openai";
-import { RecursiveCharacterTextSplitter } from "langchain/text_splitter";
-import {
- generateObject as aiGenerateObject,
- generateText as aiGenerateText,
- CoreTool,
- GenerateObjectResult,
- StepResult as AIStepResult,
-} from "ai";
-import { Buffer } from "buffer";
-import { createOllama } from "ollama-ai-provider";
-import OpenAI from "openai";
-import { encodingForModel, TiktokenModel } from "js-tiktoken";
-import { AutoTokenizer } from "@huggingface/transformers";
-import Together from "together-ai";
-import { ZodSchema } from "zod";
-import { elizaLogger } from "./index.ts";
-import { getModel, models } from "./models.ts";
-import {
- parseBooleanFromText,
- parseJsonArrayFromText,
- parseJSONObjectFromText,
- parseShouldRespondFromText,
- parseActionResponseFromText,
-} from "./parsing.ts";
-import settings from "./settings.ts";
-import {
- Content,
- IAgentRuntime,
- IImageDescriptionService,
- ITextGenerationService,
- ModelClass,
- ModelProviderName,
- ServiceType,
- SearchResponse,
- ActionResponse,
- TelemetrySettings,
- TokenizerType,
-} from "./types.ts";
-import { fal } from "@fal-ai/client";
-import { tavily } from "@tavily/core";
-
-type Tool = CoreTool;
-type StepResult = AIStepResult;
-
-/**
- * Trims the provided text context to a specified token limit using a tokenizer model and type.
- *
- * The function dynamically determines the truncation method based on the tokenizer settings
- * provided by the runtime. If no tokenizer settings are defined, it defaults to using the
- * TikToken truncation method with the "gpt-4o" model.
- *
- * @async
- * @function trimTokens
- * @param {string} context - The text to be tokenized and trimmed.
- * @param {number} maxTokens - The maximum number of tokens allowed after truncation.
- * @param {IAgentRuntime} runtime - The runtime interface providing tokenizer settings.
- *
- * @returns {Promise} A promise that resolves to the trimmed text.
- *
- * @throws {Error} Throws an error if the runtime settings are invalid or missing required fields.
- *
- * @example
- * const trimmedText = await trimTokens("This is an example text", 50, runtime);
- * console.log(trimmedText); // Output will be a truncated version of the input text.
- */
-export async function trimTokens(
- context: string,
- maxTokens: number,
- runtime: IAgentRuntime
-) {
- if (!context) return "";
- if (maxTokens <= 0) throw new Error("maxTokens must be positive");
-
- const tokenizerModel = runtime.getSetting("TOKENIZER_MODEL");
- const tokenizerType = runtime.getSetting("TOKENIZER_TYPE");
-
- if (!tokenizerModel || !tokenizerType) {
- // Default to TikToken truncation using the "gpt-4o" model if tokenizer settings are not defined
- return truncateTiktoken("gpt-4o", context, maxTokens);
- }
-
- // Choose the truncation method based on tokenizer type
- if (tokenizerType === TokenizerType.Auto) {
- return truncateAuto(tokenizerModel, context, maxTokens);
- }
-
- if (tokenizerType === TokenizerType.TikToken) {
- return truncateTiktoken(
- tokenizerModel as TiktokenModel,
- context,
- maxTokens
- );
- }
-
- elizaLogger.warn(`Unsupported tokenizer type: ${tokenizerType}`);
- return truncateTiktoken("gpt-4o", context, maxTokens);
-}
-
-async function truncateAuto(
- modelPath: string,
- context: string,
- maxTokens: number
-) {
- try {
- const tokenizer = await AutoTokenizer.from_pretrained(modelPath);
- const tokens = tokenizer.encode(context);
-
- // If already within limits, return unchanged
- if (tokens.length <= maxTokens) {
- return context;
- }
-
- // Keep the most recent tokens by slicing from the end
- const truncatedTokens = tokens.slice(-maxTokens);
-
- // Decode back to text - js-tiktoken decode() returns a string directly
- return tokenizer.decode(truncatedTokens);
- } catch (error) {
- elizaLogger.error("Error in trimTokens:", error);
- // Return truncated string if tokenization fails
- return context.slice(-maxTokens * 4); // Rough estimate of 4 chars per token
- }
-}
-
-async function truncateTiktoken(
- model: TiktokenModel,
- context: string,
- maxTokens: number
-) {
- try {
- const encoding = encodingForModel(model);
-
- // Encode the text into tokens
- const tokens = encoding.encode(context);
-
- // If already within limits, return unchanged
- if (tokens.length <= maxTokens) {
- return context;
- }
-
- // Keep the most recent tokens by slicing from the end
- const truncatedTokens = tokens.slice(-maxTokens);
-
- // Decode back to text - js-tiktoken decode() returns a string directly
- return encoding.decode(truncatedTokens);
- } catch (error) {
- elizaLogger.error("Error in trimTokens:", error);
- // Return truncated string if tokenization fails
- return context.slice(-maxTokens * 4); // Rough estimate of 4 chars per token
- }
-}
-
-/**
- * Send a message to the model for a text generateText - receive a string back and parse how you'd like
- * @param opts - The options for the generateText request.
- * @param opts.context The context of the message to be completed.
- * @param opts.stop A list of strings to stop the generateText at.
- * @param opts.model The model to use for generateText.
- * @param opts.frequency_penalty The frequency penalty to apply to the generateText.
- * @param opts.presence_penalty The presence penalty to apply to the generateText.
- * @param opts.temperature The temperature to apply to the generateText.
- * @param opts.max_context_length The maximum length of the context to apply to the generateText.
- * @returns The completed message.
- */
-
-export async function generateText({
- runtime,
- context,
- modelClass,
- tools = {},
- onStepFinish,
- maxSteps = 1,
- stop,
- customSystemPrompt,
-}: {
- runtime: IAgentRuntime;
- context: string;
- modelClass: string;
- tools?: Record;
- onStepFinish?: (event: StepResult) => Promise | void;
- maxSteps?: number;
- stop?: string[];
- customSystemPrompt?: string;
-}): Promise {
- if (!context) {
- console.error("generateText context is empty");
- return "";
- }
-
- elizaLogger.log("Generating text...");
-
- elizaLogger.info("Generating text with options:", {
- modelProvider: runtime.modelProvider,
- model: modelClass,
- });
-
- const provider = runtime.modelProvider;
- const endpoint =
- runtime.character.modelEndpointOverride || models[provider].endpoint;
- let model = models[provider].model[modelClass];
-
- // allow character.json settings => secrets to override models
- // FIXME: add MODEL_MEDIUM support
- switch (provider) {
- // if runtime.getSetting("LLAMACLOUD_MODEL_LARGE") is true and modelProvider is LLAMACLOUD, then use the large model
- case ModelProviderName.LLAMACLOUD:
- {
- switch (modelClass) {
- case ModelClass.LARGE:
- {
- model =
- runtime.getSetting("LLAMACLOUD_MODEL_LARGE") ||
- model;
- }
- break;
- case ModelClass.SMALL:
- {
- model =
- runtime.getSetting("LLAMACLOUD_MODEL_SMALL") ||
- model;
- }
- break;
- }
- }
- break;
- case ModelProviderName.TOGETHER:
- {
- switch (modelClass) {
- case ModelClass.LARGE:
- {
- model =
- runtime.getSetting("TOGETHER_MODEL_LARGE") ||
- model;
- }
- break;
- case ModelClass.SMALL:
- {
- model =
- runtime.getSetting("TOGETHER_MODEL_SMALL") ||
- model;
- }
- break;
- }
- }
- break;
- case ModelProviderName.OPENROUTER:
- {
- switch (modelClass) {
- case ModelClass.LARGE:
- {
- model =
- runtime.getSetting("LARGE_OPENROUTER_MODEL") ||
- model;
- }
- break;
- case ModelClass.SMALL:
- {
- model =
- runtime.getSetting("SMALL_OPENROUTER_MODEL") ||
- model;
- }
- break;
- }
- }
- break;
- }
-
- elizaLogger.info("Selected model:", model);
-
- const modelConfiguration = runtime.character?.settings?.modelConfig;
- const temperature =
- modelConfiguration?.temperature ||
- models[provider].settings.temperature;
- const frequency_penalty =
- modelConfiguration?.frequency_penalty ||
- models[provider].settings.frequency_penalty;
- const presence_penalty =
- modelConfiguration?.presence_penalty ||
- models[provider].settings.presence_penalty;
- const max_context_length =
- modelConfiguration?.maxInputTokens ||
- models[provider].settings.maxInputTokens;
- const max_response_length =
- modelConfiguration?.max_response_length ||
- models[provider].settings.maxOutputTokens;
- const experimental_telemetry =
- modelConfiguration?.experimental_telemetry ||
- models[provider].settings.experimental_telemetry;
-
- const apiKey = runtime.token;
-
- try {
- elizaLogger.debug(
- `Trimming context to max length of ${max_context_length} tokens.`
- );
-
- context = await trimTokens(context, max_context_length, runtime);
-
- let response: string;
-
- const _stop = stop || models[provider].settings.stop;
- elizaLogger.debug(
- `Using provider: ${provider}, model: ${model}, temperature: ${temperature}, max response length: ${max_response_length}`
- );
-
- switch (provider) {
- // OPENAI & LLAMACLOUD shared same structure.
- case ModelProviderName.OPENAI:
- case ModelProviderName.ALI_BAILIAN:
- case ModelProviderName.VOLENGINE:
- case ModelProviderName.LLAMACLOUD:
- case ModelProviderName.NANOGPT:
- case ModelProviderName.HYPERBOLIC:
- case ModelProviderName.TOGETHER:
- case ModelProviderName.AKASH_CHAT_API: {
- elizaLogger.debug("Initializing OpenAI model.");
- const openai = createOpenAI({
- apiKey,
- baseURL: endpoint,
- fetch: runtime.fetch,
- });
-
- const { text: openaiResponse } = await aiGenerateText({
- model: openai.languageModel(model),
- prompt: context,
- system:
- runtime.character.system ??
- settings.SYSTEM_PROMPT ??
- undefined,
- tools: tools,
- onStepFinish: onStepFinish,
- maxSteps: maxSteps,
- temperature: temperature,
- maxTokens: max_response_length,
- frequencyPenalty: frequency_penalty,
- presencePenalty: presence_penalty,
- experimental_telemetry: experimental_telemetry,
- });
-
- response = openaiResponse;
- elizaLogger.debug("Received response from OpenAI model.");
- break;
- }
-
- case ModelProviderName.ETERNALAI: {
- elizaLogger.debug("Initializing EternalAI model.");
- const openai = createOpenAI({
- apiKey,
- baseURL: endpoint,
- fetch: async (url: string, options: any) => {
- const fetching = await runtime.fetch(url, options);
- if (
- parseBooleanFromText(
- runtime.getSetting("ETERNALAI_LOG")
- )
- ) {
- elizaLogger.info(
- "Request data: ",
- JSON.stringify(options, null, 2)
- );
- const clonedResponse = fetching.clone();
- clonedResponse.json().then((data) => {
- elizaLogger.info(
- "Response data: ",
- JSON.stringify(data, null, 2)
- );
- });
- }
- return fetching;
- },
- });
-
- const { text: openaiResponse } = await aiGenerateText({
- model: openai.languageModel(model),
- prompt: context,
- system:
- runtime.character.system ??
- settings.SYSTEM_PROMPT ??
- undefined,
- temperature: temperature,
- maxTokens: max_response_length,
- frequencyPenalty: frequency_penalty,
- presencePenalty: presence_penalty,
- });
-
- response = openaiResponse;
- elizaLogger.debug("Received response from EternalAI model.");
- break;
- }
-
- case ModelProviderName.GOOGLE: {
- const google = createGoogleGenerativeAI({
- apiKey,
- fetch: runtime.fetch,
- });
-
- const { text: googleResponse } = await aiGenerateText({
- model: google(model),
- prompt: context,
- system:
- runtime.character.system ??
- settings.SYSTEM_PROMPT ??
- undefined,
- tools: tools,
- onStepFinish: onStepFinish,
- maxSteps: maxSteps,
- temperature: temperature,
- maxTokens: max_response_length,
- frequencyPenalty: frequency_penalty,
- presencePenalty: presence_penalty,
- experimental_telemetry: experimental_telemetry,
- });
-
- response = googleResponse;
- elizaLogger.debug("Received response from Google model.");
- break;
- }
-
- case ModelProviderName.ANTHROPIC: {
- elizaLogger.debug("Initializing Anthropic model.");
-
- const anthropic = createAnthropic({
- apiKey,
- fetch: runtime.fetch,
- });
-
- const { text: anthropicResponse } = await aiGenerateText({
- model: anthropic.languageModel(model),
- prompt: context,
- system:
- runtime.character.system ??
- settings.SYSTEM_PROMPT ??
- undefined,
- tools: tools,
- onStepFinish: onStepFinish,
- maxSteps: maxSteps,
- temperature: temperature,
- maxTokens: max_response_length,
- frequencyPenalty: frequency_penalty,
- presencePenalty: presence_penalty,
- experimental_telemetry: experimental_telemetry,
- });
-
- response = anthropicResponse;
- elizaLogger.debug("Received response from Anthropic model.");
- break;
- }
-
- case ModelProviderName.CLAUDE_VERTEX: {
- elizaLogger.debug("Initializing Claude Vertex model.");
-
- const anthropic = createAnthropic({
- apiKey,
- fetch: runtime.fetch,
- });
-
- const { text: anthropicResponse } = await aiGenerateText({
- model: anthropic.languageModel(model),
- prompt: context,
- system:
- runtime.character.system ??
- settings.SYSTEM_PROMPT ??
- undefined,
- tools: tools,
- onStepFinish: onStepFinish,
- maxSteps: maxSteps,
- temperature: temperature,
- maxTokens: max_response_length,
- frequencyPenalty: frequency_penalty,
- presencePenalty: presence_penalty,
- experimental_telemetry: experimental_telemetry,
- });
-
- response = anthropicResponse;
- elizaLogger.debug(
- "Received response from Claude Vertex model."
- );
- break;
- }
-
- case ModelProviderName.GROK: {
- elizaLogger.debug("Initializing Grok model.");
- const grok = createOpenAI({
- apiKey,
- baseURL: endpoint,
- fetch: runtime.fetch,
- });
-
- const { text: grokResponse } = await aiGenerateText({
- model: grok.languageModel(model, {
- parallelToolCalls: false,
- }),
- prompt: context,
- system:
- runtime.character.system ??
- settings.SYSTEM_PROMPT ??
- undefined,
- tools: tools,
- onStepFinish: onStepFinish,
- maxSteps: maxSteps,
- temperature: temperature,
- maxTokens: max_response_length,
- frequencyPenalty: frequency_penalty,
- presencePenalty: presence_penalty,
- experimental_telemetry: experimental_telemetry,
- });
-
- response = grokResponse;
- elizaLogger.debug("Received response from Grok model.");
- break;
- }
-
- case ModelProviderName.GROQ: {
- const groq = createGroq({ apiKey, fetch: runtime.fetch });
-
- const { text: groqResponse } = await aiGenerateText({
- model: groq.languageModel(model),
- prompt: context,
- temperature: temperature,
- system:
- runtime.character.system ??
- settings.SYSTEM_PROMPT ??
- undefined,
- tools: tools,
- onStepFinish: onStepFinish,
- maxSteps: maxSteps,
- maxTokens: max_response_length,
- frequencyPenalty: frequency_penalty,
- presencePenalty: presence_penalty,
- experimental_telemetry: experimental_telemetry,
- });
-
- response = groqResponse;
- break;
- }
-
- case ModelProviderName.LLAMALOCAL: {
- elizaLogger.debug(
- "Using local Llama model for text completion."
- );
- const textGenerationService =
- runtime.getService(
- ServiceType.TEXT_GENERATION
- );
-
- if (!textGenerationService) {
- throw new Error("Text generation service not found");
- }
-
- response = await textGenerationService.queueTextCompletion(
- context,
- temperature,
- _stop,
- frequency_penalty,
- presence_penalty,
- max_response_length
- );
- elizaLogger.debug("Received response from local Llama model.");
- break;
- }
-
- case ModelProviderName.REDPILL: {
- elizaLogger.debug("Initializing RedPill model.");
- const serverUrl = models[provider].endpoint;
- const openai = createOpenAI({
- apiKey,
- baseURL: serverUrl,
- fetch: runtime.fetch,
- });
-
- const { text: redpillResponse } = await aiGenerateText({
- model: openai.languageModel(model),
- prompt: context,
- temperature: temperature,
- system:
- runtime.character.system ??
- settings.SYSTEM_PROMPT ??
- undefined,
- tools: tools,
- onStepFinish: onStepFinish,
- maxSteps: maxSteps,
- maxTokens: max_response_length,
- frequencyPenalty: frequency_penalty,
- presencePenalty: presence_penalty,
- experimental_telemetry: experimental_telemetry,
- });
-
- response = redpillResponse;
- elizaLogger.debug("Received response from redpill model.");
- break;
- }
-
- case ModelProviderName.OPENROUTER: {
- elizaLogger.debug("Initializing OpenRouter model.");
- const serverUrl = models[provider].endpoint;
- const openrouter = createOpenAI({
- apiKey,
- baseURL: serverUrl,
- fetch: runtime.fetch,
- });
-
- const { text: openrouterResponse } = await aiGenerateText({
- model: openrouter.languageModel(model),
- prompt: context,
- temperature: temperature,
- system:
- runtime.character.system ??
- settings.SYSTEM_PROMPT ??
- undefined,
- tools: tools,
- onStepFinish: onStepFinish,
- maxSteps: maxSteps,
- maxTokens: max_response_length,
- frequencyPenalty: frequency_penalty,
- presencePenalty: presence_penalty,
- experimental_telemetry: experimental_telemetry,
- });
-
- response = openrouterResponse;
- elizaLogger.debug("Received response from OpenRouter model.");
- break;
- }
-
- case ModelProviderName.OLLAMA:
- {
- elizaLogger.debug("Initializing Ollama model.");
-
- const ollamaProvider = createOllama({
- baseURL: models[provider].endpoint + "/api",
- fetch: runtime.fetch,
- });
- const ollama = ollamaProvider(model);
-
- elizaLogger.debug("****** MODEL\n", model);
-
- const { text: ollamaResponse } = await aiGenerateText({
- model: ollama,
- prompt: context,
- tools: tools,
- onStepFinish: onStepFinish,
- temperature: temperature,
- maxSteps: maxSteps,
- maxTokens: max_response_length,
- frequencyPenalty: frequency_penalty,
- presencePenalty: presence_penalty,
- experimental_telemetry: experimental_telemetry,
- });
-
- response = ollamaResponse;
- }
- elizaLogger.debug("Received response from Ollama model.");
- break;
-
- case ModelProviderName.HEURIST: {
- elizaLogger.debug("Initializing Heurist model.");
- const heurist = createOpenAI({
- apiKey: apiKey,
- baseURL: endpoint,
- fetch: runtime.fetch,
- });
-
- const { text: heuristResponse } = await aiGenerateText({
- model: heurist.languageModel(model),
- prompt: context,
- system:
- customSystemPrompt ??
- runtime.character.system ??
- settings.SYSTEM_PROMPT ??
- undefined,
- tools: tools,
- onStepFinish: onStepFinish,
- temperature: temperature,
- maxTokens: max_response_length,
- maxSteps: maxSteps,
- frequencyPenalty: frequency_penalty,
- presencePenalty: presence_penalty,
- experimental_telemetry: experimental_telemetry,
- });
-
- response = heuristResponse;
- elizaLogger.debug("Received response from Heurist model.");
- break;
- }
- case ModelProviderName.GAIANET: {
- elizaLogger.debug("Initializing GAIANET model.");
-
- var baseURL = models[provider].endpoint;
- if (!baseURL) {
- switch (modelClass) {
- case ModelClass.SMALL:
- baseURL =
- settings.SMALL_GAIANET_SERVER_URL ||
- "https://llama3b.gaia.domains/v1";
- break;
- case ModelClass.MEDIUM:
- baseURL =
- settings.MEDIUM_GAIANET_SERVER_URL ||
- "https://llama8b.gaia.domains/v1";
- break;
- case ModelClass.LARGE:
- baseURL =
- settings.LARGE_GAIANET_SERVER_URL ||
- "https://qwen72b.gaia.domains/v1";
- break;
- }
- }
-
- elizaLogger.debug("Using GAIANET model with baseURL:", baseURL);
-
- const openai = createOpenAI({
- apiKey,
- baseURL: endpoint,
- fetch: runtime.fetch,
- });
-
- const { text: openaiResponse } = await aiGenerateText({
- model: openai.languageModel(model),
- prompt: context,
- system:
- runtime.character.system ??
- settings.SYSTEM_PROMPT ??
- undefined,
- tools: tools,
- onStepFinish: onStepFinish,
- maxSteps: maxSteps,
- temperature: temperature,
- maxTokens: max_response_length,
- frequencyPenalty: frequency_penalty,
- presencePenalty: presence_penalty,
- experimental_telemetry: experimental_telemetry,
- });
-
- response = openaiResponse;
- elizaLogger.debug("Received response from GAIANET model.");
- break;
- }
-
- case ModelProviderName.GALADRIEL: {
- elizaLogger.debug("Initializing Galadriel model.");
- const galadriel = createOpenAI({
- apiKey: apiKey,
- baseURL: endpoint,
- fetch: runtime.fetch,
- });
-
- const { text: galadrielResponse } = await aiGenerateText({
- model: galadriel.languageModel(model),
- prompt: context,
- system:
- runtime.character.system ??
- settings.SYSTEM_PROMPT ??
- undefined,
- tools: tools,
- onStepFinish: onStepFinish,
- maxSteps: maxSteps,
- temperature: temperature,
- maxTokens: max_response_length,
- frequencyPenalty: frequency_penalty,
- presencePenalty: presence_penalty,
- experimental_telemetry: experimental_telemetry,
- });
-
- response = galadrielResponse;
- elizaLogger.debug("Received response from Galadriel model.");
- break;
- }
-
- case ModelProviderName.VENICE: {
- elizaLogger.debug("Initializing Venice model.");
- const venice = createOpenAI({
- apiKey: apiKey,
- baseURL: endpoint,
- });
-
- const { text: veniceResponse } = await aiGenerateText({
- model: venice.languageModel(model),
- prompt: context,
- system:
- runtime.character.system ??
- settings.SYSTEM_PROMPT ??
- undefined,
- tools: tools,
- onStepFinish: onStepFinish,
- temperature: temperature,
- maxSteps: maxSteps,
- maxTokens: max_response_length,
- });
-
- response = veniceResponse;
- elizaLogger.debug("Received response from Venice model.");
- break;
- }
-
- case ModelProviderName.INFERA: {
- elizaLogger.debug("Initializing Infera model.");
- const apiKey = settings.INFERA_API_KEY || runtime.token;
-
- const infera = createOpenAI({
- apiKey,
- baseURL: endpoint,
- headers: {
- 'api_key': apiKey,
- 'Content-Type': 'application/json'
- }
- });
-
- const { text: inferaResponse } = await aiGenerateText({
- model: infera.languageModel(model),
- prompt: context,
- system: runtime.character.system ?? settings.SYSTEM_PROMPT ?? undefined,
- temperature: temperature,
- maxTokens: max_response_length,
- frequencyPenalty: frequency_penalty,
- presencePenalty: presence_penalty,
- });
-
- response = inferaResponse;
- elizaLogger.debug("Received response from Infera model.");
- break;
- }
-
- default: {
- const errorMessage = `Unsupported provider: ${provider}`;
- elizaLogger.error(errorMessage);
- throw new Error(errorMessage);
- }
- }
-
- return response;
- } catch (error) {
- elizaLogger.error("Error in generateText:", error);
- throw error;
- }
-}
-
-/**
- * Sends a message to the model to determine if it should respond to the given context.
- * @param opts - The options for the generateText request
- * @param opts.context The context to evaluate for response
- * @param opts.stop A list of strings to stop the generateText at
- * @param opts.model The model to use for generateText
- * @param opts.frequency_penalty The frequency penalty to apply (0.0 to 2.0)
- * @param opts.presence_penalty The presence penalty to apply (0.0 to 2.0)
- * @param opts.temperature The temperature to control randomness (0.0 to 2.0)
- * @param opts.serverUrl The URL of the API server
- * @param opts.max_context_length Maximum allowed context length in tokens
- * @param opts.max_response_length Maximum allowed response length in tokens
- * @returns Promise resolving to "RESPOND", "IGNORE", "STOP" or null
- */
-export async function generateShouldRespond({
- runtime,
- context,
- modelClass,
-}: {
- runtime: IAgentRuntime;
- context: string;
- modelClass: string;
-}): Promise<"RESPOND" | "IGNORE" | "STOP" | null> {
- let retryDelay = 1000;
- while (true) {
- try {
- elizaLogger.debug(
- "Attempting to generate text with context:",
- context
- );
- const response = await generateText({
- runtime,
- context,
- modelClass,
- });
-
- elizaLogger.debug("Received response from generateText:", response);
- const parsedResponse = parseShouldRespondFromText(response.trim());
- if (parsedResponse) {
- elizaLogger.debug("Parsed response:", parsedResponse);
- return parsedResponse;
- } else {
- elizaLogger.debug("generateShouldRespond no response");
- }
- } catch (error) {
- elizaLogger.error("Error in generateShouldRespond:", error);
- if (
- error instanceof TypeError &&
- error.message.includes("queueTextCompletion")
- ) {
- elizaLogger.error(
- "TypeError: Cannot read properties of null (reading 'queueTextCompletion')"
- );
- }
- }
-
- elizaLogger.log(`Retrying in ${retryDelay}ms...`);
- await new Promise((resolve) => setTimeout(resolve, retryDelay));
- retryDelay *= 2;
- }
-}
-
-/**
- * Splits content into chunks of specified size with optional overlapping bleed sections
- * @param content - The text content to split into chunks
- * @param chunkSize - The maximum size of each chunk in tokens
- * @param bleed - Number of characters to overlap between chunks (default: 100)
- * @returns Promise resolving to array of text chunks with bleed sections
- */
-export async function splitChunks(
- content: string,
- chunkSize: number = 512,
- bleed: number = 20
-): Promise {
- const textSplitter = new RecursiveCharacterTextSplitter({
- chunkSize: Number(chunkSize),
- chunkOverlap: Number(bleed),
- });
-
- return textSplitter.splitText(content);
-}
-
-/**
- * Sends a message to the model and parses the response as a boolean value
- * @param opts - The options for the generateText request
- * @param opts.context The context to evaluate for the boolean response
- * @param opts.stop A list of strings to stop the generateText at
- * @param opts.model The model to use for generateText
- * @param opts.frequency_penalty The frequency penalty to apply (0.0 to 2.0)
- * @param opts.presence_penalty The presence penalty to apply (0.0 to 2.0)
- * @param opts.temperature The temperature to control randomness (0.0 to 2.0)
- * @param opts.serverUrl The URL of the API server
- * @param opts.token The API token for authentication
- * @param opts.max_context_length Maximum allowed context length in tokens
- * @param opts.max_response_length Maximum allowed response length in tokens
- * @returns Promise resolving to a boolean value parsed from the model's response
- */
-export async function generateTrueOrFalse({
- runtime,
- context = "",
- modelClass,
-}: {
- runtime: IAgentRuntime;
- context: string;
- modelClass: string;
-}): Promise {
- let retryDelay = 1000;
-
- const stop = Array.from(
- new Set([
- ...(models[runtime.modelProvider].settings.stop || []),
- ["\n"],
- ])
- ) as string[];
-
- while (true) {
- try {
- const response = await generateText({
- stop,
- runtime,
- context,
- modelClass,
- });
-
- const parsedResponse = parseBooleanFromText(response.trim());
- if (parsedResponse !== null) {
- return parsedResponse;
- }
- } catch (error) {
- elizaLogger.error("Error in generateTrueOrFalse:", error);
- }
-
- await new Promise((resolve) => setTimeout(resolve, retryDelay));
- retryDelay *= 2;
- }
-}
-
-/**
- * Send a message to the model and parse the response as a string array
- * @param opts - The options for the generateText request
- * @param opts.context The context/prompt to send to the model
- * @param opts.stop Array of strings that will stop the model's generation if encountered
- * @param opts.model The language model to use
- * @param opts.frequency_penalty The frequency penalty to apply (0.0 to 2.0)
- * @param opts.presence_penalty The presence penalty to apply (0.0 to 2.0)
- * @param opts.temperature The temperature to control randomness (0.0 to 2.0)
- * @param opts.serverUrl The URL of the API server
- * @param opts.token The API token for authentication
- * @param opts.max_context_length Maximum allowed context length in tokens
- * @param opts.max_response_length Maximum allowed response length in tokens
- * @returns Promise resolving to an array of strings parsed from the model's response
- */
-export async function generateTextArray({
- runtime,
- context,
- modelClass,
-}: {
- runtime: IAgentRuntime;
- context: string;
- modelClass: string;
-}): Promise {
- if (!context) {
- elizaLogger.error("generateTextArray context is empty");
- return [];
- }
- let retryDelay = 1000;
-
- while (true) {
- try {
- const response = await generateText({
- runtime,
- context,
- modelClass,
- });
-
- const parsedResponse = parseJsonArrayFromText(response);
- if (parsedResponse) {
- return parsedResponse;
- }
- } catch (error) {
- elizaLogger.error("Error in generateTextArray:", error);
- }
-
- await new Promise((resolve) => setTimeout(resolve, retryDelay));
- retryDelay *= 2;
- }
-}
-
-export async function generateObjectDeprecated({
- runtime,
- context,
- modelClass,
-}: {
- runtime: IAgentRuntime;
- context: string;
- modelClass: string;
-}): Promise {
- if (!context) {
- elizaLogger.error("generateObjectDeprecated context is empty");
- return null;
- }
- let retryDelay = 1000;
-
- while (true) {
- try {
- // this is slightly different than generateObjectArray, in that we parse object, not object array
- const response = await generateText({
- runtime,
- context,
- modelClass,
- });
- const parsedResponse = parseJSONObjectFromText(response);
- if (parsedResponse) {
- return parsedResponse;
- }
- } catch (error) {
- elizaLogger.error("Error in generateObject:", error);
- }
-
- await new Promise((resolve) => setTimeout(resolve, retryDelay));
- retryDelay *= 2;
- }
-}
-
-export async function generateObjectArray({
- runtime,
- context,
- modelClass,
-}: {
- runtime: IAgentRuntime;
- context: string;
- modelClass: string;
-}): Promise {
- if (!context) {
- elizaLogger.error("generateObjectArray context is empty");
- return [];
- }
- let retryDelay = 1000;
-
- while (true) {
- try {
- const response = await generateText({
- runtime,
- context,
- modelClass,
- });
-
- const parsedResponse = parseJsonArrayFromText(response);
- if (parsedResponse) {
- return parsedResponse;
- }
- } catch (error) {
- elizaLogger.error("Error in generateTextArray:", error);
- }
-
- await new Promise((resolve) => setTimeout(resolve, retryDelay));
- retryDelay *= 2;
- }
-}
-
-/**
- * Send a message to the model for generateText.
- * @param opts - The options for the generateText request.
- * @param opts.context The context of the message to be completed.
- * @param opts.stop A list of strings to stop the generateText at.
- * @param opts.model The model to use for generateText.
- * @param opts.frequency_penalty The frequency penalty to apply to the generateText.
- * @param opts.presence_penalty The presence penalty to apply to the generateText.
- * @param opts.temperature The temperature to apply to the generateText.
- * @param opts.max_context_length The maximum length of the context to apply to the generateText.
- * @returns The completed message.
- */
-export async function generateMessageResponse({
- runtime,
- context,
- modelClass,
-}: {
- runtime: IAgentRuntime;
- context: string;
- modelClass: string;
-}): Promise {
- const provider = runtime.modelProvider;
- const max_context_length = models[provider].settings.maxInputTokens;
-
- context = await trimTokens(context, max_context_length, runtime);
- let retryLength = 1000; // exponential backoff
- while (true) {
- try {
- elizaLogger.log("Generating message response..");
-
- const response = await generateText({
- runtime,
- context,
- modelClass,
- });
-
- // try parsing the response as JSON, if null then try again
- const parsedContent = parseJSONObjectFromText(response) as Content;
- if (!parsedContent) {
- elizaLogger.debug("parsedContent is null, retrying");
- continue;
- }
-
- return parsedContent;
- } catch (error) {
- elizaLogger.error("ERROR:", error);
- // wait for 2 seconds
- retryLength *= 2;
- await new Promise((resolve) => setTimeout(resolve, retryLength));
- elizaLogger.debug("Retrying...");
- }
- }
-}
-
-export const generateImage = async (
- data: {
- prompt: string;
- width: number;
- height: number;
- count?: number;
- negativePrompt?: string;
- numIterations?: number;
- guidanceScale?: number;
- seed?: number;
- modelId?: string;
- jobId?: string;
- stylePreset?: string;
- hideWatermark?: boolean;
- },
- runtime: IAgentRuntime
-): Promise<{
- success: boolean;
- data?: string[];
- error?: any;
-}> => {
- const model = getModel(runtime.imageModelProvider, ModelClass.IMAGE);
- const modelSettings = models[runtime.imageModelProvider].imageSettings;
-
- elizaLogger.info("Generating image with options:", {
- imageModelProvider: model,
- });
-
- const apiKey =
- runtime.imageModelProvider === runtime.modelProvider
- ? runtime.token
- : (() => {
- // First try to match the specific provider
- switch (runtime.imageModelProvider) {
- case ModelProviderName.HEURIST:
- return runtime.getSetting("HEURIST_API_KEY");
- case ModelProviderName.TOGETHER:
- return runtime.getSetting("TOGETHER_API_KEY");
- case ModelProviderName.FAL:
- return runtime.getSetting("FAL_API_KEY");
- case ModelProviderName.OPENAI:
- return runtime.getSetting("OPENAI_API_KEY");
- case ModelProviderName.VENICE:
- return runtime.getSetting("VENICE_API_KEY");
- case ModelProviderName.LIVEPEER:
- return runtime.getSetting("LIVEPEER_GATEWAY_URL");
- default:
- // If no specific match, try the fallback chain
- return (
- runtime.getSetting("HEURIST_API_KEY") ??
- runtime.getSetting("TOGETHER_API_KEY") ??
- runtime.getSetting("FAL_API_KEY") ??
- runtime.getSetting("OPENAI_API_KEY") ??
- runtime.getSetting("VENICE_API_KEY") ??
- runtime.getSetting("LIVEPEER_GATEWAY_URL")
- );
- }
- })();
- try {
- if (runtime.imageModelProvider === ModelProviderName.HEURIST) {
- const response = await fetch(
- "http://sequencer.heurist.xyz/submit_job",
- {
- method: "POST",
- headers: {
- Authorization: `Bearer ${apiKey}`,
- "Content-Type": "application/json",
- },
- body: JSON.stringify({
- job_id: data.jobId || crypto.randomUUID(),
- model_input: {
- SD: {
- prompt: data.prompt,
- neg_prompt: data.negativePrompt,
- num_iterations: data.numIterations || 20,
- width: data.width || 512,
- height: data.height || 512,
- guidance_scale: data.guidanceScale || 3,
- seed: data.seed || -1,
- },
- },
- model_id: data.modelId || "FLUX.1-dev",
- deadline: 60,
- priority: 1,
- }),
- }
- );
-
- if (!response.ok) {
- throw new Error(
- `Heurist image generation failed: ${response.statusText}`
- );
- }
-
- const imageURL = await response.json();
- return { success: true, data: [imageURL] };
- } else if (
- runtime.imageModelProvider === ModelProviderName.TOGETHER ||
- // for backwards compat
- runtime.imageModelProvider === ModelProviderName.LLAMACLOUD
- ) {
- const together = new Together({ apiKey: apiKey as string });
- const response = await together.images.create({
- model: "black-forest-labs/FLUX.1-schnell",
- prompt: data.prompt,
- width: data.width,
- height: data.height,
- steps: modelSettings?.steps ?? 4,
- n: data.count,
- });
-
- // Add type assertion to handle the response properly
- const togetherResponse =
- response as unknown as TogetherAIImageResponse;
-
- if (
- !togetherResponse.data ||
- !Array.isArray(togetherResponse.data)
- ) {
- throw new Error("Invalid response format from Together AI");
- }
-
- // Rest of the code remains the same...
- const base64s = await Promise.all(
- togetherResponse.data.map(async (image) => {
- if (!image.url) {
- elizaLogger.error("Missing URL in image data:", image);
- throw new Error("Missing URL in Together AI response");
- }
-
- // Fetch the image from the URL
- const imageResponse = await fetch(image.url);
- if (!imageResponse.ok) {
- throw new Error(
- `Failed to fetch image: ${imageResponse.statusText}`
- );
- }
-
- // Convert to blob and then to base64
- const blob = await imageResponse.blob();
- const arrayBuffer = await blob.arrayBuffer();
- const base64 = Buffer.from(arrayBuffer).toString("base64");
-
- // Return with proper MIME type
- return `data:image/jpeg;base64,${base64}`;
- })
- );
-
- if (base64s.length === 0) {
- throw new Error("No images generated by Together AI");
- }
-
- elizaLogger.debug(`Generated ${base64s.length} images`);
- return { success: true, data: base64s };
- } else if (runtime.imageModelProvider === ModelProviderName.FAL) {
- fal.config({
- credentials: apiKey as string,
- });
-
- // Prepare the input parameters according to their schema
- const input = {
- prompt: data.prompt,
- image_size: "square" as const,
- num_inference_steps: modelSettings?.steps ?? 50,
- guidance_scale: data.guidanceScale || 3.5,
- num_images: data.count,
- enable_safety_checker:
- runtime.getSetting("FAL_AI_ENABLE_SAFETY_CHECKER") ===
- "true",
- safety_tolerance: Number(
- runtime.getSetting("FAL_AI_SAFETY_TOLERANCE") || "2"
- ),
- output_format: "png" as const,
- seed: data.seed ?? 6252023,
- ...(runtime.getSetting("FAL_AI_LORA_PATH")
- ? {
- loras: [
- {
- path: runtime.getSetting("FAL_AI_LORA_PATH"),
- scale: 1,
- },
- ],
- }
- : {}),
- };
-
- // Subscribe to the model
- const result = await fal.subscribe(model, {
- input,
- logs: true,
- onQueueUpdate: (update) => {
- if (update.status === "IN_PROGRESS") {
- elizaLogger.info(update.logs.map((log) => log.message));
- }
- },
- });
-
- // Convert the returned image URLs to base64 to match existing functionality
- const base64Promises = result.data.images.map(async (image) => {
- const response = await fetch(image.url);
- const blob = await response.blob();
- const buffer = await blob.arrayBuffer();
- const base64 = Buffer.from(buffer).toString("base64");
- return `data:${image.content_type};base64,${base64}`;
- });
-
- const base64s = await Promise.all(base64Promises);
- return { success: true, data: base64s };
- } else if (runtime.imageModelProvider === ModelProviderName.VENICE) {
- const response = await fetch(
- "https://api.venice.ai/api/v1/image/generate",
- {
- method: "POST",
- headers: {
- Authorization: `Bearer ${apiKey}`,
- "Content-Type": "application/json",
- },
- body: JSON.stringify({
- model: data.modelId || "fluently-xl",
- prompt: data.prompt,
- negative_prompt: data.negativePrompt,
- width: data.width,
- height: data.height,
- steps: data.numIterations,
- seed: data.seed,
- style_preset: data.stylePreset,
- hide_watermark: data.hideWatermark,
- }),
- }
- );
-
- const result = await response.json();
-
- if (!result.images || !Array.isArray(result.images)) {
- throw new Error("Invalid response format from Venice AI");
- }
-
- const base64s = result.images.map((base64String) => {
- if (!base64String) {
- throw new Error(
- "Empty base64 string in Venice AI response"
- );
- }
- return `data:image/png;base64,${base64String}`;
- });
-
- return { success: true, data: base64s };
- } else if (runtime.imageModelProvider === ModelProviderName.LIVEPEER) {
- if (!apiKey) {
- throw new Error("Livepeer Gateway is not defined");
- }
- try {
- const baseUrl = new URL(apiKey);
- if (!baseUrl.protocol.startsWith("http")) {
- throw new Error("Invalid Livepeer Gateway URL protocol");
- }
- const response = await fetch(
- `${baseUrl.toString()}text-to-image`,
- {
- method: "POST",
- headers: {
- "Content-Type": "application/json",
- },
- body: JSON.stringify({
- model_id:
- data.modelId || "ByteDance/SDXL-Lightning",
- prompt: data.prompt,
- width: data.width || 1024,
- height: data.height || 1024,
- }),
- }
- );
- const result = await response.json();
- if (!result.images?.length) {
- throw new Error("No images generated");
- }
- const base64Images = await Promise.all(
- result.images.map(async (image) => {
- console.log("imageUrl console log", image.url);
- let imageUrl;
- if (image.url.includes("http")) {
- imageUrl = image.url;
- } else {
- imageUrl = `${apiKey}${image.url}`;
- }
- const imageResponse = await fetch(imageUrl);
- if (!imageResponse.ok) {
- throw new Error(
- `Failed to fetch image: ${imageResponse.statusText}`
- );
- }
- const blob = await imageResponse.blob();
- const arrayBuffer = await blob.arrayBuffer();
- const base64 =
- Buffer.from(arrayBuffer).toString("base64");
- return `data:image/jpeg;base64,${base64}`;
- })
- );
- return {
- success: true,
- data: base64Images,
- };
- } catch (error) {
- console.error(error);
- return { success: false, error: error };
- }
- } else {
- let targetSize = `${data.width}x${data.height}`;
- if (
- targetSize !== "1024x1024" &&
- targetSize !== "1792x1024" &&
- targetSize !== "1024x1792"
- ) {
- targetSize = "1024x1024";
- }
- const openaiApiKey = runtime.getSetting("OPENAI_API_KEY") as string;
- if (!openaiApiKey) {
- throw new Error("OPENAI_API_KEY is not set");
- }
- const openai = new OpenAI({
- apiKey: openaiApiKey as string,
- });
- const response = await openai.images.generate({
- model,
- prompt: data.prompt,
- size: targetSize as "1024x1024" | "1792x1024" | "1024x1792",
- n: data.count,
- response_format: "b64_json",
- });
- const base64s = response.data.map(
- (image) => `data:image/png;base64,${image.b64_json}`
- );
- return { success: true, data: base64s };
- }
- } catch (error) {
- console.error(error);
- return { success: false, error: error };
- }
-};
-
-export const generateCaption = async (
- data: { imageUrl: string },
- runtime: IAgentRuntime
-): Promise<{
- title: string;
- description: string;
-}> => {
- const { imageUrl } = data;
- const imageDescriptionService =
- runtime.getService(
- ServiceType.IMAGE_DESCRIPTION
- );
-
- if (!imageDescriptionService) {
- throw new Error("Image description service not found");
- }
-
- const resp = await imageDescriptionService.describeImage(imageUrl);
- return {
- title: resp.title.trim(),
- description: resp.description.trim(),
- };
-};
-
-export const generateWebSearch = async (
- query: string,
- runtime: IAgentRuntime
-): Promise => {
- try {
- const apiKey = runtime.getSetting("TAVILY_API_KEY") as string;
- if (!apiKey) {
- throw new Error("TAVILY_API_KEY is not set");
- }
- const tvly = tavily({ apiKey });
- const response = await tvly.search(query, {
- includeAnswer: true,
- maxResults: 3, // 5 (default)
- topic: "general", // "general"(default) "news"
- searchDepth: "basic", // "basic"(default) "advanced"
- includeImages: false, // false (default) true
- });
- return response;
- } catch (error) {
- elizaLogger.error("Error:", error);
- }
-};
-/**
- * Configuration options for generating objects with a model.
- */
-export interface GenerationOptions {
- runtime: IAgentRuntime;
- context: string;
- modelClass: ModelClass;
- schema?: ZodSchema;
- schemaName?: string;
- schemaDescription?: string;
- stop?: string[];
- mode?: "auto" | "json" | "tool";
- experimental_providerMetadata?: Record;
-}
-
-/**
- * Base settings for model generation.
- */
-interface ModelSettings {
- prompt: string;
- temperature: number;
- maxTokens: number;
- frequencyPenalty: number;
- presencePenalty: number;
- stop?: string[];
- experimental_telemetry?: TelemetrySettings;
-}
-
-/**
- * Generates structured objects from a prompt using specified AI models and configuration options.
- *
- * @param {GenerationOptions} options - Configuration options for generating objects.
- * @returns {Promise} - A promise that resolves to an array of generated objects.
- * @throws {Error} - Throws an error if the provider is unsupported or if generation fails.
- */
-export const generateObject = async ({
- runtime,
- context,
- modelClass,
- schema,
- schemaName,
- schemaDescription,
- stop,
- mode = "json",
-}: GenerationOptions): Promise> => {
- if (!context) {
- const errorMessage = "generateObject context is empty";
- console.error(errorMessage);
- throw new Error(errorMessage);
- }
-
- const provider = runtime.modelProvider;
- const model = models[provider].model[modelClass];
- const temperature = models[provider].settings.temperature;
- const frequency_penalty = models[provider].settings.frequency_penalty;
- const presence_penalty = models[provider].settings.presence_penalty;
- const max_context_length = models[provider].settings.maxInputTokens;
- const max_response_length = models[provider].settings.maxOutputTokens;
- const experimental_telemetry =
- models[provider].settings.experimental_telemetry;
- const apiKey = runtime.token;
-
- try {
- context = await trimTokens(context, max_context_length, runtime);
-
- const modelOptions: ModelSettings = {
- prompt: context,
- temperature,
- maxTokens: max_response_length,
- frequencyPenalty: frequency_penalty,
- presencePenalty: presence_penalty,
- stop: stop || models[provider].settings.stop,
- experimental_telemetry: experimental_telemetry,
- };
-
- const response = await handleProvider({
- provider,
- model,
- apiKey,
- schema,
- schemaName,
- schemaDescription,
- mode,
- modelOptions,
- runtime,
- context,
- modelClass,
- });
-
- return response;
- } catch (error) {
- console.error("Error in generateObject:", error);
- throw error;
- }
-};
-
-/**
- * Interface for provider-specific generation options.
- */
-interface ProviderOptions {
- runtime: IAgentRuntime;
- provider: ModelProviderName;
- model: any;
- apiKey: string;
- schema?: ZodSchema;
- schemaName?: string;
- schemaDescription?: string;
- mode?: "auto" | "json" | "tool";
- experimental_providerMetadata?: Record;
- modelOptions: ModelSettings;
- modelClass: string;
- context: string;
-}
-
-/**
- * Handles AI generation based on the specified provider.
- *
- * @param {ProviderOptions} options - Configuration options specific to the provider.
- * @returns {Promise} - A promise that resolves to an array of generated objects.
- */
-export async function handleProvider(
- options: ProviderOptions
-): Promise> {
- const { provider, runtime, context, modelClass } = options;
- switch (provider) {
- case ModelProviderName.OPENAI:
- case ModelProviderName.ETERNALAI:
- case ModelProviderName.ALI_BAILIAN:
- case ModelProviderName.VOLENGINE:
- case ModelProviderName.LLAMACLOUD:
- case ModelProviderName.TOGETHER:
- case ModelProviderName.NANOGPT:
- case ModelProviderName.INFERA:
- case ModelProviderName.AKASH_CHAT_API:
- return await handleOpenAI(options);
- case ModelProviderName.ANTHROPIC:
- case ModelProviderName.CLAUDE_VERTEX:
- return await handleAnthropic(options);
- case ModelProviderName.GROK:
- return await handleGrok(options);
- case ModelProviderName.GROQ:
- return await handleGroq(options);
- case ModelProviderName.LLAMALOCAL:
- return await generateObjectDeprecated({
- runtime,
- context,
- modelClass,
- });
- case ModelProviderName.GOOGLE:
- return await handleGoogle(options);
- case ModelProviderName.REDPILL:
- return await handleRedPill(options);
- case ModelProviderName.OPENROUTER:
- return await handleOpenRouter(options);
- case ModelProviderName.OLLAMA:
- return await handleOllama(options);
- default: {
- const errorMessage = `Unsupported provider: ${provider}`;
- elizaLogger.error(errorMessage);
- throw new Error(errorMessage);
- }
- }
-}
-/**
- * Handles object generation for OpenAI.
- *
- * @param {ProviderOptions} options - Options specific to OpenAI.
- * @returns {Promise>} - A promise that resolves to generated objects.
- */
-async function handleOpenAI({
- model,
- apiKey,
- schema,
- schemaName,
- schemaDescription,
- mode,
- modelOptions,
-}: ProviderOptions): Promise> {
- const baseURL = models.openai.endpoint || undefined;
- const openai = createOpenAI({ apiKey, baseURL });
- return await aiGenerateObject({
- model: openai.languageModel(model),
- schema,
- schemaName,
- schemaDescription,
- mode,
- ...modelOptions,
- });
-}
-
-/**
- * Handles object generation for Anthropic models.
- *
- * @param {ProviderOptions} options - Options specific to Anthropic.
- * @returns {Promise>} - A promise that resolves to generated objects.
- */
-async function handleAnthropic({
- model,
- apiKey,
- schema,
- schemaName,
- schemaDescription,
- mode,
- modelOptions,
-}: ProviderOptions): Promise> {
- const anthropic = createAnthropic({ apiKey });
- return await aiGenerateObject({
- model: anthropic.languageModel(model),
- schema,
- schemaName,
- schemaDescription,
- mode,
- ...modelOptions,
- });
-}
-
-/**
- * Handles object generation for Grok models.
- *
- * @param {ProviderOptions} options - Options specific to Grok.
- * @returns {Promise>} - A promise that resolves to generated objects.
- */
-async function handleGrok({
- model,
- apiKey,
- schema,
- schemaName,
- schemaDescription,
- mode,
- modelOptions,
-}: ProviderOptions): Promise> {
- const grok = createOpenAI({ apiKey, baseURL: models.grok.endpoint });
- return await aiGenerateObject({
- model: grok.languageModel(model, { parallelToolCalls: false }),
- schema,
- schemaName,
- schemaDescription,
- mode,
- ...modelOptions,
- });
-}
-
-/**
- * Handles object generation for Groq models.
- *
- * @param {ProviderOptions} options - Options specific to Groq.
- * @returns {Promise>} - A promise that resolves to generated objects.
- */
-async function handleGroq({
- model,
- apiKey,
- schema,
- schemaName,
- schemaDescription,
- mode,
- modelOptions,
-}: ProviderOptions): Promise> {
- const groq = createGroq({ apiKey });
- return await aiGenerateObject({
- model: groq.languageModel(model),
- schema,
- schemaName,
- schemaDescription,
- mode,
- ...modelOptions,
- });
-}
-
-/**
- * Handles object generation for Google models.
- *
- * @param {ProviderOptions} options - Options specific to Google.
- * @returns {Promise>} - A promise that resolves to generated objects.
- */
-async function handleGoogle({
- model,
- apiKey: _apiKey,
- schema,
- schemaName,
- schemaDescription,
- mode,
- modelOptions,
-}: ProviderOptions): Promise> {
- const google = createGoogleGenerativeAI();
- return await aiGenerateObject({
- model: google(model),
- schema,
- schemaName,
- schemaDescription,
- mode,
- ...modelOptions,
- });
-}
-
-/**
- * Handles object generation for Redpill models.
- *
- * @param {ProviderOptions} options - Options specific to Redpill.
- * @returns {Promise>} - A promise that resolves to generated objects.
- */
-async function handleRedPill({
- model,
- apiKey,
- schema,
- schemaName,
- schemaDescription,
- mode,
- modelOptions,
-}: ProviderOptions): Promise> {
- const redPill = createOpenAI({ apiKey, baseURL: models.redpill.endpoint });
- return await aiGenerateObject({
- model: redPill.languageModel(model),
- schema,
- schemaName,
- schemaDescription,
- mode,
- ...modelOptions,
- });
-}
-
-/**
- * Handles object generation for OpenRouter models.
- *
- * @param {ProviderOptions} options - Options specific to OpenRouter.
- * @returns {Promise>} - A promise that resolves to generated objects.
- */
-async function handleOpenRouter({
- model,
- apiKey,
- schema,
- schemaName,
- schemaDescription,
- mode,
- modelOptions,
-}: ProviderOptions): Promise> {
- const openRouter = createOpenAI({
- apiKey,
- baseURL: models.openrouter.endpoint,
- });
- return await aiGenerateObject({
- model: openRouter.languageModel(model),
- schema,
- schemaName,
- schemaDescription,
- mode,
- ...modelOptions,
- });
-}
-
-/**
- * Handles object generation for Ollama models.
- *
- * @param {ProviderOptions} options - Options specific to Ollama.
- * @returns {Promise>} - A promise that resolves to generated objects.
- */
-async function handleOllama({
- model,
- schema,
- schemaName,
- schemaDescription,
- mode,
- modelOptions,
- provider,
-}: ProviderOptions): Promise> {
- const ollamaProvider = createOllama({
- baseURL: models[provider].endpoint + "/api",
- });
- const ollama = ollamaProvider(model);
- return await aiGenerateObject({
- model: ollama,
- schema,
- schemaName,
- schemaDescription,
- mode,
- ...modelOptions,
- });
-}
-
-// Add type definition for Together AI response
-interface TogetherAIImageResponse {
- data: Array<{
- url: string;
- content_type?: string;
- image_type?: string;
- }>;
-}
-
-export async function generateTweetActions({
- runtime,
- context,
- modelClass,
-}: {
- runtime: IAgentRuntime;
- context: string;
- modelClass: string;
-}): Promise {
- let retryDelay = 1000;
- while (true) {
- try {
- const response = await generateText({
- runtime,
- context,
- modelClass,
- });
- console.debug(
- "Received response from generateText for tweet actions:",
- response
- );
- const { actions } = parseActionResponseFromText(response.trim());
- if (actions) {
- console.debug("Parsed tweet actions:", actions);
- return actions;
- } else {
- elizaLogger.debug("generateTweetActions no valid response");
- }
- } catch (error) {
- elizaLogger.error("Error in generateTweetActions:", error);
- if (
- error instanceof TypeError &&
- error.message.includes("queueTextCompletion")
- ) {
- elizaLogger.error(
- "TypeError: Cannot read properties of null (reading 'queueTextCompletion')"
- );
- }
- }
- elizaLogger.log(`Retrying in ${retryDelay}ms...`);
- await new Promise((resolve) => setTimeout(resolve, retryDelay));
- retryDelay *= 2;
- }
-}
diff --git a/packages/core/models.ts b/packages/core/models.ts
deleted file mode 100644
index 67269b49d37..00000000000
--- a/packages/core/models.ts
+++ /dev/null
@@ -1,542 +0,0 @@
-import settings from "./settings.ts";
-import { Models, ModelProviderName, ModelClass } from "./types.ts";
-
-export const models: Models = {
- [ModelProviderName.OPENAI]: {
- endpoint: settings.OPENAI_API_URL || "https://api.openai.com/v1",
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- frequency_penalty: 0.0,
- presence_penalty: 0.0,
- temperature: 0.6,
- },
- model: {
- [ModelClass.SMALL]: settings.SMALL_OPENAI_MODEL || "gpt-4o-mini",
- [ModelClass.MEDIUM]: settings.MEDIUM_OPENAI_MODEL || "gpt-4o",
- [ModelClass.LARGE]: settings.LARGE_OPENAI_MODEL || "gpt-4o",
- [ModelClass.EMBEDDING]: settings.EMBEDDING_OPENAI_MODEL || "text-embedding-3-small",
- [ModelClass.IMAGE]: settings.IMAGE_OPENAI_MODEL || "dall-e-3",
- },
- },
- [ModelProviderName.ETERNALAI]: {
- endpoint: settings.ETERNALAI_URL,
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- frequency_penalty: 0.0,
- presence_penalty: 0.0,
- temperature: 0.6,
- },
- model: {
- [ModelClass.SMALL]:
- settings.ETERNALAI_MODEL ||
- "neuralmagic/Meta-Llama-3.1-405B-Instruct-quantized.w4a16",
- [ModelClass.MEDIUM]:
- settings.ETERNALAI_MODEL ||
- "neuralmagic/Meta-Llama-3.1-405B-Instruct-quantized.w4a16",
- [ModelClass.LARGE]:
- settings.ETERNALAI_MODEL ||
- "neuralmagic/Meta-Llama-3.1-405B-Instruct-quantized.w4a16",
- [ModelClass.EMBEDDING]: "",
- [ModelClass.IMAGE]: "",
- },
- },
- [ModelProviderName.ANTHROPIC]: {
- settings: {
- stop: [],
- maxInputTokens: 200000,
- maxOutputTokens: 4096,
- frequency_penalty: 0.4,
- presence_penalty: 0.4,
- temperature: 0.7,
- },
- endpoint: "https://api.anthropic.com/v1",
- model: {
- [ModelClass.SMALL]: settings.SMALL_ANTHROPIC_MODEL || "claude-3-haiku-20240307",
- [ModelClass.MEDIUM]: settings.MEDIUM_ANTHROPIC_MODEL || "claude-3-5-sonnet-20241022",
- [ModelClass.LARGE]: settings.LARGE_ANTHROPIC_MODEL || "claude-3-5-sonnet-20241022",
- },
- },
- [ModelProviderName.CLAUDE_VERTEX]: {
- settings: {
- stop: [],
- maxInputTokens: 200000,
- maxOutputTokens: 8192,
- frequency_penalty: 0.4,
- presence_penalty: 0.4,
- temperature: 0.7,
- },
- endpoint: "https://api.anthropic.com/v1", // TODO: check
- model: {
- [ModelClass.SMALL]: "claude-3-5-sonnet-20241022",
- [ModelClass.MEDIUM]: "claude-3-5-sonnet-20241022",
- [ModelClass.LARGE]: "claude-3-opus-20240229",
- },
- },
- [ModelProviderName.GROK]: {
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- frequency_penalty: 0.4,
- presence_penalty: 0.4,
- temperature: 0.7,
- },
- endpoint: "https://api.x.ai/v1",
- model: {
- [ModelClass.SMALL]: settings.SMALL_GROK_MODEL || "grok-2-1212",
- [ModelClass.MEDIUM]: settings.MEDIUM_GROK_MODEL || "grok-2-1212",
- [ModelClass.LARGE]: settings.LARGE_GROK_MODEL || "grok-2-1212",
- [ModelClass.EMBEDDING]: settings.EMBEDDING_GROK_MODEL || "grok-2-1212", // not sure about this one
- },
- },
- [ModelProviderName.GROQ]: {
- endpoint: "https://api.groq.com/openai/v1",
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8000,
- frequency_penalty: 0.4,
- presence_penalty: 0.4,
- temperature: 0.7,
- },
- model: {
- [ModelClass.SMALL]:
- settings.SMALL_GROQ_MODEL || "llama-3.1-8b-instant",
- [ModelClass.MEDIUM]:
- settings.MEDIUM_GROQ_MODEL || "llama-3.3-70b-versatile",
- [ModelClass.LARGE]:
- settings.LARGE_GROQ_MODEL || "llama-3.2-90b-vision-preview",
- [ModelClass.EMBEDDING]:
- settings.EMBEDDING_GROQ_MODEL || "llama-3.1-8b-instant",
- },
- },
- [ModelProviderName.LLAMACLOUD]: {
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- repetition_penalty: 0.4,
- temperature: 0.7,
- },
- imageSettings: {
- steps: 4,
- },
- endpoint: "https://api.llamacloud.com/v1",
- model: {
- [ModelClass.SMALL]: "meta-llama/Llama-3.2-3B-Instruct-Turbo",
- [ModelClass.MEDIUM]: "meta-llama-3.1-8b-instruct",
- [ModelClass.LARGE]: "meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo",
- [ModelClass.EMBEDDING]:
- "togethercomputer/m2-bert-80M-32k-retrieval",
- [ModelClass.IMAGE]: "black-forest-labs/FLUX.1-schnell",
- },
- },
- [ModelProviderName.TOGETHER]: {
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- repetition_penalty: 0.4,
- temperature: 0.7,
- },
- imageSettings: {
- steps: 4,
- },
- endpoint: "https://api.together.ai/v1",
- model: {
- [ModelClass.SMALL]: "meta-llama/Llama-3.2-3B-Instruct-Turbo",
- [ModelClass.MEDIUM]: "meta-llama-3.1-8b-instruct",
- [ModelClass.LARGE]: "meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo",
- [ModelClass.EMBEDDING]:
- "togethercomputer/m2-bert-80M-32k-retrieval",
- [ModelClass.IMAGE]: "black-forest-labs/FLUX.1-schnell",
- },
- },
- [ModelProviderName.LLAMALOCAL]: {
- settings: {
- stop: ["<|eot_id|>", "<|eom_id|>"],
- maxInputTokens: 32768,
- maxOutputTokens: 8192,
- repetition_penalty: 0.4,
- temperature: 0.7,
- },
- model: {
- [ModelClass.SMALL]:
- "NousResearch/Hermes-3-Llama-3.1-8B-GGUF/resolve/main/Hermes-3-Llama-3.1-8B.Q8_0.gguf?download=true",
- [ModelClass.MEDIUM]:
- "NousResearch/Hermes-3-Llama-3.1-8B-GGUF/resolve/main/Hermes-3-Llama-3.1-8B.Q8_0.gguf?download=true", // TODO: ?download=true
- [ModelClass.LARGE]:
- "NousResearch/Hermes-3-Llama-3.1-8B-GGUF/resolve/main/Hermes-3-Llama-3.1-8B.Q8_0.gguf?download=true",
- // "RichardErkhov/NousResearch_-_Meta-Llama-3.1-70B-gguf", // TODO:
- [ModelClass.EMBEDDING]:
- "togethercomputer/m2-bert-80M-32k-retrieval",
- },
- },
- [ModelProviderName.GOOGLE]: {
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- frequency_penalty: 0.4,
- presence_penalty: 0.4,
- temperature: 0.7,
- },
- model: {
- [ModelClass.SMALL]:
- settings.SMALL_GOOGLE_MODEL ||
- settings.GOOGLE_MODEL ||
- "gemini-2.0-flash-exp",
- [ModelClass.MEDIUM]:
- settings.MEDIUM_GOOGLE_MODEL ||
- settings.GOOGLE_MODEL ||
- "gemini-2.0-flash-exp",
- [ModelClass.LARGE]:
- settings.LARGE_GOOGLE_MODEL ||
- settings.GOOGLE_MODEL ||
- "gemini-2.0-flash-exp",
- [ModelClass.EMBEDDING]:
- settings.EMBEDDING_GOOGLE_MODEL ||
- settings.GOOGLE_MODEL ||
- "text-embedding-004",
- },
- },
- [ModelProviderName.REDPILL]: {
- endpoint: "https://api.red-pill.ai/v1",
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- frequency_penalty: 0.0,
- presence_penalty: 0.0,
- temperature: 0.6,
- },
- // Available models: https://docs.red-pill.ai/get-started/supported-models
- // To test other models, change the models below
- model: {
- [ModelClass.SMALL]:
- settings.SMALL_REDPILL_MODEL ||
- settings.REDPILL_MODEL ||
- "gpt-4o-mini",
- [ModelClass.MEDIUM]:
- settings.MEDIUM_REDPILL_MODEL ||
- settings.REDPILL_MODEL ||
- "gpt-4o",
- [ModelClass.LARGE]:
- settings.LARGE_REDPILL_MODEL ||
- settings.REDPILL_MODEL ||
- "gpt-4o",
- [ModelClass.EMBEDDING]: "text-embedding-3-small",
- },
- },
- [ModelProviderName.OPENROUTER]: {
- endpoint: "https://openrouter.ai/api/v1",
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- frequency_penalty: 0.4,
- presence_penalty: 0.4,
- temperature: 0.7,
- },
- // Available models: https://openrouter.ai/models
- // To test other models, change the models below
- model: {
- [ModelClass.SMALL]:
- settings.SMALL_OPENROUTER_MODEL ||
- settings.OPENROUTER_MODEL ||
- "nousresearch/hermes-3-llama-3.1-405b",
- [ModelClass.MEDIUM]:
- settings.MEDIUM_OPENROUTER_MODEL ||
- settings.OPENROUTER_MODEL ||
- "nousresearch/hermes-3-llama-3.1-405b",
- [ModelClass.LARGE]:
- settings.LARGE_OPENROUTER_MODEL ||
- settings.OPENROUTER_MODEL ||
- "nousresearch/hermes-3-llama-3.1-405b",
- [ModelClass.EMBEDDING]: "text-embedding-3-small",
- },
- },
- [ModelProviderName.OLLAMA]: {
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- frequency_penalty: 0.4,
- presence_penalty: 0.4,
- temperature: 0.7,
- },
- endpoint: settings.OLLAMA_SERVER_URL || "http://localhost:11434",
- model: {
- [ModelClass.SMALL]:
- settings.SMALL_OLLAMA_MODEL ||
- settings.OLLAMA_MODEL ||
- "llama3.2",
- [ModelClass.MEDIUM]:
- settings.MEDIUM_OLLAMA_MODEL ||
- settings.OLLAMA_MODEL ||
- "hermes3",
- [ModelClass.LARGE]:
- settings.LARGE_OLLAMA_MODEL ||
- settings.OLLAMA_MODEL ||
- "hermes3:70b",
- [ModelClass.EMBEDDING]:
- settings.OLLAMA_EMBEDDING_MODEL || "mxbai-embed-large",
- },
- },
- [ModelProviderName.HEURIST]: {
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- repetition_penalty: 0.4,
- temperature: 0.7,
- },
- imageSettings: {
- steps: 20,
- },
- endpoint: "https://llm-gateway.heurist.xyz",
- model: {
- [ModelClass.SMALL]:
- settings.SMALL_HEURIST_MODEL ||
- "meta-llama/llama-3-70b-instruct",
- [ModelClass.MEDIUM]:
- settings.MEDIUM_HEURIST_MODEL ||
- "meta-llama/llama-3-70b-instruct",
- [ModelClass.LARGE]:
- settings.LARGE_HEURIST_MODEL ||
- "meta-llama/llama-3.1-405b-instruct",
- [ModelClass.EMBEDDING]: "", //Add later,
- [ModelClass.IMAGE]: settings.HEURIST_IMAGE_MODEL || "PepeXL",
- },
- },
- [ModelProviderName.GALADRIEL]: {
- endpoint: "https://api.galadriel.com/v1",
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- frequency_penalty: 0.5,
- presence_penalty: 0.5,
- temperature: 0.8,
- },
- model: {
- [ModelClass.SMALL]: "llama3.1:70b",
- [ModelClass.MEDIUM]: "llama3.1:70b",
- [ModelClass.LARGE]: "llama3.1:405b",
- [ModelClass.EMBEDDING]: "gte-large-en-v1.5",
- [ModelClass.IMAGE]: "stabilityai/stable-diffusion-xl-base-1.0",
- },
- },
- [ModelProviderName.FAL]: {
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- repetition_penalty: 0.4,
- temperature: 0.7,
- },
- imageSettings: {
- steps: 28,
- },
- endpoint: "https://api.fal.ai/v1",
- model: {
- [ModelClass.SMALL]: "", // FAL doesn't provide text models
- [ModelClass.MEDIUM]: "",
- [ModelClass.LARGE]: "",
- [ModelClass.EMBEDDING]: "",
- [ModelClass.IMAGE]: "fal-ai/flux-lora",
- },
- },
- [ModelProviderName.GAIANET]: {
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- repetition_penalty: 0.4,
- temperature: 0.7,
- },
- endpoint: settings.GAIANET_SERVER_URL,
- model: {
- [ModelClass.SMALL]:
- settings.GAIANET_MODEL ||
- settings.SMALL_GAIANET_MODEL ||
- "llama3b",
- [ModelClass.MEDIUM]:
- settings.GAIANET_MODEL ||
- settings.MEDIUM_GAIANET_MODEL ||
- "llama",
- [ModelClass.LARGE]:
- settings.GAIANET_MODEL ||
- settings.LARGE_GAIANET_MODEL ||
- "qwen72b",
- [ModelClass.EMBEDDING]:
- settings.GAIANET_EMBEDDING_MODEL || "nomic-embed",
- },
- },
- [ModelProviderName.ALI_BAILIAN]: {
- endpoint: "https://dashscope.aliyuncs.com/compatible-mode/v1",
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- frequency_penalty: 0.4,
- presence_penalty: 0.4,
- temperature: 0.6,
- },
- model: {
- [ModelClass.SMALL]: "qwen-turbo",
- [ModelClass.MEDIUM]: "qwen-plus",
- [ModelClass.LARGE]: "qwen-max",
- [ModelClass.IMAGE]: "wanx-v1",
- },
- },
- [ModelProviderName.VOLENGINE]: {
- endpoint: settings.VOLENGINE_API_URL || "https://open.volcengineapi.com/api/v3/",
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- frequency_penalty: 0.4,
- presence_penalty: 0.4,
- temperature: 0.6,
- },
- model: {
- [ModelClass.SMALL]:
- settings.SMALL_VOLENGINE_MODEL ||
- settings.VOLENGINE_MODEL ||
- "doubao-lite-128k",
- [ModelClass.MEDIUM]:
- settings.MEDIUM_VOLENGINE_MODEL ||
- settings.VOLENGINE_MODEL ||
- "doubao-pro-128k",
- [ModelClass.LARGE]:
- settings.LARGE_VOLENGINE_MODEL ||
- settings.VOLENGINE_MODEL ||
- "doubao-pro-256k",
- [ModelClass.EMBEDDING]:
- settings.VOLENGINE_EMBEDDING_MODEL ||
- "doubao-embedding",
- },
- },
- [ModelProviderName.NANOGPT]: {
- endpoint: "https://nano-gpt.com/api/v1",
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- frequency_penalty: 0.0,
- presence_penalty: 0.0,
- temperature: 0.6,
- },
- model: {
- [ModelClass.SMALL]: settings.SMALL_NANOGPT_MODEL || "gpt-4o-mini",
- [ModelClass.MEDIUM]: settings.MEDIUM_NANOGPT_MODEL || "gpt-4o",
- [ModelClass.LARGE]: settings.LARGE_NANOGPT_MODEL || "gpt-4o",
- }
- },
- [ModelProviderName.HYPERBOLIC]: {
- endpoint: "https://api.hyperbolic.xyz/v1",
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- temperature: 0.6,
- },
- model: {
- [ModelClass.SMALL]:
- settings.SMALL_HYPERBOLIC_MODEL ||
- settings.HYPERBOLIC_MODEL ||
- "meta-llama/Llama-3.2-3B-Instruct",
- [ModelClass.MEDIUM]:
- settings.MEDIUM_HYPERBOLIC_MODEL ||
- settings.HYPERBOLIC_MODEL ||
- "meta-llama/Meta-Llama-3.1-70B-Instruct",
- [ModelClass.LARGE]:
- settings.LARGE_HYPERBOLIC_MODEL ||
- settings.HYPERBOLIC_MODEL ||
- "meta-llama/Meta-Llama-3.1-405-Instruct",
- [ModelClass.IMAGE]: settings.IMAGE_HYPERBOLIC_MODEL || "FLUX.1-dev",
- },
- },
- [ModelProviderName.VENICE]: {
- endpoint: "https://api.venice.ai/api/v1",
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- temperature: 0.6,
- },
- model: {
- [ModelClass.SMALL]: settings.SMALL_VENICE_MODEL || "llama-3.3-70b",
- [ModelClass.MEDIUM]: settings.MEDIUM_VENICE_MODEL || "llama-3.3-70b",
- [ModelClass.LARGE]: settings.LARGE_VENICE_MODEL || "llama-3.1-405b",
- [ModelClass.IMAGE]: settings.IMAGE_VENICE_MODEL || "fluently-xl",
- },
- },
- [ModelProviderName.AKASH_CHAT_API]: {
- endpoint: "https://chatapi.akash.network/api/v1",
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- temperature: 0.6,
- },
- model: {
- [ModelClass.SMALL]:
- settings.SMALL_AKASH_CHAT_API_MODEL ||
- "Meta-Llama-3-2-3B-Instruct",
- [ModelClass.MEDIUM]:
- settings.MEDIUM_AKASH_CHAT_API_MODEL ||
- "Meta-Llama-3-3-70B-Instruct",
- [ModelClass.LARGE]:
- settings.LARGE_AKASH_CHAT_API_MODEL ||
- "Meta-Llama-3-1-405B-Instruct-FP8",
- },
- },
- [ModelProviderName.LIVEPEER]: {
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- repetition_penalty: 0.4,
- temperature: 0.7,
- },
- // livepeer endpoint is handled from the sdk
- model: {
- [ModelClass.SMALL]: "",
- [ModelClass.MEDIUM]: "",
- [ModelClass.LARGE]: "",
- [ModelClass.EMBEDDING]: "",
- [ModelClass.IMAGE]: settings.LIVEPEER_IMAGE_MODEL || "ByteDance/SDXL-Lightning",
- },
- },
- [ModelProviderName.INFERA]: {
- endpoint: "https://api.infera.org",
- settings: {
- stop: [],
- maxInputTokens: 128000,
- maxOutputTokens: 8192,
- temperature: 0.6,
- },
- model: {
- [ModelClass.SMALL]:
- settings.SMALL_INFERA_MODEL || "llama3.2:3b",
- [ModelClass.MEDIUM]:
- settings.MEDIUM_INFERA_MODEL || "mistral-nemo:latest",
- [ModelClass.LARGE]:
- settings.LARGE_INFERA_MODEL || "mistral-small:latest",
- },
- },
-};
-
-export function getModel(provider: ModelProviderName, type: ModelClass) {
- return models[provider].model[type];
-}
-
-export function getEndpoint(provider: ModelProviderName) {
- return models[provider].endpoint;
-}
diff --git a/packages/core/package.json b/packages/core/package.json
index 3a1b74388fe..8aa76e07d7c 100644
--- a/packages/core/package.json
+++ b/packages/core/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/core",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"description": "",
"type": "module",
"main": "dist/index.js",
@@ -69,6 +69,7 @@
"@ai-sdk/google": "0.0.55",
"@ai-sdk/google-vertex": "0.0.43",
"@ai-sdk/groq": "0.0.3",
+ "@ai-sdk/mistral": "^1.0.8",
"@ai-sdk/openai": "1.0.5",
"@anthropic-ai/sdk": "0.30.1",
"@fal-ai/client": "1.2.0",
diff --git a/packages/core/src/database.ts b/packages/core/src/database.ts
index 310c44c32ab..322341a8cfd 100644
--- a/packages/core/src/database.ts
+++ b/packages/core/src/database.ts
@@ -95,6 +95,7 @@ export abstract class DatabaseAdapter implements IDatabaseAdapter {
agentId: UUID;
roomIds: UUID[];
tableName: string;
+ limit?: number;
}): Promise;
abstract getMemoryById(id: UUID): Promise;
diff --git a/packages/core/src/defaultCharacter.ts b/packages/core/src/defaultCharacter.ts
index e4a81b07e2e..8faaa64f2b4 100644
--- a/packages/core/src/defaultCharacter.ts
+++ b/packages/core/src/defaultCharacter.ts
@@ -527,4 +527,5 @@ export const defaultCharacter: Character = {
"meticulous",
"provocative",
],
+ extends: [],
};
diff --git a/packages/core/src/embedding.ts b/packages/core/src/embedding.ts
index 73cc657f00c..ce2d00b21b7 100644
--- a/packages/core/src/embedding.ts
+++ b/packages/core/src/embedding.ts
@@ -18,6 +18,7 @@ export const EmbeddingProvider = {
OpenAI: "OpenAI",
Ollama: "Ollama",
GaiaNet: "GaiaNet",
+ Heurist: "Heurist",
BGE: "BGE",
} as const;
@@ -39,7 +40,10 @@ export const getEmbeddingConfig = (): EmbeddingConfig => ({
: settings.USE_GAIANET_EMBEDDING?.toLowerCase() === "true"
? getEmbeddingModelSettings(ModelProviderName.GAIANET)
.dimensions
- : 384, // BGE
+ : settings.USE_HEURIST_EMBEDDING?.toLowerCase() === "true"
+ ? getEmbeddingModelSettings(ModelProviderName.HEURIST)
+ .dimensions
+ : 384, // BGE
model:
settings.USE_OPENAI_EMBEDDING?.toLowerCase() === "true"
? getEmbeddingModelSettings(ModelProviderName.OPENAI).name
@@ -47,7 +51,9 @@ export const getEmbeddingConfig = (): EmbeddingConfig => ({
? getEmbeddingModelSettings(ModelProviderName.OLLAMA).name
: settings.USE_GAIANET_EMBEDDING?.toLowerCase() === "true"
? getEmbeddingModelSettings(ModelProviderName.GAIANET).name
- : "BGE-small-en-v1.5",
+ : settings.USE_HEURIST_EMBEDDING?.toLowerCase() === "true"
+ ? getEmbeddingModelSettings(ModelProviderName.HEURIST).name
+ : "BGE-small-en-v1.5",
provider:
settings.USE_OPENAI_EMBEDDING?.toLowerCase() === "true"
? "OpenAI"
@@ -55,7 +61,9 @@ export const getEmbeddingConfig = (): EmbeddingConfig => ({
? "Ollama"
: settings.USE_GAIANET_EMBEDDING?.toLowerCase() === "true"
? "GaiaNet"
- : "BGE",
+ : settings.USE_HEURIST_EMBEDDING?.toLowerCase() === "true"
+ ? "Heurist"
+ : "BGE",
});
async function getRemoteEmbedding(
@@ -126,6 +134,7 @@ export function getEmbeddingType(runtime: IAgentRuntime): "local" | "remote" {
isNode &&
runtime.character.modelProvider !== ModelProviderName.OPENAI &&
runtime.character.modelProvider !== ModelProviderName.GAIANET &&
+ runtime.character.modelProvider !== ModelProviderName.HEURIST &&
!settings.USE_OPENAI_EMBEDDING;
return isLocal ? "local" : "remote";
@@ -146,6 +155,10 @@ export function getEmbeddingZeroVector(): number[] {
embeddingDimension = getEmbeddingModelSettings(
ModelProviderName.GAIANET
).dimensions; // GaiaNet dimension
+ } else if (settings.USE_HEURIST_EMBEDDING?.toLowerCase() === "true") {
+ embeddingDimension = getEmbeddingModelSettings(
+ ModelProviderName.HEURIST
+ ).dimensions; // Heurist dimension
}
return Array(embeddingDimension).fill(0);
@@ -229,6 +242,15 @@ export async function embed(runtime: IAgentRuntime, input: string) {
});
}
+ if (config.provider === EmbeddingProvider.Heurist) {
+ return await getRemoteEmbedding(input, {
+ model: config.model,
+ endpoint: getEndpoint(ModelProviderName.HEURIST),
+ apiKey: runtime.token,
+ dimensions: config.dimensions,
+ });
+ }
+
// BGE - try local first if in Node
if (isNode) {
try {
diff --git a/packages/core/src/environment.ts b/packages/core/src/environment.ts
index ed7edf3bf25..4bbe5fcb915 100644
--- a/packages/core/src/environment.ts
+++ b/packages/core/src/environment.ts
@@ -135,6 +135,7 @@ export const CharacterSchema = z.object({
prompt: z.string().optional(),
})
.optional(),
+ extends: z.array(z.string()).optional(),
});
// Type inference
diff --git a/packages/core/src/generation.ts b/packages/core/src/generation.ts
index 76d4e5a0569..77fb3986a80 100644
--- a/packages/core/src/generation.ts
+++ b/packages/core/src/generation.ts
@@ -1,5 +1,6 @@
import { createAnthropic } from "@ai-sdk/anthropic";
import { createGoogleGenerativeAI } from "@ai-sdk/google";
+import { createMistral } from "@ai-sdk/mistral";
import { createGroq } from "@ai-sdk/groq";
import { createOpenAI } from "@ai-sdk/openai";
import { RecursiveCharacterTextSplitter } from "langchain/text_splitter";
@@ -45,7 +46,7 @@ import {
IVerifiableInferenceAdapter,
VerifiableInferenceOptions,
VerifiableInferenceResult,
- VerifiableInferenceProvider,
+ //VerifiableInferenceProvider,
TelemetrySettings,
TokenizerType,
} from "./types.ts";
@@ -163,6 +164,50 @@ async function truncateTiktoken(
}
}
+/**
+ * Gets the Cloudflare Gateway base URL for a specific provider if enabled
+ * @param runtime The runtime environment
+ * @param provider The model provider name
+ * @returns The Cloudflare Gateway base URL if enabled, undefined otherwise
+ */
+function getCloudflareGatewayBaseURL(runtime: IAgentRuntime, provider: string): string | undefined {
+ const isCloudflareEnabled = runtime.getSetting("CLOUDFLARE_GW_ENABLED") === "true";
+ const cloudflareAccountId = runtime.getSetting("CLOUDFLARE_AI_ACCOUNT_ID");
+ const cloudflareGatewayId = runtime.getSetting("CLOUDFLARE_AI_GATEWAY_ID");
+
+ elizaLogger.debug("Cloudflare Gateway Configuration:", {
+ isEnabled: isCloudflareEnabled,
+ hasAccountId: !!cloudflareAccountId,
+ hasGatewayId: !!cloudflareGatewayId,
+ provider: provider
+ });
+
+ if (!isCloudflareEnabled) {
+ elizaLogger.debug("Cloudflare Gateway is not enabled");
+ return undefined;
+ }
+
+ if (!cloudflareAccountId) {
+ elizaLogger.warn("Cloudflare Gateway is enabled but CLOUDFLARE_AI_ACCOUNT_ID is not set");
+ return undefined;
+ }
+
+ if (!cloudflareGatewayId) {
+ elizaLogger.warn("Cloudflare Gateway is enabled but CLOUDFLARE_AI_GATEWAY_ID is not set");
+ return undefined;
+ }
+
+ const baseURL = `https://gateway.ai.cloudflare.com/v1/${cloudflareAccountId}/${cloudflareGatewayId}/${provider.toLowerCase()}`;
+ elizaLogger.info("Using Cloudflare Gateway:", {
+ provider,
+ baseURL,
+ accountId: cloudflareAccountId,
+ gatewayId: cloudflareGatewayId
+ });
+
+ return baseURL;
+}
+
/**
* Send a message to the model for a text generateText - receive a string back and parse how you'd like
* @param opts - The options for the generateText request.
@@ -215,7 +260,10 @@ export async function generateText({
elizaLogger.log("Using provider:", runtime.modelProvider);
// If verifiable inference is requested and adapter is provided, use it
if (verifiableInference && runtime.verifiableInferenceAdapter) {
- elizaLogger.log("Using verifiable inference adapter:", runtime.verifiableInferenceAdapter);
+ elizaLogger.log(
+ "Using verifiable inference adapter:",
+ runtime.verifiableInferenceAdapter
+ );
try {
const result: VerifiableInferenceResult =
await runtime.verifiableInferenceAdapter.generateText(
@@ -239,6 +287,16 @@ export async function generateText({
}
const provider = runtime.modelProvider;
+ elizaLogger.debug("Provider settings:", {
+ provider,
+ hasRuntime: !!runtime,
+ runtimeSettings: {
+ CLOUDFLARE_GW_ENABLED: runtime.getSetting("CLOUDFLARE_GW_ENABLED"),
+ CLOUDFLARE_AI_ACCOUNT_ID: runtime.getSetting("CLOUDFLARE_AI_ACCOUNT_ID"),
+ CLOUDFLARE_AI_GATEWAY_ID: runtime.getSetting("CLOUDFLARE_AI_GATEWAY_ID")
+ }
+ });
+
const endpoint =
runtime.character.modelEndpointOverride || getEndpoint(provider);
const modelSettings = getModelSettings(runtime.modelProvider, modelClass);
@@ -353,13 +411,16 @@ export async function generateText({
case ModelProviderName.LLAMACLOUD:
case ModelProviderName.NANOGPT:
case ModelProviderName.HYPERBOLIC:
- case ModelProviderName.NINETEEN_AI:
case ModelProviderName.TOGETHER:
+ case ModelProviderName.NINETEEN_AI:
case ModelProviderName.AKASH_CHAT_API: {
- elizaLogger.debug("Initializing OpenAI model.");
+ elizaLogger.debug("Initializing OpenAI model with Cloudflare check");
+ const baseURL = getCloudflareGatewayBaseURL(runtime, 'openai') || endpoint;
+
+ //elizaLogger.debug("OpenAI baseURL result:", { baseURL });
const openai = createOpenAI({
apiKey,
- baseURL: endpoint,
+ baseURL,
fetch: runtime.fetch,
});
@@ -391,7 +452,8 @@ export async function generateText({
apiKey,
baseURL: endpoint,
fetch: async (url: string, options: any) => {
- const chain_id = runtime.getSetting("ETERNALAI_CHAIN_ID") || "45762"
+ const chain_id =
+ runtime.getSetting("ETERNALAI_CHAIN_ID") || "45762";
if (options?.body) {
const body = JSON.parse(options.body);
body.chain_id = chain_id;
@@ -426,10 +488,7 @@ export async function generateText({
const { text: openaiResponse } = await aiGenerateText({
model: openai.languageModel(model),
prompt: context,
- system:
- runtime.character.system ??
- settings.SYSTEM_PROMPT ??
- undefined,
+ system: runtime.character.system ?? settings.SYSTEM_PROMPT ?? undefined,
temperature: temperature,
maxTokens: max_response_length,
frequencyPenalty: frequency_penalty,
@@ -469,14 +528,33 @@ export async function generateText({
break;
}
- case ModelProviderName.ANTHROPIC: {
- elizaLogger.debug("Initializing Anthropic model.");
+ case ModelProviderName.MISTRAL: {
+ const mistral = createMistral();
- const anthropic = createAnthropic({
- apiKey,
- fetch: runtime.fetch,
+ const { text: mistralResponse } = await aiGenerateText({
+ model: mistral(model),
+ prompt: context,
+ system:
+ runtime.character.system ??
+ settings.SYSTEM_PROMPT ??
+ undefined,
+ temperature: temperature,
+ maxTokens: max_response_length,
+ frequencyPenalty: frequency_penalty,
+ presencePenalty: presence_penalty,
});
+ response = mistralResponse;
+ elizaLogger.debug("Received response from Mistral model.");
+ break;
+ }
+
+ case ModelProviderName.ANTHROPIC: {
+ elizaLogger.debug("Initializing Anthropic model with Cloudflare check");
+ const baseURL = getCloudflareGatewayBaseURL(runtime, 'anthropic') || "https://api.anthropic.com/v1";
+ elizaLogger.debug("Anthropic baseURL result:", { baseURL });
+
+ const anthropic = createAnthropic({ apiKey, baseURL, fetch: runtime.fetch });
const { text: anthropicResponse } = await aiGenerateText({
model: anthropic.languageModel(model),
prompt: context,
@@ -564,26 +642,30 @@ export async function generateText({
}
case ModelProviderName.GROQ: {
- const groq = createGroq({ apiKey, fetch: runtime.fetch });
+ elizaLogger.debug("Initializing Groq model with Cloudflare check");
+ const baseURL = getCloudflareGatewayBaseURL(runtime, 'groq');
+ elizaLogger.debug("Groq baseURL result:", { baseURL });
+ const groq = createGroq({ apiKey, fetch: runtime.fetch, baseURL });
const { text: groqResponse } = await aiGenerateText({
model: groq.languageModel(model),
prompt: context,
- temperature: temperature,
+ temperature,
system:
runtime.character.system ??
settings.SYSTEM_PROMPT ??
undefined,
- tools: tools,
+ tools,
onStepFinish: onStepFinish,
- maxSteps: maxSteps,
+ maxSteps,
maxTokens: max_response_length,
frequencyPenalty: frequency_penalty,
presencePenalty: presence_penalty,
- experimental_telemetry: experimental_telemetry,
+ experimental_telemetry,
});
response = groqResponse;
+ elizaLogger.debug("Received response from Groq model.");
break;
}
@@ -790,10 +872,12 @@ export async function generateText({
case ModelProviderName.GALADRIEL: {
elizaLogger.debug("Initializing Galadriel model.");
- const headers = {}
- const fineTuneApiKey = runtime.getSetting("GALADRIEL_FINE_TUNE_API_KEY")
+ const headers = {};
+ const fineTuneApiKey = runtime.getSetting(
+ "GALADRIEL_FINE_TUNE_API_KEY"
+ );
if (fineTuneApiKey) {
- headers["Fine-Tune-Authentication"] = fineTuneApiKey
+ headers["Fine-Tune-Authentication"] = fineTuneApiKey;
}
const galadriel = createOpenAI({
headers,
@@ -824,6 +908,37 @@ export async function generateText({
break;
}
+ case ModelProviderName.INFERA: {
+ elizaLogger.debug("Initializing Infera model.");
+
+ const apiKey = settings.INFERA_API_KEY || runtime.token;
+
+ const infera = createOpenAI({
+ apiKey,
+ baseURL: endpoint,
+ headers: {
+ api_key: apiKey,
+ "Content-Type": "application/json",
+ },
+ });
+
+ const { text: inferaResponse } = await aiGenerateText({
+ model: infera.languageModel(model),
+ prompt: context,
+ system:
+ runtime.character.system ??
+ settings.SYSTEM_PROMPT ??
+ undefined,
+ temperature: temperature,
+ maxTokens: max_response_length,
+ frequencyPenalty: frequency_penalty,
+ presencePenalty: presence_penalty,
+ });
+ response = inferaResponse;
+ elizaLogger.debug("Received response from Infera model.");
+ break;
+ }
+
case ModelProviderName.VENICE: {
elizaLogger.debug("Initializing Venice model.");
const venice = createOpenAI({
@@ -850,6 +965,37 @@ export async function generateText({
break;
}
+ case ModelProviderName.DEEPSEEK: {
+ elizaLogger.debug("Initializing Deepseek model.");
+ const serverUrl = models[provider].endpoint;
+ const deepseek = createOpenAI({
+ apiKey,
+ baseURL: serverUrl,
+ fetch: runtime.fetch,
+ });
+
+ const { text: deepseekResponse } = await aiGenerateText({
+ model: deepseek.languageModel(model),
+ prompt: context,
+ temperature: temperature,
+ system:
+ runtime.character.system ??
+ settings.SYSTEM_PROMPT ??
+ undefined,
+ tools: tools,
+ onStepFinish: onStepFinish,
+ maxSteps: maxSteps,
+ maxTokens: max_response_length,
+ frequencyPenalty: frequency_penalty,
+ presencePenalty: presence_penalty,
+ experimental_telemetry: experimental_telemetry,
+ });
+
+ response = deepseekResponse;
+ elizaLogger.debug("Received response from Deepseek model.");
+ break;
+ }
+
default: {
const errorMessage = `Unsupported provider: ${provider}`;
elizaLogger.error(errorMessage);
@@ -1423,7 +1569,9 @@ export const generateImage = async (
});
return { success: true, data: base64s };
- }else if (runtime.imageModelProvider === ModelProviderName.NINETEEN_AI) {
+ } else if (
+ runtime.imageModelProvider === ModelProviderName.NINETEEN_AI
+ ) {
const response = await fetch(
"https://api.nineteen.ai/v1/text-to-image",
{
@@ -1433,13 +1581,13 @@ export const generateImage = async (
"Content-Type": "application/json",
},
body: JSON.stringify({
- model: data.modelId || "dataautogpt3/ProteusV0.4-Lightning",
+ model: model,
prompt: data.prompt,
negative_prompt: data.negativePrompt,
width: data.width,
height: data.height,
steps: data.numIterations,
- cfg_scale: data.guidanceScale || 3
+ cfg_scale: data.guidanceScale || 3,
}),
}
);
@@ -1738,9 +1886,9 @@ export async function handleProvider(
runtime,
context,
modelClass,
- verifiableInference,
- verifiableInferenceAdapter,
- verifiableInferenceOptions,
+ //verifiableInference,
+ //verifiableInferenceAdapter,
+ //verifiableInferenceOptions,
} = options;
switch (provider) {
case ModelProviderName.OPENAI:
@@ -1767,12 +1915,16 @@ export async function handleProvider(
});
case ModelProviderName.GOOGLE:
return await handleGoogle(options);
+ case ModelProviderName.MISTRAL:
+ return await handleMistral(options);
case ModelProviderName.REDPILL:
return await handleRedPill(options);
case ModelProviderName.OPENROUTER:
return await handleOpenRouter(options);
case ModelProviderName.OLLAMA:
return await handleOllama(options);
+ case ModelProviderName.DEEPSEEK:
+ return await handleDeepSeek(options);
default: {
const errorMessage = `Unsupported provider: ${provider}`;
elizaLogger.error(errorMessage);
@@ -1794,8 +1946,10 @@ async function handleOpenAI({
schemaDescription,
mode = "json",
modelOptions,
+ provider: _provider,
+ runtime,
}: ProviderOptions): Promise> {
- const baseURL = models.openai.endpoint || undefined;
+ const baseURL = getCloudflareGatewayBaseURL(runtime, 'openai') || models.openai.endpoint;
const openai = createOpenAI({ apiKey, baseURL });
return await aiGenerateObject({
model: openai.languageModel(model),
@@ -1821,8 +1975,13 @@ async function handleAnthropic({
schemaDescription,
mode = "json",
modelOptions,
+ runtime,
}: ProviderOptions): Promise> {
- const anthropic = createAnthropic({ apiKey });
+ elizaLogger.debug("Handling Anthropic request with Cloudflare check");
+ const baseURL = getCloudflareGatewayBaseURL(runtime, 'anthropic');
+ elizaLogger.debug("Anthropic handleAnthropic baseURL:", { baseURL });
+
+ const anthropic = createAnthropic({ apiKey, baseURL });
return await aiGenerateObject({
model: anthropic.languageModel(model),
schema,
@@ -1873,8 +2032,13 @@ async function handleGroq({
schemaDescription,
mode = "json",
modelOptions,
+ runtime,
}: ProviderOptions): Promise> {
- const groq = createGroq({ apiKey });
+ elizaLogger.debug("Handling Groq request with Cloudflare check");
+ const baseURL = getCloudflareGatewayBaseURL(runtime, 'groq');
+ elizaLogger.debug("Groq handleGroq baseURL:", { baseURL });
+
+ const groq = createGroq({ apiKey, baseURL });
return await aiGenerateObject({
model: groq.languageModel(model),
schema,
@@ -1911,6 +2075,31 @@ async function handleGoogle({
});
}
+/**
+ * Handles object generation for Mistral models.
+ *
+ * @param {ProviderOptions} options - Options specific to Mistral.
+ * @returns {Promise>} - A promise that resolves to generated objects.
+ */
+async function handleMistral({
+ model,
+ schema,
+ schemaName,
+ schemaDescription,
+ mode,
+ modelOptions,
+}: ProviderOptions): Promise> {
+ const mistral = createMistral();
+ return await aiGenerateObject({
+ model: mistral(model),
+ schema,
+ schemaName,
+ schemaDescription,
+ mode,
+ ...modelOptions,
+ });
+}
+
/**
* Handles object generation for Redpill models.
*
@@ -1995,6 +2184,32 @@ async function handleOllama({
});
}
+/**
+ * Handles object generation for DeepSeek models.
+ *
+ * @param {ProviderOptions} options - Options specific to DeepSeek.
+ * @returns {Promise>} - A promise that resolves to generated objects.
+ */
+async function handleDeepSeek({
+ model,
+ apiKey,
+ schema,
+ schemaName,
+ schemaDescription,
+ mode,
+ modelOptions,
+}: ProviderOptions): Promise> {
+ const openai = createOpenAI({ apiKey, baseURL: models.deepseek.endpoint });
+ return await aiGenerateObject({
+ model: openai.languageModel(model),
+ schema,
+ schemaName,
+ schemaDescription,
+ mode,
+ ...modelOptions,
+ });
+}
+
// Add type definition for Together AI response
interface TogetherAIImageResponse {
data: Array<{
diff --git a/packages/core/src/memory.ts b/packages/core/src/memory.ts
index 112352766f1..698c1c63376 100644
--- a/packages/core/src/memory.ts
+++ b/packages/core/src/memory.ts
@@ -189,11 +189,12 @@ export class MemoryManager implements IMemoryManager {
);
}
- async getMemoriesByRoomIds(params: { roomIds: UUID[] }): Promise {
+ async getMemoriesByRoomIds(params: { roomIds: UUID[], limit?: number; }): Promise {
return await this.runtime.databaseAdapter.getMemoriesByRoomIds({
tableName: this.tableName,
agentId: this.runtime.agentId,
roomIds: params.roomIds,
+ limit: params.limit
});
}
diff --git a/packages/core/src/models.ts b/packages/core/src/models.ts
index 2c54db91dda..903f865724f 100644
--- a/packages/core/src/models.ts
+++ b/packages/core/src/models.ts
@@ -1,11 +1,11 @@
import settings from "./settings.ts";
import {
- Models,
- ModelProviderName,
+ EmbeddingModelSettings,
+ ImageModelSettings,
ModelClass,
+ ModelProviderName,
+ Models,
ModelSettings,
- ImageModelSettings,
- EmbeddingModelSettings,
} from "./types.ts";
export const models: Models = {
@@ -276,7 +276,7 @@ export const models: Models = {
temperature: 0.7,
},
[ModelClass.MEDIUM]: {
- name: "meta-llama-3.1-8b-instruct",
+ name: "meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo-128K",
stop: [],
maxInputTokens: 128000,
maxOutputTokens: 8192,
@@ -332,6 +332,7 @@ export const models: Models = {
},
},
[ModelProviderName.GOOGLE]: {
+ endpoint: "https://generativelanguage.googleapis.com",
model: {
[ModelClass.SMALL]: {
name:
@@ -377,6 +378,46 @@ export const models: Models = {
},
},
},
+ [ModelProviderName.MISTRAL]: {
+ model: {
+ [ModelClass.SMALL]: {
+ name:
+ settings.SMALL_MISTRAL_MODEL ||
+ settings.MISTRAL_MODEL ||
+ "mistral-small-latest",
+ stop: [],
+ maxInputTokens: 128000,
+ maxOutputTokens: 8192,
+ frequency_penalty: 0.4,
+ presence_penalty: 0.4,
+ temperature: 0.7,
+ },
+ [ModelClass.MEDIUM]: {
+ name:
+ settings.MEDIUM_MISTRAL_MODEL ||
+ settings.MISTRAL_MODEL ||
+ "mistral-large-latest",
+ stop: [],
+ maxInputTokens: 128000,
+ maxOutputTokens: 8192,
+ frequency_penalty: 0.4,
+ presence_penalty: 0.4,
+ temperature: 0.7,
+ },
+ [ModelClass.LARGE]: {
+ name:
+ settings.LARGE_MISTRAL_MODEL ||
+ settings.MISTRAL_MODEL ||
+ "mistral-large-latest",
+ stop: [],
+ maxInputTokens: 128000,
+ maxOutputTokens: 8192,
+ frequency_penalty: 0.4,
+ presence_penalty: 0.4,
+ temperature: 0.7,
+ },
+ },
+ },
[ModelProviderName.REDPILL]: {
endpoint: "https://api.red-pill.ai/v1",
// Available models: https://docs.red-pill.ai/get-started/supported-models
@@ -544,7 +585,7 @@ export const models: Models = {
[ModelClass.LARGE]: {
name:
settings.LARGE_HEURIST_MODEL ||
- "meta-llama/llama-3.1-405b-instruct",
+ "meta-llama/llama-3.3-70b-instruct",
stop: [],
maxInputTokens: 128000,
maxOutputTokens: 8192,
@@ -552,9 +593,13 @@ export const models: Models = {
temperature: 0.7,
},
[ModelClass.IMAGE]: {
- name: settings.HEURIST_IMAGE_MODEL || "PepeXL",
+ name: settings.HEURIST_IMAGE_MODEL || "FLUX.1-dev",
steps: 20,
},
+ [ModelClass.EMBEDDING]: {
+ name: "BAAI/bge-large-en-v1.5",
+ dimensions: 1024,
+ },
},
},
[ModelProviderName.GALADRIEL]: {
@@ -815,36 +860,44 @@ export const models: Models = {
[ModelClass.IMAGE]: {
name: settings.IMAGE_VENICE_MODEL || "fluently-xl",
},
- }
+ },
},
[ModelProviderName.NINETEEN_AI]: {
endpoint: "https://api.nineteen.ai/v1",
model: {
[ModelClass.SMALL]: {
- name: settings.SMALL_NINETEEN_AI_MODEL || "unsloth/Llama-3.2-3B-Instruct",
+ name:
+ settings.SMALL_NINETEEN_AI_MODEL ||
+ "unsloth/Llama-3.2-3B-Instruct",
stop: [],
maxInputTokens: 128000,
maxOutputTokens: 8192,
- temperature: 0.6
+ temperature: 0.6,
},
[ModelClass.MEDIUM]: {
- name: settings.MEDIUM_NINETEEN_AI_MODEL || "unsloth/Meta-Llama-3.1-8B-Instruct",
+ name:
+ settings.MEDIUM_NINETEEN_AI_MODEL ||
+ "unsloth/Meta-Llama-3.1-8B-Instruct",
stop: [],
maxInputTokens: 128000,
maxOutputTokens: 8192,
temperature: 0.6,
},
[ModelClass.LARGE]: {
- name: settings.LARGE_NINETEEN_AI_MODEL || "hugging-quants/Meta-Llama-3.1-70B-Instruct-AWQ-INT4",
+ name:
+ settings.LARGE_NINETEEN_AI_MODEL ||
+ "hugging-quants/Meta-Llama-3.1-70B-Instruct-AWQ-INT4",
stop: [],
maxInputTokens: 128000,
maxOutputTokens: 8192,
temperature: 0.6,
},
[ModelClass.IMAGE]: {
- name: settings.IMAGE_NINETEEN_AI_MODEL || "dataautogpt3/ProteusV0.4-Lightning",
+ name:
+ settings.IMAGE_NINETEEN_AI_MODEL ||
+ "dataautogpt3/ProteusV0.4-Lightning",
},
- }
+ },
},
[ModelProviderName.AKASH_CHAT_API]: {
endpoint: "https://chatapi.akash.network/api/v1",
@@ -887,6 +940,64 @@ export const models: Models = {
},
},
},
+ [ModelProviderName.INFERA]: {
+ endpoint: "https://api.infera.org",
+ model: {
+ [ModelClass.SMALL]: {
+ name: settings.SMALL_INFERA_MODEL || "llama3.2:3b",
+ stop: [],
+ maxInputTokens: 128000,
+ maxOutputTokens: 8192,
+ temperature: 0.6,
+ },
+ [ModelClass.MEDIUM]: {
+ name: settings.MEDIUM_INFERA_MODEL || "mistral-nemo:latest",
+ stop: [],
+ maxInputTokens: 128000,
+ maxOutputTokens: 8192,
+ temperature: 0.6,
+ },
+ [ModelClass.LARGE]: {
+ name: settings.LARGE_INFERA_MODEL || "mistral-small:latest",
+ stop: [],
+ maxInputTokens: 128000,
+ maxOutputTokens: 8192,
+ temperature: 0.6,
+ },
+ },
+ },
+ [ModelProviderName.DEEPSEEK]: {
+ endpoint: settings.DEEPSEEK_API_URL || "https://api.deepseek.com",
+ model: {
+ [ModelClass.SMALL]: {
+ name: settings.SMALL_DEEPSEEK_MODEL || "deepseek-chat",
+ stop: [],
+ maxInputTokens: 128000,
+ maxOutputTokens: 8192,
+ frequency_penalty: 0.0,
+ presence_penalty: 0.0,
+ temperature: 0.7,
+ },
+ [ModelClass.MEDIUM]: {
+ name: settings.MEDIUM_DEEPSEEK_MODEL || "deepseek-chat",
+ stop: [],
+ maxInputTokens: 128000,
+ maxOutputTokens: 8192,
+ frequency_penalty: 0.0,
+ presence_penalty: 0.0,
+ temperature: 0.7,
+ },
+ [ModelClass.LARGE]: {
+ name: settings.LARGE_DEEPSEEK_MODEL || "deepseek-chat",
+ stop: [],
+ maxInputTokens: 128000,
+ maxOutputTokens: 8192,
+ frequency_penalty: 0.0,
+ presence_penalty: 0.0,
+ temperature: 0.7,
+ },
+ },
+ },
};
export function getModelSettings(
diff --git a/packages/core/src/parsing.ts b/packages/core/src/parsing.ts
index 107ce8ea0bd..331cd30a13b 100644
--- a/packages/core/src/parsing.ts
+++ b/packages/core/src/parsing.ts
@@ -205,3 +205,37 @@ export const parseActionResponseFromText = (
return { actions };
};
+
+/**
+ * Truncate text to fit within the character limit, ensuring it ends at a complete sentence.
+ */
+export function truncateToCompleteSentence(
+ text: string,
+ maxLength: number
+): string {
+ if (text.length <= maxLength) {
+ return text;
+ }
+
+ // Attempt to truncate at the last period within the limit
+ const lastPeriodIndex = text.lastIndexOf(".", maxLength - 1);
+ if (lastPeriodIndex !== -1) {
+ const truncatedAtPeriod = text.slice(0, lastPeriodIndex + 1).trim();
+ if (truncatedAtPeriod.length > 0) {
+ return truncatedAtPeriod;
+ }
+ }
+
+ // If no period, truncate to the nearest whitespace within the limit
+ const lastSpaceIndex = text.lastIndexOf(" ", maxLength - 1);
+ if (lastSpaceIndex !== -1) {
+ const truncatedAtSpace = text.slice(0, lastSpaceIndex).trim();
+ if (truncatedAtSpace.length > 0) {
+ return truncatedAtSpace + "...";
+ }
+ }
+
+ // Fallback: Hard truncate and add ellipsis
+ const hardTruncated = text.slice(0, maxLength - 3).trim();
+ return hardTruncated + "...";
+}
diff --git a/packages/core/src/ragknowledge.ts b/packages/core/src/ragknowledge.ts
index 0856cea67a8..4ccc56c8e13 100644
--- a/packages/core/src/ragknowledge.ts
+++ b/packages/core/src/ragknowledge.ts
@@ -1,12 +1,12 @@
import { embed } from "./embedding.ts";
+import { splitChunks } from "./generation.ts";
import elizaLogger from "./logger.ts";
import {
+ IAgentRuntime,
IRAGKnowledgeManager,
RAGKnowledgeItem,
UUID,
- IAgentRuntime
} from "./types.ts";
-import { splitChunks } from "./generation.ts";
import { stringToUuid } from "./uuid.ts";
/**
@@ -41,20 +41,62 @@ export class RAGKnowledgeManager implements IRAGKnowledgeManager {
* Common English stop words to filter out from query analysis
*/
private readonly stopWords = new Set([
- 'a', 'an', 'and', 'are', 'as', 'at', 'be', 'by', 'does', 'for', 'from', 'had',
- 'has', 'have', 'he', 'her', 'his', 'how', 'hey', 'i', 'in', 'is', 'it', 'its',
- 'of', 'on', 'or', 'that', 'the', 'this', 'to', 'was', 'what', 'when', 'where',
- 'which', 'who', 'will', 'with', 'would', 'there', 'their', 'they', 'your', 'you'
+ "a",
+ "an",
+ "and",
+ "are",
+ "as",
+ "at",
+ "be",
+ "by",
+ "does",
+ "for",
+ "from",
+ "had",
+ "has",
+ "have",
+ "he",
+ "her",
+ "his",
+ "how",
+ "hey",
+ "i",
+ "in",
+ "is",
+ "it",
+ "its",
+ "of",
+ "on",
+ "or",
+ "that",
+ "the",
+ "this",
+ "to",
+ "was",
+ "what",
+ "when",
+ "where",
+ "which",
+ "who",
+ "will",
+ "with",
+ "would",
+ "there",
+ "their",
+ "they",
+ "your",
+ "you",
]);
/**
* Filters out stop words and returns meaningful terms
*/
private getQueryTerms(query: string): string[] {
- return query.toLowerCase()
- .split(' ')
- .filter(term => term.length > 3) // Filter very short words
- .filter(term => !this.stopWords.has(term)); // Filter stop words
+ return query
+ .toLowerCase()
+ .split(" ")
+ .filter((term) => term.length > 3) // Filter very short words
+ .filter((term) => !this.stopWords.has(term)); // Filter stop words
}
/**
@@ -89,9 +131,10 @@ export class RAGKnowledgeManager implements IRAGKnowledgeManager {
}
private hasProximityMatch(text: string, terms: string[]): boolean {
- const words = text.toLowerCase().split(' ');
- const positions = terms.map(term => words.findIndex(w => w.includes(term)))
- .filter(pos => pos !== -1);
+ const words = text.toLowerCase().split(" ");
+ const positions = terms
+ .map((term) => words.findIndex((w) => w.includes(term)))
+ .filter((pos) => pos !== -1);
if (positions.length < 2) return false;
@@ -115,10 +158,11 @@ export class RAGKnowledgeManager implements IRAGKnowledgeManager {
// If id is provided, do direct lookup first
if (params.id) {
- const directResults = await this.runtime.databaseAdapter.getKnowledge({
- id: params.id,
- agentId: agentId
- });
+ const directResults =
+ await this.runtime.databaseAdapter.getKnowledge({
+ id: params.id,
+ agentId: agentId,
+ });
if (directResults.length > 0) {
return directResults;
@@ -133,7 +177,9 @@ export class RAGKnowledgeManager implements IRAGKnowledgeManager {
// Build search text with optional context
let searchText = processedQuery;
if (params.conversationContext) {
- const relevantContext = this.preprocess(params.conversationContext);
+ const relevantContext = this.preprocess(
+ params.conversationContext
+ );
searchText = `${relevantContext} ${processedQuery}`;
}
@@ -142,51 +188,65 @@ export class RAGKnowledgeManager implements IRAGKnowledgeManager {
const embedding = new Float32Array(embeddingArray);
// Get results with single query
- const results = await this.runtime.databaseAdapter.searchKnowledge({
- agentId: this.runtime.agentId,
- embedding: embedding,
- match_threshold: this.defaultRAGMatchThreshold,
- match_count: (params.limit || this.defaultRAGMatchCount) * 2,
- searchText: processedQuery
- });
+ const results =
+ await this.runtime.databaseAdapter.searchKnowledge({
+ agentId: this.runtime.agentId,
+ embedding: embedding,
+ match_threshold: this.defaultRAGMatchThreshold,
+ match_count:
+ (params.limit || this.defaultRAGMatchCount) * 2,
+ searchText: processedQuery,
+ });
// Enhanced reranking with sophisticated scoring
- const rerankedResults = results.map(result => {
- let score = result.similarity;
-
- // Check for direct query term matches
- const queryTerms = this.getQueryTerms(processedQuery);
-
- const matchingTerms = queryTerms.filter(term =>
- result.content.text.toLowerCase().includes(term));
-
- if (matchingTerms.length > 0) {
- // Much stronger boost for matches
- score *= (1 + (matchingTerms.length / queryTerms.length) * 2); // Double the boost
-
- if (this.hasProximityMatch(result.content.text, matchingTerms)) {
- score *= 1.5; // Stronger proximity boost
- }
- } else {
- // More aggressive penalty
- if (!params.conversationContext) {
- score *= 0.3; // Stronger penalty
+ const rerankedResults = results
+ .map((result) => {
+ let score = result.similarity;
+
+ // Check for direct query term matches
+ const queryTerms = this.getQueryTerms(processedQuery);
+
+ const matchingTerms = queryTerms.filter((term) =>
+ result.content.text.toLowerCase().includes(term)
+ );
+
+ if (matchingTerms.length > 0) {
+ // Much stronger boost for matches
+ score *=
+ 1 +
+ (matchingTerms.length / queryTerms.length) * 2; // Double the boost
+
+ if (
+ this.hasProximityMatch(
+ result.content.text,
+ matchingTerms
+ )
+ ) {
+ score *= 1.5; // Stronger proximity boost
+ }
+ } else {
+ // More aggressive penalty
+ if (!params.conversationContext) {
+ score *= 0.3; // Stronger penalty
+ }
}
- }
- return {
- ...result,
- score,
- matchedTerms: matchingTerms // Add for debugging
- };
- }).sort((a, b) => b.score - a.score);
+ return {
+ ...result,
+ score,
+ matchedTerms: matchingTerms, // Add for debugging
+ };
+ })
+ .sort((a, b) => b.score - a.score);
// Filter and return results
return rerankedResults
- .filter(result => result.score >= this.defaultRAGMatchThreshold)
+ .filter(
+ (result) =>
+ result.score >= this.defaultRAGMatchThreshold
+ )
.slice(0, params.limit || this.defaultRAGMatchCount);
-
- } catch(error) {
+ } catch (error) {
console.log(`[RAG Search Error] ${error}`);
return [];
}
@@ -205,7 +265,10 @@ export class RAGKnowledgeManager implements IRAGKnowledgeManager {
try {
// Process main document
const processedContent = this.preprocess(item.content.text);
- const mainEmbeddingArray = await embed(this.runtime, processedContent);
+ const mainEmbeddingArray = await embed(
+ this.runtime,
+ processedContent
+ );
const mainEmbedding = new Float32Array(mainEmbeddingArray);
@@ -217,11 +280,11 @@ export class RAGKnowledgeManager implements IRAGKnowledgeManager {
text: item.content.text,
metadata: {
...item.content.metadata,
- isMain: true
- }
+ isMain: true,
+ },
},
embedding: mainEmbedding,
- createdAt: Date.now()
+ createdAt: Date.now(),
});
// Generate and store chunks
@@ -241,11 +304,11 @@ export class RAGKnowledgeManager implements IRAGKnowledgeManager {
...item.content.metadata,
isChunk: true,
originalId: item.id,
- chunkIndex: index
- }
+ chunkIndex: index,
+ },
},
embedding: chunkEmbedding,
- createdAt: Date.now()
+ createdAt: Date.now(),
});
}
} catch (error) {
@@ -265,17 +328,19 @@ export class RAGKnowledgeManager implements IRAGKnowledgeManager {
match_threshold = this.defaultRAGMatchThreshold,
match_count = this.defaultRAGMatchCount,
embedding,
- searchText
+ searchText,
} = params;
- const float32Embedding = Array.isArray(embedding) ? new Float32Array(embedding) : embedding;
+ const float32Embedding = Array.isArray(embedding)
+ ? new Float32Array(embedding)
+ : embedding;
return await this.runtime.databaseAdapter.searchKnowledge({
agentId: params.agentId || this.runtime.agentId,
embedding: float32Embedding,
match_threshold,
match_count,
- searchText
+ searchText,
});
}
@@ -284,14 +349,17 @@ export class RAGKnowledgeManager implements IRAGKnowledgeManager {
}
async clearKnowledge(shared?: boolean): Promise {
- await this.runtime.databaseAdapter.clearKnowledge(this.runtime.agentId, shared ? shared : false);
+ await this.runtime.databaseAdapter.clearKnowledge(
+ this.runtime.agentId,
+ shared ? shared : false
+ );
}
async processFile(file: {
path: string;
content: string;
- type: 'pdf' | 'md' | 'txt';
- isShared?: boolean
+ type: "pdf" | "md" | "txt";
+ isShared?: boolean;
}): Promise {
const timeMarker = (label: string) => {
const time = (Date.now() - startTime) / 1000;
@@ -299,21 +367,26 @@ export class RAGKnowledgeManager implements IRAGKnowledgeManager {
};
const startTime = Date.now();
- let content = file.content;
+ const content = file.content;
try {
- const fileSizeKB = (new TextEncoder().encode(content)).length / 1024;
- elizaLogger.info(`[File Progress] Starting ${file.path} (${fileSizeKB.toFixed(2)} KB)`);
+ const fileSizeKB = new TextEncoder().encode(content).length / 1024;
+ elizaLogger.info(
+ `[File Progress] Starting ${file.path} (${fileSizeKB.toFixed(2)} KB)`
+ );
// Step 1: Preprocessing
- const preprocessStart = Date.now();
+ //const preprocessStart = Date.now();
const processedContent = this.preprocess(content);
- timeMarker('Preprocessing');
+ timeMarker("Preprocessing");
// Step 2: Main document embedding
- const mainEmbeddingArray = await embed(this.runtime, processedContent);
+ const mainEmbeddingArray = await embed(
+ this.runtime,
+ processedContent
+ );
const mainEmbedding = new Float32Array(mainEmbeddingArray);
- timeMarker('Main embedding');
+ timeMarker("Main embedding");
// Step 3: Create main document
await this.runtime.databaseAdapter.createKnowledge({
@@ -324,19 +397,19 @@ export class RAGKnowledgeManager implements IRAGKnowledgeManager {
metadata: {
source: file.path,
type: file.type,
- isShared: file.isShared || false
- }
+ isShared: file.isShared || false,
+ },
},
embedding: mainEmbedding,
- createdAt: Date.now()
+ createdAt: Date.now(),
});
- timeMarker('Main document storage');
+ timeMarker("Main document storage");
// Step 4: Generate chunks
const chunks = await splitChunks(processedContent, 512, 20);
const totalChunks = chunks.length;
elizaLogger.info(`Generated ${totalChunks} chunks`);
- timeMarker('Chunk generation');
+ timeMarker("Chunk generation");
// Step 5: Process chunks with larger batches
const BATCH_SIZE = 10; // Increased batch size
@@ -344,52 +417,66 @@ export class RAGKnowledgeManager implements IRAGKnowledgeManager {
for (let i = 0; i < chunks.length; i += BATCH_SIZE) {
const batchStart = Date.now();
- const batch = chunks.slice(i, Math.min(i + BATCH_SIZE, chunks.length));
+ const batch = chunks.slice(
+ i,
+ Math.min(i + BATCH_SIZE, chunks.length)
+ );
// Process embeddings in parallel
const embeddings = await Promise.all(
- batch.map(chunk => embed(this.runtime, chunk))
+ batch.map((chunk) => embed(this.runtime, chunk))
);
// Batch database operations
- await Promise.all(embeddings.map(async (embeddingArray, index) => {
- const chunkId = `${stringToUuid(file.path)}-chunk-${i + index}` as UUID;
- const chunkEmbedding = new Float32Array(embeddingArray);
-
- await this.runtime.databaseAdapter.createKnowledge({
- id: chunkId,
- agentId: this.runtime.agentId,
- content: {
- text: batch[index],
- metadata: {
- source: file.path,
- type: file.type,
- isShared: file.isShared || false,
- isChunk: true,
- originalId: stringToUuid(file.path),
- chunkIndex: i + index
- }
- },
- embedding: chunkEmbedding,
- createdAt: Date.now()
- });
- }));
+ await Promise.all(
+ embeddings.map(async (embeddingArray, index) => {
+ const chunkId =
+ `${stringToUuid(file.path)}-chunk-${i + index}` as UUID;
+ const chunkEmbedding = new Float32Array(embeddingArray);
+
+ await this.runtime.databaseAdapter.createKnowledge({
+ id: chunkId,
+ agentId: this.runtime.agentId,
+ content: {
+ text: batch[index],
+ metadata: {
+ source: file.path,
+ type: file.type,
+ isShared: file.isShared || false,
+ isChunk: true,
+ originalId: stringToUuid(file.path),
+ chunkIndex: i + index,
+ },
+ },
+ embedding: chunkEmbedding,
+ createdAt: Date.now(),
+ });
+ })
+ );
processedChunks += batch.length;
const batchTime = (Date.now() - batchStart) / 1000;
- elizaLogger.info(`[Batch Progress] Processed ${processedChunks}/${totalChunks} chunks (${batchTime.toFixed(2)}s for batch)`);
+ elizaLogger.info(
+ `[Batch Progress] Processed ${processedChunks}/${totalChunks} chunks (${batchTime.toFixed(2)}s for batch)`
+ );
}
const totalTime = (Date.now() - startTime) / 1000;
- elizaLogger.info(`[Complete] Processed ${file.path} in ${totalTime.toFixed(2)}s`);
-
+ elizaLogger.info(
+ `[Complete] Processed ${file.path} in ${totalTime.toFixed(2)}s`
+ );
} catch (error) {
- if (file.isShared && error?.code === 'SQLITE_CONSTRAINT_PRIMARYKEY') {
- elizaLogger.info(`Shared knowledge ${file.path} already exists in database, skipping creation`);
+ if (
+ file.isShared &&
+ error?.code === "SQLITE_CONSTRAINT_PRIMARYKEY"
+ ) {
+ elizaLogger.info(
+ `Shared knowledge ${file.path} already exists in database, skipping creation`
+ );
return;
}
elizaLogger.error(`Error processing file ${file.path}:`, error);
throw error;
}
}
-}
\ No newline at end of file
+}
diff --git a/packages/core/src/runtime.ts b/packages/core/src/runtime.ts
index a3f4062a728..646dc2b0ed3 100644
--- a/packages/core/src/runtime.ts
+++ b/packages/core/src/runtime.ts
@@ -1,3 +1,5 @@
+import { readFile } from "fs/promises";
+import { join } from "path";
import { names, uniqueNamesGenerator } from "unique-names-generator";
import { v4 as uuidv4 } from "uuid";
import {
@@ -17,12 +19,12 @@ import { generateText } from "./generation.ts";
import { formatGoalsAsString, getGoals } from "./goals.ts";
import { elizaLogger } from "./index.ts";
import knowledge from "./knowledge.ts";
-import { RAGKnowledgeManager } from "./ragknowledge.ts";
import { MemoryManager } from "./memory.ts";
import { formatActors, formatMessages, getActorDetails } from "./messages.ts";
import { parseJsonArrayFromText } from "./parsing.ts";
import { formatPosts } from "./posts.ts";
import { getProviders } from "./providers.ts";
+import { RAGKnowledgeManager } from "./ragknowledge.ts";
import settings from "./settings.ts";
import {
Character,
@@ -31,11 +33,12 @@ import {
IAgentRuntime,
ICacheManager,
IDatabaseAdapter,
- IRAGKnowledgeManager,
IMemoryManager,
+ IRAGKnowledgeManager,
+ IVerifiableInferenceAdapter,
KnowledgeItem,
- RAGKnowledgeItem,
- Media,
+ //RAGKnowledgeItem,
+ //Media,
ModelClass,
ModelProviderName,
Plugin,
@@ -48,11 +51,8 @@ import {
type Actor,
type Evaluator,
type Memory,
- IVerifiableInferenceAdapter,
} from "./types.ts";
import { stringToUuid } from "./uuid.ts";
-import { readFile } from 'fs/promises';
-import { join } from 'path';
/**
* Represents the runtime environment for an agent, handling message processing,
@@ -308,7 +308,7 @@ export class AgentRuntime implements IAgentRuntime {
this.ragKnowledgeManager = new RAGKnowledgeManager({
runtime: this,
- tableName: 'knowledge'
+ tableName: "knowledge",
});
(opts.managers ?? []).forEach((manager: IMemoryManager) => {
@@ -438,11 +438,13 @@ export class AgentRuntime implements IAgentRuntime {
this.character.knowledge &&
this.character.knowledge.length > 0
) {
- if(this.character.settings.ragKnowledge) {
- await this.processCharacterRAGKnowledge(this.character.knowledge);
+ if (this.character.settings.ragKnowledge) {
+ await this.processCharacterRAGKnowledge(
+ this.character.knowledge
+ );
} else {
- const stringKnowledge = this.character.knowledge.filter((item): item is string =>
- typeof item === 'string'
+ const stringKnowledge = this.character.knowledge.filter(
+ (item): item is string => typeof item === "string"
);
await this.processCharacterKnowledge(stringKnowledge);
@@ -511,19 +513,21 @@ export class AgentRuntime implements IAgentRuntime {
* then chunks the content into fragments, embeds each fragment, and creates fragment knowledge.
* An array of knowledge items or objects containing id, path, and content.
*/
- private async processCharacterRAGKnowledge(items: (string | { path: string; shared?: boolean })[]) {
+ private async processCharacterRAGKnowledge(
+ items: (string | { path: string; shared?: boolean })[]
+ ) {
let hasError = false;
for (const item of items) {
if (!item) continue;
try {
- // Check if item is marked as shared
+ // Check if item is marked as shared
let isShared = false;
let contentItem = item;
// Only treat as shared if explicitly marked
- if (typeof item === 'object' && 'path' in item) {
+ if (typeof item === "object" && "path" in item) {
isShared = item.shared === true;
contentItem = item.path;
} else {
@@ -531,25 +535,40 @@ export class AgentRuntime implements IAgentRuntime {
}
const knowledgeId = stringToUuid(contentItem);
- const fileExtension = contentItem.split('.').pop()?.toLowerCase();
+ const fileExtension = contentItem
+ .split(".")
+ .pop()
+ ?.toLowerCase();
// Check if it's a file or direct knowledge
- if (fileExtension && ['md', 'txt', 'pdf'].includes(fileExtension)) {
+ if (
+ fileExtension &&
+ ["md", "txt", "pdf"].includes(fileExtension)
+ ) {
try {
- const rootPath = join(process.cwd(), '..');
- const filePath = join(rootPath, 'characters', 'knowledge', contentItem);
- elizaLogger.info("Attempting to read file from:", filePath);
+ const rootPath = join(process.cwd(), "..");
+ const filePath = join(
+ rootPath,
+ "characters",
+ "knowledge",
+ contentItem
+ );
+ elizaLogger.info(
+ "Attempting to read file from:",
+ filePath
+ );
// Get existing knowledge first
- const existingKnowledge = await this.ragKnowledgeManager.getKnowledge({
- id: knowledgeId,
- agentId: this.agentId
- });
-
- let content: string;
-
- content = await readFile(filePath, 'utf8');
+ const existingKnowledge =
+ await this.ragKnowledgeManager.getKnowledge({
+ id: knowledgeId,
+ agentId: this.agentId,
+ });
+ const content: string = await readFile(
+ filePath,
+ "utf8"
+ );
if (!content) {
hasError = true;
continue;
@@ -557,15 +576,23 @@ export class AgentRuntime implements IAgentRuntime {
// If the file exists in DB, check if content has changed
if (existingKnowledge.length > 0) {
- const existingContent = existingKnowledge[0].content.text;
+ const existingContent =
+ existingKnowledge[0].content.text;
if (existingContent === content) {
- elizaLogger.info(`File ${contentItem} unchanged, skipping`);
+ elizaLogger.info(
+ `File ${contentItem} unchanged, skipping`
+ );
continue;
} else {
// If content changed, remove old knowledge before adding new
- await this.ragKnowledgeManager.removeKnowledge(knowledgeId);
- // Also remove any associated chunks
- await this.ragKnowledgeManager.removeKnowledge(`${knowledgeId}-chunk-*` as UUID);
+ await this.ragKnowledgeManager.removeKnowledge(
+ knowledgeId
+ );
+ // Also remove any associated chunks - this is needed for non-PostgreSQL adapters
+ // PostgreSQL adapter handles chunks internally via foreign keys
+ await this.ragKnowledgeManager.removeKnowledge(
+ `${knowledgeId}-chunk-*` as UUID
+ );
}
}
@@ -579,15 +606,14 @@ export class AgentRuntime implements IAgentRuntime {
await this.ragKnowledgeManager.processFile({
path: contentItem,
content: content,
- type: fileExtension as 'pdf' | 'md' | 'txt',
- isShared: isShared
+ type: fileExtension as "pdf" | "md" | "txt",
+ isShared: isShared,
});
-
} catch (error: any) {
hasError = true;
elizaLogger.error(
`Failed to read knowledge file ${contentItem}. Error details:`,
- error?.message || error || 'Unknown error'
+ error?.message || error || "Unknown error"
);
continue; // Continue to next item even if this one fails
}
@@ -600,13 +626,16 @@ export class AgentRuntime implements IAgentRuntime {
contentItem.slice(0, 100)
);
- const existingKnowledge = await this.ragKnowledgeManager.getKnowledge({
- id: knowledgeId,
- agentId: this.agentId
- });
+ const existingKnowledge =
+ await this.ragKnowledgeManager.getKnowledge({
+ id: knowledgeId,
+ agentId: this.agentId,
+ });
if (existingKnowledge.length > 0) {
- elizaLogger.info(`Direct knowledge ${knowledgeId} already exists, skipping`);
+ elizaLogger.info(
+ `Direct knowledge ${knowledgeId} already exists, skipping`
+ );
continue;
}
@@ -616,23 +645,25 @@ export class AgentRuntime implements IAgentRuntime {
content: {
text: contentItem,
metadata: {
- type: 'direct'
- }
- }
+ type: "direct",
+ },
+ },
});
}
} catch (error: any) {
hasError = true;
elizaLogger.error(
`Error processing knowledge item ${item}:`,
- error?.message || error || 'Unknown error'
+ error?.message || error || "Unknown error"
);
continue; // Continue to next item even if this one fails
}
}
if (hasError) {
- elizaLogger.warn('Some knowledge items failed to process, but continuing with available knowledge');
+ elizaLogger.warn(
+ "Some knowledge items failed to process, but continuing with available knowledge"
+ );
}
}
@@ -1102,21 +1133,11 @@ Text: ${attachment.text}
]);
// Check the existing memories in the database
- const existingMemories =
- await this.messageManager.getMemoriesByRoomIds({
- // filter out the current room id from rooms
- roomIds: rooms.filter((room) => room !== roomId),
- });
-
- // Sort messages by timestamp in descending order
- existingMemories.sort(
- (a, b) =>
- (b?.createdAt ?? Date.now()) - (a?.createdAt ?? Date.now())
- );
-
- // Take the most recent messages
- const recentInteractionsData = existingMemories.slice(0, 20);
- return recentInteractionsData;
+ return this.messageManager.getMemoriesByRoomIds({
+ // filter out the current room id from rooms
+ roomIds: rooms.filter((room) => room !== roomId),
+ limit: 20,
+ });
};
const recentInteractions =
@@ -1180,18 +1201,18 @@ Text: ${attachment.text}
}
let knowledgeData = [];
- let formattedKnowledge = '';
+ let formattedKnowledge = "";
- if(this.character.settings?.ragKnowledge) {
+ if (this.character.settings?.ragKnowledge) {
const recentContext = recentMessagesData
- .slice(-3) // Last 3 messages
- .map(msg => msg.content.text)
- .join(' ');
+ .slice(-3) // Last 3 messages
+ .map((msg) => msg.content.text)
+ .join(" ");
knowledgeData = await this.ragKnowledgeManager.getKnowledge({
query: message.content.text,
conversationContext: recentContext,
- limit: 5
+ limit: 5,
});
formattedKnowledge = formatKnowledge(knowledgeData);
diff --git a/packages/core/src/types.ts b/packages/core/src/types.ts
index 1411ca7b0cb..acb92b44e24 100644
--- a/packages/core/src/types.ts
+++ b/packages/core/src/types.ts
@@ -210,6 +210,7 @@ export type Models = {
[ModelProviderName.TOGETHER]: Model;
[ModelProviderName.LLAMALOCAL]: Model;
[ModelProviderName.GOOGLE]: Model;
+ [ModelProviderName.MISTRAL]: Model;
[ModelProviderName.CLAUDE_VERTEX]: Model;
[ModelProviderName.REDPILL]: Model;
[ModelProviderName.OPENROUTER]: Model;
@@ -226,6 +227,8 @@ export type Models = {
[ModelProviderName.NINETEEN_AI]: Model;
[ModelProviderName.AKASH_CHAT_API]: Model;
[ModelProviderName.LIVEPEER]: Model;
+ [ModelProviderName.DEEPSEEK]: Model;
+ [ModelProviderName.INFERA]: Model;
};
/**
@@ -241,6 +244,7 @@ export enum ModelProviderName {
TOGETHER = "together",
LLAMALOCAL = "llama_local",
GOOGLE = "google",
+ MISTRAL = "mistral",
CLAUDE_VERTEX = "claude_vertex",
REDPILL = "redpill",
OPENROUTER = "openrouter",
@@ -258,7 +262,8 @@ export enum ModelProviderName {
AKASH_CHAT_API = "akash_chat_api",
LIVEPEER = "livepeer",
LETZAI = "letzai",
- INFERA = "infera",
+ DEEPSEEK="deepseek",
+ INFERA="infera"
}
/**
@@ -643,6 +648,7 @@ export enum Clients {
LENS = "lens",
AUTO = "auto",
SLACK = "slack",
+ GITHUB = "github",
}
export interface IAgentConfig {
@@ -867,6 +873,8 @@ export type Character = {
nft?: {
prompt: string;
};
+ /**Optinal Parent characters to inherit information from */
+ extends?: string[];
};
/**
@@ -905,6 +913,7 @@ export interface IDatabaseAdapter {
tableName: string;
agentId: UUID;
roomIds: UUID[];
+ limit?: number;
}): Promise;
getCachedEmbeddings(params: {
@@ -1078,7 +1087,7 @@ export interface IMemoryManager {
): Promise<{ embedding: number[]; levenshtein_score: number }[]>;
getMemoryById(id: UUID): Promise;
- getMemoriesByRoomIds(params: { roomIds: UUID[] }): Promise;
+ getMemoriesByRoomIds(params: { roomIds: UUID[], limit?: number }): Promise;
searchMemoriesByEmbedding(
embedding: number[],
opts: {
@@ -1333,9 +1342,56 @@ export interface IAwsS3Service extends Service {
generateSignedUrl(fileName: string, expiresIn: number): Promise;
}
+export interface UploadIrysResult {
+ success: boolean;
+ url?: string;
+ error?: string;
+ data?: any;
+}
+
+export interface DataIrysFetchedFromGQL {
+ success: boolean;
+ data: any;
+ error?: string;
+}
+
+export interface GraphQLTag {
+ name: string;
+ values: any[];
+}
+
+export const enum IrysMessageType {
+ REQUEST = "REQUEST",
+ DATA_STORAGE = "DATA_STORAGE",
+ REQUEST_RESPONSE = "REQUEST_RESPONSE",
+}
+
+export const enum IrysDataType {
+ FILE = "FILE",
+ IMAGE = "IMAGE",
+ OTHER = "OTHER",
+}
+
+export interface IrysTimestamp {
+ from: number;
+ to: number;
+}
+
+export interface IIrysService extends Service {
+ getDataFromAnAgent(agentsWalletPublicKeys: string[], tags: GraphQLTag[], timestamp: IrysTimestamp): Promise;
+ workerUploadDataOnIrys(data: any, dataType: IrysDataType, messageType: IrysMessageType, serviceCategory: string[], protocol: string[], validationThreshold: number[], minimumProviders: number[], testProvider: boolean[], reputation: number[]): Promise;
+ providerUploadDataOnIrys(data: any, dataType: IrysDataType, serviceCategory: string[], protocol: string[]): Promise;
+}
+
export interface ITeeLogService extends Service {
getInstance(): ITeeLogService;
- log(agentId: string, roomId: string, userId: string, type: string, content: string): Promise;
+ log(
+ agentId: string,
+ roomId: string,
+ userId: string,
+ type: string,
+ content: string
+ ): Promise;
}
export type SearchImage = {
@@ -1373,6 +1429,7 @@ export enum ServiceType {
BUTTPLUG = "buttplug",
SLACK = "slack",
VERIFIABLE_LOGGING = "verifiable_logging",
+ IRYS = "irys",
TEE_LOG = "tee_log",
GOPLUS_SECURITY = "goplus_security",
}
@@ -1425,7 +1482,9 @@ export interface ISlackService extends Service {
* Available verifiable inference providers
*/
export enum VerifiableInferenceProvider {
+ RECLAIM = "reclaim",
OPACITY = "opacity",
+ PRIMUS = "primus",
}
/**
@@ -1496,4 +1555,4 @@ export enum TranscriptionProvider {
export enum ActionTimelineType {
ForYou = "foryou",
Following = "following",
-}
+}
\ No newline at end of file
diff --git a/packages/core/src/uuid.ts b/packages/core/src/uuid.ts
index 2227eca2132..dee5decfff2 100644
--- a/packages/core/src/uuid.ts
+++ b/packages/core/src/uuid.ts
@@ -1,5 +1,13 @@
import { sha1 } from "js-sha1";
import { UUID } from "./types.ts";
+import { z } from "zod";
+
+export const uuidSchema = z.string().uuid() as z.ZodType;
+
+export function validateUuid(value: unknown): UUID | null {
+ const result = uuidSchema.safeParse(value);
+ return result.success ? result.data : null;
+}
export function stringToUuid(target: string | number): UUID {
if (typeof target === "number") {
diff --git a/packages/core/types.ts b/packages/core/types.ts
deleted file mode 100644
index 8fb9e2814bd..00000000000
--- a/packages/core/types.ts
+++ /dev/null
@@ -1,1332 +0,0 @@
-import { Readable } from "stream";
-
-/**
- * Represents a UUID string in the format "xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx"
- */
-export type UUID = `${string}-${string}-${string}-${string}-${string}`;
-
-/**
- * Represents the content of a message or communication
- */
-export interface Content {
- /** The main text content */
- text: string;
-
- /** Optional action associated with the message */
- action?: string;
-
- /** Optional source/origin of the content */
- source?: string;
-
- /** URL of the original message/post (e.g. tweet URL, Discord message link) */
- url?: string;
-
- /** UUID of parent message if this is a reply/thread */
- inReplyTo?: UUID;
-
- /** Array of media attachments */
- attachments?: Media[];
-
- /** Additional dynamic properties */
- [key: string]: unknown;
-}
-
-/**
- * Example content with associated user for demonstration purposes
- */
-export interface ActionExample {
- /** User associated with the example */
- user: string;
-
- /** Content of the example */
- content: Content;
-}
-
-/**
- * Example conversation content with user ID
- */
-export interface ConversationExample {
- /** UUID of user in conversation */
- userId: UUID;
-
- /** Content of the conversation */
- content: Content;
-}
-
-/**
- * Represents an actor/participant in a conversation
- */
-export interface Actor {
- /** Display name */
- name: string;
-
- /** Username/handle */
- username: string;
-
- /** Additional profile details */
- details: {
- /** Short profile tagline */
- tagline: string;
-
- /** Longer profile summary */
- summary: string;
-
- /** Favorite quote */
- quote: string;
- };
-
- /** Unique identifier */
- id: UUID;
-}
-
-/**
- * Represents a single objective within a goal
- */
-export interface Objective {
- /** Optional unique identifier */
- id?: string;
-
- /** Description of what needs to be achieved */
- description: string;
-
- /** Whether objective is completed */
- completed: boolean;
-}
-
-/**
- * Status enum for goals
- */
-export enum GoalStatus {
- DONE = "DONE",
- FAILED = "FAILED",
- IN_PROGRESS = "IN_PROGRESS",
-}
-
-/**
- * Represents a high-level goal composed of objectives
- */
-export interface Goal {
- /** Optional unique identifier */
- id?: UUID;
-
- /** Room ID where goal exists */
- roomId: UUID;
-
- /** User ID of goal owner */
- userId: UUID;
-
- /** Name/title of the goal */
- name: string;
-
- /** Current status */
- status: GoalStatus;
-
- /** Component objectives */
- objectives: Objective[];
-}
-
-/**
- * Model size/type classification
- */
-export enum ModelClass {
- SMALL = "small",
- MEDIUM = "medium",
- LARGE = "large",
- EMBEDDING = "embedding",
- IMAGE = "image",
-}
-
-/**
- * Configuration for an AI model
- */
-export type Model = {
- /** Optional API endpoint */
- endpoint?: string;
-
- /** Model settings */
- settings: {
- /** Maximum input tokens */
- maxInputTokens: number;
-
- /** Maximum output tokens */
- maxOutputTokens: number;
-
- /** Optional frequency penalty */
- frequency_penalty?: number;
-
- /** Optional presence penalty */
- presence_penalty?: number;
-
- /** Optional repetition penalty */
- repetition_penalty?: number;
-
- /** Stop sequences */
- stop: string[];
-
- /** Temperature setting */
- temperature: number;
-
- /** Optional telemetry configuration (experimental) */
- experimental_telemetry?: TelemetrySettings;
- };
-
- /** Optional image generation settings */
- imageSettings?: {
- steps?: number;
- };
-
- /** Model names by size class */
- model: {
- [ModelClass.SMALL]: string;
- [ModelClass.MEDIUM]: string;
- [ModelClass.LARGE]: string;
- [ModelClass.EMBEDDING]?: string;
- [ModelClass.IMAGE]?: string;
- };
-};
-
-/**
- * Model configurations by provider
- */
-export type Models = {
- [ModelProviderName.OPENAI]: Model;
- [ModelProviderName.ETERNALAI]: Model;
- [ModelProviderName.ANTHROPIC]: Model;
- [ModelProviderName.GROK]: Model;
- [ModelProviderName.GROQ]: Model;
- [ModelProviderName.LLAMACLOUD]: Model;
- [ModelProviderName.TOGETHER]: Model;
- [ModelProviderName.LLAMALOCAL]: Model;
- [ModelProviderName.GOOGLE]: Model;
- [ModelProviderName.CLAUDE_VERTEX]: Model;
- [ModelProviderName.REDPILL]: Model;
- [ModelProviderName.OPENROUTER]: Model;
- [ModelProviderName.OLLAMA]: Model;
- [ModelProviderName.HEURIST]: Model;
- [ModelProviderName.GALADRIEL]: Model;
- [ModelProviderName.FAL]: Model;
- [ModelProviderName.GAIANET]: Model;
- [ModelProviderName.ALI_BAILIAN]: Model;
- [ModelProviderName.VOLENGINE]: Model;
- [ModelProviderName.NANOGPT]: Model;
- [ModelProviderName.HYPERBOLIC]: Model;
- [ModelProviderName.VENICE]: Model;
- [ModelProviderName.AKASH_CHAT_API]: Model;
- [ModelProviderName.LIVEPEER]: Model;
- [ModelProviderName.INFERA]: Model;
-};
-
-/**
- * Available model providers
- */
-export enum ModelProviderName {
- OPENAI = "openai",
- ETERNALAI = "eternalai",
- ANTHROPIC = "anthropic",
- GROK = "grok",
- GROQ = "groq",
- LLAMACLOUD = "llama_cloud",
- TOGETHER = "together",
- LLAMALOCAL = "llama_local",
- GOOGLE = "google",
- CLAUDE_VERTEX = "claude_vertex",
- REDPILL = "redpill",
- OPENROUTER = "openrouter",
- OLLAMA = "ollama",
- HEURIST = "heurist",
- GALADRIEL = "galadriel",
- FAL = "falai",
- GAIANET = "gaianet",
- ALI_BAILIAN = "ali_bailian",
- VOLENGINE = "volengine",
- NANOGPT = "nanogpt",
- HYPERBOLIC = "hyperbolic",
- VENICE = "venice",
- AKASH_CHAT_API = "akash_chat_api",
- LIVEPEER = "livepeer",
- INFERA = "infera",
-}
-
-/**
- * Represents the current state/context of a conversation
- */
-export interface State {
- /** ID of user who sent current message */
- userId?: UUID;
-
- /** ID of agent in conversation */
- agentId?: UUID;
-
- /** Agent's biography */
- bio: string;
-
- /** Agent's background lore */
- lore: string;
-
- /** Message handling directions */
- messageDirections: string;
-
- /** Post handling directions */
- postDirections: string;
-
- /** Current room/conversation ID */
- roomId: UUID;
-
- /** Optional agent name */
- agentName?: string;
-
- /** Optional message sender name */
- senderName?: string;
-
- /** String representation of conversation actors */
- actors: string;
-
- /** Optional array of actor objects */
- actorsData?: Actor[];
-
- /** Optional string representation of goals */
- goals?: string;
-
- /** Optional array of goal objects */
- goalsData?: Goal[];
-
- /** Recent message history as string */
- recentMessages: string;
-
- /** Recent message objects */
- recentMessagesData: Memory[];
-
- /** Optional valid action names */
- actionNames?: string;
-
- /** Optional action descriptions */
- actions?: string;
-
- /** Optional action objects */
- actionsData?: Action[];
-
- /** Optional action examples */
- actionExamples?: string;
-
- /** Optional provider descriptions */
- providers?: string;
-
- /** Optional response content */
- responseData?: Content;
-
- /** Optional recent interaction objects */
- recentInteractionsData?: Memory[];
-
- /** Optional recent interactions string */
- recentInteractions?: string;
-
- /** Optional formatted conversation */
- formattedConversation?: string;
-
- /** Optional formatted knowledge */
- knowledge?: string;
- /** Optional knowledge data */
- knowledgeData?: KnowledgeItem[];
-
- /** Additional dynamic properties */
- [key: string]: unknown;
-}
-
-/**
- * Represents a stored memory/message
- */
-export interface Memory {
- /** Optional unique identifier */
- id?: UUID;
-
- /** Associated user ID */
- userId: UUID;
-
- /** Associated agent ID */
- agentId: UUID;
-
- /** Optional creation timestamp */
- createdAt?: number;
-
- /** Memory content */
- content: Content;
-
- /** Optional embedding vector */
- embedding?: number[];
-
- /** Associated room ID */
- roomId: UUID;
-
- /** Whether memory is unique */
- unique?: boolean;
-
- /** Embedding similarity score */
- similarity?: number;
-}
-
-/**
- * Example message for demonstration
- */
-export interface MessageExample {
- /** Associated user */
- user: string;
-
- /** Message content */
- content: Content;
-}
-
-/**
- * Handler function type for processing messages
- */
-export type Handler = (
- runtime: IAgentRuntime,
- message: Memory,
- state?: State,
- options?: { [key: string]: unknown },
- callback?: HandlerCallback
-) => Promise;
-
-/**
- * Callback function type for handlers
- */
-export type HandlerCallback = (
- response: Content,
- files?: any
-) => Promise;
-
-/**
- * Validator function type for actions/evaluators
- */
-export type Validator = (
- runtime: IAgentRuntime,
- message: Memory,
- state?: State
-) => Promise;
-
-/**
- * Represents an action the agent can perform
- */
-export interface Action {
- /** Similar action descriptions */
- similes: string[];
-
- /** Detailed description */
- description: string;
-
- /** Example usages */
- examples: ActionExample[][];
-
- /** Handler function */
- handler: Handler;
-
- /** Action name */
- name: string;
-
- /** Validation function */
- validate: Validator;
-
- /** Whether to suppress the initial message when this action is used */
- suppressInitialMessage?: boolean;
-}
-
-/**
- * Example for evaluating agent behavior
- */
-export interface EvaluationExample {
- /** Evaluation context */
- context: string;
-
- /** Example messages */
- messages: Array;
-
- /** Expected outcome */
- outcome: string;
-}
-
-/**
- * Evaluator for assessing agent responses
- */
-export interface Evaluator {
- /** Whether to always run */
- alwaysRun?: boolean;
-
- /** Detailed description */
- description: string;
-
- /** Similar evaluator descriptions */
- similes: string[];
-
- /** Example evaluations */
- examples: EvaluationExample[];
-
- /** Handler function */
- handler: Handler;
-
- /** Evaluator name */
- name: string;
-
- /** Validation function */
- validate: Validator;
-}
-
-/**
- * Provider for external data/services
- */
-export interface Provider {
- /** Data retrieval function */
- get: (
- runtime: IAgentRuntime,
- message: Memory,
- state?: State
- ) => Promise;
-}
-
-/**
- * Represents a relationship between users
- */
-export interface Relationship {
- /** Unique identifier */
- id: UUID;
-
- /** First user ID */
- userA: UUID;
-
- /** Second user ID */
- userB: UUID;
-
- /** Primary user ID */
- userId: UUID;
-
- /** Associated room ID */
- roomId: UUID;
-
- /** Relationship status */
- status: string;
-
- /** Optional creation timestamp */
- createdAt?: string;
-}
-
-/**
- * Represents a user account
- */
-export interface Account {
- /** Unique identifier */
- id: UUID;
-
- /** Display name */
- name: string;
-
- /** Username */
- username: string;
-
- /** Optional additional details */
- details?: { [key: string]: any };
-
- /** Optional email */
- email?: string;
-
- /** Optional avatar URL */
- avatarUrl?: string;
-}
-
-/**
- * Room participant with account details
- */
-export interface Participant {
- /** Unique identifier */
- id: UUID;
-
- /** Associated account */
- account: Account;
-}
-
-/**
- * Represents a conversation room
- */
-export interface Room {
- /** Unique identifier */
- id: UUID;
-
- /** Room participants */
- participants: Participant[];
-}
-
-/**
- * Represents a media attachment
- */
-export type Media = {
- /** Unique identifier */
- id: string;
-
- /** Media URL */
- url: string;
-
- /** Media title */
- title: string;
-
- /** Media source */
- source: string;
-
- /** Media description */
- description: string;
-
- /** Text content */
- text: string;
-
- /** Content type */
- contentType?: string;
-};
-
-/**
- * Client interface for platform connections
- */
-export type Client = {
- /** Start client connection */
- start: (runtime: IAgentRuntime) => Promise;
-
- /** Stop client connection */
- stop: (runtime: IAgentRuntime) => Promise;
-};
-
-/**
- * Plugin for extending agent functionality
- */
-export type Plugin = {
- /** Plugin name */
- name: string;
-
- /** Plugin description */
- description: string;
-
- /** Optional actions */
- actions?: Action[];
-
- /** Optional providers */
- providers?: Provider[];
-
- /** Optional evaluators */
- evaluators?: Evaluator[];
-
- /** Optional services */
- services?: Service[];
-
- /** Optional clients */
- clients?: Client[];
-};
-
-/**
- * Available client platforms
- */
-export enum Clients {
- DISCORD = "discord",
- DIRECT = "direct",
- TWITTER = "twitter",
- TELEGRAM = "telegram",
- FARCASTER = "farcaster",
- LENS = "lens",
- AUTO = "auto",
- SLACK = "slack",
-}
-
-export interface IAgentConfig {
- [key: string]: string;
-}
-
-export type TelemetrySettings = {
- /**
- * Enable or disable telemetry. Disabled by default while experimental.
- */
- isEnabled?: boolean;
- /**
- * Enable or disable input recording. Enabled by default.
- *
- * You might want to disable input recording to avoid recording sensitive
- * information, to reduce data transfers, or to increase performance.
- */
- recordInputs?: boolean;
- /**
- * Enable or disable output recording. Enabled by default.
- *
- * You might want to disable output recording to avoid recording sensitive
- * information, to reduce data transfers, or to increase performance.
- */
- recordOutputs?: boolean;
- /**
- * Identifier for this function. Used to group telemetry data by function.
- */
- functionId?: string;
-};
-
-export interface ModelConfiguration {
- temperature?: number;
- max_response_length?: number;
- frequency_penalty?: number;
- presence_penalty?: number;
- maxInputTokens?: number;
- experimental_telemetry?: TelemetrySettings;
-}
-
-/**
- * Configuration for an agent character
- */
-export type Character = {
- /** Optional unique identifier */
- id?: UUID;
-
- /** Character name */
- name: string;
-
- /** Optional username */
- username?: string;
-
- /** Optional system prompt */
- system?: string;
-
- /** Model provider to use */
- modelProvider: ModelProviderName;
-
- /** Image model provider to use, if different from modelProvider */
- imageModelProvider?: ModelProviderName;
-
- /** Image Vision model provider to use, if different from modelProvider */
- imageVisionModelProvider?: ModelProviderName;
-
- /** Optional model endpoint override */
- modelEndpointOverride?: string;
-
- /** Optional prompt templates */
- templates?: {
- goalsTemplate?: string;
- factsTemplate?: string;
- messageHandlerTemplate?: string;
- shouldRespondTemplate?: string;
- continueMessageHandlerTemplate?: string;
- evaluationTemplate?: string;
- twitterSearchTemplate?: string;
- twitterActionTemplate?: string;
- twitterPostTemplate?: string;
- twitterMessageHandlerTemplate?: string;
- twitterShouldRespondTemplate?: string;
- farcasterPostTemplate?: string;
- lensPostTemplate?: string;
- farcasterMessageHandlerTemplate?: string;
- lensMessageHandlerTemplate?: string;
- farcasterShouldRespondTemplate?: string;
- lensShouldRespondTemplate?: string;
- telegramMessageHandlerTemplate?: string;
- telegramShouldRespondTemplate?: string;
- discordVoiceHandlerTemplate?: string;
- discordShouldRespondTemplate?: string;
- discordMessageHandlerTemplate?: string;
- slackMessageHandlerTemplate?: string;
- slackShouldRespondTemplate?: string;
- };
-
- /** Character biography */
- bio: string | string[];
-
- /** Character background lore */
- lore: string[];
-
- /** Example messages */
- messageExamples: MessageExample[][];
-
- /** Example posts */
- postExamples: string[];
-
- /** Known topics */
- topics: string[];
-
- /** Character traits */
- adjectives: string[];
-
- /** Optional knowledge base */
- knowledge?: string[];
-
- /** Supported client platforms */
- clients: Clients[];
-
- /** Available plugins */
- plugins: Plugin[];
-
- /** Optional configuration */
- settings?: {
- secrets?: { [key: string]: string };
- intiface?: boolean;
- imageSettings?: {
- steps?: number;
- width?: number;
- height?: number;
- negativePrompt?: string;
- numIterations?: number;
- guidanceScale?: number;
- seed?: number;
- modelId?: string;
- jobId?: string;
- count?: number;
- stylePreset?: string;
- hideWatermark?: boolean;
- };
- voice?: {
- model?: string; // For VITS
- url?: string; // Legacy VITS support
- elevenlabs?: {
- // New structured ElevenLabs config
- voiceId: string;
- model?: string;
- stability?: string;
- similarityBoost?: string;
- style?: string;
- useSpeakerBoost?: string;
- };
- };
- model?: string;
- modelConfig?: ModelConfiguration;
- embeddingModel?: string;
- chains?: {
- evm?: any[];
- solana?: any[];
- [key: string]: any[];
- };
- transcription?: TranscriptionProvider;
- };
-
- /** Optional client-specific config */
- clientConfig?: {
- discord?: {
- shouldIgnoreBotMessages?: boolean;
- shouldIgnoreDirectMessages?: boolean;
- shouldRespondOnlyToMentions?: boolean;
- messageSimilarityThreshold?: number;
- isPartOfTeam?: boolean;
- teamAgentIds?: string[];
- teamLeaderId?: string;
- teamMemberInterestKeywords?: string[];
- };
- telegram?: {
- shouldIgnoreBotMessages?: boolean;
- shouldIgnoreDirectMessages?: boolean;
- shouldRespondOnlyToMentions?: boolean;
- shouldOnlyJoinInAllowedGroups?: boolean;
- allowedGroupIds?: string[];
- messageSimilarityThreshold?: number;
- isPartOfTeam?: boolean;
- teamAgentIds?: string[];
- teamLeaderId?: string;
- teamMemberInterestKeywords?: string[];
- };
- slack?: {
- shouldIgnoreBotMessages?: boolean;
- shouldIgnoreDirectMessages?: boolean;
- };
- gitbook?: {
- keywords?: {
- projectTerms?: string[];
- generalQueries?: string[];
- };
- documentTriggers?: string[];
- };
- };
-
- /** Writing style guides */
- style: {
- all: string[];
- chat: string[];
- post: string[];
- };
-
- /** Optional Twitter profile */
- twitterProfile?: {
- id: string;
- username: string;
- screenName: string;
- bio: string;
- nicknames?: string[];
- };
- /** Optional NFT prompt */
- nft?: {
- prompt: string;
- };
-};
-
-/**
- * Interface for database operations
- */
-export interface IDatabaseAdapter {
- /** Database instance */
- db: any;
-
- /** Optional initialization */
- init(): Promise;
-
- /** Close database connection */
- close(): Promise;
-
- /** Get account by ID */
- getAccountById(userId: UUID): Promise;
-
- /** Create new account */
- createAccount(account: Account): Promise;
-
- /** Get memories matching criteria */
- getMemories(params: {
- roomId: UUID;
- count?: number;
- unique?: boolean;
- tableName: string;
- agentId: UUID;
- start?: number;
- end?: number;
- }): Promise;
-
- getMemoryById(id: UUID): Promise;
-
- getMemoriesByRoomIds(params: {
- tableName: string;
- agentId: UUID;
- roomIds: UUID[];
- }): Promise;
-
- getCachedEmbeddings(params: {
- query_table_name: string;
- query_threshold: number;
- query_input: string;
- query_field_name: string;
- query_field_sub_name: string;
- query_match_count: number;
- }): Promise<{ embedding: number[]; levenshtein_score: number }[]>;
-
- log(params: {
- body: { [key: string]: unknown };
- userId: UUID;
- roomId: UUID;
- type: string;
- }): Promise;
-
- getActorDetails(params: { roomId: UUID }): Promise;
-
- searchMemories(params: {
- tableName: string;
- agentId: UUID;
- roomId: UUID;
- embedding: number[];
- match_threshold: number;
- match_count: number;
- unique: boolean;
- }): Promise;
-
- updateGoalStatus(params: {
- goalId: UUID;
- status: GoalStatus;
- }): Promise;
-
- searchMemoriesByEmbedding(
- embedding: number[],
- params: {
- match_threshold?: number;
- count?: number;
- roomId?: UUID;
- agentId?: UUID;
- unique?: boolean;
- tableName: string;
- }
- ): Promise;
-
- createMemory(
- memory: Memory,
- tableName: string,
- unique?: boolean
- ): Promise;
-
- removeMemory(memoryId: UUID, tableName: string): Promise;
-
- removeAllMemories(roomId: UUID, tableName: string): Promise;
-
- countMemories(
- roomId: UUID,
- unique?: boolean,
- tableName?: string
- ): Promise;
-
- getGoals(params: {
- agentId: UUID;
- roomId: UUID;
- userId?: UUID | null;
- onlyInProgress?: boolean;
- count?: number;
- }): Promise;
-
- updateGoal(goal: Goal): Promise;
-
- createGoal(goal: Goal): Promise;
-
- removeGoal(goalId: UUID): Promise;
-
- removeAllGoals(roomId: UUID): Promise;
-
- getRoom(roomId: UUID): Promise;
-
- createRoom(roomId?: UUID): Promise;
-
- removeRoom(roomId: UUID): Promise;
-
- getRoomsForParticipant(userId: UUID): Promise;
-
- getRoomsForParticipants(userIds: UUID[]): Promise;
-
- addParticipant(userId: UUID, roomId: UUID): Promise;
-
- removeParticipant(userId: UUID, roomId: UUID): Promise;
-
- getParticipantsForAccount(userId: UUID): Promise;
-
- getParticipantsForRoom(roomId: UUID): Promise;
-
- getParticipantUserState(
- roomId: UUID,
- userId: UUID
- ): Promise<"FOLLOWED" | "MUTED" | null>;
-
- setParticipantUserState(
- roomId: UUID,
- userId: UUID,
- state: "FOLLOWED" | "MUTED" | null
- ): Promise;
-
- createRelationship(params: { userA: UUID; userB: UUID }): Promise;
-
- getRelationship(params: {
- userA: UUID;
- userB: UUID;
- }): Promise;
-
- getRelationships(params: { userId: UUID }): Promise;
-}
-
-export interface IDatabaseCacheAdapter {
- getCache(params: {
- agentId: UUID;
- key: string;
- }): Promise;
-
- setCache(params: {
- agentId: UUID;
- key: string;
- value: string;
- }): Promise;
-
- deleteCache(params: { agentId: UUID; key: string }): Promise;
-}
-
-export interface IMemoryManager {
- runtime: IAgentRuntime;
- tableName: string;
- constructor: Function;
-
- addEmbeddingToMemory(memory: Memory): Promise;
-
- getMemories(opts: {
- roomId: UUID;
- count?: number;
- unique?: boolean;
- start?: number;
- end?: number;
- }): Promise;
-
- getCachedEmbeddings(
- content: string
- ): Promise<{ embedding: number[]; levenshtein_score: number }[]>;
-
- getMemoryById(id: UUID): Promise;
- getMemoriesByRoomIds(params: { roomIds: UUID[] }): Promise;
- searchMemoriesByEmbedding(
- embedding: number[],
- opts: {
- match_threshold?: number;
- count?: number;
- roomId: UUID;
- unique?: boolean;
- }
- ): Promise;
-
- createMemory(memory: Memory, unique?: boolean): Promise;
-
- removeMemory(memoryId: UUID): Promise;
-
- removeAllMemories(roomId: UUID): Promise;
-
- countMemories(roomId: UUID, unique?: boolean): Promise;
-}
-
-export type CacheOptions = {
- expires?: number;
-};
-
-export enum CacheStore {
- REDIS = "redis",
- DATABASE = "database",
- FILESYSTEM = "filesystem",
-}
-
-export interface ICacheManager {
- get(key: string): Promise;
- set(key: string, value: T, options?: CacheOptions): Promise;
- delete(key: string): Promise;
-}
-
-export abstract class Service {
- private static instance: Service | null = null;
-
- static get serviceType(): ServiceType {
- throw new Error("Service must implement static serviceType getter");
- }
-
- public static getInstance(): T {
- if (!Service.instance) {
- Service.instance = new (this as any)();
- }
- return Service.instance as T;
- }
-
- get serviceType(): ServiceType {
- return (this.constructor as typeof Service).serviceType;
- }
-
- // Add abstract initialize method that must be implemented by derived classes
- abstract initialize(runtime: IAgentRuntime): Promise;
-}
-
-export interface IAgentRuntime {
- // Properties
- agentId: UUID;
- serverUrl: string;
- databaseAdapter: IDatabaseAdapter;
- token: string | null;
- modelProvider: ModelProviderName;
- imageModelProvider: ModelProviderName;
- imageVisionModelProvider: ModelProviderName;
- character: Character;
- providers: Provider[];
- actions: Action[];
- evaluators: Evaluator[];
- plugins: Plugin[];
-
- fetch?: typeof fetch | null;
-
- messageManager: IMemoryManager;
- descriptionManager: IMemoryManager;
- documentsManager: IMemoryManager;
- knowledgeManager: IMemoryManager;
- loreManager: IMemoryManager;
-
- cacheManager: ICacheManager;
-
- services: Map;
- // any could be EventEmitter
- // but I think the real solution is forthcoming as a base client interface
- clients: Record;
-
- initialize(): Promise;
-
- registerMemoryManager(manager: IMemoryManager): void;
-
- getMemoryManager(name: string): IMemoryManager | null;
-
- getService(service: ServiceType): T | null;
-
- registerService(service: Service): void;
-
- getSetting(key: string): string | null;
-
- // Methods
- getConversationLength(): number;
-
- processActions(
- message: Memory,
- responses: Memory[],
- state?: State,
- callback?: HandlerCallback
- ): Promise;
-
- evaluate(
- message: Memory,
- state?: State,
- didRespond?: boolean,
- callback?: HandlerCallback
- ): Promise;
-
- ensureParticipantExists(userId: UUID, roomId: UUID): Promise;
-
- ensureUserExists(
- userId: UUID,
- userName: string | null,
- name: string | null,
- source: string | null
- ): Promise;
-
- registerAction(action: Action): void;
-
- ensureConnection(
- userId: UUID,
- roomId: UUID,
- userName?: string,
- userScreenName?: string,
- source?: string
- ): Promise;
-
- ensureParticipantInRoom(userId: UUID, roomId: UUID): Promise;
-
- ensureRoomExists(roomId: UUID): Promise;
-
- composeState(
- message: Memory,
- additionalKeys?: { [key: string]: unknown }
- ): Promise;
-
- updateRecentMessageState(state: State): Promise;
-}
-
-export interface IImageDescriptionService extends Service {
- describeImage(
- imageUrl: string
- ): Promise<{ title: string; description: string }>;
-}
-
-export interface ITranscriptionService extends Service {
- transcribeAttachment(audioBuffer: ArrayBuffer): Promise;
- transcribeAttachmentLocally(
- audioBuffer: ArrayBuffer
- ): Promise;
- transcribe(audioBuffer: ArrayBuffer): Promise;
- transcribeLocally(audioBuffer: ArrayBuffer): Promise;
-}
-
-export interface IVideoService extends Service {
- isVideoUrl(url: string): boolean;
- fetchVideoInfo(url: string): Promise;
- downloadVideo(videoInfo: Media): Promise;
- processVideo(url: string, runtime: IAgentRuntime): Promise;
-}
-
-export interface ITextGenerationService extends Service {
- initializeModel(): Promise;
- queueMessageCompletion(
- context: string,
- temperature: number,
- stop: string[],
- frequency_penalty: number,
- presence_penalty: number,
- max_tokens: number
- ): Promise;
- queueTextCompletion(
- context: string,
- temperature: number,
- stop: string[],
- frequency_penalty: number,
- presence_penalty: number,
- max_tokens: number
- ): Promise;
- getEmbeddingResponse(input: string): Promise;
-}
-
-export interface IBrowserService extends Service {
- closeBrowser(): Promise;
- getPageContent(
- url: string,
- runtime: IAgentRuntime
- ): Promise<{ title: string; description: string; bodyContent: string }>;
-}
-
-export interface ISpeechService extends Service {
- getInstance(): ISpeechService;
- generate(runtime: IAgentRuntime, text: string): Promise;
-}
-
-export interface IPdfService extends Service {
- getInstance(): IPdfService;
- convertPdfToText(pdfBuffer: Buffer): Promise;
-}
-
-export interface IAwsS3Service extends Service {
- uploadFile(
- imagePath: string,
- subDirectory: string,
- useSignedUrl: boolean,
- expiresIn: number
- ): Promise<{
- success: boolean;
- url?: string;
- error?: string;
- }>;
- generateSignedUrl(fileName: string, expiresIn: number): Promise;
-}
-
-export type SearchImage = {
- url: string;
- description?: string;
-};
-
-export type SearchResult = {
- title: string;
- url: string;
- content: string;
- rawContent?: string;
- score: number;
- publishedDate?: string;
-};
-
-export type SearchResponse = {
- answer?: string;
- query: string;
- responseTime: number;
- images: SearchImage[];
- results: SearchResult[];
-};
-
-export enum ServiceType {
- IMAGE_DESCRIPTION = "image_description",
- TRANSCRIPTION = "transcription",
- VIDEO = "video",
- TEXT_GENERATION = "text_generation",
- BROWSER = "browser",
- SPEECH_GENERATION = "speech_generation",
- PDF = "pdf",
- INTIFACE = "intiface",
- AWS_S3 = "aws_s3",
- BUTTPLUG = "buttplug",
- SLACK = "slack",
-}
-
-export enum LoggingLevel {
- DEBUG = "debug",
- VERBOSE = "verbose",
- NONE = "none",
-}
-
-export type KnowledgeItem = {
- id: UUID;
- content: Content;
-};
-
-export interface ActionResponse {
- like: boolean;
- retweet: boolean;
- quote?: boolean;
- reply?: boolean;
-}
-
-export interface ISlackService extends Service {
- client: any;
-}
-
-export enum TokenizerType {
- Auto = "auto",
- TikToken = "tiktoken",
-}
-
-export enum TranscriptionProvider {
- OpenAI = "openai",
- Deepgram = "deepgram",
- Local = "local",
-}
diff --git a/packages/create-eliza-app/package.json b/packages/create-eliza-app/package.json
index 1fc01090aaf..5b4bf39fbf6 100644
--- a/packages/create-eliza-app/package.json
+++ b/packages/create-eliza-app/package.json
@@ -1,6 +1,6 @@
{
"name": "create-eliza-app",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"description": "",
"sideEffects": false,
"files": [
diff --git a/packages/plugin-0g/package.json b/packages/plugin-0g/package.json
index 757328f725d..13d3579cd50 100644
--- a/packages/plugin-0g/package.json
+++ b/packages/plugin-0g/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/plugin-0g",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
diff --git a/packages/plugin-0g/src/actions/upload.ts b/packages/plugin-0g/src/actions/upload.ts
index cb24317a516..13b72f1652b 100644
--- a/packages/plugin-0g/src/actions/upload.ts
+++ b/packages/plugin-0g/src/actions/upload.ts
@@ -8,12 +8,15 @@ import {
Content,
ActionExample,
generateObject,
+ elizaLogger,
} from "@elizaos/core";
import { Indexer, ZgFile, getFlowContract } from "@0glabs/0g-ts-sdk";
import { ethers } from "ethers";
import { composeContext } from "@elizaos/core";
import { promises as fs } from "fs";
-
+import { FileSecurityValidator } from "../utils/security";
+import { logSecurityEvent, monitorUpload, monitorFileValidation, monitorCleanup } from '../utils/monitoring';
+import path from 'path';
import { uploadTemplate } from "../templates/upload";
export interface UploadContent extends Content {
@@ -24,7 +27,7 @@ function isUploadContent(
_runtime: IAgentRuntime,
content: any
): content is UploadContent {
- console.log("Content for upload", content);
+ elizaLogger.debug("Validating upload content", { content });
return typeof content.filePath === "string";
}
@@ -41,12 +44,76 @@ export const zgUpload: Action = {
],
description: "Store data using 0G protocol",
validate: async (runtime: IAgentRuntime, message: Memory) => {
- const zgIndexerRpc = !!runtime.getSetting("ZEROG_INDEXER_RPC");
- const zgEvmRpc = !!runtime.getSetting("ZEROG_EVM_RPC");
- const zgPrivateKey = !!runtime.getSetting("ZEROG_PRIVATE_KEY");
- const flowAddr = !!runtime.getSetting("ZEROG_FLOW_ADDRESS");
- return zgIndexerRpc && zgEvmRpc && zgPrivateKey && flowAddr;
+ elizaLogger.debug("Starting ZG_UPLOAD validation", { messageId: message.id });
+
+ try {
+ const settings = {
+ indexerRpc: runtime.getSetting("ZEROG_INDEXER_RPC"),
+ evmRpc: runtime.getSetting("ZEROG_EVM_RPC"),
+ privateKey: runtime.getSetting("ZEROG_PRIVATE_KEY"),
+ flowAddr: runtime.getSetting("ZEROG_FLOW_ADDRESS")
+ };
+
+ elizaLogger.debug("Checking ZeroG settings", {
+ hasIndexerRpc: Boolean(settings.indexerRpc),
+ hasEvmRpc: Boolean(settings.evmRpc),
+ hasPrivateKey: Boolean(settings.privateKey),
+ hasFlowAddr: Boolean(settings.flowAddr)
+ });
+
+ const hasRequiredSettings = Object.entries(settings).every(([key, value]) => Boolean(value));
+
+ if (!hasRequiredSettings) {
+ const missingSettings = Object.entries(settings)
+ .filter(([_, value]) => !value)
+ .map(([key]) => key);
+
+ elizaLogger.error("Missing required ZeroG settings", {
+ missingSettings,
+ messageId: message.id
+ });
+ return false;
+ }
+
+ const config = {
+ maxFileSize: parseInt(runtime.getSetting("ZEROG_MAX_FILE_SIZE") || "10485760"),
+ allowedExtensions: runtime.getSetting("ZEROG_ALLOWED_EXTENSIONS")?.split(",") || [".pdf", ".png", ".jpg", ".jpeg", ".doc", ".docx"],
+ uploadDirectory: runtime.getSetting("ZEROG_UPLOAD_DIR") || "/tmp/zerog-uploads",
+ enableVirusScan: runtime.getSetting("ZEROG_ENABLE_VIRUS_SCAN") === "true"
+ };
+
+ // Validate config values
+ if (isNaN(config.maxFileSize) || config.maxFileSize <= 0) {
+ elizaLogger.error("Invalid ZEROG_MAX_FILE_SIZE setting", {
+ value: runtime.getSetting("ZEROG_MAX_FILE_SIZE"),
+ messageId: message.id
+ });
+ return false;
+ }
+
+ if (!config.allowedExtensions || config.allowedExtensions.length === 0) {
+ elizaLogger.error("Invalid ZEROG_ALLOWED_EXTENSIONS setting", {
+ value: runtime.getSetting("ZEROG_ALLOWED_EXTENSIONS"),
+ messageId: message.id
+ });
+ return false;
+ }
+
+ elizaLogger.info("ZG_UPLOAD action settings validated", {
+ config,
+ messageId: message.id
+ });
+ return true;
+ } catch (error) {
+ elizaLogger.error("Error validating ZG_UPLOAD settings", {
+ error: error instanceof Error ? error.message : String(error),
+ stack: error instanceof Error ? error.stack : undefined,
+ messageId: message.id
+ });
+ return false;
+ }
},
+
handler: async (
runtime: IAgentRuntime,
message: Memory,
@@ -54,90 +121,358 @@ export const zgUpload: Action = {
_options: any,
callback: HandlerCallback
) => {
- console.log("ZG_UPLOAD action called");
- if (!state) {
- state = (await runtime.composeState(message)) as State;
- } else {
- state = await runtime.updateRecentMessageState(state);
- }
-
- // Compose upload context
- const uploadContext = composeContext({
- state,
- template: uploadTemplate,
+ elizaLogger.info("ZG_UPLOAD action started", {
+ messageId: message.id,
+ hasState: Boolean(state),
+ hasCallback: Boolean(callback)
});
- // Generate upload content
- const content = await generateObject({
- runtime,
- context: uploadContext,
- modelClass: ModelClass.LARGE,
- });
+ let file: ZgFile | undefined;
+ let cleanupRequired = false;
- // Validate upload content
- if (!isUploadContent(runtime, content)) {
- console.error("Invalid content for UPLOAD action.");
- if (callback) {
- callback({
- text: "Unable to process 0G upload request. Invalid content provided.",
- content: { error: "Invalid upload content" },
+ try {
+ // Update state if needed
+ if (!state) {
+ elizaLogger.debug("No state provided, composing new state");
+ state = (await runtime.composeState(message)) as State;
+ } else {
+ elizaLogger.debug("Updating existing state");
+ state = await runtime.updateRecentMessageState(state);
+ }
+
+ // Compose upload context
+ elizaLogger.debug("Composing upload context");
+ const uploadContext = composeContext({
+ state,
+ template: uploadTemplate,
+ });
+
+ // Generate upload content
+ elizaLogger.debug("Generating upload content");
+ const content = await generateObject({
+ runtime,
+ context: uploadContext,
+ modelClass: ModelClass.LARGE,
+ });
+
+ // Validate upload content
+ if (!isUploadContent(runtime, content)) {
+ const error = "Invalid content for UPLOAD action";
+ elizaLogger.error(error, {
+ content,
+ messageId: message.id
});
+ if (callback) {
+ callback({
+ text: "Unable to process 0G upload request. Invalid content provided.",
+ content: { error }
+ });
+ }
+ return false;
}
- return false;
- }
- try {
- const zgIndexerRpc = runtime.getSetting("ZEROG_INDEXER_RPC");
- const zgEvmRpc = runtime.getSetting("ZEROG_EVM_RPC");
- const zgPrivateKey = runtime.getSetting("ZEROG_PRIVATE_KEY");
- const flowAddr = runtime.getSetting("ZEROG_FLOW_ADDRESS");
const filePath = content.filePath;
+ elizaLogger.debug("Extracted file path", { filePath, content });
+
if (!filePath) {
- console.error("File path is required");
+ const error = "File path is required";
+ elizaLogger.error(error, { messageId: message.id });
+ if (callback) {
+ callback({
+ text: "File path is required for upload.",
+ content: { error }
+ });
+ }
return false;
}
- // Check if file exists and is accessible
+ // Initialize security validator
+ const securityConfig = {
+ maxFileSize: parseInt(runtime.getSetting("ZEROG_MAX_FILE_SIZE") || "10485760"),
+ allowedExtensions: runtime.getSetting("ZEROG_ALLOWED_EXTENSIONS")?.split(",") || [".pdf", ".png", ".jpg", ".jpeg", ".doc", ".docx"],
+ uploadDirectory: runtime.getSetting("ZEROG_UPLOAD_DIR") || "/tmp/zerog-uploads",
+ enableVirusScan: runtime.getSetting("ZEROG_ENABLE_VIRUS_SCAN") === "true"
+ };
+
+ let validator: FileSecurityValidator;
try {
- await fs.access(filePath);
+ elizaLogger.debug("Initializing security validator", {
+ config: securityConfig,
+ messageId: message.id
+ });
+ validator = new FileSecurityValidator(securityConfig);
} catch (error) {
- console.error(
- `File ${filePath} does not exist or is not accessible:`,
- error
- );
+ const errorMessage = `Security validator initialization failed: ${error instanceof Error ? error.message : String(error)}`;
+ elizaLogger.error(errorMessage, {
+ config: securityConfig,
+ messageId: message.id
+ });
+ if (callback) {
+ callback({
+ text: "Upload failed: Security configuration error.",
+ content: { error: errorMessage }
+ });
+ }
return false;
}
- const file = await ZgFile.fromFilePath(filePath);
- var [tree, err] = await file.merkleTree();
- if (err === null) {
- console.log("File Root Hash: ", tree.rootHash());
- } else {
- console.log("Error getting file root hash: ", err);
+ // Validate file type
+ elizaLogger.debug("Starting file type validation", { filePath });
+ const typeValidation = await validator.validateFileType(filePath);
+ monitorFileValidation(filePath, "file_type", typeValidation.isValid, {
+ error: typeValidation.error
+ });
+ if (!typeValidation.isValid) {
+ const error = "File type validation failed";
+ elizaLogger.error(error, {
+ error: typeValidation.error,
+ filePath,
+ messageId: message.id
+ });
+ if (callback) {
+ callback({
+ text: `Upload failed: ${typeValidation.error}`,
+ content: { error: typeValidation.error }
+ });
+ }
return false;
}
- const provider = new ethers.JsonRpcProvider(zgEvmRpc);
- const signer = new ethers.Wallet(zgPrivateKey, provider);
- const indexer = new Indexer(zgIndexerRpc);
- const flowContract = getFlowContract(flowAddr, signer);
-
- var [tx, err] = await indexer.upload(
- file,
- 0,
- zgEvmRpc,
- flowContract
- );
- if (err === null) {
- console.log("File uploaded successfully, tx: ", tx);
- } else {
- console.error("Error uploading file: ", err);
+ // Validate file size
+ elizaLogger.debug("Starting file size validation", { filePath });
+ const sizeValidation = await validator.validateFileSize(filePath);
+ monitorFileValidation(filePath, "file_size", sizeValidation.isValid, {
+ error: sizeValidation.error
+ });
+ if (!sizeValidation.isValid) {
+ const error = "File size validation failed";
+ elizaLogger.error(error, {
+ error: sizeValidation.error,
+ filePath,
+ messageId: message.id
+ });
+ if (callback) {
+ callback({
+ text: `Upload failed: ${sizeValidation.error}`,
+ content: { error: sizeValidation.error }
+ });
+ }
+ return false;
+ }
+
+ // Validate file path
+ elizaLogger.debug("Starting file path validation", { filePath });
+ const pathValidation = await validator.validateFilePath(filePath);
+ monitorFileValidation(filePath, "file_path", pathValidation.isValid, {
+ error: pathValidation.error
+ });
+ if (!pathValidation.isValid) {
+ const error = "File path validation failed";
+ elizaLogger.error(error, {
+ error: pathValidation.error,
+ filePath,
+ messageId: message.id
+ });
+ if (callback) {
+ callback({
+ text: `Upload failed: ${pathValidation.error}`,
+ content: { error: pathValidation.error }
+ });
+ }
+ return false;
+ }
+
+ // Sanitize the file path
+ let sanitizedPath: string;
+ try {
+ sanitizedPath = validator.sanitizePath(filePath);
+ elizaLogger.debug("File path sanitized", {
+ originalPath: filePath,
+ sanitizedPath,
+ messageId: message.id
+ });
+ } catch (error) {
+ const errorMessage = `Failed to sanitize file path: ${error instanceof Error ? error.message : String(error)}`;
+ elizaLogger.error(errorMessage, {
+ filePath,
+ messageId: message.id
+ });
+ if (callback) {
+ callback({
+ text: "Upload failed: Invalid file path.",
+ content: { error: errorMessage }
+ });
+ }
return false;
}
- await file.close();
+ // Start upload monitoring
+ const startTime = Date.now();
+ let fileStats;
+ try {
+ fileStats = await fs.stat(sanitizedPath);
+ elizaLogger.debug("File stats retrieved", {
+ size: fileStats.size,
+ path: sanitizedPath,
+ created: fileStats.birthtime,
+ modified: fileStats.mtime,
+ messageId: message.id
+ });
+ } catch (error) {
+ const errorMessage = `Failed to get file stats: ${error instanceof Error ? error.message : String(error)}`;
+ elizaLogger.error(errorMessage, {
+ path: sanitizedPath,
+ messageId: message.id
+ });
+ if (callback) {
+ callback({
+ text: "Upload failed: Could not access file",
+ content: { error: errorMessage }
+ });
+ }
+ return false;
+ }
+
+ try {
+ // Initialize ZeroG file
+ elizaLogger.debug("Initializing ZeroG file", {
+ sanitizedPath,
+ messageId: message.id
+ });
+ file = await ZgFile.fromFilePath(sanitizedPath);
+ cleanupRequired = true;
+
+ // Generate Merkle tree
+ elizaLogger.debug("Generating Merkle tree");
+ const [merkleTree, merkleError] = await file.merkleTree();
+ if (merkleError !== null) {
+ const error = `Error getting file root hash: ${merkleError instanceof Error ? merkleError.message : String(merkleError)}`;
+ elizaLogger.error(error, { messageId: message.id });
+ if (callback) {
+ callback({
+ text: "Upload failed: Error generating file hash.",
+ content: { error }
+ });
+ }
+ return false;
+ }
+ elizaLogger.info("File root hash generated", {
+ rootHash: merkleTree.rootHash(),
+ messageId: message.id
+ });
+
+ // Initialize blockchain connection
+ elizaLogger.debug("Initializing blockchain connection");
+ const provider = new ethers.JsonRpcProvider(runtime.getSetting("ZEROG_EVM_RPC"));
+ const signer = new ethers.Wallet(runtime.getSetting("ZEROG_PRIVATE_KEY"), provider);
+ const indexer = new Indexer(runtime.getSetting("ZEROG_INDEXER_RPC"));
+ const flowContract = getFlowContract(runtime.getSetting("ZEROG_FLOW_ADDRESS"), signer);
+
+ // Upload file to ZeroG
+ elizaLogger.info("Starting file upload to ZeroG", {
+ filePath: sanitizedPath,
+ messageId: message.id
+ });
+ const [txHash, uploadError] = await indexer.upload(
+ file,
+ 0,
+ runtime.getSetting("ZEROG_EVM_RPC"),
+ flowContract
+ );
+
+ if (uploadError !== null) {
+ const error = `Error uploading file: ${uploadError instanceof Error ? uploadError.message : String(uploadError)}`;
+ elizaLogger.error(error, { messageId: message.id });
+ monitorUpload({
+ filePath: sanitizedPath,
+ size: fileStats.size,
+ duration: Date.now() - startTime,
+ success: false,
+ error: error
+ });
+ if (callback) {
+ callback({
+ text: "Upload failed: Error during file upload.",
+ content: { error }
+ });
+ }
+ return false;
+ }
+
+ // Log successful upload
+ monitorUpload({
+ filePath: sanitizedPath,
+ size: fileStats.size,
+ duration: Date.now() - startTime,
+ success: true
+ });
+
+ elizaLogger.info("File uploaded successfully", {
+ transactionHash: txHash,
+ filePath: sanitizedPath,
+ fileSize: fileStats.size,
+ duration: Date.now() - startTime,
+ messageId: message.id
+ });
+
+ if (callback) {
+ callback({
+ text: "File uploaded successfully to ZeroG.",
+ content: {
+ success: true,
+ transactionHash: txHash
+ }
+ });
+ }
+
+ return true;
+ } finally {
+ // Cleanup temporary file
+ if (cleanupRequired && file) {
+ try {
+ elizaLogger.debug("Starting file cleanup", {
+ filePath: sanitizedPath,
+ messageId: message.id
+ });
+ await file.close();
+ await fs.unlink(sanitizedPath);
+ monitorCleanup(sanitizedPath, true);
+ elizaLogger.debug("File cleanup completed successfully", {
+ filePath: sanitizedPath,
+ messageId: message.id
+ });
+ } catch (cleanupError) {
+ monitorCleanup(sanitizedPath, false, cleanupError.message);
+ elizaLogger.warn("Failed to cleanup file", {
+ error: cleanupError instanceof Error ? cleanupError.message : String(cleanupError),
+ filePath: sanitizedPath,
+ messageId: message.id
+ });
+ }
+ }
+ }
} catch (error) {
- console.error("Error getting settings for 0G upload:", error);
+ const errorMessage = error instanceof Error ? error.message : String(error);
+ logSecurityEvent("Unexpected error in upload action", "high", {
+ error: errorMessage,
+ stack: error instanceof Error ? error.stack : undefined,
+ messageId: message.id
+ });
+
+ elizaLogger.error("Unexpected error during file upload", {
+ error: errorMessage,
+ stack: error instanceof Error ? error.stack : undefined,
+ messageId: message.id
+ });
+
+ if (callback) {
+ callback({
+ text: "Upload failed due to an unexpected error.",
+ content: { error: errorMessage }
+ });
+ }
+
+ return false;
}
},
examples: [
diff --git a/packages/plugin-0g/src/utils/monitoring.ts b/packages/plugin-0g/src/utils/monitoring.ts
new file mode 100644
index 00000000000..e502a564176
--- /dev/null
+++ b/packages/plugin-0g/src/utils/monitoring.ts
@@ -0,0 +1,96 @@
+import { elizaLogger } from '@elizaos/core';
+
+export interface SecurityEvent {
+ timestamp: number;
+ event: string;
+ severity: 'low' | 'medium' | 'high';
+ details: Record;
+}
+
+export interface UploadMetrics {
+ filePath: string;
+ size: number;
+ timestamp: string;
+ duration?: number;
+ success: boolean;
+ error?: string;
+}
+
+/**
+ * Logs a security event with the specified severity and details
+ */
+export const logSecurityEvent = (
+ event: string,
+ severity: SecurityEvent['severity'],
+ details: Record
+): void => {
+ const securityEvent: SecurityEvent = {
+ timestamp: Date.now(),
+ event,
+ severity,
+ details
+ };
+
+ elizaLogger.info('Security event', securityEvent);
+
+ // For high severity events, also log as error
+ if (severity === 'high') {
+ elizaLogger.error('High severity security event', securityEvent);
+ }
+};
+
+/**
+ * Tracks upload metrics and logs them
+ */
+export const monitorUpload = (metrics: Omit): void => {
+ const uploadMetrics: UploadMetrics = {
+ ...metrics,
+ timestamp: new Date().toISOString()
+ };
+
+ elizaLogger.info('Upload metrics', uploadMetrics);
+
+ // Log errors if present
+ if (!metrics.success && metrics.error) {
+ elizaLogger.error('Upload failed', {
+ filePath: metrics.filePath,
+ error: metrics.error
+ });
+ }
+};
+
+/**
+ * Monitors file validation events
+ */
+export const monitorFileValidation = (
+ filePath: string,
+ validationType: string,
+ isValid: boolean,
+ details?: Record
+): void => {
+ const event = isValid ? 'File validation passed' : 'File validation failed';
+ const severity = isValid ? 'low' : 'medium';
+
+ logSecurityEvent(event, severity, {
+ filePath,
+ validationType,
+ ...details
+ });
+};
+
+/**
+ * Tracks cleanup operations
+ */
+export const monitorCleanup = (
+ filePath: string,
+ success: boolean,
+ error?: string
+): void => {
+ const event = success ? 'File cleanup succeeded' : 'File cleanup failed';
+ const severity = success ? 'low' : 'medium';
+
+ logSecurityEvent(event, severity, {
+ filePath,
+ error
+ });
+};
\ No newline at end of file
diff --git a/packages/plugin-0g/src/utils/security.ts b/packages/plugin-0g/src/utils/security.ts
new file mode 100644
index 00000000000..b555d7c7c55
--- /dev/null
+++ b/packages/plugin-0g/src/utils/security.ts
@@ -0,0 +1,182 @@
+import { promises as fs } from 'fs';
+import path from 'path';
+
+export interface SecurityConfig {
+ maxFileSize: number;
+ allowedExtensions: string[];
+ uploadDirectory: string;
+ enableVirusScan: boolean;
+}
+
+export interface ValidationResult {
+ isValid: boolean;
+ error?: string;
+}
+
+export class FileSecurityValidator {
+ private config: SecurityConfig;
+
+ constructor(config: SecurityConfig) {
+ if (!config.allowedExtensions || config.allowedExtensions.length === 0) {
+ throw new Error('Security configuration error: allowedExtensions must be specified');
+ }
+ if (!config.uploadDirectory) {
+ throw new Error('Security configuration error: uploadDirectory must be specified');
+ }
+ if (config.maxFileSize <= 0) {
+ throw new Error('Security configuration error: maxFileSize must be positive');
+ }
+ this.config = config;
+ }
+
+ async validateFileType(filePath: string): Promise {
+ try {
+ if (!filePath) {
+ return {
+ isValid: false,
+ error: 'Invalid file path: Path cannot be empty'
+ };
+ }
+
+ const ext = path.extname(filePath).toLowerCase();
+ if (!ext) {
+ return {
+ isValid: false,
+ error: `File type not allowed. Allowed types: ${this.config.allowedExtensions.join(', ')}`
+ };
+ }
+
+ if (!this.config.allowedExtensions.includes(ext)) {
+ return {
+ isValid: false,
+ error: `File type not allowed. Allowed types: ${this.config.allowedExtensions.join(', ')}`
+ };
+ }
+ return { isValid: true };
+ } catch (error) {
+ return {
+ isValid: false,
+ error: `Error validating file type: ${error instanceof Error ? error.message : String(error)}`
+ };
+ }
+ }
+
+ async validateFileSize(filePath: string): Promise {
+ try {
+ if (!filePath) {
+ return {
+ isValid: false,
+ error: 'Invalid file path: Path cannot be empty'
+ };
+ }
+
+ const stats = await fs.stat(filePath);
+ if (stats.size === 0) {
+ return {
+ isValid: false,
+ error: 'Invalid file: File is empty'
+ };
+ }
+
+ if (stats.size > this.config.maxFileSize) {
+ return {
+ isValid: false,
+ error: `File size exceeds limit of ${this.config.maxFileSize} bytes (file size: ${stats.size} bytes)`
+ };
+ }
+ return { isValid: true };
+ } catch (error) {
+ if (error.code === 'ENOENT') {
+ return {
+ isValid: false,
+ error: 'File not found or inaccessible'
+ };
+ }
+ if (error.code === 'EACCES') {
+ return {
+ isValid: false,
+ error: 'Permission denied: Cannot access file'
+ };
+ }
+ return {
+ isValid: false,
+ error: `Error checking file size: ${error instanceof Error ? error.message : String(error)}`
+ };
+ }
+ }
+
+ async validateFilePath(filePath: string): Promise {
+ try {
+ if (!filePath) {
+ return {
+ isValid: false,
+ error: 'Invalid file path: Path cannot be empty'
+ };
+ }
+
+ const normalizedPath = path.normalize(filePath);
+
+ // Check for directory traversal attempts
+ if (normalizedPath.includes('..')) {
+ return {
+ isValid: false,
+ error: 'Invalid file path: Directory traversal detected'
+ };
+ }
+
+ // For test files, we'll allow them to be created in the test directory
+ if (normalizedPath.includes('__test_files__')) {
+ return { isValid: true };
+ }
+
+ // For production files, ensure they're in the upload directory
+ const uploadDir = path.normalize(this.config.uploadDirectory);
+
+ // Check if upload directory exists and is accessible
+ try {
+ await fs.access(uploadDir, fs.constants.W_OK);
+ } catch (error) {
+ return {
+ isValid: false,
+ error: `Upload directory is not accessible: ${error.code === 'ENOENT' ? 'Directory does not exist' :
+ error.code === 'EACCES' ? 'Permission denied' : error.message}`
+ };
+ }
+
+ if (!normalizedPath.startsWith(uploadDir)) {
+ return {
+ isValid: false,
+ error: 'Invalid file path: File must be within the upload directory'
+ };
+ }
+
+ return { isValid: true };
+ } catch (error) {
+ return {
+ isValid: false,
+ error: `Error validating file path: ${error instanceof Error ? error.message : String(error)}`
+ };
+ }
+ }
+
+ sanitizePath(filePath: string): string {
+ try {
+ if (!filePath) {
+ throw new Error('File path cannot be empty');
+ }
+
+ // Remove any directory traversal attempts
+ const normalizedPath = path.normalize(filePath).replace(/^(\.\.[\/\\])+/, '');
+
+ // If it's a test path, preserve it
+ if (normalizedPath.includes('__test_files__') || !normalizedPath.startsWith(this.config.uploadDirectory)) {
+ return normalizedPath;
+ }
+
+ // For production paths, ensure they're in the upload directory
+ return path.join(this.config.uploadDirectory, path.basename(normalizedPath));
+ } catch (error) {
+ throw new Error(`Error sanitizing file path: ${error instanceof Error ? error.message : String(error)}`);
+ }
+ }
+}
\ No newline at end of file
diff --git a/packages/plugin-3d-generation/package.json b/packages/plugin-3d-generation/package.json
index c20d3a3e4dd..17bced90926 100644
--- a/packages/plugin-3d-generation/package.json
+++ b/packages/plugin-3d-generation/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/plugin-3d-generation",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
diff --git a/packages/plugin-abstract/package.json b/packages/plugin-abstract/package.json
index 8775a6a462f..6ba15a41d87 100644
--- a/packages/plugin-abstract/package.json
+++ b/packages/plugin-abstract/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/plugin-abstract",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
@@ -29,4 +29,4 @@
"peerDependencies": {
"whatwg-url": "7.1.0"
}
-}
\ No newline at end of file
+}
diff --git a/packages/plugin-akash/.eslintrc.js b/packages/plugin-akash/.eslintrc.js
new file mode 100644
index 00000000000..e476cac57e6
--- /dev/null
+++ b/packages/plugin-akash/.eslintrc.js
@@ -0,0 +1,29 @@
+module.exports = {
+ root: true,
+ parser: '@typescript-eslint/parser',
+ parserOptions: {
+ project: './tsconfig.json',
+ tsconfigRootDir: __dirname,
+ ecmaVersion: 2020,
+ sourceType: 'module',
+ },
+ plugins: ['@typescript-eslint'],
+ extends: [
+ 'eslint:recommended',
+ 'plugin:@typescript-eslint/recommended',
+ 'plugin:@typescript-eslint/recommended-requiring-type-checking',
+ ],
+ rules: {
+ '@typescript-eslint/no-explicit-any': 'warn',
+ '@typescript-eslint/no-unused-vars': ['error', {
+ argsIgnorePattern: '^_',
+ varsIgnorePattern: '^_',
+ ignoreRestSiblings: true,
+ }],
+ '@typescript-eslint/explicit-function-return-type': 'off',
+ '@typescript-eslint/explicit-module-boundary-types': 'off',
+ '@typescript-eslint/no-non-null-assertion': 'warn',
+ 'no-console': ['error', { allow: ['warn', 'error'] }],
+ },
+ ignorePatterns: ['dist/', 'node_modules/', '*.js', '*.mjs', '*.cjs'],
+};
\ No newline at end of file
diff --git a/packages/plugin-akash/.npmignore b/packages/plugin-akash/.npmignore
new file mode 100644
index 00000000000..078562eceab
--- /dev/null
+++ b/packages/plugin-akash/.npmignore
@@ -0,0 +1,6 @@
+*
+
+!dist/**
+!package.json
+!readme.md
+!tsup.config.ts
\ No newline at end of file
diff --git a/packages/plugin-akash/assets/akash.jpg b/packages/plugin-akash/assets/akash.jpg
new file mode 100644
index 00000000000..dd08e0e5705
Binary files /dev/null and b/packages/plugin-akash/assets/akash.jpg differ
diff --git a/packages/plugin-akash/eslint.config.mjs b/packages/plugin-akash/eslint.config.mjs
new file mode 100644
index 00000000000..92fe5bbebef
--- /dev/null
+++ b/packages/plugin-akash/eslint.config.mjs
@@ -0,0 +1,3 @@
+import eslintGlobalConfig from "../../eslint.config.mjs";
+
+export default [...eslintGlobalConfig];
diff --git a/packages/plugin-akash/jest.config.js b/packages/plugin-akash/jest.config.js
new file mode 100644
index 00000000000..a8331cee2ff
--- /dev/null
+++ b/packages/plugin-akash/jest.config.js
@@ -0,0 +1,31 @@
+module.exports = {
+ preset: 'ts-jest',
+ testEnvironment: 'node',
+ roots: ['/test'],
+ testMatch: [
+ "**/__tests__/**/*.+(ts|tsx|js)",
+ "**/?(*.)+(spec|test).+(ts|tsx|js)"
+ ],
+ transform: {
+ "^.+\\.(ts|tsx)$": "ts-jest"
+ },
+ moduleNameMapper: {
+ '^@/(.*)$': '/src/$1'
+ },
+ setupFilesAfterEnv: ['/test/setup/jest.setup.ts'],
+ globals: {
+ 'ts-jest': {
+ tsconfig: 'tsconfig.json'
+ }
+ },
+ testTimeout: 30000,
+ verbose: true,
+ collectCoverage: true,
+ coverageDirectory: "coverage",
+ coverageReporters: ["text", "lcov"],
+ coveragePathIgnorePatterns: [
+ "/node_modules/",
+ "/test/fixtures/",
+ "/test/setup/"
+ ]
+};
\ No newline at end of file
diff --git a/packages/plugin-akash/package.json b/packages/plugin-akash/package.json
new file mode 100644
index 00000000000..88a1244876f
--- /dev/null
+++ b/packages/plugin-akash/package.json
@@ -0,0 +1,51 @@
+{
+ "name": "@elizaos/plugin-akash",
+ "version": "0.1.8+build.1",
+ "description": "Akash Network Plugin for Eliza",
+ "main": "dist/index.js",
+ "types": "dist/index.d.ts",
+ "type": "module",
+ "scripts": {
+ "build": "tsup",
+ "dev": "tsup --watch",
+ "clean": "rm -rf dist",
+ "lint": "eslint .",
+ "lint:fix": "eslint . --fix",
+ "test": "vitest",
+ "test:watch": "vitest watch",
+ "test:coverage": "vitest run --coverage",
+ "test:ui": "vitest --ui"
+ },
+ "dependencies": {
+ "@akashnetwork/akash-api": "^1.4.0",
+ "@akashnetwork/akashjs": "0.10.1",
+ "@cosmjs/proto-signing": "^0.31.3",
+ "@cosmjs/stargate": "0.31.3",
+ "@elizaos/core": "workspace:*",
+ "@types/js-yaml": "^4.0.9",
+ "axios": "^1.7.9",
+ "dotenv": "^16.4.1",
+ "jsrsasign": "^11.1.0",
+ "node-fetch": "^2.7.0",
+ "zod": "^3.22.4"
+ },
+ "devDependencies": {
+ "@types/dotenv": "^8.2.0",
+ "@types/jest": "^29.5.11",
+ "@types/js-yaml": "^4.0.9",
+ "@types/node": "^20.10.5",
+ "@typescript-eslint/eslint-plugin": "^6.15.0",
+ "@typescript-eslint/parser": "^6.15.0",
+ "@vitest/coverage-v8": "^0.34.6",
+ "@vitest/ui": "^0.34.6",
+ "eslint": "^9.16.0",
+ "tsup": "^8.0.1",
+ "typescript": "^5.3.3",
+ "vite": "^5.0.10",
+ "vite-tsconfig-paths": "^4.2.2",
+ "vitest": "^0.34.6"
+ },
+ "peerDependencies": {
+ "@elizaos/core": "workspace:*"
+ }
+}
diff --git a/packages/plugin-akash/readme.md b/packages/plugin-akash/readme.md
new file mode 100644
index 00000000000..081f353f26b
--- /dev/null
+++ b/packages/plugin-akash/readme.md
@@ -0,0 +1,133 @@
+# Akash Network Plugin for Eliza
+
+A powerful plugin for interacting with the Akash Network, enabling deployment management and cloud compute operations through Eliza.
+
+## Table of Contents
+- [Installation](#installation)
+- [Configuration](#configuration)
+- [Directory Structure](#directory-structure)
+- [Available Actions](#available-actions)
+
+## Installation
+
+```bash
+pnpm add @elizaos/plugin-akash
+```
+
+## Configuration
+
+### Environment Variables
+Create a `.env` file in your project root with the following configuration:
+
+```env
+# Network Configuration
+AKASH_ENV=mainnet
+AKASH_NET=https://raw.githubusercontent.com/ovrclk/net/master/mainnet
+RPC_ENDPOINT=https://rpc.akashnet.net:443
+
+# Transaction Settings
+AKASH_GAS_PRICES=0.025uakt
+AKASH_GAS_ADJUSTMENT=1.5
+AKASH_KEYRING_BACKEND=os
+AKASH_FROM=default
+AKASH_FEES=20000uakt
+
+# Authentication
+AKASH_MNEMONIC=your_12_word_mnemonic_here
+
+# Manifest Settings
+AKASH_MANIFEST_MODE=auto # Options: auto, validate_only
+AKASH_MANIFEST_VALIDATION_LEVEL=strict # Options: strict, basic, none
+AKASH_MANIFEST_PATH=/path/to/manifests # Optional: Path to save generated manifests
+
+# Deployment Settings
+AKASH_DEPOSIT=5000000uakt # Default deployment deposit
+AKASH_SDL=deployment.yml # Default SDL file name
+```
+
+**Important Notes:**
+- `AKASH_MNEMONIC`: Your 12-word wallet mnemonic phrase (required)
+- `AKASH_MANIFEST_MODE`: Controls manifest generation behavior
+- `AKASH_MANIFEST_VALIDATION_LEVEL`: Sets SDL validation strictness
+- `AKASH_DEPOSIT`: Default deposit amount for deployments
+
+⚠️ Never commit your `.env` file with real credentials to version control!
+
+
+#### SDL (Stack Definition Language)
+```
+src/sdl/example.sdl.yml
+```
+Place your SDL configuration files here. The plugin looks for SDL files in this directory by default.
+
+#### Certificates
+```
+src/.certificates/
+```
+SSL certificates for secure provider communication are stored here.
+
+## Available Actions
+
+| Action | Description | Parameters |
+|---------------------|------------------------------------------------|---------------------------------------------|
+| CREATE_DEPLOYMENT | Create a new deployment | `sdl`, `sdlFile`, `deposit` |
+| CLOSE_DEPLOYMENT | Close an existing deployment | `dseq`, `owner` |
+| GET_PROVIDER_INFO | Get provider information | `provider` |
+| GET_DEPLOYMENT_STATUS| Check deployment status | `dseq`, `owner` |
+| GET_GPU_PRICING | Get GPU pricing comparison | `cpu`, `memory`, `storage` |
+| GET_MANIFEST | Generate deployment manifest | `sdl`, `sdlFile` |
+| GET_PROVIDERS_LIST | List available providers | `filter: { active, hasGPU, region }` |
+
+
+Each action returns a structured response with:
+```typescript
+{
+ text: string; // Human-readable response
+ content: {
+ success: boolean; // Operation success status
+ data?: any; // Action-specific data
+ error?: { // Present only on failure
+ code: string;
+ message: string;
+ };
+ metadata: { // Operation metadata
+ timestamp: string;
+ source: string;
+ action: string;
+ version: string;
+ actionId: string;
+ }
+ }
+}
+```
+
+## Error Handling
+
+The plugin includes comprehensive error handling with specific error codes:
+
+- `VALIDATION_SDL_FAILED`: SDL validation errors
+- `WALLET_NOT_INITIALIZED`: Wallet setup issues
+- `DEPLOYMENT_CREATION_FAILED`: Deployment failures
+- `API_REQUEST_FAILED`: Network/API issues
+- `MANIFEST_PARSING_FAILED`: Manifest generation errors
+- `PROVIDER_FILTER_ERROR`: Provider filtering issues
+
+## Development
+
+### Running Tests
+```bash
+pnpm test
+```
+
+### Building
+```bash
+pnpm run build
+```
+
+## License
+
+This project is licensed under the MIT License - see the LICENSE file for details.
+
+## Support
+
+For support and questions, please open an issue in the repository or contact the maintainers.
diff --git a/packages/plugin-akash/src/actions/closeDeployment.ts b/packages/plugin-akash/src/actions/closeDeployment.ts
new file mode 100644
index 00000000000..ee50e0067da
--- /dev/null
+++ b/packages/plugin-akash/src/actions/closeDeployment.ts
@@ -0,0 +1,521 @@
+import { Action, elizaLogger } from "@elizaos/core";
+import { IAgentRuntime, Memory, State, HandlerCallback, Content, ActionExample } from "@elizaos/core";
+import { DirectSecp256k1HdWallet, Registry } from "@cosmjs/proto-signing";
+import { SigningStargateClient } from "@cosmjs/stargate";
+import { getAkashTypeRegistry, getTypeUrl } from "@akashnetwork/akashjs/build/stargate";
+import { MsgCloseDeployment } from "@akashnetwork/akash-api/akash/deployment/v1beta3";
+import { validateAkashConfig } from "../environment";
+import { fetchDeployments } from "./getDeploymentApi";
+import { AkashError, AkashErrorCode } from "../error/error";
+// import { getCertificatePath } from "../utils/paths";
+import { isPluginLoaded } from "../runtime_inspect";
+
+interface CloseDeploymentContent extends Content {
+ dseq?: string;
+ closeAll?: boolean;
+}
+
+// Certificate file path
+// const CERTIFICATE_PATH = getCertificatePath(import.meta.url);
+
+// Initialize wallet and client
+async function initializeClient(runtime: IAgentRuntime) {
+ elizaLogger.info("=== Initializing Client for Deployment Closure ===");
+ const config = await validateAkashConfig(runtime);
+
+ if (!config.AKASH_MNEMONIC) {
+ throw new AkashError(
+ "AKASH_MNEMONIC is required for closing deployments",
+ AkashErrorCode.WALLET_NOT_INITIALIZED
+ );
+ }
+
+ elizaLogger.debug("Initializing wallet", {
+ rpcEndpoint: config.RPC_ENDPOINT,
+ chainId: config.AKASH_CHAIN_ID,
+ version: config.AKASH_VERSION,
+ hasMnemonic: !!config.AKASH_MNEMONIC
+ });
+
+ const wallet = await DirectSecp256k1HdWallet.fromMnemonic(config.AKASH_MNEMONIC, {
+ prefix: "akash"
+ });
+
+ const [account] = await wallet.getAccounts();
+ elizaLogger.debug("Wallet initialized successfully", {
+ address: account.address,
+ prefix: "akash"
+ });
+
+ // Initialize registry and client
+ const myRegistry = new Registry(getAkashTypeRegistry());
+ const client = await SigningStargateClient.connectWithSigner(
+ config.AKASH_NODE || "https://rpc.akash.forbole.com:443",
+ wallet,
+ { registry: myRegistry }
+ );
+
+ elizaLogger.info("Client initialization complete", {
+ nodeUrl: config.AKASH_NODE || "https://rpc.akash.forbole.com:443",
+ address: account.address
+ });
+
+ return { client, account, wallet };
+}
+
+// Verify deployment status before closing
+async function verifyDeploymentStatus(runtime: IAgentRuntime, dseq: string): Promise {
+ elizaLogger.info("Verifying deployment status", { dseq });
+
+ try {
+ const deployments = await fetchDeployments(runtime, undefined, 0, 100);
+ const deployment = deployments.results.find(d => d.dseq === dseq);
+
+ if (!deployment) {
+ throw new AkashError(
+ `Deployment not found with DSEQ: ${dseq}`,
+ AkashErrorCode.DEPLOYMENT_NOT_FOUND
+ );
+ }
+
+ if (deployment.status.toLowerCase() !== 'active') {
+ throw new AkashError(
+ `Deployment ${dseq} is not active (current status: ${deployment.status})`,
+ AkashErrorCode.DEPLOYMENT_CLOSE_FAILED
+ );
+ }
+
+ return true;
+ } catch (error) {
+ if (error instanceof AkashError) {
+ throw error;
+ }
+ throw new AkashError(
+ `Failed to verify deployment status: ${error instanceof Error ? error.message : String(error)}`,
+ AkashErrorCode.DEPLOYMENT_NOT_FOUND
+ );
+ }
+}
+
+// Close a single deployment by DSEQ
+async function closeSingleDeployment(
+ runtime: IAgentRuntime,
+ dseq: string
+): Promise {
+ elizaLogger.info("Closing single deployment", { dseq });
+
+ try {
+ // Verify deployment exists and is active
+ await verifyDeploymentStatus(runtime, dseq);
+
+ const { client, account } = await initializeClient(runtime);
+
+ // Create close deployment message
+ const message = MsgCloseDeployment.fromPartial({
+ id: {
+ dseq: dseq,
+ owner: account.address
+ }
+ });
+
+ const msgAny = {
+ typeUrl: getTypeUrl(MsgCloseDeployment),
+ value: message
+ };
+
+ // Set fee
+ const fee = {
+ amount: [{ denom: "uakt", amount: "20000" }],
+ gas: "800000"
+ };
+
+ // Send transaction
+ elizaLogger.info("Sending close deployment transaction", { dseq });
+ const result = await client.signAndBroadcast(
+ account.address,
+ [msgAny],
+ fee,
+ `close deployment ${dseq}`
+ );
+
+ if (result.code !== 0) {
+ throw new AkashError(
+ `Transaction failed: ${result.rawLog}`,
+ AkashErrorCode.DEPLOYMENT_CLOSE_FAILED,
+ { rawLog: result.rawLog }
+ );
+ }
+
+ elizaLogger.info("Deployment closed successfully", {
+ dseq,
+ transactionHash: result.transactionHash
+ });
+
+ return true;
+ } catch (error) {
+ elizaLogger.error("Failed to close deployment", {
+ dseq,
+ error: error instanceof Error ? error.message : String(error),
+ code: error instanceof AkashError ? error.code : AkashErrorCode.DEPLOYMENT_CLOSE_FAILED,
+ stack: error instanceof Error ? error.stack : undefined
+ });
+ throw error;
+ }
+}
+
+// Close all active deployments
+async function closeAllDeployments(
+ runtime: IAgentRuntime
+): Promise<{ success: string[], failed: string[] }> {
+ elizaLogger.info("Closing all active deployments");
+
+ try {
+ // Fetch active deployments
+ const deployments = await fetchDeployments(runtime, undefined, 0, 100);
+ const activeDeployments = deployments.results.filter(d =>
+ d.status.toLowerCase() === 'active'
+ );
+
+ if (activeDeployments.length === 0) {
+ elizaLogger.info("No active deployments found to close");
+ return { success: [], failed: [] };
+ }
+
+ elizaLogger.info("Found active deployments to close", {
+ count: activeDeployments.length,
+ dseqs: activeDeployments.map(d => d.dseq)
+ });
+
+ // Close each deployment
+ const results = { success: [] as string[], failed: [] as string[] };
+ for (const deployment of activeDeployments) {
+ try {
+ await closeSingleDeployment(runtime, deployment.dseq);
+ results.success.push(deployment.dseq);
+ } catch (error) {
+ elizaLogger.error("Failed to close deployment", {
+ dseq: deployment.dseq,
+ error: error instanceof Error ? error.message : String(error),
+ code: error instanceof AkashError ? error.code : AkashErrorCode.DEPLOYMENT_CLOSE_FAILED
+ });
+ results.failed.push(deployment.dseq);
+ }
+ }
+
+ elizaLogger.info("Finished closing deployments", results);
+ return results;
+ } catch (error) {
+ elizaLogger.error("Failed to close deployments", {
+ error: error instanceof Error ? error.message : String(error),
+ code: error instanceof AkashError ? error.code : AkashErrorCode.DEPLOYMENT_CLOSE_FAILED,
+ stack: error instanceof Error ? error.stack : undefined
+ });
+ throw error;
+ }
+}
+
+export const closeDeploymentAction: Action = {
+ name: "CLOSE_DEPLOYMENT",
+ similes: ["CLOSE_AKASH_DEPLOYMENT", "STOP_DEPLOYMENT", "TERMINATE_DEPLOYMENT"],
+ description: "Close an active deployment on the Akash Network",
+ examples: [[
+ {
+ user: "user",
+ content: {
+ text: "Close deployment with DSEQ 123456",
+ dseq: "123456"
+ } as CloseDeploymentContent
+ } as ActionExample,
+ {
+ user: "assistant",
+ content: {
+ text: "Closing deployment with DSEQ 123456..."
+ } as CloseDeploymentContent
+ } as ActionExample
+ ], [
+ {
+ user: "user",
+ content: {
+ text: "Close all active deployments",
+ closeAll: true
+ } as CloseDeploymentContent
+ } as ActionExample,
+ {
+ user: "assistant",
+ content: {
+ text: "Closing all active deployments..."
+ } as CloseDeploymentContent
+ } as ActionExample
+ ]],
+
+ validate: async (runtime: IAgentRuntime, message: Memory): Promise => {
+ elizaLogger.debug("=== Starting Close Deployment Validation ===");
+ elizaLogger.debug("Validating close deployment request", { message });
+
+ // Check if plugin is properly loaded
+ if (!isPluginLoaded(runtime, "akash")) {
+ elizaLogger.error("Akash plugin not properly loaded during validation");
+ return false;
+ }
+
+ try {
+ const params = message.content as Partial;
+ const config = await validateAkashConfig(runtime);
+ elizaLogger.debug("Validating parameters", { params });
+
+ // If no parameters provided, use environment defaults
+ if (!params.dseq && !params.closeAll) {
+ if (config.AKASH_CLOSE_DEP === "closeAll") {
+ params.closeAll = true;
+ } else if (config.AKASH_CLOSE_DEP === "dseq" && config.AKASH_CLOSE_DSEQ) {
+ params.dseq = config.AKASH_CLOSE_DSEQ;
+ } else {
+ throw new AkashError(
+ "Either dseq or closeAll parameter is required",
+ AkashErrorCode.VALIDATION_PARAMETER_MISSING,
+ { parameters: ["dseq", "closeAll"] }
+ );
+ }
+ }
+
+ if (params.dseq && params.closeAll) {
+ throw new AkashError(
+ "Cannot specify both dseq and closeAll parameters",
+ AkashErrorCode.VALIDATION_PARAMETER_INVALID,
+ { parameters: ["dseq", "closeAll"] }
+ );
+ }
+
+ if (params.dseq && !/^\d+$/.test(params.dseq)) {
+ throw new AkashError(
+ "DSEQ must be a numeric string",
+ AkashErrorCode.VALIDATION_PARAMETER_INVALID,
+ { parameter: "dseq", value: params.dseq }
+ );
+ }
+
+ elizaLogger.debug("Validation completed successfully");
+ return true;
+ } catch (error) {
+ elizaLogger.error("Close deployment validation failed", {
+ error: error instanceof AkashError ? {
+ code: error.code,
+ message: error.message,
+ details: error.details
+ } : String(error)
+ });
+ return false;
+ }
+ },
+
+ handler: async (
+ runtime: IAgentRuntime,
+ message: Memory,
+ state: State | undefined,
+ _options: { [key: string]: unknown } = {},
+ callback?: HandlerCallback
+ ): Promise => {
+ const actionId = Date.now().toString();
+ elizaLogger.info("=== Starting Close Deployment Request ===", {
+ actionId,
+ messageId: message.id,
+ userId: message.userId
+ });
+
+ try {
+ const config = await validateAkashConfig(runtime);
+ const params = message.content as Partial;
+
+ // If no parameters provided, use environment defaults
+ if (!params.dseq && !params.closeAll) {
+ if (config.AKASH_CLOSE_DEP === "closeAll") {
+ params.closeAll = true;
+ } else if (config.AKASH_CLOSE_DEP === "dseq" && config.AKASH_CLOSE_DSEQ) {
+ params.dseq = config.AKASH_CLOSE_DSEQ;
+ } else {
+ if (callback) {
+ elizaLogger.info("=== Preparing error callback response ===", {
+ actionId,
+ hasCallback: true,
+ errorType: 'AkashError'
+ });
+
+ const errorResponse = {
+ text: "Either DSEQ or closeAll parameter is required",
+ content: {
+ success: false,
+ error: {
+ code: AkashErrorCode.VALIDATION_PARAMETER_MISSING,
+ message: "Either dseq or closeAll parameter is required"
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'closeDeployment',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+
+ callback(errorResponse);
+ }
+ return false;
+ }
+ }
+
+ if (params.closeAll) {
+ const results = await closeAllDeployments(runtime);
+
+ if (callback) {
+ elizaLogger.info("=== Preparing callback response for bulk closure ===", {
+ hasCallback: true,
+ actionId,
+ successCount: results.success.length,
+ failedCount: results.failed.length
+ });
+
+ const callbackResponse = {
+ text: `Deployment Closure Results:\n\nSuccessfully closed: ${results.success.length} deployments${
+ results.success.length > 0 ? `\nDSEQs: ${results.success.join(', ')}` : ''
+ }${
+ results.failed.length > 0 ? `\n\nFailed to close: ${results.failed.length} deployments\nDSEQs: ${results.failed.join(', ')}` : ''
+ }`,
+ content: {
+ success: results.failed.length === 0,
+ data: {
+ successful: results.success,
+ failed: results.failed,
+ totalClosed: results.success.length,
+ totalFailed: results.failed.length
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'closeDeployment',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+
+ elizaLogger.info("=== Executing callback with response ===", {
+ actionId,
+ responseText: callbackResponse.text,
+ hasContent: !!callbackResponse.content,
+ contentKeys: Object.keys(callbackResponse.content),
+ metadata: callbackResponse.content.metadata
+ });
+
+ callback(callbackResponse);
+
+ elizaLogger.info("=== Callback executed successfully ===", {
+ actionId,
+ timestamp: new Date().toISOString()
+ });
+ }
+ return results.failed.length === 0;
+
+ } else if (params.dseq) {
+ const success = await closeSingleDeployment(runtime, params.dseq);
+
+ if (callback) {
+ elizaLogger.info("=== Preparing callback response for single closure ===", {
+ hasCallback: true,
+ actionId,
+ dseq: params.dseq,
+ success
+ });
+
+ const callbackResponse = {
+ text: success ?
+ `Successfully closed deployment DSEQ: ${params.dseq}` :
+ `Failed to close deployment DSEQ: ${params.dseq}`,
+ content: {
+ success,
+ data: {
+ dseq: params.dseq
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'closeDeployment',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+
+ elizaLogger.info("=== Executing callback with response ===", {
+ actionId,
+ responseText: callbackResponse.text,
+ hasContent: !!callbackResponse.content,
+ contentKeys: Object.keys(callbackResponse.content),
+ metadata: callbackResponse.content.metadata
+ });
+
+ callback(callbackResponse);
+
+ elizaLogger.info("=== Callback executed successfully ===", {
+ actionId,
+ timestamp: new Date().toISOString()
+ });
+ }
+ return success;
+ }
+
+ return false;
+
+ } catch (error) {
+ elizaLogger.error("Close deployment request failed", {
+ error: error instanceof Error ? error.message : String(error),
+ code: error instanceof AkashError ? error.code : AkashErrorCode.DEPLOYMENT_CLOSE_FAILED,
+ actionId
+ });
+
+ if (callback) {
+ elizaLogger.info("=== Preparing error callback response ===", {
+ actionId,
+ hasCallback: true,
+ errorType: error instanceof AkashError ? 'AkashError' : 'Error'
+ });
+
+ const errorResponse = {
+ text: `Failed to close deployment: ${error instanceof Error ? error.message : String(error)}`,
+ content: {
+ success: false,
+ error: {
+ code: error instanceof AkashError ? error.code : AkashErrorCode.DEPLOYMENT_CLOSE_FAILED,
+ message: error instanceof Error ? error.message : String(error)
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'closeDeployment',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+
+ elizaLogger.info("=== Executing error callback ===", {
+ actionId,
+ responseText: errorResponse.text,
+ hasContent: !!errorResponse.content,
+ contentKeys: Object.keys(errorResponse.content)
+ });
+
+ callback(errorResponse);
+
+ elizaLogger.info("=== Error callback executed ===", {
+ actionId,
+ timestamp: new Date().toISOString()
+ });
+ }
+
+ return false;
+ }
+ }
+};
+
+export default closeDeploymentAction;
\ No newline at end of file
diff --git a/packages/plugin-akash/src/actions/createCertificate.ts b/packages/plugin-akash/src/actions/createCertificate.ts
new file mode 100644
index 00000000000..67058e2d168
--- /dev/null
+++ b/packages/plugin-akash/src/actions/createCertificate.ts
@@ -0,0 +1,456 @@
+import { Action, elizaLogger } from "@elizaos/core";
+import { IAgentRuntime, Memory, State, HandlerCallback, Content, ActionExample } from "@elizaos/core";
+import { DirectSecp256k1HdWallet } from "@cosmjs/proto-signing";
+import { SigningStargateClient } from "@cosmjs/stargate";
+import * as cert from "@akashnetwork/akashjs/build/certificates";
+import { certificateManager } from "@akashnetwork/akashjs/build/certificates/certificate-manager";
+import { CertificatePem } from "@akashnetwork/akashjs/build/certificates/certificate-manager/CertificateManager";
+import { getAkashTypeRegistry } from "@akashnetwork/akashjs/build/stargate";
+import { validateAkashConfig } from "../environment";
+import { AkashError, AkashErrorCode, withRetry } from "../error/error";
+import * as fs from 'fs';
+import * as path from 'path';
+import { Registry } from "@cosmjs/proto-signing";
+import type { SigningStargateClient as AkashSigningStargateClient } from "@akashnetwork/akashjs/node_modules/@cosmjs/stargate";
+import { getCertificatePath } from "../utils/paths";
+
+interface CreateCertificateContent extends Content {
+ overwrite?: boolean;
+}
+
+// Certificate file path
+const CERTIFICATE_PATH = getCertificatePath(import.meta.url);
+
+// Save certificate to file
+async function saveCertificate(certificate: CertificatePem): Promise {
+ elizaLogger.debug("Saving certificate to file", { path: CERTIFICATE_PATH });
+ try {
+ // Ensure directory exists
+ const dir = path.dirname(CERTIFICATE_PATH);
+ if (!fs.existsSync(dir)) {
+ fs.mkdirSync(dir, { recursive: true });
+ }
+ const json = JSON.stringify(certificate);
+ fs.writeFileSync(CERTIFICATE_PATH, json);
+ elizaLogger.debug("Certificate saved successfully");
+ } catch (error) {
+ elizaLogger.error("Failed to save certificate", {
+ error: error instanceof Error ? error.message : String(error),
+ path: CERTIFICATE_PATH
+ });
+ throw new AkashError(
+ "Failed to save certificate",
+ AkashErrorCode.FILE_WRITE_ERROR,
+ { path: CERTIFICATE_PATH, error }
+ );
+ }
+}
+
+// Load certificate from file
+function loadCertificate(): CertificatePem {
+ elizaLogger.debug("Loading certificate from file", { path: CERTIFICATE_PATH });
+ try {
+ if (!fs.existsSync(CERTIFICATE_PATH)) {
+ throw new AkashError(
+ "Certificate file not found",
+ AkashErrorCode.CERTIFICATE_NOT_FOUND,
+ { path: CERTIFICATE_PATH }
+ );
+ }
+ const json = fs.readFileSync(CERTIFICATE_PATH, "utf8");
+ const certificate = JSON.parse(json);
+ elizaLogger.debug("Certificate loaded successfully", {
+ hasCert: !!certificate.cert,
+ hasPrivateKey: !!certificate.privateKey,
+ hasPublicKey: !!certificate.publicKey
+ });
+ return certificate;
+ } catch (error) {
+ elizaLogger.error("Failed to load certificate", {
+ error: error instanceof Error ? error.message : String(error),
+ path: CERTIFICATE_PATH
+ });
+ if (error instanceof AkashError) {
+ throw error;
+ }
+ throw new AkashError(
+ "Failed to load certificate",
+ AkashErrorCode.FILE_READ_ERROR,
+ { path: CERTIFICATE_PATH, error }
+ );
+ }
+}
+
+// Initialize wallet with proper error handling
+async function initializeWallet(mnemonic: string): Promise {
+ elizaLogger.debug("=== Initializing Wallet ===", {
+ mnemonicLength: mnemonic.split(' ').length,
+ hasMnemonic: !!mnemonic,
+ mnemonicFirstWord: mnemonic.split(' ')[0]
+ });
+
+ // Validate mnemonic format
+ const words = mnemonic.trim().split(/\s+/);
+ if (words.length !== 12 && words.length !== 24) {
+ const error = `Invalid mnemonic length: got ${words.length} words, expected 12 or 24 words`;
+ elizaLogger.error("Mnemonic validation failed", {
+ error,
+ wordCount: words.length,
+ expectedCounts: [12, 24],
+ mnemonicPreview: words.slice(0, 3).join(' ') + '...'
+ });
+ throw new AkashError(
+ error,
+ AkashErrorCode.WALLET_INITIALIZATION_FAILED,
+ {
+ wordCount: words.length,
+ expectedCounts: [12, 24]
+ }
+ );
+ }
+
+ try {
+ elizaLogger.debug("Creating wallet with mnemonic", {
+ wordCount: words.length,
+ mnemonicPreview: words.slice(0, 3).join(' ') + '...'
+ });
+
+ const wallet = await DirectSecp256k1HdWallet.fromMnemonic(mnemonic, {
+ prefix: "akash"
+ });
+ const accounts = await wallet.getAccounts();
+
+ elizaLogger.debug("Wallet initialized successfully", {
+ accountCount: accounts.length,
+ firstAccountAddress: accounts[0]?.address,
+ addressPrefix: accounts[0]?.address?.substring(0, 6)
+ });
+
+ if (!accounts.length) {
+ throw new AkashError(
+ "No accounts found in wallet",
+ AkashErrorCode.WALLET_INITIALIZATION_FAILED
+ );
+ }
+
+ return wallet;
+ } catch (error) {
+ elizaLogger.error("Wallet initialization failed", {
+ error: error instanceof Error ? error.message : String(error),
+ stack: error instanceof Error ? error.stack : undefined,
+ mnemonicLength: words.length,
+ mnemonicPreview: words.slice(0, 3).join(' ') + '...'
+ });
+
+ if (error instanceof AkashError) {
+ throw error;
+ }
+
+ throw new AkashError(
+ `Failed to initialize wallet: ${error instanceof Error ? error.message : String(error)}`,
+ AkashErrorCode.WALLET_INITIALIZATION_FAILED,
+ {
+ mnemonicLength: words.length,
+ error: error instanceof Error ? error.message : String(error)
+ }
+ );
+ }
+}
+
+// Setup client with proper error handling and fallback RPC endpoints
+async function setupClient(wallet: DirectSecp256k1HdWallet, rpcEndpoint: string): Promise {
+ // Try alternative RPC endpoints if the main one fails
+ const rpcEndpoints = [
+ rpcEndpoint,
+ "https://rpc.akashnet.net:443",
+ "https://akash-rpc.polkachu.com:443",
+ "https://akash-rpc.europlots.com:443"
+ ];
+
+ elizaLogger.info("=== Setting up Stargate Client ===", {
+ primaryRpcEndpoint: rpcEndpoint,
+ allEndpoints: rpcEndpoints,
+ walletType: wallet.constructor.name
+ });
+
+ let lastError: Error | undefined;
+ for (const endpoint of rpcEndpoints) {
+ try {
+ elizaLogger.debug("Attempting to connect to RPC endpoint", {
+ endpoint,
+ attempt: rpcEndpoints.indexOf(endpoint) + 1,
+ totalEndpoints: rpcEndpoints.length
+ });
+
+ const registry = new Registry(getAkashTypeRegistry());
+ elizaLogger.debug("Registry created for endpoint", {
+ endpoint,
+ registryType: registry.constructor.name
+ });
+
+ const client = await SigningStargateClient.connectWithSigner(
+ endpoint,
+ wallet,
+ { registry }
+ );
+
+ elizaLogger.debug("Client setup completed successfully", {
+ endpoint,
+ clientType: client.constructor.name
+ });
+
+ return client;
+ } catch (error) {
+ lastError = error as Error;
+ elizaLogger.warn("Failed to connect to RPC endpoint", {
+ endpoint,
+ error: error instanceof Error ? error.message : String(error),
+ remainingEndpoints: rpcEndpoints.slice(rpcEndpoints.indexOf(endpoint) + 1).length
+ });
+ }
+ }
+
+ throw new AkashError(
+ `Failed to connect to any RPC endpoint: ${lastError?.message}`,
+ AkashErrorCode.CLIENT_SETUP_FAILED,
+ { lastError }
+ );
+}
+
+export const createCertificateAction: Action = {
+ name: "CREATE_CERTIFICATE",
+ similes: ["GENERATE_CERTIFICATE", "SETUP_CERTIFICATE", "INIT_CERTIFICATE"],
+ description: "Create or load Akash certificate for provider interactions",
+ examples: [[
+ {
+ user: "user",
+ content: {
+ text: "Create a new certificate",
+ overwrite: true
+ } as CreateCertificateContent
+ } as ActionExample,
+ {
+ user: "assistant",
+ content: {
+ text: "Creating new certificate..."
+ } as CreateCertificateContent
+ } as ActionExample
+ ]],
+
+ validate: async (runtime: IAgentRuntime, message: Memory): Promise => {
+ elizaLogger.debug("=== Starting Certificate Validation ===");
+ try {
+ const params = message.content as Partial;
+
+ // Validate Akash configuration
+ await validateAkashConfig(runtime);
+
+ // If overwrite is specified, it must be a boolean
+ if (params.overwrite !== undefined && typeof params.overwrite !== 'boolean') {
+ throw new AkashError(
+ "Overwrite parameter must be a boolean",
+ AkashErrorCode.VALIDATION_PARAMETER_INVALID,
+ { parameter: "overwrite", value: params.overwrite }
+ );
+ }
+
+ elizaLogger.debug("Certificate validation completed successfully");
+ return true;
+ } catch (error) {
+ elizaLogger.error("Certificate validation failed", {
+ error: error instanceof AkashError ? {
+ code: error.code,
+ message: error.message,
+ details: error.details
+ } : String(error)
+ });
+ return false;
+ }
+ },
+
+ handler: async (
+ runtime: IAgentRuntime,
+ message: Memory,
+ state: State | undefined,
+ options: { callback?: HandlerCallback } = {}
+ ): Promise => {
+ const actionId = Date.now().toString();
+ elizaLogger.info("=== Starting Certificate Creation/Loading ===", { actionId });
+
+ try {
+ // First validate the parameters
+ if (!await createCertificateAction.validate(runtime, message)) {
+ const error = new AkashError(
+ "Invalid parameters provided",
+ AkashErrorCode.VALIDATION_PARAMETER_INVALID
+ );
+ if (options.callback) {
+ options.callback({
+ text: `Failed to validate parameters: ${error.message}`,
+ error: error.message,
+ content: {
+ success: false,
+ error: {
+ code: error.code,
+ message: error.message
+ }
+ }
+ });
+ }
+ return false;
+ }
+
+ const params = message.content as Partial;
+ const config = await validateAkashConfig(runtime);
+
+ try {
+ // Check if certificate exists and overwrite is not true
+ if (fs.existsSync(CERTIFICATE_PATH) && !params.overwrite) {
+ elizaLogger.info("Loading existing certificate");
+ const certificate = loadCertificate();
+
+ if (options.callback) {
+ options.callback({
+ text: "Loaded existing certificate",
+ content: {
+ success: true,
+ certificate: {
+ hasCert: !!certificate.cert,
+ hasPrivateKey: !!certificate.privateKey,
+ hasPublicKey: !!certificate.publicKey
+ }
+ }
+ });
+ }
+ return true;
+ }
+
+ // Initialize wallet
+ elizaLogger.info("Initializing wallet for certificate creation");
+ const wallet = await initializeWallet(config.AKASH_MNEMONIC);
+ const accounts = await wallet.getAccounts();
+ const address = accounts[0].address;
+ elizaLogger.debug("Wallet initialized", {
+ address,
+ accountCount: accounts.length
+ });
+
+ // Setup client
+ elizaLogger.debug("Setting up Stargate client");
+ const client = await setupClient(wallet, config.RPC_ENDPOINT);
+ elizaLogger.debug("Client setup completed");
+
+ // Generate new certificate
+ elizaLogger.info("Generating new certificate");
+ const certificate = certificateManager.generatePEM(address);
+ elizaLogger.debug("Certificate generated", {
+ hasCert: !!certificate.cert,
+ hasPrivateKey: !!certificate.privateKey,
+ hasPublicKey: !!certificate.publicKey
+ });
+
+ // Broadcast certificate
+ elizaLogger.info("Broadcasting certificate to network");
+ const result = await withRetry(async () => {
+ return await cert.broadcastCertificate(
+ certificate,
+ address,
+ client as unknown as AkashSigningStargateClient
+ );
+ });
+
+ if (result.code !== 0) {
+ throw new AkashError(
+ `Could not create certificate: ${result.rawLog}`,
+ AkashErrorCode.CERTIFICATE_CREATION_FAILED,
+ { rawLog: result.rawLog }
+ );
+ }
+
+ elizaLogger.info("Certificate broadcast successful", {
+ code: result.code,
+ txHash: result.transactionHash,
+ height: result.height,
+ gasUsed: result.gasUsed
+ });
+
+ // Save certificate
+ await saveCertificate(certificate);
+ elizaLogger.info("Certificate saved to file", { path: CERTIFICATE_PATH });
+
+ if (options.callback) {
+ options.callback({
+ text: "Certificate created and saved successfully",
+ content: {
+ success: true,
+ certificate: {
+ hasCert: !!certificate.cert,
+ hasPrivateKey: !!certificate.privateKey,
+ hasPublicKey: !!certificate.publicKey
+ },
+ transaction: {
+ hash: result.transactionHash,
+ height: result.height,
+ gasUsed: result.gasUsed
+ }
+ }
+ });
+ }
+
+ return true;
+ } catch (error) {
+ elizaLogger.error("Failed to create/load certificate", {
+ error: error instanceof Error ? error.message : String(error),
+ stack: error instanceof Error ? error.stack : undefined
+ });
+
+ if (options.callback) {
+ options.callback({
+ text: `Failed to create/load certificate: ${error instanceof Error ? error.message : String(error)}`,
+ error: error instanceof Error ? error.message : String(error),
+ content: {
+ success: false,
+ error: error instanceof AkashError ? {
+ code: error.code,
+ message: error.message,
+ details: error.details
+ } : {
+ code: AkashErrorCode.CERTIFICATE_CREATION_FAILED,
+ message: String(error)
+ }
+ }
+ });
+ }
+ return false;
+ }
+ } catch (error) {
+ elizaLogger.error("Certificate operation failed", {
+ error: error instanceof Error ? error.message : String(error),
+ code: error instanceof AkashError ? error.code : AkashErrorCode.CERTIFICATE_CREATION_FAILED,
+ actionId
+ });
+
+ if (options.callback) {
+ options.callback({
+ text: `Certificate operation failed: ${error instanceof Error ? error.message : String(error)}`,
+ error: error instanceof Error ? error.message : String(error),
+ content: {
+ success: false,
+ error: error instanceof AkashError ? {
+ code: error.code,
+ message: error.message,
+ details: error.details
+ } : {
+ code: AkashErrorCode.CERTIFICATE_CREATION_FAILED,
+ message: String(error)
+ }
+ }
+ });
+ }
+
+ return false;
+ }
+ }
+};
+
+export default createCertificateAction;
diff --git a/packages/plugin-akash/src/actions/createDeployment.ts b/packages/plugin-akash/src/actions/createDeployment.ts
new file mode 100644
index 00000000000..d64c5a6ebca
--- /dev/null
+++ b/packages/plugin-akash/src/actions/createDeployment.ts
@@ -0,0 +1,1471 @@
+import { Action, elizaLogger } from "@elizaos/core";
+import { IAgentRuntime, Memory, State, HandlerCallback, Content, ActionExample } from "@elizaos/core";
+import { MsgCreateDeployment } from "@akashnetwork/akash-api/akash/deployment/v1beta3";
+import { QueryClientImpl as QueryProviderClient, QueryProviderRequest } from "@akashnetwork/akash-api/akash/provider/v1beta3";
+import { QueryBidsRequest, QueryClientImpl as QueryMarketClient, MsgCreateLease, BidID } from "@akashnetwork/akash-api/akash/market/v1beta4";
+import * as cert from "@akashnetwork/akashjs/build/certificates";
+import { getRpc } from "@akashnetwork/akashjs/build/rpc";
+import { SDL } from "@akashnetwork/akashjs/build/sdl";
+import { getAkashTypeRegistry } from "@akashnetwork/akashjs/build/stargate";
+import { CertificatePem } from "@akashnetwork/akashjs/build/certificates/certificate-manager/CertificateManager";
+import { certificateManager } from "@akashnetwork/akashjs/build/certificates/certificate-manager";
+import { DirectSecp256k1HdWallet, Registry } from "@cosmjs/proto-signing";
+import { SigningStargateClient } from "@cosmjs/stargate";
+import { validateAkashConfig } from "../environment";
+import { AkashError, AkashErrorCode, withRetry } from "../error/error";
+import * as fs from 'fs';
+import * as path from 'path';
+import { getCertificatePath, getDefaultSDLPath } from "../utils/paths";
+// import { fileURLToPath } from 'url';
+import { inspectRuntime, isPluginLoaded } from "../runtime_inspect";
+import https from 'node:https';
+import axios from 'axios';
+
+interface CreateDeploymentContent extends Content {
+ sdl?: string;
+ sdlFile?: string;
+ deposit?: string;
+}
+
+// Certificate file path
+const CERTIFICATE_PATH = getCertificatePath(import.meta.url);
+
+// Save certificate to file
+function saveCertificate(certificate: CertificatePem) {
+ elizaLogger.debug("Saving certificate to file", { path: CERTIFICATE_PATH });
+ try {
+ // Ensure directory exists
+ const dir = path.dirname(CERTIFICATE_PATH);
+ if (!fs.existsSync(dir)) {
+ fs.mkdirSync(dir, { recursive: true });
+ }
+ const json = JSON.stringify(certificate);
+ fs.writeFileSync(CERTIFICATE_PATH, json);
+ elizaLogger.debug("Certificate saved successfully");
+ } catch (error) {
+ elizaLogger.error("Failed to save certificate", {
+ error: error instanceof Error ? error.message : String(error),
+ path: CERTIFICATE_PATH
+ });
+ throw error;
+ }
+}
+
+// Load certificate from file
+function loadCertificate(path: string): CertificatePem {
+ elizaLogger.debug("Loading certificate from file", { path });
+ try {
+ const json = fs.readFileSync(path, "utf8");
+ const certificate = JSON.parse(json);
+ elizaLogger.debug("Certificate loaded successfully", {
+ hasCert: !!certificate.cert,
+ hasPrivateKey: !!certificate.privateKey,
+ hasPublicKey: !!certificate.publicKey
+ });
+ return certificate;
+ } catch (error) {
+ elizaLogger.error("Failed to load certificate", {
+ error: error instanceof Error ? error.message : String(error),
+ path
+ });
+ throw error;
+ }
+}
+
+const DEFAULT_SDL_PATH = (() => {
+ const currentFileUrl = import.meta.url;
+ // elizaLogger.info("=== Starting SDL Path Resolution in createDeployment ===", {
+ // currentFileUrl,
+ // cwd: process.cwd(),
+ // importMetaUrl: import.meta.url
+ // });
+
+ // Use the utility function from paths.ts instead of manual resolution
+ const sdlPath = getDefaultSDLPath(currentFileUrl);
+
+ // Only log if file doesn't exist
+ if (!fs.existsSync(sdlPath)) {
+ elizaLogger.warn("Default SDL path not found", {
+ sdlPath,
+ exists: false
+ });
+ }
+
+ return sdlPath;
+})();
+
+const validateDeposit = (deposit: string): boolean => {
+ const pattern = /^\d+uakt$/;
+ return pattern.test(deposit);
+};
+
+const loadSDLFromFile = (filePath: string): string => {
+ // elizaLogger.info("=== Loading SDL File ===", {
+ // requestedPath: filePath,
+ // resolvedPath: path.resolve(filePath),
+ // defaultSdlPath: DEFAULT_SDL_PATH,
+ // cwd: process.cwd(),
+ // exists: fs.existsSync(filePath),
+ // defaultExists: fs.existsSync(DEFAULT_SDL_PATH)
+ // });
+
+ try {
+ // If path doesn't contain plugin-akash and it's not the default path, adjust it
+ if (!filePath.includes('plugin-akash') && filePath !== DEFAULT_SDL_PATH) {
+ const adjustedPath = path.join(path.dirname(DEFAULT_SDL_PATH), path.basename(filePath));
+ // elizaLogger.info("Adjusting SDL path", {
+ // originalPath: filePath,
+ // adjustedPath,
+ // exists: fs.existsSync(adjustedPath),
+ // dirExists: fs.existsSync(path.dirname(adjustedPath)),
+ // dirContents: fs.existsSync(path.dirname(adjustedPath)) ? fs.readdirSync(path.dirname(adjustedPath)) : []
+ // });
+ filePath = adjustedPath;
+ }
+
+ // Try multiple possible locations
+ const possiblePaths = [
+ filePath,
+ path.join(process.cwd(), filePath),
+ path.join(process.cwd(), 'packages', 'plugin-akash', filePath),
+ path.join(process.cwd(), 'packages', 'plugin-akash', 'src', filePath),
+ path.join(path.dirname(DEFAULT_SDL_PATH), filePath)
+ ];
+
+ // elizaLogger.info("Attempting to load SDL from possible paths", {
+ // possiblePaths,
+ // existsMap: possiblePaths.map(p => ({ path: p, exists: fs.existsSync(p) }))
+ // });
+
+ for (const tryPath of possiblePaths) {
+ if (fs.existsSync(tryPath)) {
+ const content = fs.readFileSync(tryPath, "utf8");
+ elizaLogger.info("SDL file loaded successfully from", {
+ path: tryPath
+ });
+ return content;
+ }
+ }
+
+ // If we get here, none of the paths worked
+ throw new Error(`SDL file not found in any of the possible locations`);
+ } catch (error) {
+ elizaLogger.error("Failed to read SDL file", {
+ filePath,
+ error: error instanceof Error ? error.message : String(error)
+ });
+ throw new AkashError(
+ `Failed to read SDL file: ${error instanceof Error ? error.message : String(error)}`,
+ AkashErrorCode.VALIDATION_SDL_FAILED,
+ { filePath }
+ );
+ }
+};
+
+// Preserved for future use
+/* const formatErrorMessage = (error: unknown): string => {
+ if (error instanceof AkashError) {
+ if (error.code === AkashErrorCode.WALLET_NOT_INITIALIZED) {
+ return "Akash wallet not initialized";
+ }
+ if (error.code === AkashErrorCode.DEPLOYMENT_CREATION_FAILED) {
+ return `Transaction failed: ${error.details?.rawLog || 'Unknown error'}`;
+ }
+ if (error.code === AkashErrorCode.MANIFEST_PARSING_FAILED) {
+ return "Failed to parse SDL";
+ }
+ if (error.code === AkashErrorCode.VALIDATION_PARAMETER_MISSING) {
+ return `${error.message}`;
+ }
+ if (error.code === AkashErrorCode.VALIDATION_SDL_FAILED) {
+ return `Failed to parse SDL: ${error.details?.error || error.message}`;
+ }
+ if (error.code === AkashErrorCode.VALIDATION_PARAMETER_INVALID) {
+ return `Invalid deposit format. Must be in format: uakt`;
+ }
+ return error.message;
+ }
+
+ const message = error instanceof Error ? error.message : String(error);
+ if (message.toLowerCase().includes("insufficient funds")) {
+ return "Insufficient funds";
+ }
+ if (message.toLowerCase().includes("invalid deposit")) {
+ return "Invalid deposit amount";
+ }
+ if (message.toLowerCase().includes("cannot read properties")) {
+ return "Failed to parse SDL: Invalid format";
+ }
+ return message;
+}; */
+
+async function initializeWallet(mnemonic: string) {
+ elizaLogger.debug("=== Initializing Wallet ===", {
+ mnemonicLength: mnemonic.split(' ').length,
+ hasMnemonic: !!mnemonic,
+ mnemonicFirstWord: mnemonic.split(' ')[0]
+ });
+
+ // Validate mnemonic format
+ const words = mnemonic.trim().split(/\s+/);
+ if (words.length !== 12 && words.length !== 24) {
+ const error = `Invalid mnemonic length: got ${words.length} words, expected 12 or 24 words`;
+ elizaLogger.error("Mnemonic validation failed", {
+ error,
+ wordCount: words.length,
+ expectedCounts: [12, 24],
+ mnemonicPreview: words.slice(0, 3).join(' ') + '...'
+ });
+ throw new AkashError(
+ error,
+ AkashErrorCode.WALLET_INITIALIZATION_FAILED,
+ {
+ wordCount: words.length,
+ expectedCounts: [12, 24]
+ }
+ );
+ }
+
+ try {
+ elizaLogger.debug("Creating wallet with mnemonic", {
+ wordCount: words.length,
+ mnemonicPreview: words.slice(0, 3).join(' ') + '...'
+ });
+
+ const wallet = await DirectSecp256k1HdWallet.fromMnemonic(mnemonic, {
+ prefix: "akash"
+ });
+ const accounts = await wallet.getAccounts();
+
+ elizaLogger.debug("Wallet initialized successfully", {
+ accountCount: accounts.length,
+ firstAccountAddress: accounts[0]?.address,
+ addressPrefix: accounts[0]?.address?.substring(0, 6)
+ });
+
+ if (!accounts.length) {
+ throw new AkashError(
+ "No accounts found in wallet",
+ AkashErrorCode.WALLET_INITIALIZATION_FAILED
+ );
+ }
+
+ return wallet;
+ } catch (error) {
+ elizaLogger.error("Wallet initialization failed", {
+ error: error instanceof Error ? error.message : String(error),
+ stack: error instanceof Error ? error.stack : undefined,
+ mnemonicLength: words.length,
+ mnemonicPreview: words.slice(0, 3).join(' ') + '...'
+ });
+
+ // Provide more specific error messages
+ const errorMessage = error instanceof Error ? error.message : String(error);
+ if (errorMessage.includes("Invalid mnemonic")) {
+ throw new AkashError(
+ "Invalid mnemonic format: The mnemonic phrase contains invalid words or is malformed",
+ AkashErrorCode.WALLET_INITIALIZATION_FAILED,
+ {
+ mnemonicLength: words.length,
+ error: errorMessage
+ }
+ );
+ }
+
+ throw new AkashError(
+ `Failed to initialize wallet: ${errorMessage}`,
+ AkashErrorCode.WALLET_INITIALIZATION_FAILED,
+ {
+ mnemonicLength: words.length,
+ error: errorMessage
+ }
+ );
+ }
+}
+
+async function setupClient(wallet: DirectSecp256k1HdWallet, rpcEndpoint: string) {
+ // Try alternative RPC endpoints if the main one fails
+ const rpcEndpoints = [
+ "https://akash-rpc.europlots.com:443", // New endpoint first
+ rpcEndpoint,
+ "https://rpc.akashnet.net:443",
+ "https://rpc.akash.forbole.com:443",
+ "https://rpc-akash.ecostake.com:443",
+ "https://akash-rpc.polkachu.com:443",
+ "https://akash.c29r3.xyz:443/rpc"
+ ];
+
+ elizaLogger.info("=== Setting up Stargate Client ===", {
+ primaryRpcEndpoint: rpcEndpoint,
+ allEndpoints: rpcEndpoints,
+ walletType: wallet.constructor.name,
+ preferredEndpoint: rpcEndpoints[0]
+ });
+
+ let lastError: Error | undefined;
+ for (const endpoint of rpcEndpoints) {
+ try {
+ elizaLogger.debug("Attempting to connect to RPC endpoint", {
+ endpoint,
+ attempt: rpcEndpoints.indexOf(endpoint) + 1,
+ totalEndpoints: rpcEndpoints.length
+ });
+
+ const registry = new Registry(getAkashTypeRegistry());
+ elizaLogger.debug("Registry created for endpoint", {
+ endpoint,
+ registryType: registry.constructor.name
+ });
+
+ const client = await SigningStargateClient.connectWithSigner(
+ endpoint,
+ wallet,
+ { registry }
+ );
+
+ // Check if client is connected by attempting to get the height
+ try {
+ const height = await client.getHeight();
+ elizaLogger.info("Stargate client setup successful", {
+ endpoint,
+ height,
+ clientType: client.constructor.name,
+ attempt: rpcEndpoints.indexOf(endpoint) + 1
+ });
+ return client;
+ } catch (heightError) {
+ elizaLogger.error("Failed to get chain height", {
+ endpoint,
+ attempt: rpcEndpoints.indexOf(endpoint) + 1,
+ error: heightError instanceof Error ? heightError.message : String(heightError)
+ });
+ lastError = heightError instanceof Error ? heightError : new Error(String(heightError));
+ continue;
+ }
+ } catch (error) {
+ elizaLogger.error("Failed to connect to RPC endpoint", {
+ endpoint,
+ attempt: rpcEndpoints.indexOf(endpoint) + 1,
+ error: error instanceof Error ? error.message : String(error),
+ stack: error instanceof Error ? error.stack : undefined
+ });
+ lastError = error instanceof Error ? error : new Error(String(error));
+ continue;
+ }
+ }
+
+ // If we get here, all endpoints failed
+ elizaLogger.error("All RPC endpoints failed", {
+ endpoints: rpcEndpoints,
+ lastError: lastError?.message,
+ totalAttempts: rpcEndpoints.length
+ });
+ throw new AkashError(
+ `Failed to setup client: ${lastError?.message}`,
+ AkashErrorCode.CLIENT_SETUP_FAILED,
+ { rpcEndpoint: rpcEndpoints.join(", ") }
+ );
+}
+
+async function fetchBid(dseq: number, owner: string, rpcEndpoint: string) {
+ elizaLogger.info("=== Starting Bid Fetch Process ===", {
+ dseq,
+ owner,
+ ownerPrefix: owner.substring(0, 6)
+ });
+
+ const maxRetries = 3;
+ let lastError: Error | undefined;
+
+ for (let retry = 0; retry < maxRetries; retry++) {
+ try {
+ elizaLogger.debug("Connecting to RPC for bid fetch", {
+ rpcEndpoint,
+ attempt: retry + 1,
+ maxRetries
+ });
+
+ const rpc = await getRpc(rpcEndpoint);
+ elizaLogger.debug("RPC connection established", {
+ rpcType: rpc.constructor.name,
+ attempt: retry + 1
+ });
+
+ const client = new QueryMarketClient(rpc);
+ const request = QueryBidsRequest.fromPartial({
+ filters: {
+ owner: owner,
+ dseq: dseq
+ }
+ });
+
+ const startTime = Date.now();
+ const timeout = 1000 * 60 * 5; // 5 minutes timeout
+ elizaLogger.debug("Starting bid polling loop", {
+ timeout: "5 minutes",
+ pollInterval: "5 seconds",
+ attempt: retry + 1
+ });
+
+ while (Date.now() - startTime < timeout) {
+ const elapsedTime = Math.round((Date.now() - startTime) / 1000);
+ elizaLogger.debug("Polling for bids", {
+ dseq,
+ owner: owner.substring(0, 6),
+ elapsedSeconds: elapsedTime,
+ remainingSeconds: Math.round(timeout/1000 - elapsedTime),
+ attempt: retry + 1
+ });
+
+ try {
+ await new Promise(resolve => setTimeout(resolve, 5000));
+ const bids = await client.Bids(request);
+
+ if (bids.bids.length > 0 && bids.bids[0].bid !== undefined) {
+ elizaLogger.info("Bid found successfully", {
+ dseq,
+ owner: owner.substring(0, 6),
+ bidCount: bids.bids.length,
+ elapsedSeconds: elapsedTime,
+ attempt: retry + 1
+ });
+ elizaLogger.debug("Bid details", {
+ bid: bids.bids[0].bid,
+ provider: bids.bids[0].bid?.bidId?.provider
+ });
+ return bids.bids[0].bid;
+ }
+ } catch (pollError) {
+ // Log but continue polling if it's a temporary error
+ elizaLogger.warn("Temporary error during bid polling", {
+ error: pollError instanceof Error ? pollError.message : String(pollError),
+ dseq,
+ attempt: retry + 1,
+ willRetry: true
+ });
+ continue;
+ }
+ }
+
+ elizaLogger.error("Bid fetch timeout", {
+ dseq,
+ owner: owner.substring(0, 6),
+ timeout: "5 minutes",
+ attempt: retry + 1
+ });
+ throw new AkashError(
+ `Could not fetch bid for deployment ${dseq}. Timeout reached.`,
+ AkashErrorCode.BID_FETCH_TIMEOUT,
+ { dseq, owner }
+ );
+ } catch (error) {
+ lastError = error instanceof Error ? error : new Error(String(error));
+ elizaLogger.error("Error during bid fetch", {
+ error: error instanceof Error ? error.message : String(error),
+ stack: error instanceof Error ? error.stack : undefined,
+ dseq,
+ owner: owner.substring(0, 6),
+ attempt: retry + 1,
+ hasMoreRetries: retry < maxRetries - 1
+ });
+
+ if (retry < maxRetries - 1) {
+ // Wait before retrying (exponential backoff)
+ const delay = Math.pow(2, retry) * 1000;
+ elizaLogger.info("Retrying bid fetch after delay", {
+ delay,
+ nextAttempt: retry + 2,
+ maxRetries
+ });
+ await new Promise(resolve => setTimeout(resolve, delay));
+ continue;
+ }
+ }
+ }
+
+ // If we get here, all retries failed
+ elizaLogger.error("All bid fetch attempts failed", {
+ dseq,
+ owner: owner.substring(0, 6),
+ attempts: maxRetries,
+ finalError: lastError?.message
+ });
+ throw lastError || new Error("Failed to fetch bid after all retries");
+}
+
+async function createLease(deployment: any, wallet: DirectSecp256k1HdWallet, client: SigningStargateClient, rpcEndpoint: string): Promise {
+ const { dseq, owner } = deployment.id;
+ elizaLogger.info("Starting lease creation", { dseq, owner });
+
+ try {
+ elizaLogger.debug("Fetching bid for lease creation");
+ const bid = await fetchBid(dseq, owner, rpcEndpoint);
+ const accounts = await wallet.getAccounts();
+
+ if (bid.bidId === undefined) {
+ elizaLogger.error("Invalid bid - missing bidId", { dseq, owner });
+ throw new AkashError("Bid ID is undefined", AkashErrorCode.INVALID_BID);
+ }
+
+ elizaLogger.debug("Creating lease message", {
+ dseq,
+ owner,
+ bidId: bid.bidId
+ });
+
+ const lease = {
+ bidId: bid.bidId
+ };
+
+ const fee = {
+ amount: [{ denom: "uakt", amount: "50000" }],
+ gas: "2000000"
+ };
+
+ const msg = {
+ typeUrl: `/${MsgCreateLease.$type}`,
+ value: MsgCreateLease.fromPartial(lease)
+ };
+
+ elizaLogger.info("Broadcasting lease creation transaction");
+ const tx = await client.signAndBroadcast(accounts[0].address, [msg], fee, "create lease");
+
+ if (tx.code !== 0) {
+ elizaLogger.error("Lease creation failed", {
+ dseq,
+ owner,
+ code: tx.code,
+ rawLog: tx.rawLog
+ });
+ throw new AkashError(
+ `Could not create lease: ${tx.rawLog}`,
+ AkashErrorCode.LEASE_CREATION_FAILED,
+ { rawLog: tx.rawLog }
+ );
+ }
+
+ elizaLogger.info("Lease created successfully", {
+ dseq,
+ owner,
+ txHash: tx.transactionHash
+ });
+
+ return {
+ id: BidID.toJSON(bid.bidId)
+ };
+ } catch (error) {
+ elizaLogger.error("Error during lease creation", {
+ error,
+ dseq,
+ owner
+ });
+ throw error;
+ }
+}
+
+interface LeaseStatus {
+ services: Record;
+}
+
+async function queryLeaseStatus(lease: any, providerUri: string, certificate: CertificatePem): Promise {
+ const id = lease.id;
+ elizaLogger.info("Querying lease status", {
+ dseq: id?.dseq,
+ gseq: id?.gseq,
+ oseq: id?.oseq,
+ providerUri
+ });
+
+ if (id === undefined) {
+ elizaLogger.error("Invalid lease - missing ID");
+ throw new AkashError("Lease ID is undefined", AkashErrorCode.INVALID_LEASE);
+ }
+
+ const leasePath = `/lease/${id.dseq}/${id.gseq}/${id.oseq}/status`;
+ elizaLogger.debug("Setting up request", {
+ providerUri,
+ leasePath,
+ hasCert: !!certificate.cert,
+ hasKey: !!certificate.privateKey
+ });
+
+ const MAX_RETRIES = 3;
+ const INITIAL_RETRY_DELAY = 3000;
+ let retryCount = 0;
+
+ while (retryCount < MAX_RETRIES) {
+ try {
+ const url = new URL(providerUri);
+ const fullUrl = `${url.protocol}//${url.hostname}${url.port ? ':' + url.port : ''}${leasePath}`;
+
+ elizaLogger.debug("Making request", {
+ url: fullUrl,
+ method: 'GET',
+ hasCertificate: !!certificate,
+ retryCount
+ });
+
+ const agent = new https.Agent({
+ cert: certificate.cert,
+ key: certificate.privateKey,
+ rejectUnauthorized: false,
+ keepAlive: false,
+ timeout: 10000
+ });
+
+ try {
+ const response = await fetch(fullUrl, {
+ method: 'GET',
+ headers: {
+ 'Content-Type': 'application/json',
+ 'Accept': 'application/json'
+ },
+ // @ts-expect-error - TypeScript's fetch types don't include Node's agent support, but it exists at runtime
+ agent,
+ signal: AbortSignal.timeout(10000)
+ });
+
+ if (response.status !== 200) {
+ elizaLogger.warn("Non-OK response from lease status query", {
+ statusCode: response.status,
+ statusText: response.statusText,
+ dseq: id.dseq,
+ url: fullUrl,
+ retryCount
+ });
+
+ if (response.status === 404) {
+ elizaLogger.debug("Deployment not ready yet (404)", {
+ dseq: id.dseq,
+ retryCount
+ });
+ return undefined;
+ }
+ throw new Error(`Could not query lease status: ${response.status}`);
+ }
+
+ const data = await response.json() as LeaseStatus;
+ elizaLogger.debug("Lease status received", {
+ dseq: id.dseq,
+ dataLength: JSON.stringify(data).length,
+ hasServices: !!data.services,
+ serviceCount: Object.keys(data.services || {}).length
+ });
+ return data;
+ } finally {
+ agent.destroy();
+ }
+ } catch (error) {
+ elizaLogger.warn("Error during lease status query", {
+ error: error instanceof Error ? error.message : String(error),
+ stack: error instanceof Error ? error.stack : undefined,
+ dseq: id.dseq,
+ providerUri,
+ retryCount
+ });
+
+ if (retryCount < MAX_RETRIES - 1) {
+ const delay = INITIAL_RETRY_DELAY * Math.pow(2, retryCount);
+ elizaLogger.debug("Retrying after error", {
+ delay,
+ nextRetry: retryCount + 1,
+ maxRetries: MAX_RETRIES
+ });
+ await new Promise(r => setTimeout(r, delay));
+ retryCount++;
+ continue;
+ }
+
+ // On final retry, if it's a network error or 404, return undefined
+ if (error instanceof Error &&
+ ((error as any).code === 'ECONNABORTED' ||
+ (error as any).code === 'ETIMEDOUT' ||
+ ((error as any).response && (error as any).response.status === 404))) {
+ elizaLogger.info("Returning undefined after max retries", {
+ dseq: id.dseq,
+ error: error.message
+ });
+ return undefined;
+ }
+
+ throw error;
+ }
+ }
+
+ elizaLogger.info("Max retries reached, returning undefined", {
+ dseq: id.dseq,
+ maxRetries: MAX_RETRIES
+ });
+ return undefined;
+}
+
+async function sendManifest(sdl: SDL, lease: any, certificate: CertificatePem, rpcEndpoint: string) {
+ elizaLogger.info("Starting manifest send process");
+ if (lease.id === undefined) {
+ elizaLogger.error("Invalid lease - missing ID");
+ throw new AkashError("Lease ID is undefined", AkashErrorCode.INVALID_LEASE);
+ }
+
+ try {
+ const { dseq, provider } = lease.id;
+ elizaLogger.debug("Getting provider info", { provider });
+
+ const rpc = await getRpc(rpcEndpoint);
+ const client = new QueryProviderClient(rpc);
+ const request = QueryProviderRequest.fromPartial({
+ owner: provider
+ });
+
+ const tx = await client.Provider(request);
+
+ if (tx.provider === undefined) {
+ elizaLogger.error("Provider not found", { provider });
+ throw new AkashError(
+ `Could not find provider ${provider}`,
+ AkashErrorCode.PROVIDER_NOT_FOUND
+ );
+ }
+
+ const providerInfo = tx.provider;
+ elizaLogger.debug("Provider info retrieved", {
+ provider,
+ hostUri: providerInfo.hostUri
+ });
+
+ const manifest = sdl.manifestSortedJSON();
+ const path = `/deployment/${dseq}/manifest`;
+
+ elizaLogger.info("Sending manifest to provider", {
+ dseq,
+ provider,
+ manifestLength: manifest.length
+ });
+
+ const uri = new URL(providerInfo.hostUri);
+
+ const httpsAgent = new https.Agent({
+ cert: certificate.cert,
+ key: certificate.privateKey,
+ rejectUnauthorized: false,
+ keepAlive: false,
+ timeout: 10000
+ });
+
+ try {
+ const fullUrl = `${uri.protocol}//${uri.hostname}${uri.port ? ':' + uri.port : ''}${path}`;
+ elizaLogger.debug("Making manifest request", {
+ url: fullUrl,
+ method: 'PUT',
+ manifestLength: manifest.length
+ });
+
+ const response = await axios.put(fullUrl, manifest, {
+ headers: {
+ 'Content-Type': 'application/json',
+ 'Accept': 'application/json'
+ },
+ httpsAgent,
+ timeout: 10000,
+ validateStatus: null // Don't throw on any status code
+ });
+
+ if (response.status !== 200) {
+ elizaLogger.error("Failed to send manifest", {
+ statusCode: response.status,
+ statusText: response.statusText,
+ dseq
+ });
+ throw new Error(`Failed to send manifest: ${response.status} ${response.statusText}`);
+ }
+
+ elizaLogger.info("Manifest sent successfully", { dseq });
+ } finally {
+ httpsAgent.destroy();
+ }
+
+ // Wait for deployment to start
+ elizaLogger.info("Waiting for deployment to start", { dseq });
+ const startTime = Date.now();
+ const timeout = 1000 * 60 * 10; // 10 minutes timeout
+ let consecutiveErrors = 0;
+ const MAX_CONSECUTIVE_ERRORS = 5;
+
+ while (Date.now() - startTime < timeout) {
+ const elapsedTime = Math.round((Date.now() - startTime) / 1000);
+ elizaLogger.debug("Checking deployment status", {
+ dseq,
+ elapsedTime: `${elapsedTime}s`,
+ remainingTime: `${Math.round(timeout/1000 - elapsedTime)}s`,
+ consecutiveErrors
+ });
+
+ try {
+ const status = await queryLeaseStatus(lease, providerInfo.hostUri, certificate);
+
+ if (status === undefined) {
+ consecutiveErrors++;
+ elizaLogger.debug("Status check returned undefined", {
+ dseq,
+ consecutiveErrors,
+ maxConsecutiveErrors: MAX_CONSECUTIVE_ERRORS
+ });
+
+ if (consecutiveErrors >= MAX_CONSECUTIVE_ERRORS) {
+ elizaLogger.warn("Too many consecutive undefined status responses", {
+ dseq,
+ consecutiveErrors
+ });
+ // Don't throw, just continue waiting
+ consecutiveErrors = 0;
+ }
+
+ await new Promise(resolve => setTimeout(resolve, 3000));
+ continue;
+ }
+
+ // Reset error counter on successful status check
+ consecutiveErrors = 0;
+
+ for (const [name, service] of Object.entries<{ uris?: string[] }>(status.services)) {
+ if (service.uris) {
+ const rawUrl = service.uris[0];
+ // Ensure URL has protocol
+ const serviceUrl = rawUrl.startsWith('http') ? rawUrl : `http://${rawUrl}`;
+ elizaLogger.info("Service is available", {
+ name,
+ rawUrl,
+ serviceUrl,
+ dseq
+ });
+ return serviceUrl;
+ }
+ }
+ } catch (error) {
+ consecutiveErrors++;
+ const errorMessage = error instanceof Error ? error.message : String(error);
+ elizaLogger.warn("Error checking deployment status", {
+ error: errorMessage,
+ dseq,
+ consecutiveErrors,
+ maxConsecutiveErrors: MAX_CONSECUTIVE_ERRORS
+ });
+
+ if (consecutiveErrors >= MAX_CONSECUTIVE_ERRORS) {
+ elizaLogger.error("Too many consecutive errors checking deployment status", {
+ dseq,
+ consecutiveErrors,
+ error: errorMessage
+ });
+ throw new AkashError(
+ "Too many consecutive errors checking deployment status",
+ AkashErrorCode.DEPLOYMENT_START_TIMEOUT,
+ { dseq, error: errorMessage }
+ );
+ }
+ }
+
+ await new Promise(resolve => setTimeout(resolve, 3000));
+ }
+
+ elizaLogger.error("Deployment start timeout", {
+ dseq,
+ timeout: "10 minutes"
+ });
+ throw new AkashError(
+ "Could not start deployment. Timeout reached.",
+ AkashErrorCode.DEPLOYMENT_START_TIMEOUT
+ );
+ } catch (error) {
+ elizaLogger.error("Error during manifest send process", {
+ error: error instanceof Error ? error.message : String(error),
+ stack: error instanceof Error ? error.stack : undefined,
+ dseq: lease.id.dseq
+ });
+ throw error;
+ }
+}
+
+async function loadOrCreateCertificate(wallet: DirectSecp256k1HdWallet, client: SigningStargateClient): Promise {
+ elizaLogger.info("=== Starting Certificate Creation/Loading Process ===");
+ try {
+ const accounts = await wallet.getAccounts();
+ const address = accounts[0].address;
+ elizaLogger.debug("Got wallet address for certificate", {
+ address,
+ addressLength: address.length,
+ addressPrefix: address.substring(0, 6)
+ });
+
+ // Check if certificate exists
+ if (fs.existsSync(CERTIFICATE_PATH)) {
+ elizaLogger.info("Found existing certificate file", { path: CERTIFICATE_PATH });
+ const cert = loadCertificate(CERTIFICATE_PATH);
+ elizaLogger.debug("Loaded existing certificate", {
+ hasCert: !!cert.cert,
+ hasPrivateKey: !!cert.privateKey,
+ hasPublicKey: !!cert.publicKey,
+ certLength: cert.cert?.length,
+ privateKeyLength: cert.privateKey?.length,
+ publicKeyLength: cert.publicKey?.length
+ });
+ return cert;
+ }
+
+ // Create new certificate exactly like the example
+ elizaLogger.info("No existing certificate found, creating new one", { address });
+ const certificate = certificateManager.generatePEM(address);
+ elizaLogger.debug("Certificate generated", {
+ hasCert: !!certificate.cert,
+ hasPrivateKey: !!certificate.privateKey,
+ hasPublicKey: !!certificate.publicKey,
+ certLength: certificate.cert?.length,
+ privateKeyLength: certificate.privateKey?.length,
+ publicKeyLength: certificate.publicKey?.length
+ });
+
+ // Broadcast certificate
+ elizaLogger.info("Broadcasting certificate to network", {
+ address,
+ certLength: certificate.cert?.length,
+ publicKeyLength: certificate.publicKey?.length
+ });
+
+ const result = await cert.broadcastCertificate(
+ certificate,
+ address,
+ client as any
+ ).catch(error => {
+ elizaLogger.error("Certificate broadcast failed", {
+ error: error instanceof Error ? error.message : String(error),
+ stack: error instanceof Error ? error.stack : undefined,
+ address,
+ certLength: certificate.cert?.length
+ });
+ throw error;
+ });
+
+ if (result.code !== 0) {
+ const error = `Could not create certificate: ${result.rawLog}`;
+ elizaLogger.error("Certificate broadcast returned error code", {
+ code: result.code,
+ rawLog: result.rawLog,
+ address,
+ txHash: result.transactionHash
+ });
+ throw new AkashError(
+ error,
+ AkashErrorCode.CERTIFICATE_CREATION_FAILED,
+ { rawLog: result.rawLog }
+ );
+ }
+
+ elizaLogger.info("Certificate broadcast successful", {
+ code: result.code,
+ txHash: result.transactionHash,
+ height: result.height,
+ gasUsed: result.gasUsed
+ });
+
+ // Save certificate
+ saveCertificate(certificate);
+ elizaLogger.info("Certificate saved to file", { path: CERTIFICATE_PATH });
+
+ elizaLogger.info("Certificate process completed successfully", {
+ hasCert: !!certificate.cert,
+ hasPrivateKey: !!certificate.privateKey,
+ hasPublicKey: !!certificate.publicKey,
+ path: CERTIFICATE_PATH
+ });
+
+ return certificate;
+ } catch (error) {
+ elizaLogger.error("Certificate creation/broadcast process failed", {
+ error: error instanceof Error ? error.message : String(error),
+ stack: error instanceof Error ? error.stack : undefined,
+ path: CERTIFICATE_PATH
+ });
+ throw error;
+ }
+}
+
+async function parseSDL(sdlContent: string): Promise {
+ try {
+ // Clean up SDL content by taking only the part after the YAML document separator
+ const yamlSeparatorIndex = sdlContent.indexOf('---');
+ if (yamlSeparatorIndex === -1) {
+ throw new Error("No YAML document separator (---) found in SDL");
+ }
+
+ // Extract only the actual YAML content
+ const cleanSDL = sdlContent.substring(yamlSeparatorIndex);
+
+ elizaLogger.info("Starting SDL parsing process", {
+ originalLength: sdlContent.length,
+ cleanLength: cleanSDL.length,
+ yamlSeparatorIndex,
+ cleanContent: cleanSDL.substring(0, 200) + '...',
+ firstLine: cleanSDL.split('\n')[0],
+ lastLine: cleanSDL.split('\n').slice(-1)[0],
+ lineCount: cleanSDL.split('\n').length,
+ hasVersion: cleanSDL.includes('version: "2.0"'),
+ hasServices: cleanSDL.includes('services:'),
+ hasProfiles: cleanSDL.includes('profiles:'),
+ hasDeployment: cleanSDL.includes('deployment:'),
+ charCodes: cleanSDL.substring(0, 50).split('').map(c => c.charCodeAt(0))
+ });
+
+ // Try to parse SDL with clean content - exactly like the example
+ const parsedSDL = SDL.fromString(cleanSDL, "beta3");
+ elizaLogger.debug("Initial SDL parsing successful", {
+ hasVersion: !!parsedSDL.version,
+ hasServices: !!parsedSDL.services,
+ hasProfiles: !!parsedSDL.profiles,
+ hasDeployment: !!parsedSDL.deployments,
+ serviceCount: Object.keys(parsedSDL.services || {}).length,
+ profileCount: Object.keys(parsedSDL.profiles || {}).length
+ });
+
+ // Get groups and version like the example
+ const groups = parsedSDL.groups();
+ const version = await parsedSDL.manifestVersion();
+
+ elizaLogger.info("SDL validation completed", {
+ groupCount: groups.length,
+ version,
+ groups: JSON.stringify(groups)
+ });
+
+ return parsedSDL;
+ } catch (error) {
+ elizaLogger.error("Failed to parse SDL", {
+ error: error instanceof Error ? error.message : String(error),
+ stack: error instanceof Error ? error.stack : undefined,
+ sdlContent: sdlContent.substring(0, 200) + '...',
+ sdlLength: sdlContent.length
+ });
+ throw error;
+ }
+}
+
+export const createDeploymentAction: Action = {
+ name: "CREATE_DEPLOYMENT",
+ similes: ["DEPLOY", "START_DEPLOYMENT", "LAUNCH"],
+ description: "Create a new deployment on Akash Network",
+ examples: [[
+ {
+ user: "user",
+ content: {
+ text: "Deploy SDL on Akash Network",
+ sdl: "version: \"2.0\"\n\nservices:\n web:\n image: nginx\n expose:\n - port: 80\n as: 80\n to:\n - global: true"
+ } as CreateDeploymentContent
+ } as ActionExample
+ ]],
+
+ validate: async (runtime: IAgentRuntime, message: Memory): Promise => {
+ elizaLogger.debug("=== Starting Deployment Validation ===");
+ elizaLogger.debug("Validating deployment request", { message });
+
+ // Check if plugin is properly loaded
+ if (!isPluginLoaded(runtime, "akash")) {
+ elizaLogger.error("Akash plugin not properly loaded during validation");
+ return false;
+ }
+
+ try {
+ const params = message.content as Partial;
+ elizaLogger.debug("Checking SDL content", { params });
+
+ // Get SDL content either from direct string, specified file, or default file
+ let sdlContent: string;
+ if (params.sdl) {
+ sdlContent = params.sdl;
+ } else if (params.sdlFile) {
+ sdlContent = loadSDLFromFile(params.sdlFile);
+ } else {
+ sdlContent = loadSDLFromFile(DEFAULT_SDL_PATH);
+ }
+
+ if (params.deposit && !validateDeposit(params.deposit)) {
+ throw new AkashError(
+ "Invalid deposit format",
+ AkashErrorCode.VALIDATION_PARAMETER_INVALID,
+ { parameter: "deposit", value: params.deposit }
+ );
+ }
+
+ elizaLogger.debug("Validating SDL format");
+ try {
+ // Clean up SDL content by taking only the part after the YAML document separator
+ const yamlSeparatorIndex = sdlContent.indexOf('---');
+ if (yamlSeparatorIndex === -1) {
+ throw new Error("No YAML document separator (---) found in SDL");
+ }
+
+ // Extract only the actual YAML content
+ const cleanSDL = sdlContent.substring(yamlSeparatorIndex);
+
+ // Use exact same approach as example for validation
+ const sdl = SDL.fromString(cleanSDL, "beta3");
+ await sdl.manifestVersion(); // Verify we can get the version
+ elizaLogger.debug("SDL format validation successful", {
+ groups: sdl.groups(),
+ groupCount: sdl.groups().length
+ });
+ } catch (sdlError) {
+ elizaLogger.error("SDL format validation failed", { error: sdlError });
+ throw new AkashError(
+ `Invalid SDL format: ${sdlError instanceof Error ? sdlError.message : String(sdlError)}`,
+ AkashErrorCode.VALIDATION_SDL_FAILED,
+ { sdl: sdlContent }
+ );
+ }
+
+ elizaLogger.debug("Validation completed successfully");
+ return true;
+ } catch (error) {
+ elizaLogger.error("Deployment validation failed", {
+ error: error instanceof AkashError ? {
+ category: error.category,
+ code: error.code,
+ message: error.message,
+ details: error.details
+ } : String(error)
+ });
+ return false;
+ }
+ },
+
+ handler: async (
+ runtime: IAgentRuntime,
+ message: Memory,
+ state: State | undefined,
+ _options: { [key: string]: unknown; } = {},
+ callback?: HandlerCallback
+ ): Promise => {
+ const actionId = Date.now().toString();
+ elizaLogger.info("=== Starting Deployment Creation ===", {
+ actionId,
+ messageId: message.id,
+ userId: message.userId
+ });
+
+ // Inspect runtime to verify plugin and action registration
+ inspectRuntime(runtime);
+
+ try {
+ elizaLogger.debug("=== Validating Akash Configuration ===");
+ const config = await validateAkashConfig(runtime);
+ elizaLogger.debug("Configuration validated successfully", {
+ rpcEndpoint: config.RPC_ENDPOINT,
+ chainId: config.AKASH_CHAIN_ID,
+ version: config.AKASH_VERSION,
+ hasMnemonic: !!config.AKASH_MNEMONIC
+ });
+
+ const params = message.content as CreateDeploymentContent;
+ elizaLogger.debug("=== Processing Deployment Parameters ===", {
+ hasSDL: !!params.sdl,
+ hasSDLFile: !!params.sdlFile,
+ hasDeposit: !!params.deposit
+ });
+
+ // Get SDL content either from direct string, specified file, or default file
+ let sdlContent: string;
+ let sdlSource: string;
+ if (params.sdl) {
+ sdlContent = params.sdl;
+ sdlSource = 'direct';
+ } else if (params.sdlFile) {
+ sdlContent = loadSDLFromFile(params.sdlFile);
+ sdlSource = 'file';
+ } else {
+ sdlContent = loadSDLFromFile(DEFAULT_SDL_PATH);
+ sdlSource = 'default';
+ }
+ elizaLogger.debug("SDL content loaded", {
+ source: sdlSource,
+ contentLength: sdlContent.length
+ });
+
+ if (params.deposit && !validateDeposit(params.deposit)) {
+ elizaLogger.error("Invalid deposit format", {
+ deposit: params.deposit
+ });
+ throw new AkashError(
+ "Invalid deposit format",
+ AkashErrorCode.VALIDATION_PARAMETER_INVALID,
+ { parameter: "deposit", value: params.deposit }
+ );
+ }
+
+ // Initialize wallet from mnemonic
+ elizaLogger.info("=== Initializing Wallet and Client ===");
+ const wallet = await initializeWallet(config.AKASH_MNEMONIC);
+ const accounts = await wallet.getAccounts();
+ const address = accounts[0].address;
+ elizaLogger.debug("Wallet initialized", {
+ address,
+ accountCount: accounts.length
+ });
+
+ // Setup client
+ elizaLogger.debug("Setting up Stargate client");
+ const client = await setupClient(wallet, config.RPC_ENDPOINT);
+ elizaLogger.debug("Client setup completed", {
+ rpcEndpoint: config.RPC_ENDPOINT
+ });
+
+ // Load or create certificate
+ elizaLogger.info("=== Setting up Certificate ===");
+ const certificate = await loadOrCreateCertificate(wallet, client);
+ elizaLogger.debug("Certificate setup completed", {
+ hasCert: !!certificate.cert,
+ hasPrivateKey: !!certificate.privateKey,
+ hasPublicKey: !!certificate.publicKey
+ });
+
+ // Parse SDL
+ elizaLogger.info("=== Parsing SDL Configuration ===");
+ let sdl: SDL;
+ try {
+ sdl = await parseSDL(sdlContent);
+ elizaLogger.debug("SDL parsed successfully", {
+ groupCount: sdl.groups().length,
+ groups: sdl.groups(),
+ version: await sdl.manifestVersion()
+ });
+ } catch (sdlError) {
+ elizaLogger.error("SDL parsing failed", {
+ error: sdlError instanceof Error ? sdlError.message : String(sdlError),
+ sdlContent
+ });
+ throw new AkashError(
+ `SDL parsing failed: ${sdlError instanceof Error ? sdlError.message : String(sdlError)}`,
+ AkashErrorCode.MANIFEST_PARSING_FAILED,
+ {
+ sdl: sdlContent,
+ actionId
+ }
+ );
+ }
+
+ elizaLogger.info("=== Creating Deployment Message ===");
+ const blockHeight = await client.getHeight();
+ elizaLogger.debug("Current block height", { blockHeight });
+
+ const deployment = {
+ id: {
+ owner: address,
+ dseq: blockHeight
+ },
+ groups: sdl.groups(),
+ deposit: {
+ denom: "uakt",
+ amount: params.deposit?.replace("uakt", "") || config.AKASH_DEPOSIT.replace("uakt", "")
+ },
+ version: await sdl.manifestVersion(),
+ depositor: address
+ };
+
+ elizaLogger.debug("Deployment object created", {
+ owner: deployment.id.owner,
+ dseq: deployment.id.dseq,
+ groupCount: deployment.groups.length,
+ groups: deployment.groups,
+ deposit: deployment.deposit,
+ version: deployment.version
+ });
+
+ const msg = {
+ typeUrl: "/akash.deployment.v1beta3.MsgCreateDeployment",
+ value: MsgCreateDeployment.fromPartial(deployment)
+ };
+
+ // Broadcast transaction with retry for network issues
+ elizaLogger.info("=== Broadcasting Deployment Transaction ===", {
+ owner: address,
+ dseq: blockHeight,
+ deposit: params.deposit || config.AKASH_DEPOSIT,
+ groups: deployment.groups
+ });
+
+ const result = await withRetry(async () => {
+ elizaLogger.debug("Attempting to sign and broadcast transaction", {
+ attempt: 'current',
+ fees: config.AKASH_DEPOSIT,
+ gas: "800000",
+ groups: deployment.groups
+ });
+
+ const txResult = await client.signAndBroadcast(
+ address,
+ [msg],
+ {
+ amount: [{ denom: "uakt", amount: config.AKASH_DEPOSIT.replace("uakt", "") }],
+ gas: "800000",
+ }
+ );
+
+ elizaLogger.debug("Transaction broadcast result", {
+ code: txResult.code,
+ height: txResult.height,
+ transactionHash: txResult.transactionHash,
+ gasUsed: txResult.gasUsed,
+ gasWanted: txResult.gasWanted,
+ rawLog: txResult.rawLog
+ });
+
+ if (txResult.code !== 0) {
+ elizaLogger.error("Transaction failed", {
+ code: txResult.code,
+ rawLog: txResult.rawLog,
+ groups: deployment.groups
+ });
+ throw new AkashError(
+ `Transaction failed: ${txResult.rawLog}`,
+ AkashErrorCode.DEPLOYMENT_CREATION_FAILED,
+ {
+ rawLog: txResult.rawLog,
+ dseq: blockHeight,
+ owner: address,
+ actionId,
+ groups: deployment.groups
+ }
+ );
+ }
+
+ return txResult;
+ });
+
+ elizaLogger.info("=== Deployment Created Successfully ===", {
+ txHash: result.transactionHash,
+ owner: address,
+ dseq: blockHeight,
+ actionId,
+ height: result.height,
+ gasUsed: result.gasUsed
+ });
+
+ // Create lease
+ elizaLogger.debug("=== Creating Lease ===");
+ const lease = await createLease(deployment, wallet, client, config.RPC_ENDPOINT);
+ elizaLogger.debug("Lease created", {
+ leaseId: lease.id,
+ dseq: deployment.id.dseq
+ });
+
+ // Send manifest
+ elizaLogger.debug("=== Sending Manifest ===");
+ const serviceUrl = await sendManifest(sdl, lease, certificate, config.RPC_ENDPOINT);
+ elizaLogger.debug("Manifest sent successfully", {
+ serviceUrl
+ });
+
+ if (callback) {
+ elizaLogger.info("=== Preparing callback response for deployment creation ===", {
+ hasCallback: true,
+ actionId,
+ dseq: String(blockHeight)
+ });
+
+ const callbackResponse = {
+ text: `Deployment created and started successfully\nDSEQ: ${blockHeight}\nOwner: ${address}\nTx Hash: ${result.transactionHash}\nService URL: ${serviceUrl}`,
+ content: {
+ success: true,
+ data: {
+ txHash: result.transactionHash,
+ owner: address,
+ dseq: String(blockHeight),
+ serviceUrl
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'createDeployment',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+
+ elizaLogger.info("=== Executing callback with response ===", {
+ actionId,
+ responseText: callbackResponse.text,
+ hasContent: !!callbackResponse.content,
+ contentKeys: Object.keys(callbackResponse.content),
+ metadata: callbackResponse.content.metadata
+ });
+
+ callback(callbackResponse);
+
+ elizaLogger.info("=== Callback executed successfully ===", {
+ actionId,
+ timestamp: new Date().toISOString()
+ });
+ }
+
+ elizaLogger.info("=== Deployment Process Completed Successfully ===", {
+ actionId,
+ txHash: result.transactionHash,
+ dseq: blockHeight
+ });
+
+ return true;
+ } catch (error) {
+ elizaLogger.error("=== Deployment Creation Failed ===", {
+ error: error instanceof AkashError ? {
+ category: error.category,
+ code: error.code,
+ message: error.message,
+ details: error.details
+ } : String(error),
+ actionId,
+ stack: error instanceof Error ? error.stack : undefined
+ });
+
+ if (callback) {
+ elizaLogger.info("=== Preparing error callback response ===", {
+ actionId,
+ hasCallback: true,
+ errorType: error instanceof AkashError ? 'AkashError' : 'Error'
+ });
+
+ const errorResponse = {
+ text: "Failed to create deployment",
+ content: {
+ success: false,
+ error: {
+ code: error instanceof AkashError ? error.code : AkashErrorCode.DEPLOYMENT_CREATION_FAILED,
+ message: error instanceof Error ? error.message : String(error)
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'createDeployment',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+
+ elizaLogger.info("=== Executing error callback ===", {
+ actionId,
+ responseText: errorResponse.text,
+ hasContent: !!errorResponse.content,
+ contentKeys: Object.keys(errorResponse.content)
+ });
+
+ callback(errorResponse);
+
+ elizaLogger.info("=== Error callback executed successfully ===", {
+ actionId,
+ timestamp: new Date().toISOString()
+ });
+ }
+
+ return false;
+ }
+ },
+};
+
+export default createDeploymentAction;
\ No newline at end of file
diff --git a/packages/plugin-akash/src/actions/estimateGas.ts b/packages/plugin-akash/src/actions/estimateGas.ts
new file mode 100644
index 00000000000..309c6c2c817
--- /dev/null
+++ b/packages/plugin-akash/src/actions/estimateGas.ts
@@ -0,0 +1,356 @@
+import { Action, elizaLogger } from "@elizaos/core";
+import { IAgentRuntime, Memory, State, HandlerCallback, Content, ActionExample } from "@elizaos/core";
+import { DirectSecp256k1HdWallet, Registry, EncodeObject } from "@cosmjs/proto-signing";
+import { SigningStargateClient } from "@cosmjs/stargate";
+import { MsgCloseDeployment } from "@akashnetwork/akash-api/akash/deployment/v1beta3";
+import { getAkashTypeRegistry, getTypeUrl } from "@akashnetwork/akashjs/build/stargate";
+import { validateAkashConfig } from "../environment";
+import { AkashError, AkashErrorCode } from "../error/error";
+import { encodingForModel } from "js-tiktoken";
+
+/*
+interface AkashMessage {
+ typeUrl: string;
+ value: {
+ id?: {
+ owner: string;
+ dseq: string;
+ };
+ [key: string]: unknown;
+ };
+}
+*/
+
+interface EstimateGasContent extends Content {
+ text: string;
+ dseq?: string;
+ operation: "close" | "create" | "update";
+ message?: EncodeObject;
+}
+
+function getTotalTokensFromString(str: string): number {
+ try {
+ const encoding = encodingForModel("gpt-3.5-turbo");
+ return encoding.encode(str).length;
+ } catch (error) {
+ elizaLogger.warn("Failed to count tokens", { error });
+ return 0;
+ }
+}
+
+export const estimateGas: Action = {
+ name: "ESTIMATE_GAS",
+ similes: ["CALCULATE_GAS", "GET_GAS_ESTIMATE", "CHECK_GAS"],
+ description: "Estimate gas for a transaction on Akash Network",
+ examples: [[
+ {
+ user: "user",
+ content: {
+ text: "Can you estimate gas for closing deployment with DSEQ 123456?",
+ operation: "close"
+ } as EstimateGasContent
+ } as ActionExample
+ ]],
+
+ validate: async (runtime: IAgentRuntime, message: Memory): Promise => {
+ elizaLogger.debug("Validating gas estimation request", { message });
+ try {
+ const params = message.content as Partial;
+ const config = await validateAkashConfig(runtime);
+
+ // Extract DSEQ from text if present
+ if (params.text && !params.dseq) {
+ const dseqMatch = params.text.match(/dseq\s*(?::|=|\s)\s*(\d+)/i) ||
+ params.text.match(/deployment\s+(?:number|sequence|#)?\s*(\d+)/i) ||
+ params.text.match(/(\d{6,})/); // Matches standalone numbers of 6+ digits
+ if (dseqMatch) {
+ params.dseq = dseqMatch[1];
+ elizaLogger.debug("Extracted DSEQ from text", {
+ text: params.text,
+ extractedDseq: params.dseq
+ });
+ }
+ }
+
+ // If no operation provided, check environment configuration
+ if (!params.operation) {
+ if (config.AKASH_GAS_OPERATION) {
+ params.operation = config.AKASH_GAS_OPERATION as "close" | "create" | "update";
+ elizaLogger.info("Using operation from environment", { operation: params.operation });
+ } else {
+ throw new AkashError(
+ "Operation type is required (close, create, or update)",
+ AkashErrorCode.VALIDATION_PARAMETER_MISSING,
+ { parameter: "operation" }
+ );
+ }
+ }
+
+ // For close operations, check DSEQ from various sources
+ if (params.operation === "close") {
+ if (!params.dseq) {
+ if (config.AKASH_GAS_DSEQ) {
+ params.dseq = config.AKASH_GAS_DSEQ;
+ elizaLogger.info("Using DSEQ from environment", { dseq: params.dseq });
+ } else {
+ throw new AkashError(
+ "Deployment sequence (dseq) is required for close operation",
+ AkashErrorCode.VALIDATION_PARAMETER_MISSING,
+ { parameter: "dseq" }
+ );
+ }
+ }
+ }
+
+ // For create/update operations, check message
+ if ((params.operation === "create" || params.operation === "update") && !params.message) {
+ throw new AkashError(
+ "Message is required for create/update operations",
+ AkashErrorCode.VALIDATION_PARAMETER_MISSING,
+ { parameter: "message" }
+ );
+ }
+
+ return true;
+ } catch (error) {
+ elizaLogger.error("Gas estimation validation failed", {
+ error: error instanceof AkashError ? {
+ code: error.code,
+ message: error.message,
+ details: error.details
+ } : String(error)
+ });
+ return false;
+ }
+ },
+
+ handler: async (
+ runtime: IAgentRuntime,
+ message: Memory,
+ state: State | undefined,
+ options: { [key: string]: unknown } = {},
+ callback?: HandlerCallback
+ ): Promise => {
+ const actionId = Date.now().toString();
+ elizaLogger.info("Starting gas estimation", { actionId });
+
+ elizaLogger.debug("=== Handler Parameters ===", {
+ hasRuntime: !!runtime,
+ hasMessage: !!message,
+ hasState: !!state,
+ hasOptions: !!options,
+ hasCallback: !!callback,
+ actionId
+ });
+
+ try {
+ const config = await validateAkashConfig(runtime);
+ const params = message.content as Partial;
+
+ // Initialize wallet and get address
+ const wallet = await DirectSecp256k1HdWallet.fromMnemonic(config.AKASH_MNEMONIC, { prefix: "akash" });
+ const [account] = await wallet.getAccounts();
+
+ // Initialize client with Akash registry
+ const myRegistry = new Registry(getAkashTypeRegistry());
+ const client = await SigningStargateClient.connectWithSigner(
+ config.RPC_ENDPOINT,
+ wallet,
+ { registry: myRegistry }
+ );
+
+ let msg: EncodeObject;
+ switch (params.operation) {
+ case "close":
+ msg = {
+ typeUrl: getTypeUrl(MsgCloseDeployment),
+ value: MsgCloseDeployment.fromPartial({
+ id: {
+ owner: account.address,
+ dseq: params.dseq
+ }
+ })
+ };
+ break;
+ case "create":
+ case "update":
+ if (!params.message) {
+ if (callback) {
+ callback({
+ text: `Message is required for ${params.operation} operations.`,
+ content: {
+ success: false,
+ error: {
+ code: AkashErrorCode.VALIDATION_PARAMETER_MISSING,
+ message: "Missing message",
+ help: `Please provide a message object for the ${params.operation} operation.`
+ }
+ }
+ });
+ }
+ return false;
+ }
+ msg = params.message;
+ break;
+ default:
+ if (callback) {
+ callback({
+ text: `Invalid operation type: ${params.operation}. Must be one of: close, create, or update.`,
+ content: {
+ success: false,
+ error: {
+ code: AkashErrorCode.VALIDATION_PARAMETER_INVALID,
+ message: "Invalid operation",
+ help: "Specify a valid operation type: 'close', 'create', or 'update'."
+ }
+ }
+ });
+ }
+ return false;
+ }
+
+ // Estimate gas
+ elizaLogger.info("Estimating gas for operation", {
+ operation: params.operation,
+ dseq: params.dseq,
+ owner: account.address
+ });
+
+ const gasEstimate = await client.simulate(
+ account.address,
+ [msg],
+ `Estimate gas for ${params.operation} operation`
+ );
+
+ elizaLogger.info("Gas estimation completed", {
+ gasEstimate,
+ operation: params.operation,
+ dseq: params.dseq,
+ owner: account.address,
+ actionId
+ });
+
+ if (callback) {
+ elizaLogger.info("=== Preparing callback response for gas estimation ===", {
+ hasCallback: true,
+ actionId,
+ operation: params.operation,
+ dseq: params.dseq
+ });
+
+ const operationText = params.operation === "close" ? `closing deployment ${params.dseq}` : params.operation;
+ const estimateData = {
+ gasEstimate,
+ operation: params.operation,
+ dseq: params.dseq,
+ owner: account.address,
+ message: msg
+ };
+
+ let responseText = `I've estimated the gas for ${operationText}:\n`;
+ responseText += `• Gas Required: ${gasEstimate} units\n`;
+ responseText += `• Operation: ${params.operation}\n`;
+ if (params.dseq) {
+ responseText += `• DSEQ: ${params.dseq}\n`;
+ }
+ responseText += `• Owner: ${account.address}`;
+
+ const response = {
+ text: responseText,
+ content: {
+ success: true,
+ data: estimateData,
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'estimateGas',
+ version: '1.0.0',
+ actionId,
+ tokenCount: getTotalTokensFromString(responseText)
+ }
+ }
+ };
+
+ elizaLogger.info("=== Executing callback with response ===", {
+ actionId,
+ responseText: response.text,
+ hasContent: !!response.content,
+ contentKeys: Object.keys(response.content),
+ metadata: response.content.metadata
+ });
+
+ callback(response);
+
+ elizaLogger.info("=== Callback executed successfully ===", {
+ actionId,
+ timestamp: new Date().toISOString()
+ });
+ } else {
+ elizaLogger.warn("=== No callback provided for gas estimation ===", {
+ actionId,
+ operation: params.operation,
+ dseq: params.dseq
+ });
+ }
+
+ return true;
+ } catch (error) {
+ elizaLogger.error("Gas estimation failed", {
+ error: error instanceof Error ? error.message : String(error),
+ actionId
+ });
+
+ if (callback) {
+ elizaLogger.info("=== Preparing error callback response ===", {
+ actionId,
+ hasCallback: true,
+ errorType: error instanceof AkashError ? 'AkashError' : 'Error'
+ });
+
+ const errorResponse = {
+ code: error instanceof AkashError ? error.code : AkashErrorCode.API_ERROR,
+ message: error instanceof Error ? error.message : String(error),
+ details: error instanceof AkashError ? error.details : undefined
+ };
+
+ const response = {
+ text: `Failed to estimate gas: ${errorResponse.message}`,
+ content: {
+ success: false,
+ error: errorResponse,
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'estimateGas',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+
+ elizaLogger.info("=== Executing error callback ===", {
+ actionId,
+ errorResponse,
+ hasContent: !!response.content,
+ contentKeys: Object.keys(response.content)
+ });
+
+ callback(response);
+
+ elizaLogger.info("=== Error callback executed ===", {
+ actionId,
+ timestamp: new Date().toISOString()
+ });
+ } else {
+ elizaLogger.warn("=== No callback provided for error handling ===", {
+ actionId,
+ errorMessage: error instanceof Error ? error.message : String(error)
+ });
+ }
+
+ return false;
+ }
+ }
+};
+
+export default estimateGas;
\ No newline at end of file
diff --git a/packages/plugin-akash/src/actions/getDeploymentApi.ts b/packages/plugin-akash/src/actions/getDeploymentApi.ts
new file mode 100644
index 00000000000..417a9fc508a
--- /dev/null
+++ b/packages/plugin-akash/src/actions/getDeploymentApi.ts
@@ -0,0 +1,499 @@
+import { Action, elizaLogger } from "@elizaos/core";
+import { IAgentRuntime, Memory, State, HandlerCallback, Content, ActionExample } from "@elizaos/core";
+import { DirectSecp256k1HdWallet } from "@cosmjs/proto-signing";
+import { validateAkashConfig } from "../environment";
+import { AkashError, AkashErrorCode } from "../error/error";
+import * as fs from 'fs';
+import * as path from 'path';
+import { getDeploymentsPath } from "../utils/paths";
+
+export interface DeploymentInfo {
+ owner: string;
+ dseq: string;
+ status: string;
+ createdHeight: number;
+ cpuUnits: number;
+ gpuUnits: number;
+ memoryQuantity: number;
+ storageQuantity: number;
+}
+
+export interface DeploymentListResponse {
+ count: number;
+ results: DeploymentInfo[];
+}
+
+interface GetDeploymentsContent extends Content {
+ status?: 'active' | 'closed';
+ skip?: number;
+ limit?: number;
+}
+
+async function sleep(ms: number) {
+ return new Promise(resolve => setTimeout(resolve, ms));
+}
+
+async function fetchWithRetry(url: string, options: RequestInit, retries = 3, delay = 1000): Promise {
+ for (let i = 0; i < retries; i++) {
+ try {
+ const response = await fetch(url, options);
+ if (response.ok) {
+ return response;
+ }
+
+ const error = await response.text();
+ elizaLogger.warn(`API request failed (attempt ${i + 1}/${retries})`, {
+ status: response.status,
+ error
+ });
+
+ if (i < retries - 1) {
+ await sleep(delay * Math.pow(2, i)); // Exponential backoff
+ continue;
+ }
+
+ throw new AkashError(
+ `API request failed after ${retries} attempts: ${response.status} - ${error}`,
+ AkashErrorCode.API_ERROR
+ );
+ } catch (error) {
+ if (i === retries - 1) {
+ throw error;
+ }
+ elizaLogger.warn(`API request error (attempt ${i + 1}/${retries})`, {
+ error: error instanceof Error ? error.message : String(error)
+ });
+ await sleep(delay * Math.pow(2, i));
+ }
+ }
+ throw new AkashError(
+ `Failed to fetch after ${retries} retries`,
+ AkashErrorCode.API_ERROR
+ );
+}
+
+export async function initializeWallet(runtime: IAgentRuntime): Promise<{wallet: DirectSecp256k1HdWallet | null, address: string}> {
+ try {
+ // Validate configuration and get mnemonic
+ const config = await validateAkashConfig(runtime);
+
+ elizaLogger.info("Initializing wallet with config", {
+ hasMnemonic: !!config.AKASH_MNEMONIC,
+ hasWalletAddress: !!config.AKASH_WALLET_ADDRESS
+ });
+
+ // First try to get the wallet address directly
+ if (config.AKASH_WALLET_ADDRESS) {
+ elizaLogger.info("Using provided wallet address", {
+ address: config.AKASH_WALLET_ADDRESS
+ });
+ return {
+ wallet: null,
+ address: config.AKASH_WALLET_ADDRESS
+ };
+ }
+
+ // If no wallet address, create wallet from mnemonic
+ if (!config.AKASH_MNEMONIC) {
+ throw new AkashError(
+ "Neither AKASH_WALLET_ADDRESS nor AKASH_MNEMONIC provided",
+ AkashErrorCode.WALLET_NOT_INITIALIZED
+ );
+ }
+
+ try {
+ elizaLogger.info("Creating wallet from mnemonic");
+ const wallet = await DirectSecp256k1HdWallet.fromMnemonic(config.AKASH_MNEMONIC, {
+ prefix: "akash"
+ });
+
+ // Get account address
+ const accounts = await wallet.getAccounts();
+ const address = accounts[0].address;
+
+ elizaLogger.info("Wallet initialized from mnemonic", {
+ address,
+ accountCount: accounts.length
+ });
+
+ return { wallet, address };
+ } catch (error) {
+ throw new AkashError(
+ `Failed to initialize wallet: ${error instanceof Error ? error.message : String(error)}`,
+ AkashErrorCode.WALLET_NOT_INITIALIZED,
+ { originalError: error instanceof Error ? error.message : String(error) }
+ );
+ }
+ } catch (error) {
+ // Ensure all errors are properly wrapped as AkashError
+ if (error instanceof AkashError) {
+ throw error;
+ }
+ throw new AkashError(
+ `Failed to initialize wallet: ${error instanceof Error ? error.message : String(error)}`,
+ AkashErrorCode.WALLET_NOT_INITIALIZED,
+ { originalError: error instanceof Error ? error.message : String(error) }
+ );
+ }
+}
+
+export async function fetchDeployments(
+ runtime: IAgentRuntime,
+ status?: 'active' | 'closed',
+ skip = 0,
+ limit = 10
+): Promise {
+ elizaLogger.info("Initializing deployment fetch", {
+ status: status || 'all',
+ skip,
+ limit
+ });
+
+ try {
+ // Initialize wallet and get address
+ const { address } = await initializeWallet(runtime);
+
+ if (!address) {
+ throw new AkashError(
+ "Failed to get wallet address",
+ AkashErrorCode.WALLET_NOT_INITIALIZED
+ );
+ }
+
+ elizaLogger.info("Fetching deployments from API", {
+ address,
+ status: status || 'all',
+ skip,
+ limit
+ });
+
+ // Map status for API compatibility
+ const apiStatus = status;
+
+ // Don't include status in URL if not specified
+ const params = new URLSearchParams();
+ if (apiStatus) {
+ params.append('status', apiStatus);
+ }
+ params.append('reverseSorting', 'true');
+ const url = `https://console-api.akash.network/v1/addresses/${address}/deployments/${skip}/${limit}?${params.toString()}`;
+ elizaLogger.debug("Making API request", { url });
+
+ const response = await fetchWithRetry(url, {
+ headers: {
+ 'accept': 'application/json'
+ }
+ });
+
+ const data = await response.json() as DeploymentListResponse;
+ elizaLogger.info("Deployments fetched successfully", {
+ count: data.count,
+ resultCount: data.results.length,
+ status: status || 'all'
+ });
+
+ // Keep status as-is from API
+ data.results = data.results.map(deployment => ({
+ ...deployment,
+ status: deployment.status.toLowerCase()
+ }));
+
+ // Save deployments to files, organized by their actual status
+ const deploymentDir = getDeploymentsPath(import.meta.url);
+ elizaLogger.info("Using deployments directory", { deploymentDir });
+
+ // Create base deployments directory if it doesn't exist
+ if (!fs.existsSync(deploymentDir)) {
+ elizaLogger.info("Creating deployments directory", { deploymentDir });
+ fs.mkdirSync(deploymentDir, { recursive: true });
+ }
+
+ // Group deployments by status
+ const deploymentsByStatus = data.results.reduce((acc, deployment) => {
+ const status = deployment.status.toLowerCase();
+ if (!acc[status]) {
+ acc[status] = [];
+ }
+ acc[status].push(deployment);
+ return acc;
+ }, {} as Record);
+
+ // Save deployments by status
+ for (const [status, deployments] of Object.entries(deploymentsByStatus)) {
+ const statusDir = path.join(deploymentDir, status);
+ elizaLogger.info("Processing status directory", { statusDir, status, deploymentCount: deployments.length });
+
+ // Ensure status directory exists
+ if (!fs.existsSync(statusDir)) {
+ elizaLogger.info("Creating status directory", { statusDir });
+ fs.mkdirSync(statusDir, { recursive: true });
+ }
+
+ // Save all deployments for this status in parallel
+ await Promise.all(deployments.map(async (deployment) => {
+ const filePath = path.join(statusDir, `${deployment.dseq}.json`);
+ elizaLogger.debug("Saving deployment file", { filePath, dseq: deployment.dseq });
+ await saveDeploymentInfo(deployment, filePath);
+ }));
+ }
+
+ return data;
+ } catch (error) {
+ elizaLogger.error("Failed to fetch deployments", {
+ error: error instanceof Error ? error.message : String(error),
+ stack: error instanceof Error ? error.stack : undefined
+ });
+ throw error;
+ }
+}
+
+export async function saveDeploymentInfo(deploymentInfo: DeploymentInfo, filePath: string): Promise {
+ elizaLogger.info("Saving deployment info", {
+ dseq: deploymentInfo.dseq,
+ owner: deploymentInfo.owner,
+ filePath
+ });
+
+ try {
+ // Ensure directory exists
+ const dir = path.dirname(filePath);
+ if (!fs.existsSync(dir)) {
+ fs.mkdirSync(dir, { recursive: true });
+ }
+
+ // Save deployment info
+ fs.writeFileSync(filePath, JSON.stringify(deploymentInfo, null, 2), 'utf8');
+ elizaLogger.debug("Deployment info saved successfully");
+ } catch (error) {
+ elizaLogger.error("Failed to save deployment info", {
+ error: error instanceof Error ? error.message : String(error),
+ stack: error instanceof Error ? error.stack : undefined,
+ filePath
+ });
+ throw error;
+ }
+}
+
+export async function loadDeploymentInfo(filePath: string): Promise {
+ elizaLogger.info("Loading deployment info", { filePath });
+
+ try {
+ if (!fs.existsSync(filePath)) {
+ throw new AkashError(
+ `Deployment info file not found: ${filePath}`,
+ AkashErrorCode.FILE_NOT_FOUND
+ );
+ }
+
+ const data = fs.readFileSync(filePath, 'utf8');
+ const deploymentInfo = JSON.parse(data) as DeploymentInfo;
+ elizaLogger.debug("Deployment info loaded successfully", {
+ dseq: deploymentInfo.dseq,
+ owner: deploymentInfo.owner
+ });
+
+ return deploymentInfo;
+ } catch (error) {
+ elizaLogger.error("Failed to load deployment info", {
+ error: error instanceof Error ? error.message : String(error),
+ stack: error instanceof Error ? error.stack : undefined,
+ filePath
+ });
+ throw error;
+ }
+}
+
+export const getDeploymentApiAction: Action = {
+ name: "GET_DEPLOYMENTS",
+ similes: ["LIST_DEPLOYMENTS", "FETCH_DEPLOYMENTS", "SHOW_DEPLOYMENTS"],
+ description: "Fetch deployments from Akash Network",
+ examples: [[
+ {
+ user: "user",
+ content: {
+ text: "Get all deployments",
+ } as GetDeploymentsContent
+ } as ActionExample,
+ {
+ user: "assistant",
+ content: {
+ text: "Fetching all deployments..."
+ } as GetDeploymentsContent
+ } as ActionExample
+ ], [
+ {
+ user: "user",
+ content: {
+ text: "Get active deployments",
+ status: "active"
+ } as GetDeploymentsContent
+ } as ActionExample,
+ {
+ user: "assistant",
+ content: {
+ text: "Fetching active deployments..."
+ } as GetDeploymentsContent
+ } as ActionExample
+ ]],
+
+ validate: async (runtime: IAgentRuntime, message: Memory): Promise => {
+ elizaLogger.debug("Validating get deployments request", { message });
+ try {
+ const params = message.content as Partial;
+
+ if (params.status && !['active', 'closed'].includes(params.status)) {
+ throw new AkashError(
+ "Status must be either 'active' or 'closed'",
+ AkashErrorCode.VALIDATION_PARAMETER_INVALID,
+ { parameter: "status", value: params.status }
+ );
+ }
+
+ if (params.skip !== undefined && (typeof params.skip !== 'number' || params.skip < 0)) {
+ throw new AkashError(
+ "Skip must be a non-negative number",
+ AkashErrorCode.VALIDATION_PARAMETER_INVALID,
+ { parameter: "skip", value: params.skip }
+ );
+ }
+
+ if (params.limit !== undefined && (typeof params.limit !== 'number' || params.limit <= 0)) {
+ throw new AkashError(
+ "Limit must be a positive number",
+ AkashErrorCode.VALIDATION_PARAMETER_INVALID,
+ { parameter: "limit", value: params.limit }
+ );
+ }
+
+ return true;
+ } catch (error) {
+ elizaLogger.error("Get deployments validation failed", {
+ error: error instanceof AkashError ? {
+ code: error.code,
+ message: error.message,
+ details: error.details
+ } : String(error)
+ });
+ return false;
+ }
+ },
+
+ handler: async (
+ runtime: IAgentRuntime,
+ message: Memory,
+ state: State | undefined,
+ _options: { [key: string]: unknown } = {},
+ callback?: HandlerCallback
+ ): Promise => {
+ const actionId = Date.now().toString();
+ elizaLogger.info("Starting deployment API request", { actionId });
+
+ try {
+ // const config = await validateAkashConfig(runtime);
+ const params = message.content as Partial;
+
+ // Fetch deployments
+ const deployments = await fetchDeployments(
+ runtime,
+ params.status,
+ params.skip,
+ params.limit
+ );
+
+ if (callback) {
+ elizaLogger.info("=== Preparing callback response for deployments ===", {
+ hasCallback: true,
+ actionId,
+ deploymentCount: deployments.count
+ });
+
+ const callbackResponse = {
+ text: `Found ${deployments.count} deployment${deployments.count !== 1 ? 's' : ''}${params.status ? ` with status: ${params.status}` : ''}\n\nDeployments:\n${deployments.results.map(dep =>
+ `- DSEQ: ${dep.dseq}\n Status: ${dep.status}\n CPU: ${dep.cpuUnits} units\n Memory: ${dep.memoryQuantity} units\n Storage: ${dep.storageQuantity} units`
+ ).join('\n\n')}`,
+ content: {
+ success: true,
+ data: {
+ deployments: deployments.results,
+ total: deployments.count,
+ status: params.status || 'all'
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getDeploymentApi',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+
+ elizaLogger.info("=== Executing callback with response ===", {
+ actionId,
+ responseText: callbackResponse.text,
+ hasContent: !!callbackResponse.content,
+ contentKeys: Object.keys(callbackResponse.content),
+ metadata: callbackResponse.content.metadata
+ });
+
+ callback(callbackResponse);
+
+ elizaLogger.info("=== Callback executed successfully ===", {
+ actionId,
+ timestamp: new Date().toISOString()
+ });
+ }
+
+ return true;
+ } catch (error) {
+ elizaLogger.error("Get deployments request failed", {
+ error: error instanceof Error ? error.message : String(error),
+ actionId
+ });
+
+ if (callback) {
+ elizaLogger.info("=== Preparing error callback response ===", {
+ actionId,
+ hasCallback: true,
+ errorType: error instanceof AkashError ? 'AkashError' : 'Error'
+ });
+
+ const errorResponse = {
+ text: `Failed to get deployments: ${error instanceof Error ? error.message : String(error)}`,
+ content: {
+ success: false,
+ error: {
+ code: error instanceof AkashError ? error.code : AkashErrorCode.API_ERROR,
+ message: error instanceof Error ? error.message : String(error)
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getDeploymentApi',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+
+ elizaLogger.info("=== Executing error callback ===", {
+ actionId,
+ responseText: errorResponse.text,
+ hasContent: !!errorResponse.content,
+ contentKeys: Object.keys(errorResponse.content)
+ });
+
+ callback(errorResponse);
+
+ elizaLogger.info("=== Error callback executed ===", {
+ actionId,
+ timestamp: new Date().toISOString()
+ });
+ }
+
+ return false;
+ }
+ }
+};
+
+export default getDeploymentApiAction;
\ No newline at end of file
diff --git a/packages/plugin-akash/src/actions/getDeploymentStatus.ts b/packages/plugin-akash/src/actions/getDeploymentStatus.ts
new file mode 100644
index 00000000000..98a90dccf99
--- /dev/null
+++ b/packages/plugin-akash/src/actions/getDeploymentStatus.ts
@@ -0,0 +1,493 @@
+import { Action, elizaLogger } from "@elizaos/core";
+import { IAgentRuntime, Memory, State, HandlerCallback, Content, ActionExample } from "@elizaos/core";
+import { DirectSecp256k1HdWallet } from "@cosmjs/proto-signing";
+import { QueryDeploymentRequest, QueryClientImpl as DeploymentQueryClient } from "@akashnetwork/akash-api/akash/deployment/v1beta3";
+import { getRpc } from "@akashnetwork/akashjs/build/rpc";
+import { validateAkashConfig } from "../environment";
+import { AkashError, AkashErrorCode } from "../error/error";
+
+interface GetDeploymentStatusContent extends Content {
+ text: string;
+ dseq?: string;
+}
+
+interface DeploymentGroup {
+ groupId?: {
+ owner: string;
+ dseq: string;
+ gseq: number;
+ };
+ state: string;
+ resources: Array<{
+ resources: {
+ cpu: {
+ units: {
+ val: string;
+ };
+ };
+ memory: {
+ quantity: {
+ val: string;
+ };
+ };
+ storage: Array<{
+ quantity: {
+ val: string;
+ };
+ }>;
+ };
+ count: number;
+ price: {
+ denom: string;
+ amount: string;
+ };
+ }>;
+}
+
+interface DeploymentResponse {
+ deploymentId?: {
+ owner: string;
+ dseq: string;
+ };
+ state: string;
+ version: string;
+ createdAt: string;
+ escrowAccount?: {
+ balance?: {
+ denom: string;
+ amount: string;
+ };
+ };
+ groups?: DeploymentGroup[];
+}
+
+enum DeploymentState {
+ UNKNOWN = 0,
+ ACTIVE = 1,
+ CLOSED = 2,
+ INSUFFICIENT_FUNDS = 3,
+}
+
+export const getDeploymentStatusAction: Action = {
+ name: "GET_DEPLOYMENT_STATUS",
+ similes: ["CHECK_DEPLOYMENT", "DEPLOYMENT_STATUS", "DEPLOYMENT_STATE", "CHECK DSEQ"],
+ description: "Get the current status of a deployment on Akash Network",
+ examples: [[
+ {
+ user: "user",
+ content: {
+ text: "Can you check the deployment status of the DSEQ 123456?",
+ } as GetDeploymentStatusContent
+ } as ActionExample
+ ]],
+
+ validate: async (runtime: IAgentRuntime, message: Memory): Promise => {
+ elizaLogger.debug("Validating get deployment status request", { message });
+ try {
+ const params = message.content as Partial;
+ const config = await validateAkashConfig(runtime);
+
+ // Extract DSEQ from text if present
+ if (params.text && !params.dseq) {
+ // Pattern to match DSEQ followed by numbers
+ const dseqMatch = params.text.match(/(?:DSEQ|dseq)\s*(\d+)/i);
+ if (dseqMatch) {
+ params.dseq = dseqMatch[1];
+ elizaLogger.debug("Extracted DSEQ from text", {
+ text: params.text,
+ extractedDseq: params.dseq
+ });
+ }
+ }
+
+ // If no dseq provided, check environment configuration
+ if (!params.dseq) {
+ if (config.AKASH_DEP_STATUS === "dseq" && config.AKASH_DEP_DSEQ) {
+ params.dseq = config.AKASH_DEP_DSEQ;
+ } else if (config.AKASH_DEP_STATUS === "param_passed") {
+ elizaLogger.info("DSEQ parameter is required when AKASH_DEP_STATUS is set to param_passed", {
+ current_status: config.AKASH_DEP_STATUS
+ });
+ return true; // Allow validation to pass, we'll handle the missing parameter in the handler
+ } else {
+ elizaLogger.info("No DSEQ provided and no valid environment configuration found", {
+ dep_status: config.AKASH_DEP_STATUS,
+ dep_dseq: config.AKASH_DEP_DSEQ
+ });
+ return true; // Allow validation to pass, we'll handle the missing configuration in the handler
+ }
+ }
+
+ // If dseq is provided, validate its format
+ if (params.dseq && !/^\d+$/.test(params.dseq)) {
+ throw new AkashError(
+ "Invalid DSEQ format. Must be a numeric string",
+ AkashErrorCode.VALIDATION_PARAMETER_INVALID,
+ { parameter: "dseq", value: params.dseq }
+ );
+ }
+
+ return true;
+ } catch (error) {
+ elizaLogger.error("Get deployment status validation failed", {
+ error: error instanceof AkashError ? {
+ code: error.code,
+ message: error.message,
+ details: error.details
+ } : String(error)
+ });
+ return false;
+ }
+ },
+
+ handler: async (
+ runtime: IAgentRuntime,
+ message: Memory,
+ state: State | undefined,
+ _options: { [key: string]: unknown } = {},
+ callback?: HandlerCallback
+ ): Promise => {
+ const actionId = Date.now().toString();
+ elizaLogger.info("Starting deployment status request", { actionId });
+
+ try {
+ const config = await validateAkashConfig(runtime);
+ const params = message.content as Partial;
+ let dseqSource = "parameter"; // Track where the DSEQ came from
+
+ // Handle missing dseq parameter based on environment configuration
+ if (!params.dseq) {
+ if (config.AKASH_DEP_STATUS === "dseq") {
+ if (config.AKASH_DEP_DSEQ) {
+ params.dseq = config.AKASH_DEP_DSEQ;
+ dseqSource = "environment";
+ } else {
+ if (callback) {
+ callback({
+ text: "AKASH_DEP_DSEQ is not set in your environment. Please set a valid deployment sequence number.",
+ content: {
+ success: false,
+ error: {
+ code: AkashErrorCode.VALIDATION_PARAMETER_MISSING,
+ message: "Missing AKASH_DEP_DSEQ",
+ help: "When AKASH_DEP_STATUS is set to 'dseq', you must also set AKASH_DEP_DSEQ in your .env file."
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getDeploymentStatus',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ });
+ }
+ return false;
+ }
+ } else if (config.AKASH_DEP_STATUS === "param_passed") {
+ if (callback) {
+ callback({
+ text: "DSEQ parameter is required. Please provide a deployment sequence number.",
+ content: {
+ success: false,
+ error: {
+ code: AkashErrorCode.VALIDATION_PARAMETER_MISSING,
+ message: "Missing required parameter: dseq",
+ help: "You need to provide a deployment sequence number (dseq) to check its status."
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getDeploymentStatus',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ });
+ }
+ return false;
+ } else {
+ if (callback) {
+ callback({
+ text: "No deployment configuration found. Please set AKASH_DEP_STATUS and AKASH_DEP_DSEQ in your environment or provide a dseq parameter.",
+ content: {
+ success: false,
+ error: {
+ code: AkashErrorCode.VALIDATION_PARAMETER_MISSING,
+ message: "Missing configuration",
+ help: "Set AKASH_DEP_STATUS='dseq' and AKASH_DEP_DSEQ in your .env file, or set AKASH_DEP_STATUS='param_passed' and provide dseq parameter in your request."
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getDeploymentStatus',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ });
+ }
+ return false;
+ }
+ }
+
+ // Initialize wallet from mnemonic
+ const wallet = await DirectSecp256k1HdWallet.fromMnemonic(config.AKASH_MNEMONIC, { prefix: "akash" });
+ const [account] = await wallet.getAccounts();
+
+ // Initialize query client
+ const queryClient = new DeploymentQueryClient(await getRpc(config.RPC_ENDPOINT));
+
+ // Query deployment
+ elizaLogger.info("Querying deployment status", {
+ dseq: params.dseq,
+ owner: account.address
+ });
+
+ try {
+ const request = QueryDeploymentRequest.fromPartial({
+ id: {
+ owner: account.address,
+ dseq: params.dseq
+ }
+ });
+
+ const response = await queryClient.Deployment(request);
+
+ if (!response.deployment) {
+ // Different messages based on DSEQ source
+ if (dseqSource === "environment") {
+ if (callback) {
+ callback({
+ text: "The deployment sequence number in your environment configuration was not found. Please check AKASH_DEP_DSEQ value.",
+ content: {
+ success: false,
+ error: {
+ code: AkashErrorCode.DEPLOYMENT_NOT_FOUND,
+ message: "Invalid AKASH_DEP_DSEQ",
+ help: "Update AKASH_DEP_DSEQ in your .env file with a valid deployment sequence number, or switch to AKASH_DEP_STATUS='param_passed' to provide DSEQ as a parameter.",
+ current_dseq: params.dseq
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getDeploymentStatus',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ });
+ }
+ } else {
+ throw new AkashError(
+ "Deployment not found",
+ AkashErrorCode.DEPLOYMENT_NOT_FOUND,
+ {
+ dseq: params.dseq,
+ owner: account.address,
+ actionId
+ }
+ );
+ }
+ return false;
+ }
+
+ // Format deployment status
+ const deployment = response.deployment as unknown as DeploymentResponse;
+ const status = {
+ owner: deployment.deploymentId?.owner,
+ dseq: deployment.deploymentId?.dseq,
+ state: deployment.state,
+ version: deployment.version,
+ createdAt: deployment.createdAt,
+ balance: deployment.escrowAccount?.balance,
+ groups: deployment.groups?.map((group: DeploymentGroup) => ({
+ groupId: group.groupId,
+ state: group.state,
+ resources: group.resources
+ }))
+ };
+
+ elizaLogger.info("Deployment status retrieved successfully", {
+ dseq: params.dseq,
+ state: status.state,
+ owner: status.owner,
+ actionId
+ });
+
+ if (callback) {
+ // Convert numeric state to readable string
+ const stateString = DeploymentState[status.state as keyof typeof DeploymentState] || 'UNKNOWN';
+
+ const formattedBalance = deployment.escrowAccount?.balance
+ ? `${deployment.escrowAccount.balance.amount}${deployment.escrowAccount.balance.denom}`
+ : 'No balance information';
+
+ elizaLogger.info("=== Preparing callback response for deployment status ===", {
+ hasCallback: true,
+ actionId,
+ dseq: params.dseq
+ });
+
+ const callbackResponse = {
+ text: `Deployment ${params.dseq} Status:\nState: ${stateString}\nBalance: ${formattedBalance}\nCreated At: ${status.createdAt}`,
+ content: {
+ success: true,
+ data: {
+ deployment: status,
+ queryResponse: response.deployment
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getDeploymentStatus',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+
+ elizaLogger.info("=== Executing callback with response ===", {
+ actionId,
+ responseText: callbackResponse.text,
+ hasContent: !!callbackResponse.content,
+ contentKeys: Object.keys(callbackResponse.content),
+ metadata: callbackResponse.content.metadata
+ });
+
+ callback(callbackResponse);
+
+ elizaLogger.info("=== Callback executed successfully ===", {
+ actionId,
+ timestamp: new Date().toISOString()
+ });
+ }
+
+ return true;
+ } catch (queryError) {
+ // Handle query errors differently based on DSEQ source
+ if (dseqSource === "environment") {
+ elizaLogger.warn("Failed to query deployment from environment configuration", {
+ dseq: params.dseq,
+ error: queryError instanceof Error ? queryError.message : String(queryError)
+ });
+ if (callback) {
+ callback({
+ text: "Could not find deployment with the configured DSEQ. Please check your environment settings.",
+ content: {
+ success: false,
+ error: {
+ code: AkashErrorCode.API_ERROR,
+ message: "Invalid AKASH_DEP_DSEQ configuration",
+ help: "Verify that AKASH_DEP_DSEQ contains a valid deployment sequence number, or switch to AKASH_DEP_STATUS='param_passed' to provide DSEQ as a parameter.",
+ current_dseq: params.dseq
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getDeploymentStatus',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ });
+ }
+ } else {
+ elizaLogger.error("Failed to query deployment", {
+ error: queryError instanceof Error ? queryError.message : String(queryError),
+ actionId
+ });
+ if (callback) {
+ elizaLogger.info("=== Preparing error callback response ===", {
+ actionId,
+ hasCallback: true,
+ errorType: queryError instanceof AkashError ? 'AkashError' : 'Error'
+ });
+
+ const errorResponse = {
+ text: `Failed to get deployment status: ${queryError instanceof Error ? queryError.message : String(queryError)}`,
+ content: {
+ success: false,
+ error: {
+ code: queryError instanceof AkashError ? queryError.code : AkashErrorCode.API_ERROR,
+ message: queryError instanceof Error ? queryError.message : String(queryError)
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getDeploymentStatus',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+
+ elizaLogger.info("=== Executing error callback ===", {
+ actionId,
+ responseText: errorResponse.text,
+ hasContent: !!errorResponse.content,
+ contentKeys: Object.keys(errorResponse.content)
+ });
+
+ callback(errorResponse);
+
+ elizaLogger.info("=== Error callback executed ===", {
+ actionId,
+ timestamp: new Date().toISOString()
+ });
+ }
+ }
+ return false;
+ }
+ } catch (error) {
+ elizaLogger.error("Get deployment status request failed", {
+ error: error instanceof Error ? error.message : String(error),
+ actionId
+ });
+
+ if (callback) {
+ elizaLogger.info("=== Preparing error callback response ===", {
+ actionId,
+ hasCallback: true,
+ errorType: error instanceof AkashError ? 'AkashError' : 'Error'
+ });
+
+ const errorResponse = {
+ text: `Failed to get deployment status: ${error instanceof Error ? error.message : String(error)}`,
+ content: {
+ success: false,
+ error: {
+ code: error instanceof AkashError ? error.code : AkashErrorCode.API_ERROR,
+ message: error instanceof Error ? error.message : String(error)
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getDeploymentStatus',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+
+ elizaLogger.info("=== Executing error callback ===", {
+ actionId,
+ responseText: errorResponse.text,
+ hasContent: !!errorResponse.content,
+ contentKeys: Object.keys(errorResponse.content)
+ });
+
+ callback(errorResponse);
+
+ elizaLogger.info("=== Error callback executed ===", {
+ actionId,
+ timestamp: new Date().toISOString()
+ });
+ }
+
+ return false;
+ }
+ }
+};
+
+export default getDeploymentStatusAction;
\ No newline at end of file
diff --git a/packages/plugin-akash/src/actions/getGPUPricing.ts b/packages/plugin-akash/src/actions/getGPUPricing.ts
new file mode 100644
index 00000000000..35e407f00f1
--- /dev/null
+++ b/packages/plugin-akash/src/actions/getGPUPricing.ts
@@ -0,0 +1,225 @@
+import { Action, elizaLogger } from "@elizaos/core";
+import { IAgentRuntime, Memory, State, HandlerCallback, Content, ActionExample } from "@elizaos/core";
+import { getConfig } from "../environment";
+
+interface GetGPUPricingContent extends Content {
+ cpu?: number; // CPU units in millicores (e.g., 1000 = 1 CPU)
+ memory?: number; // Memory in bytes (e.g., 1000000000 = 1GB)
+ storage?: number; // Storage in bytes (e.g., 1000000000 = 1GB)
+}
+
+interface PricingResponse {
+ spec: {
+ cpu: number;
+ memory: number;
+ storage: number;
+ };
+ akash: number;
+ aws: number;
+ gcp: number;
+ azure: number;
+}
+
+// Get configuration with defaults
+const config = getConfig(process.env.AKASH_ENV);
+const PRICING_API_URL = config.AKASH_PRICING_API_URL;
+const DEFAULT_CPU = parseInt(config.AKASH_DEFAULT_CPU || "1000");
+const DEFAULT_MEMORY = parseInt(config.AKASH_DEFAULT_MEMORY || "1000000000");
+const DEFAULT_STORAGE = parseInt(config.AKASH_DEFAULT_STORAGE || "1000000000");
+
+// Custom error class for GPU pricing errors
+class GPUPricingError extends Error {
+ constructor(message: string, public code: string) {
+ super(message);
+ this.name = 'GPUPricingError';
+ }
+}
+
+export const getGPUPricingAction: Action = {
+ name: "GET_GPU_PRICING",
+ similes: ["GET_PRICING", "COMPARE_PRICES", "CHECK_PRICING"],
+ description: "Get GPU pricing comparison between Akash and major cloud providers",
+ examples: [[
+ {
+ user: "user",
+ content: {
+ text: "Get GPU pricing for 2 CPUs, 2GB memory, and 10GB storage",
+ cpu: 2000,
+ memory: 2000000000,
+ storage: 10000000000
+ } as GetGPUPricingContent
+ } as ActionExample
+ ], [
+ {
+ user: "user",
+ content: {
+ text: "Compare GPU prices across providers"
+ } as GetGPUPricingContent
+ } as ActionExample
+ ]],
+
+ validate: async (runtime: IAgentRuntime, message: Memory): Promise => {
+ elizaLogger.debug("Validating GPU pricing request", { message });
+ try {
+ const params = message.content as Partial;
+
+ // Validate CPU if provided
+ if (params.cpu !== undefined && (isNaN(params.cpu) || params.cpu <= 0)) {
+ throw new GPUPricingError("CPU units must be a positive number", "INVALID_CPU");
+ }
+
+ // Validate memory if provided
+ if (params.memory !== undefined && (isNaN(params.memory) || params.memory <= 0)) {
+ throw new GPUPricingError("Memory must be a positive number", "INVALID_MEMORY");
+ }
+
+ // Validate storage if provided
+ if (params.storage !== undefined && (isNaN(params.storage) || params.storage <= 0)) {
+ throw new GPUPricingError("Storage must be a positive number", "INVALID_STORAGE");
+ }
+
+ return true;
+ } catch (error) {
+ elizaLogger.error("GPU pricing validation failed", {
+ error: error instanceof GPUPricingError ? {
+ code: error.code,
+ message: error.message
+ } : String(error)
+ });
+ return false;
+ }
+ },
+
+ handler: async (
+ runtime: IAgentRuntime,
+ message: Memory,
+ state: State | undefined,
+ _options: { [key: string]: unknown; } = {},
+ callback?: HandlerCallback
+ ): Promise => {
+ const actionId = Date.now().toString();
+ elizaLogger.info("Starting GPU pricing request", { actionId });
+
+ try {
+ const params = message.content as GetGPUPricingContent;
+
+ // Use provided values or defaults
+ const requestBody = {
+ cpu: params.cpu || DEFAULT_CPU,
+ memory: params.memory || DEFAULT_MEMORY,
+ storage: params.storage || DEFAULT_STORAGE
+ };
+
+ elizaLogger.info("Fetching pricing information", {
+ specs: requestBody,
+ apiUrl: PRICING_API_URL
+ });
+
+ // Make API request using fetch
+ const response = await fetch(PRICING_API_URL, {
+ method: 'POST',
+ headers: {
+ 'Content-Type': 'application/json',
+ 'Accept': 'application/json'
+ },
+ body: JSON.stringify(requestBody)
+ });
+
+ if (!response.ok) {
+ throw new GPUPricingError(
+ `API request failed with status ${response.status}: ${response.statusText}`,
+ "API_ERROR"
+ );
+ }
+
+ const data = await response.json() as PricingResponse;
+
+ // Calculate savings percentages
+ const savings = {
+ vs_aws: ((data.aws - data.akash) / data.aws * 100).toFixed(2),
+ vs_gcp: ((data.gcp - data.akash) / data.gcp * 100).toFixed(2),
+ vs_azure: ((data.azure - data.akash) / data.azure * 100).toFixed(2)
+ };
+
+ elizaLogger.info("Pricing information retrieved successfully", {
+ specs: data.spec,
+ pricing: {
+ akash: data.akash,
+ aws: data.aws,
+ gcp: data.gcp,
+ azure: data.azure
+ },
+ savings
+ });
+
+ if (callback) {
+ const callbackResponse = {
+ text: `GPU Pricing Comparison\nAkash: $${data.akash}\nAWS: $${data.aws} (${savings.vs_aws}% savings)\nGCP: $${data.gcp} (${savings.vs_gcp}% savings)\nAzure: $${data.azure} (${savings.vs_azure}% savings)`,
+ content: {
+ success: true,
+ data: {
+ specs: {
+ cpu: data.spec.cpu,
+ memory: data.spec.memory,
+ storage: data.spec.storage
+ },
+ pricing: {
+ akash: data.akash,
+ aws: data.aws,
+ gcp: data.gcp,
+ azure: data.azure
+ },
+ savings: {
+ vs_aws: `${savings.vs_aws}%`,
+ vs_gcp: `${savings.vs_gcp}%`,
+ vs_azure: `${savings.vs_azure}%`
+ }
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getGPUPricing',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+
+ callback(callbackResponse);
+ }
+
+ return true;
+ } catch (error) {
+ elizaLogger.error("GPU pricing request failed", {
+ error: error instanceof Error ? error.message : String(error),
+ actionId
+ });
+
+ if (callback) {
+ const errorResponse = {
+ text: "Failed to get GPU pricing information",
+ content: {
+ success: false,
+ error: {
+ code: error instanceof GPUPricingError ? error.code : 'UNKNOWN_ERROR',
+ message: error instanceof Error ? error.message : String(error)
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getGPUPricing',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+
+ callback(errorResponse);
+ }
+
+ return false;
+ }
+ }
+};
+
+export default getGPUPricingAction;
diff --git a/packages/plugin-akash/src/actions/getManifest.ts b/packages/plugin-akash/src/actions/getManifest.ts
new file mode 100644
index 00000000000..5d6e116f9e9
--- /dev/null
+++ b/packages/plugin-akash/src/actions/getManifest.ts
@@ -0,0 +1,361 @@
+import { Action, elizaLogger } from "@elizaos/core";
+import { IAgentRuntime, Memory, State, HandlerCallback, Content, ActionExample } from "@elizaos/core";
+import { SDL } from "@akashnetwork/akashjs/build/sdl";
+import { validateAkashConfig } from "../environment";
+import { AkashError, AkashErrorCode } from "../error/error";
+import * as fs from 'fs';
+import * as path from 'path';
+import yaml from 'js-yaml';
+// import { getAkashTypeRegistry } from "@akashnetwork/akashjs/build/stargate";
+import { getDefaultSDLPath } from "../utils/paths";
+
+interface GetManifestContent extends Content {
+ sdl?: string;
+ sdlFile?: string;
+}
+
+// elizaLogger.info("Default SDL path initialized", { DEFAULT_SDL_PATH });
+// elizaLogger.info("Loading SDL from file", { filePath });
+// elizaLogger.info("Resolved SDL file path", { resolvedPath });
+// elizaLogger.error("SDL file not found", { resolvedPath });
+// elizaLogger.info("SDL file loaded successfully", { content });
+// elizaLogger.error("Failed to read SDL file", { error });
+// elizaLogger.error("SDL validation failed", { error });
+// elizaLogger.info("Using provided SDL content");
+// elizaLogger.info("Loading SDL from file", { path: params.sdlFile });
+// elizaLogger.info("Loading default SDL", { path: DEFAULT_SDL_PATH });
+// elizaLogger.debug("Parsing SDL content and generating manifest");
+
+const DEFAULT_SDL_PATH = (() => {
+ const currentFileUrl = import.meta.url;
+ const sdlPath = getDefaultSDLPath(currentFileUrl);
+
+ // Only log if file doesn't exist
+ if (!fs.existsSync(sdlPath)) {
+ elizaLogger.warn("Default SDL path not found", {
+ sdlPath,
+ exists: false
+ });
+ }
+
+ return sdlPath;
+})();
+
+const loadSDLFromFile = (filePath: string): string => {
+ try {
+ // If path doesn't contain plugin-akash and it's not the default path, adjust it
+ if (!filePath.includes('plugin-akash') && filePath !== DEFAULT_SDL_PATH) {
+ const adjustedPath = path.join(path.dirname(DEFAULT_SDL_PATH), path.basename(filePath));
+ filePath = adjustedPath;
+ }
+
+ // Try multiple possible locations
+ const possiblePaths = [
+ filePath,
+ path.join(process.cwd(), filePath),
+ path.join(process.cwd(), 'packages', 'plugin-akash', filePath),
+ path.join(process.cwd(), 'packages', 'plugin-akash', 'src', filePath),
+ path.join(path.dirname(DEFAULT_SDL_PATH), filePath)
+ ];
+
+ for (const tryPath of possiblePaths) {
+ if (fs.existsSync(tryPath)) {
+ const content = fs.readFileSync(tryPath, "utf8");
+ elizaLogger.info("SDL file loaded successfully from", {
+ path: tryPath
+ });
+ return content;
+ }
+ }
+
+ // If we get here, none of the paths worked
+ throw new AkashError(
+ `SDL file not found in any of the possible locations`,
+ AkashErrorCode.VALIDATION_SDL_FAILED,
+ {
+ filePath,
+ triedPaths: possiblePaths
+ }
+ );
+ } catch (error) {
+ elizaLogger.error("Failed to read SDL file", {
+ filePath,
+ error: error instanceof Error ? error.message : String(error)
+ });
+ throw new AkashError(
+ `Failed to read SDL file: ${error instanceof Error ? error.message : String(error)}`,
+ AkashErrorCode.VALIDATION_SDL_FAILED,
+ { filePath }
+ );
+ }
+};
+
+const validateSDL = (sdlContent: string, validationLevel: string = "strict"): boolean => {
+ try {
+ // First try to parse as YAML
+ const parsed = yaml.load(sdlContent);
+ if (!parsed || typeof parsed !== 'object') {
+ throw new Error('Invalid SDL format: not a valid YAML object');
+ }
+
+ if (validationLevel === "none") {
+ // elizaLogger.debug("Skipping SDL validation (validation level: none)");
+ return true;
+ }
+
+ // Required sections based on validation level
+ const requiredSections = ['version', 'services'];
+ const sectionsToCheck = validationLevel === "strict" ?
+ [...requiredSections, 'profiles', 'deployment'] :
+ requiredSections;
+
+ for (const section of sectionsToCheck) {
+ if (!(section in parsed)) {
+ throw new Error(`Invalid SDL format: missing required section '${section}'`);
+ }
+ }
+
+ // elizaLogger.debug("SDL validation successful", {
+ // validationLevel,
+ // checkedSections: sectionsToCheck
+ // });
+ return true;
+ } catch (error) {
+ elizaLogger.error("SDL validation failed", {
+ error: error instanceof Error ? error.message : String(error),
+ validationLevel
+ });
+ return false;
+ }
+};
+
+export const getManifestAction: Action = {
+ name: "GET_MANIFEST",
+ similes: ["LOAD_MANIFEST", "READ_MANIFEST", "PARSE_MANIFEST"],
+ description: "Load and validate SDL to generate a manifest for Akash deployments",
+ examples: [[
+ {
+ user: "user",
+ content: {
+ text: "Get manifest from SDL file",
+ sdlFile: "deployment.yml"
+ } as GetManifestContent
+ } as ActionExample
+ ]],
+
+ validate: async (runtime: IAgentRuntime, message: Memory): Promise => {
+ elizaLogger.debug("Validating manifest request", { message });
+ try {
+ const params = message.content as Partial;
+ const config = await validateAkashConfig(runtime);
+
+ // Either SDL content or file path must be provided
+ if (!params.sdl && !params.sdlFile && !config.AKASH_SDL) {
+ throw new AkashError(
+ "Either SDL content, file path, or AKASH_SDL environment variable must be provided",
+ AkashErrorCode.VALIDATION_PARAMETER_MISSING,
+ { parameters: ["sdl", "sdlFile", "AKASH_SDL"] }
+ );
+ }
+
+ // If SDL content is provided, validate it
+ if (params.sdl) {
+ const validationLevel = config.AKASH_MANIFEST_VALIDATION_LEVEL || "strict";
+ if (!validateSDL(params.sdl, validationLevel)) {
+ throw new AkashError(
+ "Invalid SDL format",
+ AkashErrorCode.VALIDATION_SDL_FAILED
+ );
+ }
+ }
+
+ return true;
+ } catch (error) {
+ elizaLogger.error("Manifest validation failed", {
+ error: error instanceof AkashError ? {
+ code: error.code,
+ message: error.message,
+ details: error.details
+ } : String(error)
+ });
+ return false;
+ }
+ },
+
+ handler: async (
+ runtime: IAgentRuntime,
+ message: Memory,
+ state: State | undefined,
+ _options: { [key: string]: unknown; } = {},
+ callback?: HandlerCallback
+ ): Promise => {
+ const actionId = Date.now().toString();
+ elizaLogger.info("Starting manifest operation", { actionId });
+
+ try {
+ const config = await validateAkashConfig(runtime);
+ const params = message.content as Partial;
+
+ let sdlContent: string;
+ try {
+ // Load SDL content based on priority: params.sdl > params.sdlFile > config.AKASH_SDL
+ if (params.sdl) {
+ sdlContent = params.sdl;
+ elizaLogger.info("Using provided SDL content");
+ } else if (params.sdlFile) {
+ sdlContent = loadSDLFromFile(params.sdlFile);
+ elizaLogger.info("Loaded SDL from file", { path: params.sdlFile });
+ } else {
+ const sdlPath = config.AKASH_SDL || DEFAULT_SDL_PATH;
+ sdlContent = loadSDLFromFile(sdlPath);
+ elizaLogger.info("Using SDL from environment", { path: sdlPath });
+ }
+
+ // Validate based on environment settings
+ const validationLevel = config.AKASH_MANIFEST_VALIDATION_LEVEL || "strict";
+ const isValid = validateSDL(sdlContent, validationLevel);
+
+ if (!isValid) {
+ throw new AkashError(
+ "SDL validation failed",
+ AkashErrorCode.VALIDATION_SDL_FAILED
+ );
+ }
+
+ // Check manifest mode
+ const manifestMode = config.AKASH_MANIFEST_MODE || "auto";
+ if (manifestMode === "validate_only") {
+ elizaLogger.info("Validation successful (validate_only mode)");
+ if (callback) {
+ const callbackResponse = {
+ text: "SDL validation successful",
+ content: {
+ success: true,
+ data: {
+ validationLevel,
+ mode: manifestMode
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getManifest',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+ callback(callbackResponse);
+ }
+ return true;
+ }
+
+ // Generate manifest
+ const sdl = new SDL(yaml.load(sdlContent) as any);
+ const manifest = sdl.manifest();
+
+ // Save manifest if path is specified
+ if (config.AKASH_MANIFEST_PATH) {
+ const manifestPath = path.join(
+ config.AKASH_MANIFEST_PATH,
+ `manifest-${Date.now()}.yaml`
+ );
+ fs.writeFileSync(manifestPath, yaml.dump(manifest), 'utf8');
+ elizaLogger.info("Manifest saved", { path: manifestPath });
+ }
+
+ if (callback) {
+ const callbackResponse = {
+ text: "Manifest generated successfully",
+ content: {
+ success: true,
+ data: {
+ manifest,
+ settings: {
+ mode: manifestMode,
+ validationLevel,
+ outputPath: config.AKASH_MANIFEST_PATH
+ }
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getManifest',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+ callback(callbackResponse);
+ }
+
+ return true;
+ } catch (error) {
+ const formattedError = error instanceof Error ? error.message : String(error);
+ elizaLogger.error("Manifest operation failed", {
+ error: formattedError,
+ settings: {
+ mode: config.AKASH_MANIFEST_MODE || "auto",
+ validationLevel: config.AKASH_MANIFEST_VALIDATION_LEVEL || "strict",
+ outputPath: config.AKASH_MANIFEST_PATH
+ }
+ });
+
+ if (callback) {
+ const errorResponse = {
+ text: "Failed to process manifest",
+ content: {
+ success: false,
+ error: error instanceof AkashError ? {
+ code: error.code,
+ message: error.message,
+ details: error.details
+ } : {
+ code: AkashErrorCode.MANIFEST_PARSING_FAILED,
+ message: formattedError
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getManifest',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+ callback(errorResponse);
+ }
+ return false;
+ }
+ } catch (error) {
+ elizaLogger.error("Manifest operation failed", {
+ error: error instanceof Error ? error.message : String(error),
+ actionId
+ });
+
+ if (callback) {
+ const errorResponse = {
+ text: "Manifest operation failed",
+ content: {
+ success: false,
+ error: {
+ code: AkashErrorCode.MANIFEST_PARSING_FAILED,
+ message: error instanceof Error ? error.message : String(error)
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getManifest',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+ callback(errorResponse);
+ }
+
+ return false;
+ }
+ }
+};
+
+export default getManifestAction;
diff --git a/packages/plugin-akash/src/actions/getProviderInfo.ts b/packages/plugin-akash/src/actions/getProviderInfo.ts
new file mode 100644
index 00000000000..0203a4a62f9
--- /dev/null
+++ b/packages/plugin-akash/src/actions/getProviderInfo.ts
@@ -0,0 +1,369 @@
+import { Action, elizaLogger } from "@elizaos/core";
+import { IAgentRuntime, Memory, State, HandlerCallback, Content, ActionExample } from "@elizaos/core";
+import { QueryProviderRequest, QueryClientImpl as ProviderQueryClient } from "@akashnetwork/akash-api/akash/provider/v1beta3";
+import { getRpc } from "@akashnetwork/akashjs/build/rpc";
+import { AkashError, AkashErrorCode } from "../error/error";
+import { validateAkashConfig } from "../environment";
+
+interface GetProviderInfoContent extends Content {
+ text: string;
+ provider?: string;
+}
+
+interface ProviderResponse {
+ provider?: {
+ owner: string;
+ hostUri: string;
+ attributes: Array<{
+ key: string;
+ value: string;
+ }>;
+ info?: {
+ email: string;
+ website: string;
+ capabilities: string[];
+ };
+ status?: ProviderStatus;
+ };
+}
+
+interface ProviderStatus {
+ cluster?: {
+ nodes: Array<{
+ name: string;
+ capacity: {
+ cpu: string;
+ memory: string;
+ storage: string;
+ };
+ allocatable: {
+ cpu: string;
+ memory: string;
+ storage: string;
+ };
+ }>;
+ };
+ leases?: {
+ active: number;
+ pending: number;
+ available: number;
+ };
+}
+
+const sleep = (ms: number) => new Promise(resolve => setTimeout(resolve, ms));
+
+export const getProviderInfoAction: Action = {
+ name: "GET_PROVIDER_INFO",
+ similes: ["CHECK_PROVIDER", "PROVIDER_INFO", "PROVIDER_STATUS", "CHECK PROVIDER"],
+ description: "Get detailed information about a provider on Akash Network",
+ examples: [[
+ {
+ user: "user",
+ content: {
+ text: "Can you check the provider info for akash1ccktptfkvdc67msasmesuy5m7gpc76z75kukpz?",
+ } as GetProviderInfoContent
+ } as ActionExample
+ ]],
+
+ validate: async (runtime: IAgentRuntime, message: Memory): Promise => {
+ elizaLogger.debug("Validating get provider info request", { message });
+ try {
+ const params = message.content as Partial;
+ const config = await validateAkashConfig(runtime);
+
+ // Extract provider address from text if present
+ if (params.text && !params.provider) {
+ // Pattern to match akash1 followed by address characters
+ const providerMatch = params.text.match(/akash1[a-zA-Z0-9]{38}/);
+ if (providerMatch) {
+ params.provider = providerMatch[0];
+ elizaLogger.debug("Extracted provider address from text", {
+ text: params.text,
+ extractedProvider: params.provider
+ });
+ }
+ }
+
+ // If still no provider specified, use environment default
+ if (!params.provider && config.AKASH_PROVIDER_INFO) {
+ params.provider = config.AKASH_PROVIDER_INFO;
+ }
+
+ if (!params.provider) {
+ throw new AkashError(
+ "Provider address is required",
+ AkashErrorCode.VALIDATION_PARAMETER_MISSING,
+ { parameter: "provider" }
+ );
+ }
+
+ // Validate provider address format
+ if (!params.provider.startsWith("akash1")) {
+ throw new AkashError(
+ "Invalid provider address format. Must start with 'akash1'",
+ AkashErrorCode.VALIDATION_PARAMETER_INVALID,
+ { parameter: "provider", value: params.provider }
+ );
+ }
+
+ return true;
+ } catch (error) {
+ elizaLogger.error("Get provider info validation failed", {
+ error: error instanceof AkashError ? {
+ code: error.code,
+ message: error.message,
+ details: error.details
+ } : String(error)
+ });
+ return false;
+ }
+ },
+
+ handler: async (
+ runtime: IAgentRuntime,
+ message: Memory,
+ state: State | undefined,
+ options: { [key: string]: unknown } = {},
+ callback?: HandlerCallback
+ ): Promise => {
+ const actionId = Date.now().toString();
+ elizaLogger.info("Starting provider info request", { actionId });
+
+ elizaLogger.debug("=== Handler Parameters ===", {
+ hasRuntime: !!runtime,
+ hasMessage: !!message,
+ hasState: !!state,
+ hasOptions: !!options,
+ hasCallback: !!callback,
+ actionId
+ });
+
+ try {
+ const config = await validateAkashConfig(runtime);
+ const params = message.content as Partial;
+
+ // If no provider specified, use environment default
+ if (!params.provider && config.AKASH_PROVIDER_INFO) {
+ params.provider = config.AKASH_PROVIDER_INFO;
+ }
+
+ if (!params.provider) {
+ throw new AkashError(
+ "Provider address is required",
+ AkashErrorCode.VALIDATION_PARAMETER_MISSING,
+ { parameter: "provider" }
+ );
+ }
+
+ // Query provider information
+ elizaLogger.info("Querying provider information", {
+ provider: params.provider,
+ actionId
+ });
+
+ const queryClient = new ProviderQueryClient(await getRpc(config.RPC_ENDPOINT));
+ const request = QueryProviderRequest.fromPartial({
+ owner: params.provider
+ });
+
+ try {
+ const response = await queryClient.Provider(request) as ProviderResponse;
+
+ if (!response.provider) {
+ throw new AkashError(
+ "Failed to query provider: Provider not found",
+ AkashErrorCode.PROVIDER_NOT_FOUND,
+ {
+ provider: params.provider,
+ actionId
+ }
+ );
+ }
+
+ // Add a delay before querying status
+ await sleep(2000); // 2 second delay
+
+ // Query provider status from their API
+ elizaLogger.info("Querying provider status", {
+ hostUri: response.provider.hostUri,
+ actionId
+ });
+
+ const hostUri = response.provider.hostUri.replace(/^https?:\/\//, '');
+ elizaLogger.debug("Making provider status request", { url: `https://${hostUri}/status` });
+
+ try {
+ const statusResponse = await fetch(`https://${hostUri}/status`, {
+ headers: {
+ 'Accept': 'application/json'
+ },
+ signal: AbortSignal.timeout(5000)
+ });
+
+ if (!statusResponse.ok) {
+ elizaLogger.debug("Provider status not available", {
+ status: statusResponse.status,
+ provider: params.provider,
+ hostUri: response.provider.hostUri,
+ actionId
+ });
+ } else {
+ const statusData = await statusResponse.json();
+ response.provider.status = statusData;
+ }
+ } catch (statusError) {
+ elizaLogger.debug("Provider status fetch failed", {
+ error: statusError instanceof Error ? statusError.message : String(statusError),
+ provider: params.provider,
+ hostUri: response.provider.hostUri,
+ actionId
+ });
+ }
+
+ // Format provider information
+ const info = {
+ owner: response.provider.owner,
+ hostUri: response.provider.hostUri,
+ attributes: response.provider.attributes,
+ info: response.provider.info,
+ status: response.provider.status ? {
+ nodes: response.provider.status.cluster?.nodes.map(node => ({
+ name: node.name,
+ capacity: node.capacity,
+ allocatable: node.allocatable
+ })),
+ leases: response.provider.status.leases
+ } : undefined
+ };
+
+ elizaLogger.info("Provider information retrieved successfully", {
+ provider: params.provider,
+ hostUri: response.provider.hostUri,
+ hasStatus: !!response.provider.status,
+ actionId
+ });
+
+ if (callback) {
+ elizaLogger.info("=== Preparing callback response for provider info ===", {
+ hasCallback: true,
+ actionId,
+ provider: params.provider
+ });
+
+ const callbackResponse = {
+ text: `Provider ${params.provider} information:\nHost URI: ${info.hostUri}\nOwner: ${info.owner}${info.info ? `\nEmail: ${info.info.email}\nWebsite: ${info.info.website}` : ''}\nAttributes: ${info.attributes.map(attr => `${attr.key}: ${attr.value}`).join(', ')}`,
+ content: {
+ success: true,
+ data: {
+ provider: info,
+ queryResponse: response.provider
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getProviderInfo',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+
+ elizaLogger.info("=== Executing callback with response ===", {
+ actionId,
+ responseText: callbackResponse.text,
+ hasContent: !!callbackResponse.content,
+ contentKeys: Object.keys(callbackResponse.content),
+ metadata: callbackResponse.content.metadata
+ });
+
+ callback(callbackResponse);
+
+ elizaLogger.info("=== Callback executed successfully ===", {
+ actionId,
+ timestamp: new Date().toISOString()
+ });
+ }
+
+ return true;
+ } catch (queryError) {
+ // Handle specific error cases
+ const errorMessage = queryError instanceof Error ? queryError.message : String(queryError);
+
+ if (errorMessage.toLowerCase().includes("invalid address")) {
+ throw new AkashError(
+ "Failed to query provider: Invalid address format",
+ AkashErrorCode.VALIDATION_PARAMETER_INVALID,
+ {
+ provider: params.provider,
+ error: errorMessage,
+ actionId
+ }
+ );
+ }
+
+ // For all other query errors, treat as provider not found
+ throw new AkashError(
+ "Failed to query provider: Provider not found or not accessible",
+ AkashErrorCode.PROVIDER_NOT_FOUND,
+ {
+ provider: params.provider,
+ error: errorMessage,
+ actionId
+ }
+ );
+ }
+ } catch (error) {
+ elizaLogger.error("Get provider info request failed", {
+ error: error instanceof Error ? error.message : String(error),
+ actionId
+ });
+
+ if (callback) {
+ elizaLogger.info("=== Preparing error callback response ===", {
+ actionId,
+ hasCallback: true,
+ errorType: error instanceof AkashError ? 'AkashError' : 'Error'
+ });
+
+ const errorResponse = {
+ code: error instanceof AkashError ? error.code : AkashErrorCode.API_ERROR,
+ message: error instanceof Error ? error.message : String(error),
+ details: error instanceof AkashError ? error.details : undefined
+ };
+
+ const response = {
+ text: `Failed to get provider information: ${errorResponse.message}`,
+ content: {
+ success: false,
+ error: errorResponse,
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getProviderInfo',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+
+ elizaLogger.info("=== Executing error callback ===", {
+ actionId,
+ errorResponse,
+ hasContent: !!response.content,
+ contentKeys: Object.keys(response.content)
+ });
+
+ callback(response);
+
+ elizaLogger.info("=== Error callback executed ===", {
+ actionId,
+ timestamp: new Date().toISOString()
+ });
+ }
+
+ return false;
+ }
+ }
+};
+
+export default getProviderInfoAction;
\ No newline at end of file
diff --git a/packages/plugin-akash/src/actions/getProvidersList.ts b/packages/plugin-akash/src/actions/getProvidersList.ts
new file mode 100644
index 00000000000..52e3c0fe911
--- /dev/null
+++ b/packages/plugin-akash/src/actions/getProvidersList.ts
@@ -0,0 +1,333 @@
+import { Action, elizaLogger } from "@elizaos/core";
+import { IAgentRuntime, Memory, State, HandlerCallback, Content, ActionExample } from "@elizaos/core";
+import { AkashError, AkashErrorCode } from "../error/error";
+import { validateAkashConfig } from "../environment";
+
+interface GetProvidersListContent extends Content {
+ filter?: {
+ active?: boolean;
+ hasGPU?: boolean;
+ region?: string;
+ };
+}
+
+interface ProviderAttributes {
+ key: string;
+ value: string;
+}
+
+interface ProviderInfo {
+ owner: string;
+ hostUri: string;
+ attributes: ProviderAttributes[];
+ active: boolean;
+ uptime: number;
+ leaseCount: number;
+ info?: {
+ email?: string;
+ website?: string;
+ capabilities?: string[];
+ };
+ status?: {
+ available: boolean;
+ error?: string;
+ lastCheckTime: string;
+ resources?: {
+ cpu: {
+ total: number;
+ available: number;
+ };
+ memory: {
+ total: number;
+ available: number;
+ };
+ storage: {
+ total: number;
+ available: number;
+ };
+ };
+ };
+}
+
+const API_BASE_URL = "https://console-api.akash.network/v1";
+
+async function fetchProviders(): Promise {
+ try {
+ const response = await fetch(`${API_BASE_URL}/providers`, {
+ headers: {
+ 'Accept': 'application/json'
+ }
+ });
+
+ if (!response.ok) {
+ throw new AkashError(
+ "Failed to fetch providers list: Invalid response from API",
+ AkashErrorCode.API_RESPONSE_INVALID,
+ {
+ status: response.status,
+ statusText: response.statusText
+ }
+ );
+ }
+
+ const data = await response.json();
+ return data;
+ } catch (error) {
+ if (error instanceof AkashError) {
+ throw error;
+ }
+ throw new AkashError(
+ `Failed to fetch providers list: ${error instanceof Error ? error.message : String(error)}`,
+ AkashErrorCode.API_REQUEST_FAILED,
+ {
+ error: error instanceof Error ? error.message : String(error)
+ }
+ );
+ }
+}
+
+function filterProviders(providers: ProviderInfo[], filter?: GetProvidersListContent['filter']): ProviderInfo[] {
+ if (!filter) return providers;
+
+ try {
+ let filtered = [...providers];
+
+ if (filter.active !== undefined) {
+ filtered = filtered.filter(p => {
+ const isActive = p.active && p.status?.available !== false;
+ return isActive === filter.active;
+ });
+ }
+
+ if (filter.hasGPU) {
+ filtered = filtered.filter(p =>
+ p.attributes.some(attr =>
+ attr.key.toLowerCase().includes('gpu') &&
+ attr.value.toLowerCase() !== 'false' &&
+ attr.value !== '0'
+ )
+ );
+ }
+
+ if (filter.region) {
+ const regionFilter = filter.region.toLowerCase();
+ filtered = filtered.filter(p =>
+ p.attributes.some(attr =>
+ attr.key.toLowerCase() === 'region' &&
+ attr.value.toLowerCase().includes(regionFilter)
+ )
+ );
+ }
+
+ return filtered;
+ } catch (error) {
+ throw new AkashError(
+ "Failed to apply provider filters",
+ AkashErrorCode.PROVIDER_FILTER_ERROR,
+ { filter, error: error instanceof Error ? error.message : String(error) }
+ );
+ }
+}
+
+export const getProvidersListAction: Action = {
+ name: "GET_PROVIDERS_LIST",
+ similes: ["LIST_PROVIDERS", "FETCH_PROVIDERS", "GET_ALL_PROVIDERS"],
+ description: "Get a list of all available providers on the Akash Network with their details and status",
+ examples: [[
+ {
+ user: "user",
+ content: {
+ text: "Get a list of all active providers"
+ } as GetProvidersListContent
+ } as ActionExample,
+ {
+ user: "assistant",
+ content: {
+ text: "Fetching list of active Akash providers...",
+ filter: {
+ active: true
+ }
+ } as GetProvidersListContent
+ } as ActionExample
+ ], [
+ {
+ user: "user",
+ content: {
+ text: "Show me all GPU providers in the US region",
+ filter: {
+ hasGPU: true,
+ region: "us"
+ }
+ } as GetProvidersListContent
+ } as ActionExample
+ ]],
+
+ validate: async (runtime: IAgentRuntime, message: Memory): Promise => {
+ elizaLogger.debug("Validating get providers list request", { message });
+ try {
+ const params = message.content as Partial;
+
+ // Validate filter parameters if provided
+ if (params.filter) {
+ if (params.filter.region && typeof params.filter.region !== 'string') {
+ throw new AkashError(
+ "Region filter must be a string",
+ AkashErrorCode.VALIDATION_PARAMETER_INVALID,
+ { parameter: "filter.region" }
+ );
+ }
+
+ if (params.filter.active !== undefined && typeof params.filter.active !== 'boolean') {
+ throw new AkashError(
+ "Active filter must be a boolean",
+ AkashErrorCode.VALIDATION_PARAMETER_INVALID,
+ { parameter: "filter.active" }
+ );
+ }
+
+ if (params.filter.hasGPU !== undefined && typeof params.filter.hasGPU !== 'boolean') {
+ throw new AkashError(
+ "HasGPU filter must be a boolean",
+ AkashErrorCode.VALIDATION_PARAMETER_INVALID,
+ { parameter: "filter.hasGPU" }
+ );
+ }
+ }
+
+ return true;
+ } catch (error) {
+ elizaLogger.error("Get providers list validation failed", {
+ error: error instanceof AkashError ? {
+ code: error.code,
+ message: error.message,
+ details: error.details
+ } : String(error)
+ });
+ return false;
+ }
+ },
+
+ handler: async (
+ runtime: IAgentRuntime,
+ message: Memory,
+ state: State | undefined,
+ _options: { [key: string]: unknown; } = {},
+ callback?: HandlerCallback
+ ): Promise => {
+ const actionId = Date.now().toString();
+ elizaLogger.info("Starting providers list request", { actionId });
+
+ try {
+ await validateAkashConfig(runtime);
+ const params = message.content as GetProvidersListContent;
+
+ elizaLogger.info("Fetching providers list", {
+ filter: params.filter,
+ actionId
+ });
+
+ // Fetch providers
+ const allProviders = await fetchProviders();
+
+ // Apply filters
+ const filteredProviders = filterProviders(allProviders, params.filter);
+
+ elizaLogger.info("Providers list retrieved successfully", {
+ totalProviders: allProviders.length,
+ filteredProviders: filteredProviders.length,
+ filter: params.filter,
+ actionId
+ });
+
+ if (callback) {
+ const callbackResponse = {
+ text: `Retrieved ${filteredProviders.length} providers${params.filter ? ' (filtered)' : ''} from total ${allProviders.length}`,
+ content: {
+ success: true,
+ data: {
+ summary: {
+ total: allProviders.length,
+ filtered: filteredProviders.length,
+ activeCount: filteredProviders.filter(p => p.active && p.status?.available !== false).length,
+ gpuCount: filteredProviders.filter(p =>
+ p.attributes.some(attr =>
+ attr.key.toLowerCase().includes('gpu') &&
+ attr.value.toLowerCase() !== 'false' &&
+ attr.value !== '0'
+ )
+ ).length
+ },
+ providers: filteredProviders.map(p => ({
+ owner: p.owner,
+ hostUri: p.hostUri,
+ active: p.active && p.status?.available !== false,
+ uptime: p.uptime,
+ leaseCount: p.leaseCount,
+ attributes: p.attributes,
+ info: {
+ ...p.info,
+ capabilities: p.info?.capabilities || [],
+ region: p.attributes.find(a => a.key.toLowerCase() === 'region')?.value || 'unknown'
+ },
+ resources: p.status?.resources || {
+ cpu: { total: 0, available: 0 },
+ memory: { total: 0, available: 0 },
+ storage: { total: 0, available: 0 }
+ },
+ status: {
+ available: p.status?.available || false,
+ lastCheckTime: p.status?.lastCheckTime || new Date().toISOString(),
+ error: p.status?.error
+ }
+ }))
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getProvidersList',
+ version: '1.0.0',
+ actionId,
+ filters: params.filter || {}
+ }
+ }
+ };
+
+ callback(callbackResponse);
+ }
+
+ return true;
+ } catch (error) {
+ elizaLogger.error("Get providers list request failed", {
+ error: error instanceof Error ? error.message : String(error),
+ code: error instanceof AkashError ? error.code : undefined,
+ actionId
+ });
+
+ if (callback) {
+ const errorResponse = {
+ text: "Failed to get providers list",
+ content: {
+ success: false,
+ error: {
+ code: error instanceof AkashError ? error.code : AkashErrorCode.API_REQUEST_FAILED,
+ message: error instanceof Error ? error.message : String(error)
+ },
+ metadata: {
+ timestamp: new Date().toISOString(),
+ source: 'akash-plugin',
+ action: 'getProvidersList',
+ version: '1.0.0',
+ actionId
+ }
+ }
+ };
+
+ callback(errorResponse);
+ }
+
+ return false;
+ }
+ }
+};
+export default getProvidersListAction;
+
diff --git a/packages/plugin-akash/src/environment.ts b/packages/plugin-akash/src/environment.ts
new file mode 100644
index 00000000000..12a8332b087
--- /dev/null
+++ b/packages/plugin-akash/src/environment.ts
@@ -0,0 +1,259 @@
+import { IAgentRuntime, elizaLogger } from "@elizaos/core";
+import { z } from "zod";
+
+// Add ENV variable at the top
+let ENV: string = "mainnet";
+
+// Log environment information
+elizaLogger.info("Environment sources", {
+ shellVars: Object.keys(process.env).filter(key => key.startsWith('AKASH_')),
+});
+
+export const akashEnvSchema = z.object({
+ AKASH_MNEMONIC: z.string()
+ .min(1, "Wallet mnemonic is required")
+ .refine(
+ (mnemonic) => {
+ const words = mnemonic.trim().split(/\s+/);
+ return words.length === 12 || words.length === 24;
+ },
+ {
+ message: "Mnemonic must be 12 or 24 words",
+ path: ["AKASH_MNEMONIC"]
+ }
+ ),
+ AKASH_WALLET_ADDRESS: z.string()
+ .min(1, "Wallet address is required")
+ .regex(/^akash[a-zA-Z0-9]{39}$/, "Invalid Akash wallet address format")
+ .optional(),
+ AKASH_NET: z.string().min(1, "Network configuration URL is required"),
+ AKASH_VERSION: z.string().min(1, "Akash version is required"),
+ AKASH_CHAIN_ID: z.string().min(1, "Chain ID is required"),
+ AKASH_NODE: z.string().min(1, "Node URL is required"),
+ RPC_ENDPOINT: z.string().min(1, "RPC endpoint is required"),
+ AKASH_GAS_PRICES: z.string().min(1, "Gas prices are required"),
+ AKASH_GAS_ADJUSTMENT: z.string().min(1, "Gas adjustment is required"),
+ AKASH_KEYRING_BACKEND: z.string().min(1, "Keyring backend is required"),
+ AKASH_FROM: z.string().min(1, "Key name is required"),
+ AKASH_FEES: z.string().min(1, "Transaction fees are required"),
+ AKASH_DEPOSIT: z.string().min(1, "Deposit is required be careful with the value not too low generally around 500000uakt"),
+ AKASH_PRICING_API_URL: z.string().optional(),
+ AKASH_DEFAULT_CPU: z.string().optional(),
+ AKASH_DEFAULT_MEMORY: z.string().optional(),
+ AKASH_DEFAULT_STORAGE: z.string().optional(),
+ AKASH_SDL: z.string().optional(),
+ AKASH_CLOSE_DEP: z.string().optional(),
+ AKASH_CLOSE_DSEQ: z.string().optional(),
+ AKASH_PROVIDER_INFO: z.string().optional(),
+ AKASH_DEP_STATUS: z.string().optional(),
+ AKASH_DEP_DSEQ: z.string().optional(),
+ AKASH_GAS_OPERATION: z.string().optional(),
+ AKASH_GAS_DSEQ: z.string().optional(),
+ // Manifest Configuration
+ AKASH_MANIFEST_MODE: z.string()
+ .optional()
+ .refine(
+ (mode) => !mode || ["auto", "manual", "validate_only"].includes(mode),
+ {
+ message: "AKASH_MANIFEST_MODE must be one of: auto, manual, validate_only"
+ }
+ ),
+ AKASH_MANIFEST_PATH: z.string()
+ .optional(),
+ AKASH_MANIFEST_VALIDATION_LEVEL: z.string()
+ .optional()
+ .refine(
+ (level) => !level || ["strict", "lenient", "none"].includes(level),
+ {
+ message: "AKASH_MANIFEST_VALIDATION_LEVEL must be one of: strict, lenient, none"
+ }
+ ),
+});
+
+export type AkashConfig = z.infer;
+
+export function getConfig(
+ env: string | undefined | null = ENV ||
+ process.env.AKASH_ENV
+) {
+ ENV = env || "mainnet";
+ switch (env) {
+ case "mainnet":
+ return {
+ AKASH_NET: "https://raw.githubusercontent.com/ovrclk/net/master/mainnet",
+ RPC_ENDPOINT: "https://rpc.akashnet.net:443",
+ AKASH_GAS_PRICES: "0.025uakt",
+ AKASH_GAS_ADJUSTMENT: "1.5",
+ AKASH_KEYRING_BACKEND: "os",
+ AKASH_FROM: "default",
+ AKASH_FEES: "20000uakt",
+ AKASH_WALLET_ADDRESS: process.env.AKASH_WALLET_ADDRESS || "",
+ AKASH_PRICING_API_URL: process.env.AKASH_PRICING_API_URL || "https://console-api.akash.network/v1/pricing",
+ AKASH_DEFAULT_CPU: process.env.AKASH_DEFAULT_CPU || "1000",
+ AKASH_DEFAULT_MEMORY: process.env.AKASH_DEFAULT_MEMORY || "1000000000",
+ AKASH_DEFAULT_STORAGE: process.env.AKASH_DEFAULT_STORAGE || "1000000000",
+ AKASH_SDL: process.env.AKASH_SDL || "example.sdl.yml",
+ AKASH_CLOSE_DEP: process.env.AKASH_CLOSE_DEP || "closeAll",
+ AKASH_CLOSE_DSEQ: process.env.AKASH_CLOSE_DSEQ || "",
+ AKASH_PROVIDER_INFO: process.env.AKASH_PROVIDER_INFO || "",
+ AKASH_DEP_STATUS: process.env.AKASH_DEP_STATUS || "param_passed",
+ AKASH_DEP_DSEQ: process.env.AKASH_DEP_DSEQ || "",
+ AKASH_GAS_OPERATION: process.env.AKASH_GAS_OPERATION || "close",
+ AKASH_GAS_DSEQ: process.env.AKASH_GAS_DSEQ || "",
+ AKASH_MANIFEST_MODE: process.env.AKASH_MANIFEST_MODE || "auto",
+ AKASH_MANIFEST_PATH: process.env.AKASH_MANIFEST_PATH || "",
+ AKASH_MANIFEST_VALIDATION_LEVEL: process.env.AKASH_MANIFEST_VALIDATION_LEVEL || "strict",
+ AKASH_DEPOSIT: process.env.AKASH_DEPOSIT || "500000uakt"
+ };
+ case "testnet":
+ return {
+ AKASH_NET: "https://raw.githubusercontent.com/ovrclk/net/master/testnet",
+ RPC_ENDPOINT: "https://rpc.sandbox-01.aksh.pw",
+ AKASH_GAS_PRICES: "0.025uakt",
+ AKASH_GAS_ADJUSTMENT: "1.5",
+ AKASH_KEYRING_BACKEND: "test",
+ AKASH_FROM: "default",
+ AKASH_FEES: "20000uakt",
+ AKASH_WALLET_ADDRESS: process.env.AKASH_WALLET_ADDRESS || "",
+ AKASH_PRICING_API_URL: process.env.AKASH_PRICING_API_URL || "https://console-api.akash.network/v1/pricing",
+ AKASH_DEFAULT_CPU: process.env.AKASH_DEFAULT_CPU || "1000",
+ AKASH_DEFAULT_MEMORY: process.env.AKASH_DEFAULT_MEMORY || "1000000000",
+ AKASH_DEFAULT_STORAGE: process.env.AKASH_DEFAULT_STORAGE || "1000000000",
+ AKASH_SDL: process.env.AKASH_SDL || "example.sdl.yml",
+ AKASH_CLOSE_DEP: process.env.AKASH_CLOSE_DEP || "closeAll",
+ AKASH_CLOSE_DSEQ: process.env.AKASH_CLOSE_DSEQ || "",
+ AKASH_PROVIDER_INFO: process.env.AKASH_PROVIDER_INFO || "",
+ AKASH_DEP_STATUS: process.env.AKASH_DEP_STATUS || "param_passed",
+ AKASH_DEP_DSEQ: process.env.AKASH_DEP_DSEQ || "",
+ AKASH_GAS_OPERATION: process.env.AKASH_GAS_OPERATION || "close",
+ AKASH_GAS_DSEQ: process.env.AKASH_GAS_DSEQ || "",
+ AKASH_MANIFEST_MODE: process.env.AKASH_MANIFEST_MODE || "auto",
+ AKASH_MANIFEST_PATH: process.env.AKASH_MANIFEST_PATH || "",
+ AKASH_MANIFEST_VALIDATION_LEVEL: process.env.AKASH_MANIFEST_VALIDATION_LEVEL || "strict",
+ AKASH_DEPOSIT: process.env.AKASH_DEPOSIT || "500000uakt"
+ };
+ default:
+ return {
+ AKASH_NET: "https://raw.githubusercontent.com/ovrclk/net/master/mainnet",
+ RPC_ENDPOINT: "https://rpc.akashnet.net:443",
+ AKASH_GAS_PRICES: "0.025uakt",
+ AKASH_GAS_ADJUSTMENT: "1.5",
+ AKASH_KEYRING_BACKEND: "os",
+ AKASH_FROM: "default",
+ AKASH_FEES: "20000uakt",
+ AKASH_WALLET_ADDRESS: process.env.AKASH_WALLET_ADDRESS || "",
+ AKASH_PRICING_API_URL: process.env.AKASH_PRICING_API_URL || "https://console-api.akash.network/v1/pricing",
+ AKASH_DEFAULT_CPU: process.env.AKASH_DEFAULT_CPU || "1000",
+ AKASH_DEFAULT_MEMORY: process.env.AKASH_DEFAULT_MEMORY || "1000000000",
+ AKASH_DEFAULT_STORAGE: process.env.AKASH_DEFAULT_STORAGE || "1000000000",
+ AKASH_SDL: process.env.AKASH_SDL || "example.sdl.yml",
+ AKASH_CLOSE_DEP: process.env.AKASH_CLOSE_DEP || "closeAll",
+ AKASH_CLOSE_DSEQ: process.env.AKASH_CLOSE_DSEQ || "",
+ AKASH_PROVIDER_INFO: process.env.AKASH_PROVIDER_INFO || "",
+ AKASH_DEP_STATUS: process.env.AKASH_DEP_STATUS || "param_passed",
+ AKASH_DEP_DSEQ: process.env.AKASH_DEP_DSEQ || "",
+ AKASH_GAS_OPERATION: process.env.AKASH_GAS_OPERATION || "close",
+ AKASH_GAS_DSEQ: process.env.AKASH_GAS_DSEQ || "",
+ AKASH_MANIFEST_MODE: process.env.AKASH_MANIFEST_MODE || "auto",
+ AKASH_MANIFEST_PATH: process.env.AKASH_MANIFEST_PATH || "",
+ AKASH_MANIFEST_VALIDATION_LEVEL: process.env.AKASH_MANIFEST_VALIDATION_LEVEL || "strict",
+ AKASH_DEPOSIT: process.env.AKASH_DEPOSIT || "500000uakt"
+ };
+ }
+}
+
+export async function validateAkashConfig(
+ runtime: IAgentRuntime
+): Promise {
+ try {
+ // Log environment information
+ // elizaLogger.info("Environment configuration details", {
+ // shellMnemonic: process.env.AKASH_MNEMONIC,
+ // runtimeMnemonic: runtime.getSetting("AKASH_MNEMONIC"),
+ // envVars: {
+ // fromShell: Object.keys(process.env).filter(key => key.startsWith('AKASH_')),
+ // fromRuntime: Object.keys(runtime)
+ // .filter(key => typeof runtime.getSetting === 'function' && runtime.getSetting(key))
+ // .filter(key => key.startsWith('AKASH_'))
+ // }
+ // });
+
+ const envConfig = getConfig(
+ runtime.getSetting("AKASH_ENV") ?? undefined
+ );
+
+ // Fetch dynamic values from the network configuration
+ const akashNet = process.env.AKASH_NET || runtime.getSetting("AKASH_NET") || envConfig.AKASH_NET;
+ const version = await fetch(`${akashNet}/version.txt`).then(res => res.text());
+ const chainId = await fetch(`${akashNet}/chain-id.txt`).then(res => res.text());
+ const node = await fetch(`${akashNet}/rpc-nodes.txt`).then(res => res.text().then(text => text.split('\n')[0]));
+
+ // Prioritize shell environment variables over runtime settings
+ const mnemonic = process.env.AKASH_MNEMONIC || runtime.getSetting("AKASH_MNEMONIC");
+
+ // elizaLogger.debug("SDL configuration", {
+ // fromShell: process.env.AKASH_SDL,
+ // fromRuntime: runtime.getSetting("AKASH_SDL"),
+ // fromConfig: envConfig.AKASH_SDL
+ // });
+
+ if (!mnemonic) {
+ throw new Error(
+ "AKASH_MNEMONIC not found in environment variables or runtime settings.\n" +
+ "Please ensure AKASH_MNEMONIC is set in your shell environment or runtime settings"
+ );
+ }
+
+ // Clean the mnemonic string - handle quotes and whitespace
+ const cleanMnemonic = mnemonic
+ .trim()
+ .replace(/^["']|["']$/g, '') // Remove surrounding quotes
+ .replace(/\n/g, ' ')
+ .replace(/\r/g, ' ')
+ .replace(/\s+/g, ' ');
+
+ const mnemonicWords = cleanMnemonic.split(' ').filter(word => word.length > 0);
+
+ if (mnemonicWords.length !== 12 && mnemonicWords.length !== 24) {
+ throw new Error(
+ `Invalid AKASH_MNEMONIC length: got ${mnemonicWords.length} words, expected 12 or 24 words.\n` +
+ `Words found: ${mnemonicWords.join(', ')}`
+ );
+ }
+
+ const config = {
+ AKASH_MNEMONIC: cleanMnemonic,
+ AKASH_NET: akashNet,
+ AKASH_VERSION: version,
+ AKASH_CHAIN_ID: chainId,
+ AKASH_NODE: node,
+ RPC_ENDPOINT: process.env.RPC_ENDPOINT || runtime.getSetting("RPC_ENDPOINT") || envConfig.RPC_ENDPOINT,
+ AKASH_GAS_PRICES: process.env.AKASH_GAS_PRICES || runtime.getSetting("AKASH_GAS_PRICES") || envConfig.AKASH_GAS_PRICES,
+ AKASH_GAS_ADJUSTMENT: process.env.AKASH_GAS_ADJUSTMENT || runtime.getSetting("AKASH_GAS_ADJUSTMENT") || envConfig.AKASH_GAS_ADJUSTMENT,
+ AKASH_KEYRING_BACKEND: process.env.AKASH_KEYRING_BACKEND || runtime.getSetting("AKASH_KEYRING_BACKEND") || envConfig.AKASH_KEYRING_BACKEND,
+ AKASH_FROM: process.env.AKASH_FROM || runtime.getSetting("AKASH_FROM") || envConfig.AKASH_FROM,
+ AKASH_FEES: process.env.AKASH_FEES || runtime.getSetting("AKASH_FEES") || envConfig.AKASH_FEES,
+ AKASH_PRICING_API_URL: process.env.AKASH_PRICING_API_URL || runtime.getSetting("AKASH_PRICING_API_URL") || envConfig.AKASH_PRICING_API_URL,
+ AKASH_DEFAULT_CPU: process.env.AKASH_DEFAULT_CPU || runtime.getSetting("AKASH_DEFAULT_CPU") || envConfig.AKASH_DEFAULT_CPU,
+ AKASH_DEFAULT_MEMORY: process.env.AKASH_DEFAULT_MEMORY || runtime.getSetting("AKASH_DEFAULT_MEMORY") || envConfig.AKASH_DEFAULT_MEMORY,
+ AKASH_DEFAULT_STORAGE: process.env.AKASH_DEFAULT_STORAGE || runtime.getSetting("AKASH_DEFAULT_STORAGE") || envConfig.AKASH_DEFAULT_STORAGE,
+ AKASH_SDL: process.env.AKASH_SDL || runtime.getSetting("AKASH_SDL") || envConfig.AKASH_SDL,
+ AKASH_CLOSE_DEP: process.env.AKASH_CLOSE_DEP || runtime.getSetting("AKASH_CLOSE_DEP") || envConfig.AKASH_CLOSE_DEP,
+ AKASH_CLOSE_DSEQ: process.env.AKASH_CLOSE_DSEQ || runtime.getSetting("AKASH_CLOSE_DSEQ") || envConfig.AKASH_CLOSE_DSEQ,
+ AKASH_PROVIDER_INFO: process.env.AKASH_PROVIDER_INFO || runtime.getSetting("AKASH_PROVIDER_INFO") || envConfig.AKASH_PROVIDER_INFO,
+ AKASH_DEP_STATUS: process.env.AKASH_DEP_STATUS || runtime.getSetting("AKASH_DEP_STATUS") || envConfig.AKASH_DEP_STATUS,
+ AKASH_DEP_DSEQ: process.env.AKASH_DEP_DSEQ || runtime.getSetting("AKASH_DEP_DSEQ") || envConfig.AKASH_DEP_DSEQ,
+ AKASH_GAS_OPERATION: process.env.AKASH_GAS_OPERATION || runtime.getSetting("AKASH_GAS_OPERATION") || envConfig.AKASH_GAS_OPERATION,
+ AKASH_GAS_DSEQ: process.env.AKASH_GAS_DSEQ || runtime.getSetting("AKASH_GAS_DSEQ") || envConfig.AKASH_GAS_DSEQ,
+ AKASH_MANIFEST_MODE: process.env.AKASH_MANIFEST_MODE || runtime.getSetting("AKASH_MANIFEST_MODE") || envConfig.AKASH_MANIFEST_MODE,
+ AKASH_MANIFEST_PATH: process.env.AKASH_MANIFEST_PATH || runtime.getSetting("AKASH_MANIFEST_PATH") || envConfig.AKASH_MANIFEST_PATH,
+ AKASH_MANIFEST_VALIDATION_LEVEL: process.env.AKASH_MANIFEST_VALIDATION_LEVEL || runtime.getSetting("AKASH_MANIFEST_VALIDATION_LEVEL") || envConfig.AKASH_MANIFEST_VALIDATION_LEVEL,
+ AKASH_DEPOSIT: process.env.AKASH_DEPOSIT || runtime.getSetting("AKASH_DEPOSIT") || envConfig.AKASH_DEPOSIT
+ };
+
+ return akashEnvSchema.parse(config);
+ } catch (error) {
+ const errorMessage = error instanceof Error ? error.message : String(error);
+ throw new Error(`Failed to validate Akash configuration: ${errorMessage}`);
+ }
+}
diff --git a/packages/plugin-akash/src/error/error.ts b/packages/plugin-akash/src/error/error.ts
new file mode 100644
index 00000000000..fb6d56416b3
--- /dev/null
+++ b/packages/plugin-akash/src/error/error.ts
@@ -0,0 +1,125 @@
+
+export enum AkashErrorCategory {
+ WALLET = 'WALLET',
+ DEPLOYMENT = 'DEPLOYMENT',
+ LEASE = 'LEASE',
+ PROVIDER = 'PROVIDER',
+ MANIFEST = 'MANIFEST',
+ NETWORK = 'NETWORK',
+ TRANSACTION = 'TRANSACTION',
+ VALIDATION = 'VALIDATION',
+ SDK = 'SDK',
+ API = 'API',
+ FILE = 'FILE'
+}
+
+export enum AkashErrorCode {
+ // Wallet Errors (1000-1999)
+ WALLET_NOT_INITIALIZED = 1000,
+ WALLET_CONNECTION_FAILED = 1001,
+ WALLET_INSUFFICIENT_FUNDS = 1002,
+ WALLET_UNAUTHORIZED = 1003,
+ WALLET_SIGNATURE_FAILED = 1004,
+ WALLET_MESSAGE_INVALID = 1005,
+ WALLET_INITIALIZATION_FAILED = "WALLET_INITIALIZATION_FAILED",
+ CLIENT_SETUP_FAILED = "CLIENT_SETUP_FAILED",
+
+ // Certificate Errors (1500-1599)
+ CERTIFICATE_CREATION_FAILED = 1500,
+ CERTIFICATE_BROADCAST_FAILED = 1501,
+ CERTIFICATE_NOT_FOUND = 1502,
+
+ // Deployment Errors (2000-2999)
+ DEPLOYMENT_NOT_FOUND = 2000,
+ DEPLOYMENT_CREATION_FAILED = 2001,
+ DEPLOYMENT_UPDATE_FAILED = 2002,
+ DEPLOYMENT_CLOSE_FAILED = 2003,
+ DEPLOYMENT_START_TIMEOUT = 2004,
+
+ // Lease Errors (3000-3999)
+ LEASE_NOT_FOUND = 3000,
+ LEASE_CREATION_FAILED = 3001,
+ LEASE_CLOSE_FAILED = 3002,
+ LEASE_INVALID_STATE = 3003,
+ LEASE_BID_NOT_FOUND = 3004,
+ LEASE_QUERY_FAILED = 3005,
+ LEASE_STATUS_ERROR = 3006,
+ LEASE_VALIDATION_FAILED = 3007,
+ INVALID_LEASE = 3008,
+
+ // Provider Errors (4000-4999)
+ PROVIDER_NOT_FOUND = 4000,
+ PROVIDER_UNREACHABLE = 4001,
+ PROVIDER_RESPONSE_ERROR = 4002,
+ PROVIDER_LIST_ERROR = 4003,
+ PROVIDER_FILTER_ERROR = 4004,
+
+ // Manifest Errors (5000-5999)
+ MANIFEST_INVALID = 5000,
+ MANIFEST_PARSING_FAILED = 5001,
+ MANIFEST_DEPLOYMENT_FAILED = 5002,
+ MANIFEST_VALIDATION_FAILED = 5003,
+
+ // Bid Errors (6000-6999)
+ BID_FETCH_TIMEOUT = 6000,
+ INVALID_BID = 6001,
+
+ // SDL Errors (7000-7999)
+ SDL_PARSING_FAILED = 7000,
+
+ // Validation Errors (8000-8999)
+ VALIDATION_PARAMETER_MISSING = 8000,
+ VALIDATION_PARAMETER_INVALID = 8001,
+ VALIDATION_STATE_INVALID = 8002,
+ VALIDATION_SDL_FAILED = 8003,
+ VALIDATION_CONFIG_INVALID = 8004,
+
+ // Generic Errors (9000-9999)
+ INSUFFICIENT_FUNDS = 9000,
+
+ // API Errors (10000-10999)
+ API_ERROR = 10000,
+ API_RESPONSE_INVALID = 10001,
+ API_REQUEST_FAILED = 10002,
+ API_TIMEOUT = 10003,
+
+ // File System Errors (11000-11999)
+ FILE_NOT_FOUND = 11000,
+ FILE_READ_ERROR = 11001,
+ FILE_WRITE_ERROR = 11002,
+ FILE_PERMISSION_ERROR = 11003,
+
+ // Network Errors (12000-12999)
+ RPC_CONNECTION_FAILED = 12000
+}
+
+export class AkashError extends Error {
+ constructor(
+ message: string,
+ public code: AkashErrorCode,
+ public details?: Record,
+ public category: string = "akash"
+ ) {
+ super(message);
+ this.name = "AkashError";
+ }
+}
+
+export async function withRetry(
+ fn: () => Promise,
+ maxRetries: number = 3,
+ delay: number = 1000
+): Promise {
+ let lastError: Error | undefined;
+ for (let i = 0; i < maxRetries; i++) {
+ try {
+ return await fn();
+ } catch (error) {
+ lastError = error as Error;
+ if (i < maxRetries - 1) {
+ await new Promise(resolve => setTimeout(resolve, delay * Math.pow(2, i)));
+ }
+ }
+ }
+ throw lastError;
+}
diff --git a/packages/plugin-akash/src/index.ts b/packages/plugin-akash/src/index.ts
new file mode 100644
index 00000000000..ffa07f75f6b
--- /dev/null
+++ b/packages/plugin-akash/src/index.ts
@@ -0,0 +1,68 @@
+import { Plugin} from "@elizaos/core";
+import { createDeploymentAction } from "./actions/createDeployment";
+import { closeDeploymentAction } from "./actions/closeDeployment";
+import { getProviderInfoAction } from "./actions/getProviderInfo";
+import { getDeploymentStatusAction } from "./actions/getDeploymentStatus";
+import { estimateGas } from "./actions/estimateGas";
+import { getDeploymentApiAction } from "./actions/getDeploymentApi";
+import { getGPUPricingAction } from "./actions/getGPUPricing";
+import { getManifestAction } from "./actions/getManifest";
+import { getProvidersListAction } from "./actions/getProvidersList";
+
+const actions = [
+ createDeploymentAction,
+ closeDeploymentAction,
+ getProviderInfoAction,
+ getDeploymentStatusAction,
+ estimateGas,
+ getDeploymentApiAction,
+ getGPUPricingAction,
+ getManifestAction,
+ getProvidersListAction,
+];
+
+// Initial banner
+console.log("\n┌════════════════════════════════════════┐");
+console.log("│ AKASH NETWORK PLUGIN │");
+console.log("├────────────────────────────────────────┤");
+console.log("│ Initializing Akash Network Plugin... │");
+console.log("│ Version: 0.1.0 │");
+console.log("└════════════════════════════════════════┘");
+
+// Format action registration message
+const formatActionInfo = (action: any) => {
+ const name = action.name.padEnd(25);
+ const similes = (action.similes?.join(", ") || "none").padEnd(60);
+ const hasHandler = action.handler ? "✓" : "✗";
+ const hasValidator = action.validate ? "✓" : "✗";
+ const hasExamples = action.examples?.length > 0 ? "✓" : "✗";
+
+ return `│ ${name} │ ${hasHandler} │ ${hasValidator} │ ${hasExamples} │ ${similes} │`;
+};
+
+// Log registered actions
+console.log("\n┌───────────────────────────┬───┬───┬───┬───────────────────────────────────────────────────────────┐");
+console.log("│ Action │ H │ V │ E │ Similes │");
+console.log("├───────────────────────────┼───┼───┼───┼────────────────────────────────────────────────────────────┤");
+actions.forEach(action => {
+ console.log(formatActionInfo(action));
+});
+console.log("└───────────────────────────┴───┴───┴───┴──────────────────────────────────────────────────────────┘");
+
+// Plugin status
+console.log("\n┌─────────────────────────────────────┐");
+console.log("│ Plugin Status │");
+console.log("├─────────────────────────────────────┤");
+console.log(`│ Name : akash │`);
+console.log(`│ Actions : ${actions.length.toString().padEnd(24)} │`);
+console.log(`│ Status : Loaded & Ready │`);
+console.log("└─────────────────────────────────────┘\n");
+
+export const akashPlugin: Plugin = {
+ name: "akash",
+ description: "Akash Network Plugin for deploying and managing cloud compute",
+ actions: actions,
+ evaluators: []
+};
+
+export default akashPlugin;
\ No newline at end of file
diff --git a/packages/plugin-akash/src/providers/wallet.ts b/packages/plugin-akash/src/providers/wallet.ts
new file mode 100644
index 00000000000..397c37ba5a0
--- /dev/null
+++ b/packages/plugin-akash/src/providers/wallet.ts
@@ -0,0 +1,108 @@
+import { DirectSecp256k1HdWallet } from "@cosmjs/proto-signing";
+import { SigningStargateClient } from "@cosmjs/stargate";
+import { elizaLogger, IAgentRuntime, Memory } from "@elizaos/core";
+// import { IAgentRuntime, Memory } from "@elizaos/core/src/types";
+import { validateAkashConfig } from "../environment";
+import { getAkashTypeRegistry } from "@akashnetwork/akashjs/build/stargate";
+import {
+ AkashProvider,
+ AkashWalletState,
+ AkashError,
+ AKASH_ERROR_CODES,
+} from "../types";
+
+// Use a proper UUID for the wallet room
+const WALLET_ROOM_ID = "00000000-0000-0000-0000-000000000001";
+
+export const walletProvider: AkashProvider = {
+ type: "AKASH_WALLET",
+ version: "1.0.0",
+ name: "wallet",
+ description: "Akash wallet provider",
+
+ initialize: async (runtime: IAgentRuntime): Promise => {
+ elizaLogger.info("Initializing Akash wallet provider");
+ try {
+ const mnemonic = runtime.getSetting("AKASH_MNEMONIC");
+ if (!mnemonic) {
+ throw new Error("AKASH_MNEMONIC not found in environment variables");
+ }
+
+ const config = await validateAkashConfig(runtime);
+
+ // Create wallet from mnemonic
+ const wallet = await DirectSecp256k1HdWallet.fromMnemonic(config.AKASH_MNEMONIC, {
+ prefix: "akash",
+ });
+
+ // Get the wallet address
+ const [account] = await wallet.getAccounts();
+ const address = account.address;
+
+ // Create signing client with registry
+ const client = await SigningStargateClient.connectWithSigner(
+ config.RPC_ENDPOINT,
+ wallet,
+ { registry: getAkashTypeRegistry() as any }
+ );
+
+ // Store wallet info in memory manager
+ const state: AkashWalletState = {
+ wallet,
+ client,
+ address,
+ };
+
+ // Create memory object
+ const memory: Memory = {
+ id: WALLET_ROOM_ID,
+ userId: runtime.agentId,
+ agentId: runtime.agentId,
+ roomId: WALLET_ROOM_ID,
+ content: {
+ type: "wallet_state",
+ text: `Akash wallet initialized with address: ${address}`,
+ data: state,
+ },
+ createdAt: Date.now(),
+ };
+
+ await runtime.messageManager.createMemory(memory);
+
+ elizaLogger.info("Akash wallet provider initialized successfully", {
+ address,
+ });
+ } catch (error) {
+ elizaLogger.error("Failed to initialize Akash wallet provider", {
+ error: error instanceof Error ? error.message : String(error)
+ });
+ throw error;
+ }
+ },
+
+ get: async (runtime: IAgentRuntime, _message?: Memory): Promise => {
+ const memories = await runtime.messageManager.getMemories({
+ roomId: WALLET_ROOM_ID,
+ count: 1,
+ });
+
+ const state = memories[0]?.content?.data;
+ if (!state) {
+ throw new AkashError(
+ "Akash wallet not initialized",
+ AKASH_ERROR_CODES.WALLET_NOT_INITIALIZED
+ );
+ }
+ return state as AkashWalletState;
+ },
+
+ validate: async (_runtime: IAgentRuntime, _message?: Memory): Promise => {
+ return true;
+ },
+
+ process: async (_runtime: IAgentRuntime, _message?: Memory): Promise => {
+ // No processing needed for wallet provider
+ }
+};
+
+export default walletProvider;
diff --git a/packages/plugin-akash/src/runtime_inspect.ts b/packages/plugin-akash/src/runtime_inspect.ts
new file mode 100644
index 00000000000..25b5aee39fd
--- /dev/null
+++ b/packages/plugin-akash/src/runtime_inspect.ts
@@ -0,0 +1,90 @@
+import { elizaLogger } from "@elizaos/core";
+import type { IAgentRuntime, Plugin, Action } from "@elizaos/core";
+
+/**
+ * Utility to inspect runtime plugin loading
+ */
+export function inspectRuntime(runtime: IAgentRuntime) {
+ elizaLogger.info("=== Runtime Plugin Inspection ===");
+
+ // Check if runtime has plugins array
+ const hasPlugins = !!(runtime as any).plugins;
+ elizaLogger.info("Runtime plugins status:", {
+ hasPluginsArray: hasPlugins,
+ pluginCount: hasPlugins ? (runtime as any).plugins.length : 0
+ });
+
+ // If plugins exist, check for our plugin
+ if (hasPlugins) {
+ const plugins = (runtime as any).plugins as Plugin[];
+ const akashPlugin = plugins.find(p => p.name === "akash");
+
+ elizaLogger.info("Akash plugin status:", {
+ isLoaded: !!akashPlugin,
+ pluginDetails: akashPlugin ? {
+ name: akashPlugin.name,
+ actionCount: akashPlugin.actions?.length || 0,
+ actions: akashPlugin.actions?.map(a => a.name) || []
+ } : null
+ });
+ }
+
+ // Check registered actions
+ const hasActions = !!(runtime as any).actions;
+ if (hasActions) {
+ const actions = (runtime as any).actions as Action[];
+ const akashActions = actions.filter((action: Action) =>
+ action.name === "CREATE_DEPLOYMENT" ||
+ (action.similes || []).includes("CREATE_DEPLOYMENT")
+ );
+
+ elizaLogger.info("Akash actions status:", {
+ totalActions: actions.length,
+ akashActionsCount: akashActions.length,
+ akashActions: akashActions.map((action: Action) => ({
+ name: action.name,
+ similes: action.similes
+ }))
+ });
+ }
+}
+
+/**
+ * Helper to check if a plugin is properly loaded
+ */
+export function isPluginLoaded(runtime: IAgentRuntime, pluginName: string): boolean {
+ // Check plugins array
+ const plugins = (runtime as any).plugins as Plugin[];
+ if (!plugins) {
+ elizaLogger.warn(`No plugins array found in runtime`);
+ return false;
+ }
+
+ // Look for our plugin
+ const plugin = plugins.find(p => p.name === pluginName);
+ if (!plugin) {
+ elizaLogger.warn(`Plugin ${pluginName} not found in runtime plugins`);
+ return false;
+ }
+
+ // Check if actions are registered
+ const actions = (runtime as any).actions as Action[];
+ if (!actions || !actions.length) {
+ elizaLogger.warn(`No actions found in runtime`);
+ return false;
+ }
+
+ // Check if plugin's actions are registered
+ const pluginActions = plugin.actions || [];
+ const registeredActions = pluginActions.every(pluginAction =>
+ actions.some((action: Action) => action.name === pluginAction.name)
+ );
+
+ if (!registeredActions) {
+ elizaLogger.warn(`Not all ${pluginName} actions are registered in runtime`);
+ return false;
+ }
+
+ elizaLogger.info(`Plugin ${pluginName} is properly loaded and registered`);
+ return true;
+}
\ No newline at end of file
diff --git a/packages/plugin-akash/src/sdl/example.sdl.yml b/packages/plugin-akash/src/sdl/example.sdl.yml
new file mode 100644
index 00000000000..6e6ac836886
--- /dev/null
+++ b/packages/plugin-akash/src/sdl/example.sdl.yml
@@ -0,0 +1,33 @@
+---
+version: "2.0"
+services:
+ web:
+ image: baktun/hello-akash-world:1.0.0
+ expose:
+ - port: 3000
+ as: 80
+ to:
+ - global: true
+profiles:
+ compute:
+ web:
+ resources:
+ cpu:
+ units: 0.5
+ memory:
+ size: 512Mi
+ storage:
+ size: 512Mi
+ placement:
+ dcloud:
+ pricing:
+ web:
+ denom: uakt
+ amount: 20000
+
+
+deployment:
+ web:
+ dcloud:
+ profile: web
+ count: 1
diff --git a/packages/plugin-akash/src/types.ts b/packages/plugin-akash/src/types.ts
new file mode 100644
index 00000000000..b784290b499
--- /dev/null
+++ b/packages/plugin-akash/src/types.ts
@@ -0,0 +1,167 @@
+import { DirectSecp256k1HdWallet} from "@cosmjs/proto-signing";
+import { SigningStargateClient } from "@cosmjs/stargate";
+// import { Provider } from "@elizaos/core";
+import { IAgentRuntime, Memory } from "@elizaos/core";
+import { SDL } from "@akashnetwork/akashjs/build/sdl";
+import { MsgCreateDeployment } from "@akashnetwork/akash-api/akash/deployment/v1beta3";
+import { MsgCreateLease} from "@akashnetwork/akash-api/akash/market/v1beta4";
+
+// Core wallet state type
+export interface AkashWalletState {
+ wallet: DirectSecp256k1HdWallet;
+ client: SigningStargateClient;
+ address: string;
+ certificate?: {
+ cert: string;
+ privateKey: string;
+ publicKey: string;
+ };
+}
+
+// Provider type extending core Provider
+export interface AkashProvider {
+ type: string;
+ version: string;
+ name: string;
+ description: string;
+ initialize: (runtime: IAgentRuntime) => Promise;
+ get: (runtime: IAgentRuntime, message?: Memory) => Promise;
+ validate: (runtime: IAgentRuntime, message?: Memory) => Promise;
+ process: (runtime: IAgentRuntime, message?: Memory) => Promise;
+}
+
+// Registry type for Akash
+export type AkashRegistryTypes = [string, any][];
+
+// Deployment related types
+export interface AkashDeploymentId {
+ owner: string;
+ dseq: string;
+}
+
+export interface AkashDeployment {
+ id: AkashDeploymentId;
+ sdl: SDL;
+ deposit: string;
+ msg?: MsgCreateDeployment;
+}
+
+// Lease related types
+export interface AkashLeaseId {
+ owner: string;
+ dseq: string;
+ provider: string;
+ gseq: number;
+ oseq: number;
+}
+
+export interface AkashLease {
+ id: AkashLeaseId;
+ state?: string;
+ manifestData?: any;
+ msg?: MsgCreateLease;
+}
+
+// Provider types
+export interface AkashProviderInfo {
+ owner: string;
+ hostUri: string;
+ attributes: Array<{
+ key: string;
+ value: string;
+ }>;
+}
+
+// Bid types
+export interface AkashBidId {
+ owner: string;
+ dseq: string;
+ gseq: number;
+ oseq: number;
+ provider: string;
+}
+
+export interface AkashBid {
+ id: AkashBidId;
+ state: string;
+ price: {
+ denom: string;
+ amount: string;
+ };
+}
+
+// Error handling types
+export enum AKASH_ERROR_CODES {
+ WALLET_NOT_INITIALIZED = "WALLET_NOT_INITIALIZED",
+ INVALID_MNEMONIC = "INVALID_MNEMONIC",
+ INVALID_ADDRESS = "INVALID_ADDRESS",
+ INSUFFICIENT_FUNDS = "INSUFFICIENT_FUNDS",
+ DEPLOYMENT_FAILED = "DEPLOYMENT_FAILED",
+ LEASE_FAILED = "LEASE_FAILED",
+ PROVIDER_NOT_FOUND = "PROVIDER_NOT_FOUND",
+ NETWORK_ERROR = "NETWORK_ERROR",
+ CERTIFICATE_ERROR = "CERTIFICATE_ERROR",
+ MANIFEST_ERROR = "MANIFEST_ERROR",
+ BID_ERROR = "BID_ERROR",
+ MANIFEST_FAILED = "MANIFEST_FAILED",
+ PROVIDER_ERROR = "PROVIDER_ERROR"
+}
+
+export class AkashError extends Error {
+ constructor(
+ message: string,
+ public code: AKASH_ERROR_CODES,
+ public originalError?: Error
+ ) {
+ super(message);
+ this.name = "AkashError";
+ }
+}
+
+// Provider configuration
+export interface AkashConfig {
+ AKASH_MNEMONIC: string;
+ RPC_ENDPOINT: string;
+ CHAIN_ID?: string;
+ GAS_PRICE?: string;
+ GAS_ADJUSTMENT?: number;
+ CERTIFICATE_PATH?: string;
+}
+
+// Message types
+export interface AkashMessage {
+ type: string;
+ value: any;
+}
+
+// Response types
+export interface AkashTxResponse {
+ code: number;
+ height: number;
+ txhash: string;
+ rawLog: string;
+ data?: string;
+ gasUsed: number;
+ gasWanted: number;
+}
+
+// Provider state types
+export interface AkashProviderState {
+ isInitialized: boolean;
+ lastSync: number;
+ balance?: string;
+ address?: string;
+ certificate?: {
+ cert: string;
+ privateKey: string;
+ publicKey: string;
+ };
+}
+
+// Memory room constants
+export const AKASH_MEMORY_ROOMS = {
+ WALLET: "00000000-0000-0000-0000-000000000001",
+ DEPLOYMENT: "00000000-0000-0000-0000-000000000002",
+ LEASE: "00000000-0000-0000-0000-000000000003",
+ CERTIFICATE: "00000000-0000-0000-0000-000000000004"
+} as const;
diff --git a/packages/plugin-akash/src/utils/paths.ts b/packages/plugin-akash/src/utils/paths.ts
new file mode 100644
index 00000000000..c74151b2f7a
--- /dev/null
+++ b/packages/plugin-akash/src/utils/paths.ts
@@ -0,0 +1,133 @@
+import * as path from 'path';
+import { fileURLToPath } from 'url';
+import { elizaLogger } from "@elizaos/core";
+import { existsSync } from 'fs';
+import { getConfig } from '../environment';
+
+export const getPluginRoot = (importMetaUrl: string) => {
+ // elizaLogger.info("=== Starting Plugin Root Resolution ===", {
+ // importMetaUrl,
+ // isFileProtocol: importMetaUrl.startsWith('file://'),
+ // urlSegments: importMetaUrl.split('/')
+ // });
+
+ const currentFileUrl = importMetaUrl;
+ const currentFilePath = fileURLToPath(currentFileUrl);
+ const currentDir = path.dirname(currentFilePath);
+
+ // Find plugin-akash directory by walking up until we find it
+ let dir = currentDir;
+ while (dir && path.basename(dir) !== 'plugin-akash' && dir !== '/') {
+ dir = path.dirname(dir);
+ }
+
+ if (!dir || dir === '/') {
+ elizaLogger.error("Could not find plugin-akash directory", {
+ currentFilePath,
+ currentDir,
+ searchPath: dir
+ });
+ throw new Error("Could not find plugin-akash directory");
+ }
+
+ // elizaLogger.info("Plugin Root Path Details", {
+ // currentFilePath,
+ // currentDir,
+ // pluginRoot: dir,
+ // exists: existsSync(dir),
+ // parentDir: path.dirname(dir),
+ // parentExists: existsSync(path.dirname(dir)),
+ // parentContents: existsSync(path.dirname(dir)) ? fs.readdirSync(path.dirname(dir)) : []
+ // });
+
+ return dir;
+};
+
+export const getSrcPath = (importMetaUrl: string) => {
+ // elizaLogger.info("=== Resolving Src Path ===");
+ const pluginRoot = getPluginRoot(importMetaUrl);
+ const srcPath = path.join(pluginRoot, 'src');
+
+ // elizaLogger.info("Src Path Details", {
+ // pluginRoot,
+ // srcPath,
+ // exists: existsSync(srcPath),
+ // contents: existsSync(srcPath) ? fs.readdirSync(srcPath) : [],
+ // absolutePath: path.resolve(srcPath),
+ // relativeToCwd: path.relative(process.cwd(), srcPath)
+ // });
+
+ return srcPath;
+};
+
+export const getCertificatePath = (importMetaUrl: string) => {
+ const srcPath = getSrcPath(importMetaUrl);
+ const certPath = path.join(srcPath, '.certificates', 'cert.json');
+
+ // elizaLogger.debug("Certificate Path Resolution", {
+ // srcPath,
+ // certPath,
+ // exists: existsSync(certPath)
+ // });
+
+ return certPath;
+};
+
+export const getDefaultSDLPath = (importMetaUrl: string) => {
+ // elizaLogger.info("=== Resolving SDL Path ===");
+ const pluginRoot = getPluginRoot(importMetaUrl);
+ const srcPath = getSrcPath(importMetaUrl);
+ const config = getConfig(process.env.AKASH_ENV);
+ const sdlFileName = config.AKASH_SDL;
+ const sdlPath = path.join(srcPath, 'sdl', sdlFileName);
+ // const sdlDir = path.dirname(sdlPath);
+
+ // Only log if file doesn't exist as a warning
+ if (!existsSync(sdlPath)) {
+ // elizaLogger.warn("SDL file not found at expected path", {
+ // sdlPath,
+ // exists: false
+ // });
+ }
+
+ // Try to find SDL file in nearby directories
+ const searchPaths = [
+ sdlPath,
+ path.join(srcPath, sdlFileName),
+ path.join(pluginRoot, sdlFileName),
+ path.join(pluginRoot, 'sdl', sdlFileName),
+ path.join(pluginRoot, 'src', 'sdl', sdlFileName)
+ ];
+
+ // Only log if we find the file
+ for (const searchPath of searchPaths) {
+ if (existsSync(searchPath)) {
+ // elizaLogger.info("Found SDL file at", { path: searchPath });
+ return searchPath;
+ }
+ }
+
+ return sdlPath;
+};
+
+// Helper function to ensure a path includes plugin-akash
+export const ensurePluginPath = (filePath: string, importMetaUrl: string) => {
+ if (!filePath.includes('plugin-akash')) {
+ const srcPath = getSrcPath(importMetaUrl);
+ return path.join(srcPath, path.basename(filePath));
+ }
+ return filePath;
+};
+
+export function getDeploymentsPath(importMetaUrl: string): string {
+ const srcPath = getSrcPath(importMetaUrl);
+ const deploymentsPath = path.join(srcPath, 'deployments');
+
+ // elizaLogger.debug("Deployments Path Resolution", {
+ // srcPath,
+ // deploymentsPath,
+ // exists: existsSync(deploymentsPath)
+ // });
+
+ return deploymentsPath;
+}
\ No newline at end of file
diff --git a/packages/plugin-akash/tsconfig.json b/packages/plugin-akash/tsconfig.json
new file mode 100644
index 00000000000..e535bee0d71
--- /dev/null
+++ b/packages/plugin-akash/tsconfig.json
@@ -0,0 +1,39 @@
+{
+ "extends": "../../tsconfig.json",
+ "compilerOptions": {
+ "outDir": "dist",
+ "rootDir": "src",
+ "module": "ESNext",
+ "target": "ESNext",
+ "lib": [
+ "ESNext",
+ "DOM"
+ ],
+ "moduleResolution": "Bundler",
+ "allowImportingTsExtensions": true,
+ "emitDeclarationOnly": true,
+ "isolatedModules": true,
+ "esModuleInterop": true,
+ "skipLibCheck": true,
+ "strict": true,
+ "declaration": true,
+ "sourceMap": true,
+ "types": [
+ "vitest/globals",
+ "node",
+ "jest"
+ ],
+ "baseUrl": ".",
+ "preserveSymlinks": true
+ },
+ "include": [
+ "src/**/*",
+ "test/actions/getDeploymentApi.test.ts"
+ ],
+ "exclude": [
+ "node_modules",
+ "dist",
+ "test",
+ "../../packages/core/**/*"
+ ]
+}
\ No newline at end of file
diff --git a/packages/plugin-akash/tsup.config.ts b/packages/plugin-akash/tsup.config.ts
new file mode 100644
index 00000000000..a2b714de910
--- /dev/null
+++ b/packages/plugin-akash/tsup.config.ts
@@ -0,0 +1,10 @@
+import { defineConfig } from "tsup";
+
+export default defineConfig({
+ entry: ["src/index.ts"],
+ format: ["esm"],
+ dts: true,
+ splitting: false,
+ sourcemap: true,
+ clean: true,
+});
diff --git a/packages/plugin-akash/vitest.config.ts b/packages/plugin-akash/vitest.config.ts
new file mode 100644
index 00000000000..2b76c168780
--- /dev/null
+++ b/packages/plugin-akash/vitest.config.ts
@@ -0,0 +1,27 @@
+import { defineConfig } from 'vitest/config';
+import path from 'path';
+
+export default defineConfig({
+ test: {
+ globals: true,
+ environment: 'node',
+ include: ['test/**/*.{test,spec}.{js,mjs,cjs,ts,mts,cts,jsx,tsx}'],
+ exclude: ['node_modules', 'dist', '.idea', '.git', '.cache'],
+ root: '.',
+ reporters: ['verbose'],
+ coverage: {
+ reporter: ['text', 'json', 'html'],
+ exclude: [
+ 'node_modules/',
+ 'test/fixtures/',
+ 'test/setup/'
+ ]
+ },
+ setupFiles: ['./test/setup/vitest.setup.ts']
+ },
+ resolve: {
+ alias: {
+ '@': path.resolve(__dirname, './src')
+ }
+ }
+});
\ No newline at end of file
diff --git a/packages/plugin-allora/package.json b/packages/plugin-allora/package.json
index 3422ddb92ef..969d14668e8 100644
--- a/packages/plugin-allora/package.json
+++ b/packages/plugin-allora/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/plugin-allora",
- "version": "0.1.7-alpha.1",
+ "version": "0.1.8+build.1",
"main": "dist/index.js",
"type": "module",
"types": "dist/index.d.ts",
@@ -20,4 +20,4 @@
"peerDependencies": {
"whatwg-url": "7.1.0"
}
-}
\ No newline at end of file
+}
diff --git a/packages/plugin-anyone/package.json b/packages/plugin-anyone/package.json
index 371c1c99cc9..a829629324b 100644
--- a/packages/plugin-anyone/package.json
+++ b/packages/plugin-anyone/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/plugin-anyone",
- "version": "0.1.7-alpha.2",
+ "version": "0.1.8+build.1",
"main": "dist/index.js",
"type": "module",
"types": "dist/index.d.ts",
@@ -18,4 +18,4 @@
"peerDependencies": {
"whatwg-url": "7.1.0"
}
-}
\ No newline at end of file
+}
diff --git a/packages/plugin-anyone/src/actions/startAnyone.ts b/packages/plugin-anyone/src/actions/startAnyone.ts
index 9edc260ae71..855837c4f8b 100644
--- a/packages/plugin-anyone/src/actions/startAnyone.ts
+++ b/packages/plugin-anyone/src/actions/startAnyone.ts
@@ -24,7 +24,8 @@ export const startAnyone: Action = {
_callback: HandlerCallback
): Promise => {
await AnyoneClientService.initialize();
- const anon = AnyoneClientService.getInstance();
+ //lint says unused
+ //const anon = AnyoneClientService.getInstance();
const proxyService = AnyoneProxyService.getInstance();
await proxyService.initialize();
diff --git a/packages/plugin-aptos/package.json b/packages/plugin-aptos/package.json
index c2271365c80..abd65d16eb0 100644
--- a/packages/plugin-aptos/package.json
+++ b/packages/plugin-aptos/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/plugin-aptos",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
diff --git a/packages/plugin-arthera/package.json b/packages/plugin-arthera/package.json
index db58990809f..95fe3f5557e 100644
--- a/packages/plugin-arthera/package.json
+++ b/packages/plugin-arthera/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/plugin-arthera",
- "version": "0.1.8-alpha.1",
+ "version": "0.1.8+build.1",
"main": "dist/index.js",
"type": "module",
"types": "dist/index.d.ts",
diff --git a/packages/plugin-asterai/.npmignore b/packages/plugin-asterai/.npmignore
new file mode 100644
index 00000000000..0468b4b3648
--- /dev/null
+++ b/packages/plugin-asterai/.npmignore
@@ -0,0 +1,6 @@
+*
+
+!dist/**
+!package.json
+!readme.md
+!tsup.config.ts
diff --git a/packages/plugin-asterai/README.md b/packages/plugin-asterai/README.md
new file mode 100644
index 00000000000..c84bf32f3a5
--- /dev/null
+++ b/packages/plugin-asterai/README.md
@@ -0,0 +1,80 @@
+# @elizaos/plugin-asterai
+
+A plugin for interacting with [asterai](https://asterai.io) plugins and agents.
+
+## Description
+
+This plugin provides functionality to allow Eliza agents to interact with
+asterai plugins and agents.
+
+This will expand your Eliza character's utility by giving it access to all
+the functionality of asterai's ecosystem of marketplace and private plugins
+and agents.
+
+## Installation
+
+```bash
+pnpm install @elizaos/plugin-asterai
+```
+
+## Configuration
+
+The plugin requires the following environment variables to be set:
+
+```typescript
+ASTERAI_AGENT_ID=
+ASTERAI_PUBLIC_QUERY_KEY=
+```
+
+## Usage
+
+### Basic Integration
+
+```typescript
+import { asteraiPlugin } from '@elizaos/plugin-asterai';
+```
+
+### Example Usage
+
+The plugin supports natural language for interacting with the asterai agent
+through your Eliza character.
+
+For example, if your asterai agent can fetch weather data:
+
+```typescript
+"Hey Eliza, how's the weather in LA?"
+```
+
+Eliza will then query the asterai agent to fetch the information.
+
+## Development Guide
+
+### Setting Up Development Environment
+
+1. Clone the repository
+2. Install dependencies:
+
+```bash
+pnpm install
+```
+
+3. Build the plugin:
+
+```bash
+pnpm run build
+```
+
+4. Run tests:
+
+```bash
+pnpm run test
+```
+
+## Contributing
+
+Contributions are welcome! Please see the [CONTRIBUTING.md](CONTRIBUTING.md) file for more information.
+
+## License
+
+This plugin is part of the Eliza project. See the main project repository for license information.
+
diff --git a/packages/plugin-asterai/eslint.config.mjs b/packages/plugin-asterai/eslint.config.mjs
new file mode 100644
index 00000000000..92fe5bbebef
--- /dev/null
+++ b/packages/plugin-asterai/eslint.config.mjs
@@ -0,0 +1,3 @@
+import eslintGlobalConfig from "../../eslint.config.mjs";
+
+export default [...eslintGlobalConfig];
diff --git a/packages/plugin-asterai/package.json b/packages/plugin-asterai/package.json
new file mode 100644
index 00000000000..9ddcc805eab
--- /dev/null
+++ b/packages/plugin-asterai/package.json
@@ -0,0 +1,46 @@
+{
+ "name": "@elizaos/plugin-asterai",
+ "version": "0.1.8+build.1",
+ "type": "module",
+ "main": "dist/index.js",
+ "module": "dist/index.js",
+ "types": "dist/index.d.ts",
+ "exports": {
+ "./package.json": "./package.json",
+ ".": {
+ "import": {
+ "@elizaos/source": "./src/index.ts",
+ "types": "./dist/index.d.ts",
+ "default": "./dist/index.js"
+ }
+ }
+ },
+ "files": [
+ "dist"
+ ],
+ "dependencies": {
+ "@asterai/client": "0.1.6",
+ "@elizaos/core": "workspace:*",
+ "bignumber.js": "9.1.2",
+ "bs58": "6.0.0",
+ "elliptic": "6.6.1",
+ "node-cache": "5.1.2",
+ "sha3": "2.1.4",
+ "uuid": "11.0.3",
+ "zod": "3.23.8"
+ },
+ "devDependencies": {
+ "@types/elliptic": "6.4.18",
+ "@types/uuid": "10.0.0",
+ "tsup": "8.3.5"
+ },
+ "scripts": {
+ "lines": "find . \\( -name '*.cdc' -o -name '*.ts' \\) -not -path '*/node_modules/*' -not -path '*/tests/*' -not -path '*/deps/*' -not -path '*/dist/*' -not -path '*/imports*' | xargs wc -l",
+ "build": "tsup --format esm --dts",
+ "dev": "tsup --format esm --dts --watch",
+ "lint": "eslint --fix --cache ."
+ },
+ "peerDependencies": {
+ "whatwg-url": "7.1.0"
+ }
+}
diff --git a/packages/plugin-asterai/src/actions/query.ts b/packages/plugin-asterai/src/actions/query.ts
new file mode 100644
index 00000000000..c59fbbe632c
--- /dev/null
+++ b/packages/plugin-asterai/src/actions/query.ts
@@ -0,0 +1,72 @@
+import {
+ elizaLogger,
+ type Action,
+ type ActionExample,
+ type HandlerCallback,
+ type IAgentRuntime,
+ type Memory,
+ type State,
+} from "@elizaos/core";
+import { validateAsteraiConfig } from "../environment";
+import {getInitAsteraiClient} from "../index.ts";
+
+export const queryAction = {
+ name: "QUERY_ASTERAI_AGENT",
+ similes: [
+ "MESSAGE_ASTERAI_AGENT",
+ "TALK_TO_ASTERAI_AGENT",
+ "SEND_MESSAGE_TO_ASTERAI_AGENT",
+ "COMMUNICATE_WITH_ASTERAI_AGENT",
+ ],
+ description:
+ "Call this action to send a message to the asterai agent which " +
+ "has access to external plugins and functionality to answer " +
+ "the user you are assisting, to help perform a workflow task, etc.",
+ validate: async (runtime: IAgentRuntime, _message: Memory) => {
+ const config = await validateAsteraiConfig(runtime);
+ getInitAsteraiClient(
+ config.ASTERAI_AGENT_ID,
+ config.ASTERAI_PUBLIC_QUERY_KEY
+ );
+ return true;
+ },
+ handler: async (
+ runtime: IAgentRuntime,
+ message: Memory,
+ _state: State,
+ _options: { [key: string]: unknown },
+ callback?: HandlerCallback
+ ): Promise => {
+ const config = await validateAsteraiConfig(runtime);
+ const asteraiClient = getInitAsteraiClient(
+ config.ASTERAI_AGENT_ID,
+ config.ASTERAI_PUBLIC_QUERY_KEY
+ );
+ elizaLogger.debug("called QUERY_ASTERAI_AGENT action with message:", message.content);
+ const response = await asteraiClient.query({
+ query: message.content.text
+ });
+ const textResponse = await response.text();
+ callback({
+ text: textResponse
+ });
+ return true;
+ },
+ examples: [
+ [
+ {
+ user: "{{user1}}",
+ content: {
+ text: "How's the weather in LA?",
+ },
+ },
+ {
+ user: "{{user2}}",
+ content: {
+ text: "Let me check that for you, just a moment.",
+ action: "QUERY_ASTERAI_AGENT",
+ },
+ },
+ ],
+ ] as ActionExample[][],
+} as Action;
diff --git a/packages/plugin-asterai/src/environment.ts b/packages/plugin-asterai/src/environment.ts
new file mode 100644
index 00000000000..a15c6f919bf
--- /dev/null
+++ b/packages/plugin-asterai/src/environment.ts
@@ -0,0 +1,39 @@
+import { IAgentRuntime } from "@elizaos/core";
+import { z } from "zod";
+
+const envSchema = z.object({
+ ASTERAI_AGENT_ID: z
+ .string()
+ .min(1, "ASTERAI_AGENT_ID is required"),
+ ASTERAI_PUBLIC_QUERY_KEY: z
+ .string()
+ .min(1, "ASTERAI_PUBLIC_QUERY_KEY is required"),
+});
+
+export type AsteraiConfig = z.infer;
+
+export async function validateAsteraiConfig(
+ runtime: IAgentRuntime
+): Promise {
+ try {
+ const config = {
+ ASTERAI_AGENT_ID:
+ runtime.getSetting("ASTERAI_AGENT_ID") ||
+ process.env.ASTERAI_AGENT_ID,
+ ASTERAI_PUBLIC_QUERY_KEY:
+ runtime.getSetting("ASTERAI_PUBLIC_QUERY_KEY") || process.env.ASTERAI_PUBLIC_QUERY_KEY,
+ };
+
+ return envSchema.parse(config);
+ } catch (error) {
+ if (error instanceof z.ZodError) {
+ const errorMessages = error.errors
+ .map((err) => `${err.path.join(".")}: ${err.message}`)
+ .join("\n");
+ throw new Error(
+ `Asterai plugin configuration validation failed:\n${errorMessages}`
+ );
+ }
+ throw error;
+ }
+}
diff --git a/packages/plugin-asterai/src/index.ts b/packages/plugin-asterai/src/index.ts
new file mode 100644
index 00000000000..7f77f154b26
--- /dev/null
+++ b/packages/plugin-asterai/src/index.ts
@@ -0,0 +1,33 @@
+import {asteraiProvider} from "./providers/asterai.provider.ts";
+import type { Plugin } from "@elizaos/core";
+import { queryAction } from "./actions/query";
+import { AsteraiClient } from "@asterai/client";
+
+export * from "./environment";
+export * from "./providers/asterai.provider";
+
+let asteraiClient: AsteraiClient | null = null;
+
+export const getInitAsteraiClient = (
+ agentId: string,
+ publicQueryKey: string
+): AsteraiClient => {
+ if (!asteraiClient) {
+ asteraiClient = new AsteraiClient({
+ appId: agentId,
+ queryKey: publicQueryKey,
+ })
+ }
+ return asteraiClient;
+};
+
+export const asteraiPlugin: Plugin = {
+ name: "asterai",
+ description: "asterai Plugin for Eliza",
+ providers: [asteraiProvider],
+ actions: [queryAction],
+ evaluators: [],
+ services: [],
+};
+
+export default asteraiPlugin;
diff --git a/packages/plugin-asterai/src/providers/asterai.provider.ts b/packages/plugin-asterai/src/providers/asterai.provider.ts
new file mode 100644
index 00000000000..f8bfc0084a6
--- /dev/null
+++ b/packages/plugin-asterai/src/providers/asterai.provider.ts
@@ -0,0 +1,63 @@
+import {
+ elizaLogger,
+ IAgentRuntime,
+ Memory,
+ Provider,
+ State, UUID,
+} from "@elizaos/core";
+import {validateAsteraiConfig} from "../environment.ts";
+import {getInitAsteraiClient} from "../index.ts";
+
+const asteraiProvider: Provider = {
+ get: async (
+ runtime: IAgentRuntime,
+ message: Memory,
+ _state?: State
+ ): Promise => {
+ const hasConfiguredEnv =
+ !!runtime.getSetting("ASTERAI_AGENT_ID") &&
+ !!runtime.getSetting("ASTERAI_PUBLIC_QUERY_KEY");
+ if (!hasConfiguredEnv) {
+ elizaLogger.error(
+ "ASTERAI_AGENT_ID or ASTERAI_PUBLIC_QUERY_KEY " +
+ "not configured, skipping provider"
+ );
+ return null;
+ }
+ const config = await validateAsteraiConfig(runtime);
+ const asteraiClient = getInitAsteraiClient(
+ config.ASTERAI_AGENT_ID,
+ config.ASTERAI_PUBLIC_QUERY_KEY
+ );
+ if (!asteraiClient) {
+ elizaLogger.error("asteraiClient is not initialised");
+ return null;
+ }
+ const agentId = runtime.getSetting("ASTERAI_AGENT_ID") as UUID;
+ let agentSummaryMemory = await runtime.knowledgeManager.getMemoryById(agentId);
+ if (!agentSummaryMemory) {
+ // Fetch & set summary memory.
+ const summary = await asteraiClient.fetchSummary();
+ elizaLogger.debug("asterai agent summary fetched:", summary);
+ await runtime.knowledgeManager.createMemory({
+ id: agentId,
+ userId: message.userId,
+ agentId: message.agentId,
+ roomId: message.roomId,
+ createdAt: Date.now(),
+ content: {
+ text: summary
+ }
+ });
+ agentSummaryMemory = await runtime.knowledgeManager.getMemoryById(agentId);
+ }
+ if (!agentSummaryMemory) {
+ elizaLogger.error("failed to initialise agent's summary memory");
+ return null;
+ }
+ return agentSummaryMemory.content.text;
+ },
+};
+
+// Module exports
+export { asteraiProvider };
diff --git a/packages/plugin-coinprice/tsconfig.json b/packages/plugin-asterai/tsconfig.json
similarity index 100%
rename from packages/plugin-coinprice/tsconfig.json
rename to packages/plugin-asterai/tsconfig.json
diff --git a/packages/plugin-asterai/tsup.config.ts b/packages/plugin-asterai/tsup.config.ts
new file mode 100644
index 00000000000..7f072ccb784
--- /dev/null
+++ b/packages/plugin-asterai/tsup.config.ts
@@ -0,0 +1,35 @@
+import { defineConfig } from "tsup";
+
+export default defineConfig({
+ entry: ["src/index.ts"],
+ outDir: "dist",
+ sourcemap: true,
+ clean: true,
+ format: ["esm"], // Ensure you're targeting CommonJS
+ loader: {
+ ".cdc": "text",
+ },
+ external: [
+ "dotenv", // Externalize dotenv to prevent bundling
+ "fs", // Externalize fs to use Node.js built-in module
+ "path", // Externalize other built-ins if necessary
+ "@reflink/reflink",
+ "@node-llama-cpp",
+ "https",
+ "http",
+ "agentkeepalive",
+ "safe-buffer",
+ "base-x",
+ "bs58",
+ "borsh",
+ "stream",
+ "buffer",
+ "querystring",
+ "amqplib",
+ // Add other modules you want to externalize
+ "@onflow/fcl",
+ "@onflow/types",
+ "sha3",
+ "elliptic",
+ ],
+});
diff --git a/packages/plugin-autonome/.npmignore b/packages/plugin-autonome/.npmignore
new file mode 100644
index 00000000000..078562eceab
--- /dev/null
+++ b/packages/plugin-autonome/.npmignore
@@ -0,0 +1,6 @@
+*
+
+!dist/**
+!package.json
+!readme.md
+!tsup.config.ts
\ No newline at end of file
diff --git a/packages/plugin-autonome/eslint.config.mjs b/packages/plugin-autonome/eslint.config.mjs
new file mode 100644
index 00000000000..92fe5bbebef
--- /dev/null
+++ b/packages/plugin-autonome/eslint.config.mjs
@@ -0,0 +1,3 @@
+import eslintGlobalConfig from "../../eslint.config.mjs";
+
+export default [...eslintGlobalConfig];
diff --git a/packages/plugin-autonome/package.json b/packages/plugin-autonome/package.json
new file mode 100644
index 00000000000..97e01cf663c
--- /dev/null
+++ b/packages/plugin-autonome/package.json
@@ -0,0 +1,24 @@
+{
+ "name": "@elizaos/plugin-autonome",
+ "version": "0.1.8+build.1",
+ "main": "dist/index.js",
+ "type": "module",
+ "types": "dist/index.d.ts",
+ "dependencies": {
+ "@coral-xyz/anchor": "0.30.1",
+ "@elizaos/core": "workspace:*",
+ "@elizaos/plugin-tee": "workspace:*",
+ "@elizaos/plugin-trustdb": "workspace:*",
+ "axios": "^1.7.9"
+ },
+ "scripts": {
+ "build": "tsup --format esm --dts",
+ "dev": "tsup --format esm --dts --watch",
+ "lint": "eslint --fix --cache .",
+ "test": "vitest run"
+ },
+ "peerDependencies": {
+ "form-data": "4.0.1",
+ "whatwg-url": "7.1.0"
+ }
+}
diff --git a/packages/plugin-autonome/src/actions/launchAgent.ts b/packages/plugin-autonome/src/actions/launchAgent.ts
new file mode 100644
index 00000000000..f53eaddc5f5
--- /dev/null
+++ b/packages/plugin-autonome/src/actions/launchAgent.ts
@@ -0,0 +1,174 @@
+import axios from "axios";
+import {
+ ActionExample,
+ composeContext,
+ Content,
+ elizaLogger,
+ generateObjectDeprecated,
+ HandlerCallback,
+ IAgentRuntime,
+ Memory,
+ ModelClass,
+ State,
+ type Action,
+} from "@elizaos/core";
+
+export interface LaunchAgentContent extends Content {
+ name: string;
+ config: string;
+}
+
+function isLaunchAgentContent(content: any): content is LaunchAgentContent {
+ elizaLogger.log("Content for launchAgent", content);
+ return typeof content.name === "string" && typeof content.config === "string";
+}
+
+const launchTemplate = `Respond with a JSON markdown block containing only the extracted values. Use null for any values that cannot be determined.
+
+Example response:
+\`\`\`json
+{
+ "name": "xiaohuo",
+}
+\`\`\`
+
+{{recentMessages}}
+
+Given the recent messages, extract the following information about the requested agent launch:
+- Agent name
+- Character json config
+`;
+
+export default {
+ name: "LAUNCH_AGENT",
+ similes: ["CREATE_AGENT", "DEPLOY_AGENT", "DEPLOY_ELIZA", "DEPLOY_BOT"],
+ validate: async (_runtime: IAgentRuntime, _message: Memory) => {
+ return true;
+ },
+ description: "Launch an Eliza agent",
+ handler: async (
+ runtime: IAgentRuntime,
+ message: Memory,
+ state: State,
+ _options: { [key: string]: unknown },
+ callback?: HandlerCallback
+ ): Promise => {
+ elizaLogger.log("Starting LAUNCH_AGENT handler...");
+ // Initialize or update state
+ if (!state) {
+ state = (await runtime.composeState(message)) as State;
+ } else {
+ state = await runtime.updateRecentMessageState(state);
+ }
+
+ // Compose launch context
+ const launchContext = composeContext({
+ state,
+ template: launchTemplate,
+ });
+
+ // Generate launch content
+ const content = await generateObjectDeprecated({
+ runtime,
+ context: launchContext,
+ modelClass: ModelClass.LARGE,
+ });
+
+ // Validate launch content
+ if (!isLaunchAgentContent(content)) {
+ elizaLogger.error("Invalid launch content", content);
+ if (callback) {
+ callback({
+ text: "Unable to process launch agent request. Invalid content provided.",
+ content: { error: "Invalid launch agent content" },
+ });
+ }
+ return false;
+ }
+
+ const autonomeJwt = runtime.getSetting("AUTONOME_JWT_TOKEN");
+ const autonomeRpc = runtime.getSetting("AUTONOME_RPC");
+
+ const requestBody = {
+ name: content.name,
+ config: content.config,
+ creationMethod: 2,
+ envList: {},
+ templateId: "Eliza",
+ };
+
+ const sendPostRequest = async () => {
+ try {
+ const response = await axios.post(autonomeRpc, requestBody, {
+ headers: {
+ Authorization: `Bearer ${autonomeJwt}`,
+ "Content-Type": "application/json",
+ },
+ });
+ return response;
+ } catch (error) {
+ console.error("Error making RPC call:", error);
+ }
+ };
+
+ try {
+ const resp = await sendPostRequest();
+ if (resp && resp.data && resp.data.app && resp.data.app.id) {
+ elizaLogger.log(
+ "Launching successful, please find your agent on"
+ );
+ elizaLogger.log(
+ "https://dev.autonome.fun/autonome/" +
+ resp.data.app.id +
+ "/details"
+ );
+ }
+ if (callback) {
+ callback({
+ text: `Successfully launch agent ${content.name}`,
+ content: {
+ success: true,
+ appId:
+ "https://dev.autonome.fun/autonome/" +
+ resp.data.app.id +
+ "/details",
+ },
+ });
+ }
+ return true;
+ } catch (error) {
+ if (callback) {
+ elizaLogger.error("Error during launching agent");
+ elizaLogger.error(error);
+ callback({
+ text: `Error launching agent: ${error.message}`,
+ content: { error: error.message },
+ });
+ }
+ }
+ },
+ examples: [
+ [
+ {
+ user: "{{user1}}",
+ content: {
+ text: "Launch an agent, name is xiaohuo",
+ },
+ },
+ {
+ user: "{{user2}}",
+ content: {
+ text: "I'll launch the agent now...",
+ action: "LAUNCH_AGENT",
+ },
+ },
+ {
+ user: "{{user2}}",
+ content: {
+ text: "Successfully launch agent, id is ba2e8369-e256-4a0d-9f90-9c64e306dc9f",
+ },
+ },
+ ],
+ ] as ActionExample[][],
+} satisfies Action;
+
diff --git a/packages/plugin-autonome/src/index.ts b/packages/plugin-autonome/src/index.ts
new file mode 100644
index 00000000000..bbf49808982
--- /dev/null
+++ b/packages/plugin-autonome/src/index.ts
@@ -0,0 +1,12 @@
+import { Plugin } from "@elizaos/core";
+import launchAgent from "./actions/launchAgent";
+
+export const autonomePlugin: Plugin = {
+ name: "autonome",
+ description: "Autonome Plugin for Eliza",
+ actions: [launchAgent],
+ evaluators: [],
+ providers: [],
+};
+
+export default autonomePlugin;
diff --git a/packages/plugin-autonome/tsconfig.json b/packages/plugin-autonome/tsconfig.json
new file mode 100644
index 00000000000..73993deaaf7
--- /dev/null
+++ b/packages/plugin-autonome/tsconfig.json
@@ -0,0 +1,10 @@
+{
+ "extends": "../core/tsconfig.json",
+ "compilerOptions": {
+ "outDir": "dist",
+ "rootDir": "src"
+ },
+ "include": [
+ "src/**/*.ts"
+ ]
+}
\ No newline at end of file
diff --git a/packages/plugin-autonome/tsup.config.ts b/packages/plugin-autonome/tsup.config.ts
new file mode 100644
index 00000000000..a47c9eb64b0
--- /dev/null
+++ b/packages/plugin-autonome/tsup.config.ts
@@ -0,0 +1,19 @@
+import { defineConfig } from "tsup";
+
+export default defineConfig({
+ entry: ["src/index.ts"],
+ outDir: "dist",
+ sourcemap: true,
+ clean: true,
+ format: ["esm"], // Ensure you're targeting CommonJS
+ external: [
+ "dotenv", // Externalize dotenv to prevent bundling
+ "fs", // Externalize fs to use Node.js built-in module
+ "path", // Externalize other built-ins if necessary
+ "@reflink/reflink",
+ "@node-llama-cpp",
+ "https",
+ "http",
+ "agentkeepalive",
+ ],
+});
diff --git a/packages/plugin-avail/src/actions/submitData.ts b/packages/plugin-avail/src/actions/submitData.ts
index 978ae5a3dcb..e90622105ad 100644
--- a/packages/plugin-avail/src/actions/submitData.ts
+++ b/packages/plugin-avail/src/actions/submitData.ts
@@ -9,16 +9,13 @@ import {
type Action,
elizaLogger,
composeContext,
- generateObject,
generateObjectDeprecated,
} from "@elizaos/core";
import { validateAvailConfig } from "../environment";
import {
- getDecimals,
+ //getDecimals,
initialize,
- formatNumberToBalance,
getKeyringFromSeed,
- isValidAddress,
} from "avail-js-sdk";
import { ISubmittableResult } from "@polkadot/types/types/extrinsic";
import { H256 } from "@polkadot/types/interfaces/runtime";
@@ -67,7 +64,7 @@ export default {
"SUBMIT_DATA_ON_AVAIL_NETWORK",
"SUBMIT_DATA_TO_AVAIL_NETWORK",
],
- validate: async (runtime: IAgentRuntime, message: Memory) => {
+ validate: async (runtime: IAgentRuntime, _message: Memory) => {
await validateAvailConfig(runtime);
return true;
},
@@ -116,14 +113,14 @@ export default {
if (content.data != null) {
try {
const SEED = runtime.getSetting("AVAIL_SEED")!;
- const ACCOUNT = runtime.getSetting("AVAIL_ADDRESS")!;
+ //const ACCOUNT = runtime.getSetting("AVAIL_ADDRESS")!;
const ENDPOINT = runtime.getSetting("AVAIL_RPC_URL");
const APP_ID = runtime.getSetting("AVAIL_APP_ID");
const api = await initialize(ENDPOINT);
const keyring = getKeyringFromSeed(SEED);
const options = { app_id: APP_ID, nonce: -1 };
- const decimals = getDecimals(api);
+ //const decimals = getDecimals(api);
const data = content.data;
const submitDataInfo = await api.tx.dataAvailability
diff --git a/packages/plugin-avail/src/actions/transfer.ts b/packages/plugin-avail/src/actions/transfer.ts
index df3b04cbe8f..8745048a964 100644
--- a/packages/plugin-avail/src/actions/transfer.ts
+++ b/packages/plugin-avail/src/actions/transfer.ts
@@ -9,7 +9,6 @@ import {
type Action,
elizaLogger,
composeContext,
- generateObject,
generateObjectDeprecated,
} from "@elizaos/core";
import { validateAvailConfig } from "../environment";
@@ -77,7 +76,7 @@ export default {
"SEND_AVAIL_TOKEN_ON_AVAIL_DA",
"PAY_ON_AVAIL",
],
- validate: async (runtime: IAgentRuntime, message: Memory) => {
+ validate: async (runtime: IAgentRuntime, _message: Memory) => {
await validateAvailConfig(runtime);
return true;
},
@@ -128,7 +127,7 @@ export default {
if (content.amount != null && content.recipient != null) {
try {
const SEED = runtime.getSetting("AVAIL_SEED")!;
- const PUBLIC_KEY = runtime.getSetting("AVAIL_ADDRESS")!;
+ //const PUBLIC_KEY = runtime.getSetting("AVAIL_ADDRESS")!;
const ENDPOINT = runtime.getSetting("AVAIL_RPC_URL");
const api = await initialize(ENDPOINT);
diff --git a/packages/plugin-avalanche/package.json b/packages/plugin-avalanche/package.json
index 9a10cc11698..8d06db57037 100644
--- a/packages/plugin-avalanche/package.json
+++ b/packages/plugin-avalanche/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/plugin-avalanche",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
diff --git a/packages/plugin-binance/package.json b/packages/plugin-binance/package.json
index 1f8bbeee11d..1e90cf85c38 100644
--- a/packages/plugin-binance/package.json
+++ b/packages/plugin-binance/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/plugin-binance",
- "version": "0.1.0",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
@@ -32,4 +32,4 @@
"dev": "tsup --format esm --dts --watch",
"lint": "eslint --fix --cache ."
}
-}
\ No newline at end of file
+}
diff --git a/packages/plugin-bootstrap/package.json b/packages/plugin-bootstrap/package.json
index ec3ba9749b8..8d4adf2ca66 100644
--- a/packages/plugin-bootstrap/package.json
+++ b/packages/plugin-bootstrap/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/plugin-bootstrap",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
diff --git a/packages/plugin-coinbase/__tests__/commerce.test.ts b/packages/plugin-coinbase/__tests__/commerce.test.ts
new file mode 100644
index 00000000000..3556dc62ea3
--- /dev/null
+++ b/packages/plugin-coinbase/__tests__/commerce.test.ts
@@ -0,0 +1,116 @@
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { coinbaseCommercePlugin, createCharge } from '../src/plugins/commerce';
+import { IAgentRuntime, Memory, State } from '@elizaos/core';
+
+// Mock fetch
+global.fetch = vi.fn();
+
+// Mock runtime
+const mockRuntime = {
+ getSetting: vi.fn().mockReturnValue('test-api-key'),
+ getProvider: vi.fn().mockReturnValue({ apiKey: 'test-api-key' }),
+ character: {
+ name: 'test-character'
+ }
+};
+
+describe('Coinbase Commerce Plugin', () => {
+ beforeEach(() => {
+ vi.clearAllMocks();
+ });
+
+ describe('createCharge', () => {
+ it('should create a charge successfully', async () => {
+ const mockResponse = {
+ data: {
+ id: 'test-charge-id',
+ name: 'Test Charge',
+ description: 'Test Description',
+ pricing_type: 'fixed_price',
+ local_price: {
+ amount: '100',
+ currency: 'USD'
+ }
+ }
+ };
+
+ (global.fetch as any).mockResolvedValueOnce({
+ ok: true,
+ json: () => Promise.resolve(mockResponse)
+ });
+
+ const params = {
+ name: 'Test Charge',
+ description: 'Test Description',
+ pricing_type: 'fixed_price',
+ local_price: {
+ amount: '100',
+ currency: 'USD'
+ }
+ };
+
+ const result = await createCharge('test-api-key', params);
+ expect(result).toEqual(mockResponse.data);
+ expect(global.fetch).toHaveBeenCalledWith(
+ 'https://api.commerce.coinbase.com/charges',
+ {
+ method: 'POST',
+ headers: {
+ 'Content-Type': 'application/json',
+ 'X-CC-Api-Key': 'test-api-key'
+ },
+ body: JSON.stringify(params)
+ }
+ );
+ });
+
+ it('should handle errors when creating charge', async () => {
+ (global.fetch as any).mockResolvedValueOnce({
+ ok: false,
+ statusText: 'Bad Request'
+ });
+
+ const params = {
+ name: 'Test Charge',
+ description: 'Test Description',
+ pricing_type: 'fixed_price',
+ local_price: {
+ amount: '100',
+ currency: 'USD'
+ }
+ };
+
+ await expect(createCharge('test-api-key', params))
+ .rejects
+ .toThrow('Failed to create charge: Bad Request');
+ });
+ });
+
+ describe('coinbaseCommercePlugin', () => {
+ it('should have correct plugin properties', () => {
+ expect(coinbaseCommercePlugin.name).toBe('coinbaseCommerce');
+ expect(coinbaseCommercePlugin.actions).toBeDefined();
+ expect(Array.isArray(coinbaseCommercePlugin.actions)).toBe(true);
+ });
+
+ it('should validate plugin actions', async () => {
+ const mockMessage: Memory = {
+ id: '1',
+ user: 'test-user',
+ content: { text: 'test message' },
+ timestamp: new Date(),
+ type: 'text'
+ };
+
+ const createChargeAction = coinbaseCommercePlugin.actions.find(
+ action => action.name === 'CREATE_CHARGE'
+ );
+
+ expect(createChargeAction).toBeDefined();
+ if (createChargeAction) {
+ const result = await createChargeAction.validate(mockRuntime as any, mockMessage);
+ expect(result).toBe(true);
+ }
+ });
+ });
+});
diff --git a/packages/plugin-coinbase/__tests__/utils.test.ts b/packages/plugin-coinbase/__tests__/utils.test.ts
new file mode 100644
index 00000000000..fb5b36f019b
--- /dev/null
+++ b/packages/plugin-coinbase/__tests__/utils.test.ts
@@ -0,0 +1,64 @@
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { getWalletDetails } from '../src/utils';
+import { Coinbase, Wallet } from '@coinbase/coinbase-sdk';
+
+vi.mock('@coinbase/coinbase-sdk');
+
+// Mock the runtime
+const mockRuntime = {
+ getSetting: vi.fn()
+ .mockReturnValueOnce('test-seed') // COINBASE_GENERATED_WALLET_HEX_SEED
+ .mockReturnValueOnce('test-wallet-id'), // COINBASE_GENERATED_WALLET_ID
+ getProvider: vi.fn().mockReturnValue({ apiKey: 'test-api-key' }),
+ character: {
+ name: 'test-character'
+ }
+};
+
+// Mock Wallet class
+const mockWallet = {
+ getDefaultAddress: vi.fn().mockResolvedValue('0x123'),
+ getNetworkId: vi.fn().mockReturnValue('eth-mainnet'),
+ listBalances: vi.fn().mockResolvedValue([
+ ['ETH', { toString: () => '1.0' }]
+ ]),
+ getTransactions: vi.fn().mockResolvedValue([]),
+ export: vi.fn().mockReturnValue({
+ seed: 'test-seed',
+ walletId: 'test-wallet-id'
+ })
+};
+
+describe('Utils', () => {
+ describe('getWalletDetails', () => {
+ beforeEach(() => {
+ vi.clearAllMocks();
+ (Coinbase as any).networks = {
+ EthereumMainnet: 'eth-mainnet'
+ };
+ (Wallet as any).import = vi.fn().mockResolvedValue(mockWallet);
+ });
+
+ it('should fetch wallet details successfully', async () => {
+ const result = await getWalletDetails(mockRuntime as any);
+
+ expect(result).toEqual({
+ balances: [{ asset: 'ETH', amount: '1.0' }],
+ transactions: []
+ });
+
+ expect(Wallet.import).toHaveBeenCalledWith({
+ seed: 'test-seed',
+ walletId: 'test-wallet-id'
+ });
+ });
+
+ it('should handle errors when fetching wallet details', async () => {
+ (Wallet as any).import = vi.fn().mockRejectedValue(new Error('Unable to retrieve wallet details.'));
+
+ await expect(getWalletDetails(mockRuntime as any))
+ .rejects
+ .toThrow('Unable to retrieve wallet details.');
+ });
+ });
+});
diff --git a/packages/plugin-coinbase/package.json b/packages/plugin-coinbase/package.json
index 73ff823b52a..b78d9cb84f1 100644
--- a/packages/plugin-coinbase/package.json
+++ b/packages/plugin-coinbase/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/plugin-coinbase",
- "version": "0.1.7",
+ "version": "0.1.8+build.1",
"type": "module",
"main": "dist/index.js",
"module": "dist/index.js",
@@ -28,11 +28,14 @@
},
"devDependencies": {
"tsup": "8.3.5",
- "@types/node": "^20.0.0"
+ "@types/node": "^20.0.0",
+ "vitest": "^1.0.0"
},
"scripts": {
"build": "tsup --format esm --dts",
"dev": "tsup --format esm --dts --watch",
- "lint": "eslint --fix --cache ."
+ "lint": "eslint --fix --cache .",
+ "test": "vitest run",
+ "test:watch": "vitest"
}
}
diff --git a/packages/plugin-coinbase/src/plugins/commerce.ts b/packages/plugin-coinbase/src/plugins/commerce.ts
index 7dacdc0fcb6..ca249b53d68 100644
--- a/packages/plugin-coinbase/src/plugins/commerce.ts
+++ b/packages/plugin-coinbase/src/plugins/commerce.ts
@@ -82,7 +82,7 @@ export async function getAllCharges(apiKey: string) {
// Function to fetch details of a specific charge
export async function getChargeDetails(apiKey: string, chargeId: string) {
elizaLogger.debug("Starting getChargeDetails function");
- const getUrl = `${url}${chargeId}`;
+ const getUrl = `${url}/${chargeId}`;
try {
const response = await fetch(getUrl, {
@@ -204,8 +204,8 @@ export const createCoinbaseChargeAction: Action = {
text: `Charge created successfully: ${chargeResponse.hosted_url}`,
attachments: [
{
- id: crypto.randomUUID(),
- url: chargeResponse.id,
+ id: chargeResponse.id,
+ url: chargeResponse.hosted_url,
title: "Coinbase Commerce Charge",
description: `Charge ID: ${chargeResponse.id}`,
text: `Pay here: ${chargeResponse.hosted_url}`,
@@ -351,6 +351,7 @@ export const getAllChargesAction: Action = {
callback(
{
text: `Successfully fetched all charges. Total charges: ${charges.length}`,
+ attachments: charges,
},
[]
);
@@ -439,17 +440,20 @@ export const getChargeDetailsAction: Action = {
elizaLogger.info("Fetched charge details:", chargeDetails);
+ const chargeData = chargeDetails.data;
+
callback(
{
text: `Successfully fetched charge details for ID: ${charge.id}`,
attachments: [
{
- id: crypto.randomUUID(),
- url: chargeDetails.hosted_url,
+ id: chargeData.id,
+ url: chargeData.hosted_url,
title: `Charge Details for ${charge.id}`,
- description: `Details: ${JSON.stringify(chargeDetails, null, 2)}`,
source: "coinbase",
- text: "",
+ description: JSON.stringify(chargeDetails, null, 2),
+ text: `Pay here: ${chargeData.hosted_url}`,
+ contentType: "application/json",
},
],
},
diff --git a/packages/plugin-coingecko/README.md b/packages/plugin-coingecko/README.md
index ded984b61c4..fcb79d8a558 100644
--- a/packages/plugin-coingecko/README.md
+++ b/packages/plugin-coingecko/README.md
@@ -4,7 +4,9 @@ A plugin for fetching cryptocurrency price data from the CoinGecko API.
## Overview
-The Plugin CoinGecko provides a simple interface to get real-time cryptocurrency prices. It integrates with CoinGecko's API to fetch current prices for various cryptocurrencies in different fiat currencies.
+The Plugin CoinGecko provides a simple interface to get real-time cryptocurrency data. It integrates with CoinGecko's API to fetch current prices, market data, trending coins, and top gainers/losers for various cryptocurrencies in different fiat currencies.
+
+This plugin uses the [CoinGecko Pro API](https://docs.coingecko.com/reference/introduction). Please refer to their documentation for detailed information about rate limits, available endpoints, and response formats.
## Installation
@@ -18,7 +20,8 @@ Set up your environment with the required CoinGecko API key:
| Variable Name | Description |
| ------------------- | ---------------------- |
-| `COINGECKO_API_KEY` | Your CoinGecko API key |
+| `COINGECKO_API_KEY` | Your CoinGecko Pro API key |
+| `COINGECKO_PRO_API_KEY` | Your CoinGecko Pro API key |
## Usage
@@ -27,23 +30,69 @@ import { coingeckoPlugin } from "@elizaos/plugin-coingecko";
// Initialize the plugin
const plugin = coingeckoPlugin;
-
-// The plugin provides the GET_PRICE action which can be used to fetch prices
-// Supported coins: BTC, ETH, USDC, and more
```
## Actions
### GET_PRICE
-Fetches the current price of a cryptocurrency.
+Fetches the current price and market data for one or more cryptocurrencies.
-Examples:
+Features:
+- Multiple currency support (e.g., USD, EUR, JPY)
+- Optional market cap data
+- Optional 24h volume data
+- Optional 24h price change data
+- Optional last update timestamp
+Examples:
- "What's the current price of Bitcoin?"
-- "Check ETH price in EUR"
-- "What's USDC worth?"
+- "Check ETH price in EUR with market cap"
+- "Show me BTC and ETH prices in USD and EUR"
+- "What's USDC worth with 24h volume and price change?"
+
+### GET_TRENDING
+
+Fetches the current trending cryptocurrencies on CoinGecko.
-## License
+Features:
+- Includes trending coins with market data
+- Optional NFT inclusion
+- Optional category inclusion
-MIT
+Examples:
+- "What's trending in crypto?"
+- "Show me trending coins only"
+- "What are the hot cryptocurrencies right now?"
+
+### GET_TOP_GAINERS_LOSERS
+
+Fetches the top gaining and losing cryptocurrencies by price change.
+
+Features:
+- Customizable time range (1h, 24h, 7d, 14d, 30d, 60d, 1y)
+- Configurable number of top coins to include
+- Multiple currency support
+- Market cap ranking included
+
+Examples:
+- "Show me the biggest gainers and losers today"
+- "What are the top movers in EUR for the past week?"
+- "Show me monthly performance of top 100 coins"
+
+## Response Format
+
+All actions return structured data including:
+- Formatted text for easy reading
+- Raw data for programmatic use
+- Request parameters used
+- Error details when applicable
+
+## Error Handling
+
+The plugin handles various error scenarios:
+- Rate limiting
+- API key validation
+- Invalid parameters
+- Network issues
+- Pro plan requirements
\ No newline at end of file
diff --git a/packages/plugin-coingecko/package.json b/packages/plugin-coingecko/package.json
index fb1fe8b8307..3ace49624ed 100644
--- a/packages/plugin-coingecko/package.json
+++ b/packages/plugin-coingecko/package.json
@@ -1,6 +1,6 @@
{
"name": "@elizaos/plugin-coingecko",
- "version": "0.1.7-alpha.2",
+ "version": "0.1.8+build.1",
"main": "dist/index.js",
"type": "module",
"types": "dist/index.d.ts",
@@ -14,4 +14,4 @@
"dev": "tsup --format esm --dts --watch",
"test": "vitest run"
}
-}
\ No newline at end of file
+}
diff --git a/packages/plugin-coingecko/src/actions/getMarkets.ts b/packages/plugin-coingecko/src/actions/getMarkets.ts
new file mode 100644
index 00000000000..5a32ad903ce
--- /dev/null
+++ b/packages/plugin-coingecko/src/actions/getMarkets.ts
@@ -0,0 +1,308 @@
+import {
+ ActionExample,
+ composeContext,
+ Content,
+ elizaLogger,
+ generateObject,
+ HandlerCallback,
+ IAgentRuntime,
+ Memory,
+ ModelClass,
+ State,
+ type Action
+} from "@elizaos/core";
+import axios from "axios";
+import { z } from "zod";
+import { getApiConfig, validateCoingeckoConfig } from "../environment";
+import { getCategoriesData } from '../providers/categoriesProvider';
+import { getMarketsTemplate } from "../templates/markets";
+
+interface CategoryItem {
+ category_id: string;
+ name: string;
+}
+
+export function formatCategory(category: string | undefined, categories: CategoryItem[]): string | undefined {
+ if (!category) return undefined;
+
+ const normalizedInput = category.toLowerCase().trim();
+
+ // First try to find exact match by category_id
+ const exactMatch = categories.find(c => c.category_id === normalizedInput);
+ if (exactMatch) {
+ return exactMatch.category_id;
+ }
+
+ // Then try to find match by name
+ const nameMatch = categories.find(c =>
+ c.name.toLowerCase() === normalizedInput ||
+ c.name.toLowerCase().replace(/[^a-z0-9]+/g, '-') === normalizedInput
+ );
+ if (nameMatch) {
+ return nameMatch.category_id;
+ }
+
+ // Try to find partial matches
+ const partialMatch = categories.find(c =>
+ c.name.toLowerCase().includes(normalizedInput) ||
+ c.category_id.includes(normalizedInput)
+ );
+ if (partialMatch) {
+ return partialMatch.category_id;
+ }
+
+ return undefined;
+}
+
+/**
+ * Interface for CoinGecko /coins/markets endpoint response
+ * @see https://docs.coingecko.com/reference/coins-markets
+ */
+export interface CoinMarketData {
+ id: string;
+ symbol: string;
+ name: string;
+ image: string;
+ current_price: number;
+ market_cap: number;
+ market_cap_rank: number;
+ fully_diluted_valuation: number;
+ total_volume: number;
+ high_24h: number;
+ low_24h: number;
+ price_change_24h: number;
+ price_change_percentage_24h: number;
+ market_cap_change_24h: number;
+ market_cap_change_percentage_24h: number;
+ circulating_supply: number;
+ total_supply: number;
+ max_supply: number;
+ ath: number;
+ ath_change_percentage: number;
+ ath_date: string;
+ atl: number;
+ atl_change_percentage: number;
+ atl_date: string;
+ last_updated: string;
+}
+
+export const GetMarketsSchema = z.object({
+ vs_currency: z.string().default('usd'),
+ category: z.string().optional(),
+ order: z.enum(['market_cap_desc', 'market_cap_asc', 'volume_desc', 'volume_asc']).default('market_cap_desc'),
+ per_page: z.number().min(1).max(250).default(20),
+ page: z.number().min(1).default(1),
+ sparkline: z.boolean().default(false)
+});
+
+export type GetMarketsContent = z.infer & Content;
+
+export const isGetMarketsContent = (obj: any): obj is GetMarketsContent => {
+ return GetMarketsSchema.safeParse(obj).success;
+};
+
+export default {
+ name: "GET_MARKETS",
+ similes: [
+ "MARKET_OVERVIEW",
+ "TOP_RANKINGS",
+ "MARKET_LEADERBOARD",
+ "CRYPTO_RANKINGS",
+ "BEST_PERFORMING_COINS",
+ "TOP_MARKET_CAPS"
+ ],
+ validate: async (runtime: IAgentRuntime, message: Memory) => {
+ await validateCoingeckoConfig(runtime);
+ return true;
+ },
+ // Comprehensive endpoint for market rankings, supports up to 250 coins per request
+ description: "Get ranked list of top cryptocurrencies sorted by market metrics (without specifying coins)",
+ handler: async (
+ runtime: IAgentRuntime,
+ message: Memory,
+ state: State,
+ _options: { [key: string]: unknown },
+ callback?: HandlerCallback
+ ): Promise => {
+ elizaLogger.log("Starting CoinGecko GET_MARKETS handler...");
+
+ if (!state) {
+ state = (await runtime.composeState(message)) as State;
+ } else {
+ state = await runtime.updateRecentMessageState(state);
+ }
+
+ try {
+ const config = await validateCoingeckoConfig(runtime);
+ const { baseUrl, apiKey } = getApiConfig(config);
+
+ // Get categories through the provider
+ const categories = await getCategoriesData(runtime);
+
+ // Compose markets context with categories
+ const marketsContext = composeContext({
+ state,
+ template: getMarketsTemplate.replace('{{categories}}',
+ categories.map(c => `- ${c.name} (ID: ${c.category_id})`).join('\n')
+ ),
+ });
+
+ const result = await generateObject({
+ runtime,
+ context: marketsContext,
+ modelClass: ModelClass.SMALL,
+ schema: GetMarketsSchema
+ });
+
+ if (!isGetMarketsContent(result.object)) {
+ elizaLogger.error("Invalid market data format received");
+ return false;
+ }
+
+ const content = result.object;
+ elizaLogger.log("Content from template:", content);
+
+ // If template returns null, this is not a markets request
+ if (!content) {
+ return false;
+ }
+
+ const formattedCategory = formatCategory(content.category, categories);
+ if (content.category && !formattedCategory) {
+ throw new Error(`Invalid category: ${content.category}. Please choose from the available categories.`);
+ }
+
+ elizaLogger.log("Making API request with params:", {
+ url: `${baseUrl}/coins/markets`,
+ category: formattedCategory,
+ vs_currency: content.vs_currency,
+ order: content.order,
+ per_page: content.per_page,
+ page: content.page
+ });
+
+ const response = await axios.get(
+ `${baseUrl}/coins/markets`,
+ {
+ headers: {
+ 'accept': 'application/json',
+ 'x-cg-pro-api-key': apiKey
+ },
+ params: {
+ vs_currency: content.vs_currency,
+ category: formattedCategory,
+ order: content.order,
+ per_page: content.per_page,
+ page: content.page,
+ sparkline: content.sparkline
+ }
+ }
+ );
+
+ if (!response.data?.length) {
+ throw new Error("No market data received from CoinGecko API");
+ }
+
+ const formattedData = response.data.map(coin => ({
+ name: coin.name,
+ symbol: coin.symbol.toUpperCase(),
+ marketCapRank: coin.market_cap_rank,
+ currentPrice: coin.current_price,
+ priceChange24h: coin.price_change_24h,
+ priceChangePercentage24h: coin.price_change_percentage_24h,
+ marketCap: coin.market_cap,
+ volume24h: coin.total_volume,
+ high24h: coin.high_24h,
+ low24h: coin.low_24h,
+ circulatingSupply: coin.circulating_supply,
+ totalSupply: coin.total_supply,
+ maxSupply: coin.max_supply,
+ lastUpdated: coin.last_updated
+ }));
+
+ const categoryDisplay = content.category ?
+ `${categories.find(c => c.category_id === formattedCategory)?.name.toUpperCase() || content.category.toUpperCase()} ` : '';
+
+ const responseText = [
+ `Top ${formattedData.length} ${categoryDisplay}Cryptocurrencies by ${content.order === 'volume_desc' || content.order === 'volume_asc' ? 'Volume' : 'Market Cap'}:`,
+ ...formattedData.map((coin, index) =>
+ `${index + 1}. ${coin.name} (${coin.symbol})` +
+ ` | $${coin.currentPrice.toLocaleString()}` +
+ ` | ${coin.priceChangePercentage24h.toFixed(2)}%` +
+ ` | MCap: $${(coin.marketCap / 1e9).toFixed(2)}B`
+ )
+ ].join('\n');
+
+ elizaLogger.success("Market data retrieved successfully!");
+
+ if (callback) {
+ callback({
+ text: responseText,
+ content: {
+ markets: formattedData,
+ params: {
+ vs_currency: content.vs_currency,
+ category: content.category,
+ order: content.order,
+ per_page: content.per_page,
+ page: content.page
+ },
+ timestamp: new Date().toISOString()
+ }
+ });
+ }
+
+ return true;
+ } catch (error) {
+ elizaLogger.error("Error in GET_MARKETS handler:", error);
+
+ let errorMessage;
+ if (error.response?.status === 429) {
+ errorMessage = "Rate limit exceeded. Please try again later.";
+ } else if (error.response?.status === 403) {
+ errorMessage = "This endpoint requires a CoinGecko Pro API key. Please upgrade your plan to access this data.";
+ } else if (error.response?.status === 400) {
+ errorMessage = "Invalid request parameters. Please check your input.";
+ } else {
+ errorMessage = `Error fetching market data: ${error.message}`;
+ }
+
+ if (callback) {
+ callback({
+ text: errorMessage,
+ error: {
+ message: error.message,
+ statusCode: error.response?.status,
+ params: error.config?.params,
+ requiresProPlan: error.response?.status === 403
+ }
+ });
+ }
+ return false;
+ }
+ },
+
+ examples: [
+ [
+ {
+ user: "{{user1}}",
+ content: {
+ text: "Show me the top cryptocurrencies by market cap",
+ },
+ },
+ {
+ user: "{{agent}}",
+ content: {
+ text: "I'll fetch the current market data for top cryptocurrencies.",
+ action: "GET_MARKETS",
+ },
+ },
+ {
+ user: "{{agent}}",
+ content: {
+ text: "Here are the top cryptocurrencies:\n1. Bitcoin (BTC) | $45,000 | +2.5% | MCap: $870.5B\n{{dynamic}}",
+ },
+ },
+ ],
+ ] as ActionExample[][],
+} as Action;
\ No newline at end of file
diff --git a/packages/plugin-coingecko/src/actions/getPrice.ts b/packages/plugin-coingecko/src/actions/getPrice.ts
index deb923b2e91..7e47db4f3f2 100644
--- a/packages/plugin-coingecko/src/actions/getPrice.ts
+++ b/packages/plugin-coingecko/src/actions/getPrice.ts
@@ -3,7 +3,7 @@ import {
composeContext,
Content,
elizaLogger,
- generateObjectDeprecated,
+ generateObject,
HandlerCallback,
IAgentRuntime,
Memory,
@@ -12,28 +12,65 @@ import {
type Action,
} from "@elizaos/core";
import axios from "axios";
-import { validateCoingeckoConfig } from "../environment";
+import { z } from "zod";
+import { getApiConfig, validateCoingeckoConfig } from "../environment";
+import { getCoinsData } from "../providers/coinsProvider";
import { getPriceTemplate } from "../templates/price";
-import { normalizeCoinId } from "../utils/coin";
-export interface GetPriceContent extends Content {
- coinId: string;
- currency: string;
+interface CurrencyData {
+ [key: string]: number;
+ usd?: number;
+ eur?: number;
+ usd_market_cap?: number;
+ eur_market_cap?: number;
+ usd_24h_vol?: number;
+ eur_24h_vol?: number;
+ usd_24h_change?: number;
+ eur_24h_change?: number;
+ last_updated_at?: number;
+}
+
+interface PriceResponse {
+ [coinId: string]: CurrencyData;
+}
+
+export const GetPriceSchema = z.object({
+ coinIds: z.union([z.string(), z.array(z.string())]),
+ currency: z.union([z.string(), z.array(z.string())]).default(["usd"]),
+ include_market_cap: z.boolean().default(false),
+ include_24hr_vol: z.boolean().default(false),
+ include_24hr_change: z.boolean().default(false),
+ include_last_updated_at: z.boolean().default(false)
+});
+
+export type GetPriceContent = z.infer & Content;
+
+export const isGetPriceContent = (obj: any): obj is GetPriceContent => {
+ return GetPriceSchema.safeParse(obj).success;
+};
+
+function formatCoinIds(input: string | string[]): string {
+ if (Array.isArray(input)) {
+ return input.join(',');
+ }
+ return input;
}
export default {
name: "GET_PRICE",
similes: [
- "CHECK_PRICE",
- "PRICE_CHECK",
- "GET_CRYPTO_PRICE",
- "CHECK_CRYPTO_PRICE",
+ "COIN_PRICE_CHECK",
+ "SPECIFIC_COINS_PRICE",
+ "COIN_PRICE_LOOKUP",
+ "SELECTED_COINS_PRICE",
+ "PRICE_DETAILS",
+ "COIN_PRICE_DATA"
],
validate: async (runtime: IAgentRuntime, message: Memory) => {
await validateCoingeckoConfig(runtime);
return true;
},
- description: "Get the current price of a cryptocurrency from CoinGecko",
+ description: "Get price and basic market data for one or more specific cryptocurrencies (by name/symbol)",
handler: async (
runtime: IAgentRuntime,
message: Memory,
@@ -43,7 +80,6 @@ export default {
): Promise => {
elizaLogger.log("Starting CoinGecko GET_PRICE handler...");
- // Initialize or update state
if (!state) {
state = (await runtime.composeState(message)) as State;
} else {
@@ -51,78 +87,194 @@ export default {
}
try {
- // Compose price check context
elizaLogger.log("Composing price context...");
const priceContext = composeContext({
state,
template: getPriceTemplate,
});
- elizaLogger.log("Composing content...");
- const content = (await generateObjectDeprecated({
+ elizaLogger.log("Generating content from template...");
+ const result = await generateObject({
runtime,
context: priceContext,
modelClass: ModelClass.LARGE,
- })) as unknown as GetPriceContent;
+ schema: GetPriceSchema
+ });
- // Validate content structure first
- if (!content || typeof content !== "object") {
- throw new Error("Invalid response format from model");
+ if (!isGetPriceContent(result.object)) {
+ elizaLogger.error("Invalid price request format");
+ return false;
}
- // Get and validate coin ID
- const coinId = content.coinId
- ? normalizeCoinId(content.coinId)
- : null;
- if (!coinId) {
- throw new Error(
- `Unsupported or invalid cryptocurrency: ${content.coinId}`
- );
- }
+ const content = result.object;
+ elizaLogger.log("Generated content:", content);
+
+ // Format currencies for API request
+ const currencies = Array.isArray(content.currency) ? content.currency : [content.currency];
+ const vs_currencies = currencies.join(',').toLowerCase();
- // Normalize currency
- const currency = (content.currency || "usd").toLowerCase();
+ // Format coin IDs for API request
+ const coinIds = formatCoinIds(content.coinIds);
+
+ elizaLogger.log("Formatted request parameters:", { coinIds, vs_currencies });
// Fetch price from CoinGecko
const config = await validateCoingeckoConfig(runtime);
- elizaLogger.log(`Fetching price for ${coinId} in ${currency}...`);
+ const { baseUrl, apiKey } = getApiConfig(config);
- const response = await axios.get(
- `https://api.coingecko.com/api/v3/simple/price`,
+ elizaLogger.log(`Fetching prices for ${coinIds} in ${vs_currencies}...`);
+ elizaLogger.log("API request URL:", `${baseUrl}/simple/price`);
+ elizaLogger.log("API request params:", {
+ ids: coinIds,
+ vs_currencies,
+ include_market_cap: content.include_market_cap,
+ include_24hr_vol: content.include_24hr_vol,
+ include_24hr_change: content.include_24hr_change,
+ include_last_updated_at: content.include_last_updated_at
+ });
+
+ const response = await axios.get