Skip to content

Commit

Permalink
provider name
Browse files Browse the repository at this point in the history
  • Loading branch information
lalalune committed Mar 3, 2025
1 parent 958112d commit e174c25
Show file tree
Hide file tree
Showing 16 changed files with 26 additions and 4 deletions.
8 changes: 4 additions & 4 deletions packages/agent/src/server/index.ts
Original file line number Diff line number Diff line change
Expand Up @@ -137,16 +137,16 @@ export class AgentServer {
try {
switch (route.type) {
case "GET":
this.app.get(routePath, (req, res) => route.handler(req, res));
this.app.get(routePath, (req, res) => route.handler(req, res, runtime));
break;
case "POST":
this.app.post(routePath, (req, res) => route.handler(req, res));
this.app.post(routePath, (req, res) => route.handler(req, res, runtime));
break;
case "PUT":
this.app.put(routePath, (req, res) => route.handler(req, res));
this.app.put(routePath, (req, res) => route.handler(req, res, runtime));
break;
case "DELETE":
this.app.delete(routePath, (req, res) => route.handler(req, res));
this.app.delete(routePath, (req, res) => route.handler(req, res, runtime));
break;
default:
logger.error(`Unknown route type: ${route.type} for path ${routePath}`);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -249,6 +249,7 @@ async function runActions(
}

export const dataProvider: Provider = {
name: "data",
async get(
runtime: IAgentRuntime,
message: Memory,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@ const recommendationsPrompt = `<user_recommendations_provider>
</user_recommendations_provider>` as const;

export const recommendationsProvider: Provider = {
name: "recommendations",
async get(runtime, message) {
const recommendationsManager =
runtime.getMemoryManager("recommendations")!;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -252,6 +252,7 @@ export class WalletProvider {
}

export const walletProvider: Provider = {
name: "degen-solana-wallet",
get: async (
runtime: IAgentRuntime,
_message: Memory,
Expand Down
1 change: 1 addition & 0 deletions packages/core/src/providers/facts.ts
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ function formatFacts(facts: Memory[]) {
}

const factsProvider: Provider = {
name: "facts",
get: async (runtime: IAgentRuntime, message: Memory, state?: State) => {
const recentMessagesData = state?.recentMessagesData?.slice(-10);

Expand Down
1 change: 1 addition & 0 deletions packages/core/src/providers/options.ts
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@ interface OptionObject {
}

export const optionsProvider: Provider = {
name: "options",
get: async (
runtime: IAgentRuntime,
message: Memory,
Expand Down
1 change: 1 addition & 0 deletions packages/core/src/providers/relationships.ts
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,7 @@ async function formatRelationships(runtime: IAgentRuntime, relationships: Relati
}

const relationshipsProvider: Provider = {
name: "relationships",
get: async (runtime: IAgentRuntime, message: Memory, state?: State) => {
// Get all relationships for the current user
const relationships = await getRelationships({
Expand Down
1 change: 1 addition & 0 deletions packages/core/src/providers/roles.ts
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@ import { logger } from "../logger";
import { ChannelType, type IAgentRuntime, type Memory, type Provider, type State, type UUID } from "../types";

export const roleProvider: Provider = {
name: "roles",
get: async (
runtime: IAgentRuntime,
message: Memory,
Expand Down
1 change: 1 addition & 0 deletions packages/core/src/providers/settings.ts
Original file line number Diff line number Diff line change
Expand Up @@ -98,6 +98,7 @@ function generateStatusMessage(
* Updated to use world metadata instead of cache
*/
export const settingsProvider: Provider = {
name: "settings",
get: async (
runtime: IAgentRuntime,
message: Memory,
Expand Down
1 change: 1 addition & 0 deletions packages/core/src/providers/time.ts
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
import type { IAgentRuntime, Memory, Provider, State } from "../types";

const timeProvider: Provider = {
name: "time",
get: async (_runtime: IAgentRuntime, _message: Memory, _state?: State) => {
const currentDate = new Date();

Expand Down
2 changes: 2 additions & 0 deletions packages/core/src/types.ts
Original file line number Diff line number Diff line change
Expand Up @@ -426,6 +426,8 @@ export interface Evaluator {
* Provider for external data/services
*/
export interface Provider {
/** Provider name */
name: string;
/** Data retrieval function */
get: (runtime: IAgentRuntime, message: Memory, state?: State) => Promise<any>;
}
Expand Down
1 change: 1 addition & 0 deletions packages/plugin-discord/src/providers/channelState.ts
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ import type {
} from "discord.js";

const channelStateProvider: Provider = {
name: "channelState",
get: async (runtime: IAgentRuntime, message: Memory, state?: State) => {
const room = await runtime.databaseAdapter.getRoom(message.roomId);
if(!room) {
Expand Down
1 change: 1 addition & 0 deletions packages/plugin-discord/src/providers/voiceState.ts
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ import { getVoiceConnection } from "@discordjs/voice";
import type { IAgentRuntime, Memory, Provider, State } from "@elizaos/core";
import { ChannelType } from "@elizaos/core";
const voiceStateProvider: Provider = {
name: "voiceState",
get: async (runtime: IAgentRuntime, message: Memory, state?: State) => {
// Voice doesn't get a discord message, so we need to use the channel for guild data
const room = await runtime.databaseAdapter.getRoom(message.roomId);
Expand Down
1 change: 1 addition & 0 deletions packages/plugin-solana/src/providers/wallet.ts
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ import type { WalletPortfolio } from '../types';
import { SOLANA_WALLET_DATA_CACHE_KEY } from '../constants';

export const walletProvider: Provider = {
name: "solana-wallet",
get: async (
runtime: IAgentRuntime,
_message: Memory,
Expand Down
4 changes: 4 additions & 0 deletions packages/plugin-tee/src/providers/deriveKeyProvider.ts
Original file line number Diff line number Diff line change
Expand Up @@ -166,6 +166,7 @@ class PhalaDeriveKeyProvider extends DeriveKeyProvider {
}

const phalaDeriveKeyProvider: Provider = {
name: "phala-derive-key",
get: async (runtime: IAgentRuntime, _message?: Memory, _state?: State) => {
const teeMode = runtime.getSetting('TEE_MODE');
const provider = new PhalaDeriveKeyProvider(teeMode);
Expand Down Expand Up @@ -212,6 +213,7 @@ const phalaDeriveKeyProvider: Provider = {
class MarlinDeriveKeyProvider extends DeriveKeyProvider {}

const marlinDeriveKeyProvider: Provider = {
name: "marlin-derive-key",
get: async (_runtime: IAgentRuntime, _message?: Memory, _state?: State) => {
return 'Marlin Derive Key Provider';
},
Expand All @@ -227,6 +229,7 @@ const marlinDeriveKeyProvider: Provider = {
class FleekDeriveKeyProvider extends DeriveKeyProvider {}

const fleekDeriveKeyProvider: Provider = {
name: "fleek-derive-key",
get: async (_runtime: IAgentRuntime, _message?: Memory, _state?: State) => {
return 'Fleek Derive Key Provider';
},
Expand All @@ -242,6 +245,7 @@ const fleekDeriveKeyProvider: Provider = {
class SgxGramineDeriveKeyProvider extends DeriveKeyProvider {}

const sgxGramineDeriveKeyProvider: Provider = {
name: "sgx-gramine-derive-key",
get: async (_runtime: IAgentRuntime, _message?: Memory, _state?: State) => {
return 'SGX Gramine Derive Key Provider';
},
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -76,6 +76,7 @@ class PhalaRemoteAttestationProvider extends RemoteAttestationProvider {

// Keep the original provider for backwards compatibility
const phalaRemoteAttestationProvider: Provider = {
name: "phala-remote-attestation",
get: async (runtime: IAgentRuntime, message: Memory, _state?: State) => {
const teeMode = runtime.getSetting('TEE_MODE');
const provider = new PhalaRemoteAttestationProvider(teeMode);
Expand Down Expand Up @@ -117,6 +118,7 @@ const phalaRemoteAttestationProvider: Provider = {
class MarlinRemoteAttestationProvider extends RemoteAttestationProvider {}

const marlinRemoteAttestationProvider: Provider = {
name: "marlin-remote-attestation",
get: async (_runtime: IAgentRuntime, _message?: Memory, _state?: State) => {
return 'Marlin Remote Attestation Provider';
},
Expand All @@ -132,6 +134,7 @@ const marlinRemoteAttestationProvider: Provider = {
class FleekRemoteAttestationProvider extends RemoteAttestationProvider {}

const fleekRemoteAttestationProvider: Provider = {
name: "fleek-remote-attestation",
get: async (_runtime: IAgentRuntime, _message?: Memory, _state?: State) => {
return 'Fleek Remote Attestation Provider';
},
Expand Down Expand Up @@ -213,6 +216,7 @@ class SgxAttestationProvider extends RemoteAttestationProvider {
}

const sgxAttestationProvider: Provider = {
name: "sgx-gramine-remote-attestation",
get: async (runtime: IAgentRuntime, _message: Memory, _state?: State) => {
const provider = new SgxAttestationProvider();
const agentId = runtime.agentId;
Expand Down

0 comments on commit e174c25

Please sign in to comment.