Update development branch

.env.sample

@@ -1,2 +1,3 @@
 NODE_ENV=development
-GOOGLE_GEMINI_API_KEY=
+GOOGLE_GEMINI_API_KEY=
+X_AI_API_KEY=

.github/workflows/run-tests.yml

@@ -19,7 +19,7 @@ jobs:
         runs-on: ubuntu-latest
         steps:
             - name: Checkout code
-              uses: actions/checkout@v4
+              uses: actions/checkout@v5
 
             - name: Install Node.js
               uses: actions/setup-node@v4

app/index.ts

@@ -1,7 +1,8 @@
 import express from 'express'
 import { processAddress, getPortfolioVelcroV3 } from '../lib/utils/portfolio'
-import { callGemini } from '../lib/utils/geminiAI'
+import { callGemini } from '../lib/utils/llm/gemini'
 import { simplePrompt } from '../lib/utils/prompts'
+import { LlmProcessOutput, LlmProcessProps, callGrok, llmMockProcess } from '../lib'
 
 const port = 3420
 
@@ -10,12 +11,19 @@ const app = express()
 app.use(express.json())
 
 app.post('/', async (req, res) => {
-    const { address } = req.body
+    const { address, llm } = req.body
+
+    const llmChoices: { [choice: string]: (props: LlmProcessProps) => Promise<LlmProcessOutput> } =
+        {
+            gemini: callGemini,
+            grok: callGrok
+        }
+
     const data = await processAddress({
         address,
         getPortfolio: getPortfolioVelcroV3,
         makePrompt: simplePrompt,
-        llmProcessor: callGemini
+        llmProcessor: llmChoices[llm] || llmMockProcess
     })
     res.send(data)
 })

dist/lib/index.d.ts

@@ -1,6 +1,11 @@
 export * from './types';
 export * from './utils/portfolio';
-export * from './utils/mockedAI';
-export * from './utils/geminiAI';
+export * from './utils/llm/mockedAI';
+export * from './utils/llm/gemini';
+export * from './utils/llm/grok';
+export * from './utils/llm/structures';
 export * from './utils/prompts';
+export * from './utils/strategies';
+export * from './utils/errors';
+export * from './utils/timing';
 //# sourceMappingURL=index.d.ts.map

dist/lib/types/types.d.ts

@@ -1,22 +1,43 @@
+import { TokenResult } from 'ambire-common/dist/src/libs/portfolio/interfaces';
+export type PortfolioLibToken = Pick<TokenResult, 'symbol' | 'address' | 'chainId' | 'decimals' | 'amount' | 'priceIn'>;
+export type NetworkPortfolioLibResponse = {
+    tokens: PortfolioLibToken[];
+};
 export type PortfolioToken = {
     address: string;
     balance: number;
     balanceUSD: number;
     symbol: string;
-    network: string;
+    decimals: number;
+};
+export type PortfolioNetworkInfo = {
+    name: string;
+    chainId: string;
+    platformId: string;
+    explorerUrl: string;
+    iconUrls: string[];
 };
 export type PortfolioForNetwork = {
-    network: string;
+    network: PortfolioNetworkInfo;
     tokens: PortfolioToken[];
 };
 export declare enum StrategyRisk {
     LOW = "low",
-    MEDIUM = "medium",
-    HIGH = "high"
+    MODERATE = "moderate",
+    HIGH = "high",
+    OPPORTUNISTIC = "opportunistic"
 }
 export type StrategyAction = {
     tokens: string;
     description: string;
+    platforms?: Array<{
+        name: string;
+        url: string;
+    }>;
+    networks?: string[];
+    operations?: string[];
+    apy?: string;
+    flags?: string[];
 };
 export type Strategy = {
     name: string;
@@ -26,20 +47,36 @@ export type Strategy = {
 export type LlmProcessProps = {
     prompt: string;
     model?: string;
+    llmOptionsOverride?: {
+        [x: string]: any;
+    };
+    timeout?: number;
+    timeoutMsg?: string;
+};
+export type LlmProcessOutput = {
+    llm: {
+        provider: string;
+        model: string;
+    };
+    response: Strategy[] | null;
+    inputTokens: number;
+    outputTokens: number;
+    error?: string | null;
 };
-export type LlmProcessOutput = Strategy[] | null;
 export type ProcessAddressProps = {
     address: string;
     getPortfolio: (address: string) => Promise<PortfolioForNetwork[]>;
     makePrompt: (props: PromptProps) => Promise<string>;
-    llmProcessor: (props: LlmProcessProps) => Promise<Strategy[] | null>;
+    llmProcessor: (props: LlmProcessProps) => Promise<LlmProcessOutput>;
+    model?: string;
+    llmOptionsOverride?: any;
 };
 export type PromptProps = {
     portfolio: PortfolioForNetwork[];
 };
 export type AuraResponse_01 = {
     address: string;
     portfolio: PortfolioForNetwork[];
-    strategies: LlmProcessOutput;
+    strategies: LlmProcessOutput[];
 };
 //# sourceMappingURL=types.d.ts.map

dist/lib/utils/errors.d.ts

@@ -0,0 +1,2 @@
+export declare function stringifyError(error: unknown): string;
+//# sourceMappingURL=errors.d.ts.map

dist/lib/utils/llm/gemini.d.ts

@@ -0,0 +1,7 @@
+import { LlmProcessOutput, LlmProcessProps } from '../../types';
+export declare const GEMINI_MODELS: {
+    gemini20flashExp: string;
+    gemini25proPreview: string;
+};
+export declare function callGemini(llmInput: LlmProcessProps): Promise<LlmProcessOutput>;
+//# sourceMappingURL=gemini.d.ts.map