1+ import {
2+ ActionExample ,
3+ composeContext ,
4+ Content ,
5+ elizaLogger ,
6+ generateObjectDeprecated ,
7+ HandlerCallback ,
8+ IAgentRuntime ,
9+ Memory ,
10+ ModelClass ,
11+ State ,
12+ type Action ,
13+ } from "@elizaos/core" ;
14+ import axios from "axios" ;
15+ import { validateStargazeConfig } from "../environment" ;
16+ import { debugLog } from "../utils/debug" ;
17+
18+ export interface GetCollectionStatsContent extends Content {
19+ collectionAddr : string ;
20+ }
21+
22+ const COLLECTION_STATS_QUERY = `
23+ query CollectionStats($collectionAddr: String!) {
24+ collection(address: $collectionAddr) {
25+ contractAddress
26+ name
27+ stats {
28+ numOwners
29+ bestOffer
30+ volumeTotal
31+ volume24Hour
32+ salesCountTotal
33+ tokensMintedPercent
34+ uniqueOwnerPercent
35+ change24HourPercent
36+ marketCap
37+ mintCount24hour
38+ mintVolume24hour
39+ volumeUsdTotal
40+ volumeUsd24hour
41+ }
42+ }
43+ }` ;
44+
45+ // Add template for content generation
46+ const getCollectionStatsTemplate = `Given the message, extract the collection address for fetching Stargaze stats.
47+
48+ Format the response as a JSON object with this field:
49+ - collectionAddr: the collection address or name (required)
50+
51+ Example response for "Show me stats for ammelia collection":
52+ \`\`\`json
53+ {
54+ "collectionAddr": "ammelia"
55+ }
56+ \`\`\`
57+
58+ Example response for "Show me stats for stars10n0m58ztlr9wvwkgjuek2m2k0dn5pgrhfw9eahg9p8e5qtvn964suc995j collection":
59+ \`\`\`json
60+ {
61+ "collectionAddr": "stars10n0m58ztlr9wvwkgjuek2m2k0dn5pgrhfw9eahg9p8e5qtvn964suc995j"
62+ }
63+ \`\`\`
64+
65+ {{recentMessages}}
66+
67+ Extract the collection address from the above messages and respond with the appropriate JSON.` ;
68+
69+ export default {
70+ name : "GET_COLLECTION_STATS" ,
71+ similes : [ "CHECK_COLLECTION_STATS" , "COLLECTION_INFO" ] ,
72+ validate : async ( runtime : IAgentRuntime , message : Memory ) => {
73+ elizaLogger . log ( "🔄 Validating Stargaze configuration..." ) ;
74+ try {
75+ const config = await validateStargazeConfig ( runtime ) ;
76+ debugLog . validation ( config ) ;
77+ return true ;
78+ } catch ( error ) {
79+ debugLog . error ( error ) ;
80+ return false ;
81+ }
82+ } ,
83+ description : "Get detailed statistics for a Stargaze collection" ,
84+ handler : async (
85+ runtime : IAgentRuntime ,
86+ message : Memory ,
87+ state : State ,
88+ _options : { [ key : string ] : unknown } ,
89+ callback ?: HandlerCallback
90+ ) : Promise < boolean > => {
91+ elizaLogger . log ( "🚀 Starting Stargaze GET_COLLECTION_STATS handler..." ) ;
92+
93+ if ( ! state ) {
94+ elizaLogger . log ( "Creating new state..." ) ;
95+ state = ( await runtime . composeState ( message ) ) as State ;
96+ } else {
97+ elizaLogger . log ( "Updating existing state..." ) ;
98+ state = await runtime . updateRecentMessageState ( state ) ;
99+ }
100+
101+ try {
102+ elizaLogger . log ( "Composing collection stats context..." ) ;
103+ const statsContext = composeContext ( {
104+ state,
105+ template : getCollectionStatsTemplate ,
106+ } ) ;
107+
108+ elizaLogger . log ( "Generating content from context..." ) ;
109+ const content = ( await generateObjectDeprecated ( {
110+ runtime,
111+ context : statsContext ,
112+ modelClass : ModelClass . LARGE ,
113+ } ) ) as unknown as GetCollectionStatsContent ;
114+
115+ if ( ! content || ! content . collectionAddr ) {
116+ throw new Error ( "Invalid or missing collection address in parsed content" ) ;
117+ }
118+
119+ debugLog . validation ( content ) ;
120+
121+ const config = await validateStargazeConfig ( runtime ) ;
122+
123+ const requestData = {
124+ query : COLLECTION_STATS_QUERY ,
125+ variables : {
126+ collectionAddr : content . collectionAddr ,
127+ } ,
128+ } ;
129+
130+ debugLog . request ( 'POST' , config . STARGAZE_ENDPOINT , requestData ) ;
131+
132+ const response = await axios . post (
133+ config . STARGAZE_ENDPOINT ,
134+ requestData ,
135+ {
136+ headers : {
137+ 'Content-Type' : 'application/json' ,
138+ }
139+ }
140+ ) ;
141+
142+ debugLog . response ( response ) ;
143+
144+ const stats = response . data ?. data ?. collection ?. stats ;
145+ const name = response . data ?. data ?. collection ?. name ;
146+
147+ if ( ! stats ) {
148+ throw new Error ( "No stats found for collection" ) ;
149+ }
150+
151+ // Format numerical values
152+ const formatValue = ( value : number ) =>
153+ value ? Number ( value ) . toLocaleString ( undefined , {
154+ maximumFractionDigits : 2
155+ } ) : '0' ;
156+
157+ // Format percentage values
158+ const formatPercent = ( value : number ) =>
159+ value ? `${ Number ( value ) . toFixed ( 2 ) } : '0%' ;
160+
161+ if ( callback ) {
162+ const message = {
163+ text : `Collection Stats for ${ name } ${ content . collectionAddr }
164+ - Total Volume: ${ formatValue ( stats . volumeUsdTotal ) }
165+ - 24h Volume: ${ formatValue ( stats . volumeUsd24hour ) }
166+ - Total Sales: ${ formatValue ( stats . salesCountTotal ) }
167+ - Unique Owners: ${ formatValue ( stats . numOwners ) }
168+ - Owner Ratio: ${ formatPercent ( stats . uniqueOwnerPercent ) }
169+ - Minted: ${ formatPercent ( stats . tokensMintedPercent ) }
170+ - 24h Change: ${ formatPercent ( stats . change24HourPercent ) }
171+ - 24h Mints: ${ formatValue ( stats . mintCount24hour ) }
172+ - Market Cap: ${ formatValue ( stats . marketCap ) } ,
173+ content : stats ,
174+ } ;
175+ elizaLogger . log ( "✅ Sending callback with collection stats:" , message ) ;
176+ callback ( message ) ;
177+ }
178+
179+ return true ;
180+ } catch ( error ) {
181+ debugLog . error ( error ) ;
182+ if ( callback ) {
183+ callback ( {
184+ text : `Error fetching collection stats: ${ error } ,
185+ content : { error : error } ,
186+ } ) ;
187+ }
188+ return false ;
189+ }
190+ } ,
191+ examples : [ [
192+ {
193+ user : "{{user1}}" ,
194+ content : {
195+ text : "Show me stats for collection ammelia" ,
196+ } ,
197+ } ,
198+ {
199+ user : "{{agent}}" ,
200+ content : {
201+ text : "I'll check the stats for collection ammelia..." ,
202+ action : "GET_COLLECTION_STATS" ,
203+ } ,
204+ } ,
205+ {
206+ user : "{{user1}}" ,
207+ content : {
208+ text : "Show me stats for collection {collection address}" ,
209+ } ,
210+ } ,
211+ {
212+ user : "{{agent}}" ,
213+ content : {
214+ text : "I'll check the stats for collection {collection address}..." ,
215+ action : "GET_COLLECTION_STATS" ,
216+ } ,
217+ } ,
218+ ] ] ,
219+ } as Action ;
0 commit comments