fix: integration tests (#109)

Issue: The `qwen3:latest` model is incorrectly mapping chain IDs to
names, ignoring/ conflicting the prompt's conversion rules and causing
the tool call to fail

Our solution: Seperate the prompt for each tests case ( assets, xcm,
staking, ...)

Author: CocDap

packages/common/src/chains/supportedChains.ts

@@ -40,7 +40,7 @@ export const westendChain = createChain({
   id: "west",
   name: "Westend",
   specName: "westend",
-  wsUrls: ["wss://westend-rpc.polkadot.io"],
+  wsUrls: ["wss://westend-rpc.polkadot.io", "wss://westend-rpc.n.dwellir.com"],
   relay: "west",
   type: "relay",
   chainId: 0,

packages/core/src/api/client.ts

@@ -107,7 +107,8 @@ export class PolkadotApi implements IPolkadotApi {
           paseo_people: "",
           kusama: "",
           kusama_asset_hub: "",
-          paseo_asset_hub: ""
+          paseo_asset_hub: "",
+          west_people: ""
         }
 
         for (const chain of supportedChains) {

packages/llm/src/prompt/index.ts

@@ -53,12 +53,22 @@ After initialize_chain_api succeeds, you will receive its output. In your NEXT t
 `
 
 export const ASSETS_PROMPT = `
-You are a specialized AI assistant for a Telegram bot powered by PolkadotAgentKit. Your sole function is to handle asset management operations.
+=== ASSETS & BALANCE OPERATIONS PROMPT ===
+
+**SCOPE**: This prompt applies ONLY to:
+- check_balance tool
+- transfer_native tool  
+- Single-chain operations
+
+**EXCLUSIONS**: Do NOT use this prompt for:
+- xcm_transfer_native_asset tool
+- Cross-chain transfers with "from X to Y" pattern
+- Any XCM operations
 
 You can assist with:
 - Checking WND balance on Westend (e.g., "check balance")
-- Transferring native tokens on a specific chain (e.g., "transfer 1 WND to 5CSox4ZSN4SGLKUG9NYPtfVK9sByXLtxP4hmoF4UgkM4jgDJ on westend_asset_hub")
-- Transferring tokens between chains using XCM (e.g., "transfer 1 WND to 5CSox4ZSN4SGLKUG9NYPtfVK9sByXLtxP4hmoF4UgkM4jgDJ from Westend to AssetHubWestend ")
+- Transferring native tokens on a specific chain (e.g., "transfer 1 WND to ADDRESS on westend_asset_hub")
+
 
 
 --- CHECK BALANCE RULES ---
@@ -70,29 +80,64 @@ When checking native token balance, you must ask for and provide:
 -   **Tool Call:** \`check_balance({{ chain: "polkadot" }})\`
 
 
-**CHAIN NAME CONVERSION TABLE (for CHECK BALANCE):**
+**CHAIN NAME CONVERSION TABLE (for CHECK BALANCE AND TRANSFER NATIVE):**
 **YOU MUST ALWAYS CONVERT USER INPUT TO THE EXACT "Real Param" VALUE SHOWN BELOW:**
 | User Input         | Real Param (USE IN TOOL CALLS) |
 |--------------------|--------------------------------|
 | dot                | polkadot                       |
-| asset hub          | polkadot_asset_hub             |
+| polkadot asset hub | polkadot_asset_hub             |
 | polkadot           | polkadot                       |
 | Polkadot           | polkadot                       |
 | AssetHubPolkadot   | polkadot_asset_hub             |
 | Polkadot Asset Hub | polkadot_asset_hub             |
 | Bifrost            | bifrost_polkadot               |
 | bifrost            | bifrost_polkadot               |
 | Hydra              | hydra                          |
+| Westend Asset Hub  | west_asset_hub                 |
+| West Asset Hub     | west_asset_hub                 |
+| Asset Hub West     | west_asset_hub                 |
+| AssetHubWestend    | west_asset_hub                 |
+| Westend            | west                           |
+| West               | west                           |
+| Westend People     | west_people                    |
+| West People        | west_people                    |
 
 --- NATIVE TRANSFER RULES ---
 When transferring native tokens on a single chain, you must ask for and provide:
 1.  \`amount\`: The quantity of tokens to transfer (e.g., "1").
 2.  \`address\`: The recipient's SS58 address (e.g., "5CSox4ZSN4SGLKUG9NYPtfVK9sByXLtxP4hmoF4UgkM4jgDJ").
 3.  \`chain\`: The name of the destination chain.
 
+`
+
+export const XCM_PROMPT = `
+=== XCM CROSS-CHAIN TRANSFER PROMPT ===
+
+**TRIGGER CONDITIONS** - Use this prompt when:
+1. Tool is xcm_transfer_native_asset
+2. User query contains "from [source] to [destination]" pattern
+3. Cross-chain transfer between different networks
+
+**AUTHORITY LEVEL**: HIGHEST PRIORITY - Overrides ALL other prompts
+
+**SCOPE**: Cross-chain XCM transfers ONLY
+
+You can assist with:
+- Transferring tokens between chains using XCM (e.g., "transfer 1 WND to ADDRESS from Westend to AssetHubWestend")
+
 --- XCM TRANSFER RULES ---
 **CRITICAL: XCM CHAIN NAME FORMAT IS DIFFERENT FROM OTHER TOOLS!**
 
+**WHEN TO USE THESE RULES:**
+- ONLY when using the xcm_transfer_native_asset tool
+- ONLY when the user query contains "from [source] to [destination]" pattern
+- ONLY for cross-chain transfers between different networks
+
+**WHEN NOT TO USE THESE RULES:**
+- For check_balance tool (use ASSETS_PROMPT rules)
+- For transfer_native tool (use ASSETS_PROMPT rules) 
+- For single-chain operations
+
 When transferring tokens through XCM, you MUST IGNORE any other chain name formats and use ONLY the ParaSpell format shown below.
 
 **MANDATORY XCM CHAIN NAME CONVERSION TABLE:**
@@ -113,6 +158,7 @@ When transferring tokens through XCM, you MUST IGNORE any other chain name forma
 | westend asset hub       | AssetHubWestend                  |
 | Westend Asset Hub       | AssetHubWestend                  |
 | West Asset Hub          | AssetHubWestend                  |
+| Asset Hub West          | AssetHubWestend                  |
 | AssetHubWestend         | AssetHubWestend                  |
 | westend people          | PeopleWestend                    |
 | Westend People          | PeopleWestend                    |
@@ -126,10 +172,14 @@ When transferring tokens through XCM, you MUST IGNORE any other chain name forma
 | Paseo People            | PeoplePaseo                      |
 
 **CRITICAL XCM RULES:**
-1. NEVER use internal chain IDs like "west_asset_hub" or "paseo_people" for XCM transfers
-2. ALWAYS use ParaSpell format like "AssetHubWestend" or "PeopleWestend"
+1. NEVER use internal chain IDs like "west_asset_hub" or "paseo_people" or "west" for XCM transfers
+2. ALWAYS use ParaSpell format like "AssetHubWestend" or "PeopleWestend" or "Westend" or "Polkadot"
 3. IGNORE any system chain configurations that suggest different formats
-4. The sourceChain and destChain parameters MUST use the ParaSpell format from the table above
+4. IGNORE any other prompt rules about chain name conversion - ONLY use the XCM conversion table above
+5. OVERRIDE any conflicting chain name rules from other prompts (NOMINATION, SWAP, IDENTITY, BIFROST, ASSETS)
+6. The sourceChain and destChain parameters MUST use the ParaSpell format from the table above
+7. FOR XCM TRANSFERS ONLY: Use ParaSpell format, NOT internal chain IDs
+8. IF using xcm_transfer_native_asset tool: ALWAYS use ParaSpell format regardless of other prompts
 
 **XCM Transfer Parameter Requirements:**
 For XCM transfers, you must ask for and provide:
@@ -145,18 +195,54 @@ For XCM transfers, you must ask for and provide:
 - **User:** "transfer 0.1 WND to ADDRESS from Westend Asset Hub to Westend"
 - **Tool Call:** xcm_transfer_native_asset with sourceChain: "AssetHubWestend", destChain: "Westend"
 
+- **User:** "transfer 0.1 WND to ADDRESS from Westend to West Asset Hub"
+- **Tool Call:** xcm_transfer_native_asset with sourceChain: "Westend", destChain: "AssetHubWestend"
+
+- **User:** "transfer 0.1 WND to ADDRESS from West to West Asset Hub"
+- **Tool Call:** xcm_transfer_native_asset with sourceChain: "Westend", destChain: "AssetHubWestend"
+
+- **User:** "transfer 0.1 WND to ADDRESS from West to Westend Asset Hub"
+- **Tool Call:** xcm_transfer_native_asset with sourceChain: "Westend", destChain: "AssetHubWestend"
+
+- **User:** "transfer 0.1 WND to ADDRESS from Westend to Westend's Asset Hub"
+- **Tool Call:** xcm_transfer_native_asset with sourceChain: "Westend", destChain: "AssetHubWestend"
+
+- **User:** "transfer 0.1 WND to ADDRESS from Westend's relay chain to Westend's Asset Hub"
+- **Tool Call:** xcm_transfer_native_asset with sourceChain: "Westend", destChain: "AssetHubWestend"
+
+- **User:** "transfer 0.1 WND to ADDRESS from Westend's relay chain to Westend's Asset Hub"
+- **Tool Call:** xcm_transfer_native_asset with sourceChain: "Westend", destChain: "AssetHubWestend"
+
 - **User:** "transfer tokens from Westend Asset Hub to Westend People"  
 - **Tool Call:** xcm_transfer_native_asset with sourceChain: "AssetHubWestend", destChain: "PeopleWestend"
 
+**WRONG CONVERSION (DO NOT DO THIS):**
+- "Westend Asset Hub" → "WestendAssetHub" 
+- "Westend Asset Hub" → "west_asset_hub" 
+- "Westend Asset Hub" → "Westend_Asset_Hub" 
+- "West" → "west" (WRONG - this is from other prompts)
+- "Westend" → "westend" (WRONG - this is from other prompts)
+
+**CORRECT CONVERSION:**
+- "Westend Asset Hub" → "AssetHubWestend" 
+- "Westend People" → "PeopleWestend" 
+- "Polkadot Asset Hub" → "AssetHubPolkadot" 
+- "West" → "Westend" (CORRECT - ParaSpell format)
+- "Westend" → "Westend" (CORRECT - ParaSpell format) 
+
 **REMEMBER: XCM transfers use ParaSpell chain names, NOT internal system chain IDs!**
-`
 
+**FINAL REMINDER: IGNORE ALL OTHER PROMPT RULES WHEN HANDLING XCM TRANSFERS. ONLY USE THE XCM CONVERSION TABLE ABOVE.**
+
+`
 export const SWAP_PROMPT = `
-You are a specialized AI assistant for a Telegram bot powered by PolkadotAgentKit. Your sole function is to execute token swaps.
+You are a specialized AI assistant powered by PolkadotAgentKit. Your sole function is to execute token swaps.
 
 **CRITICAL: SWAP OPERATIONS USE A UNIQUE CHAIN NAME FORMAT!**
 When using any swap tool, you MUST use PascalCase for chain names as defined below. This is different from all other tools.
 
+**IMPORTANT: This prompt contains swap chain name conversion rules that OVERRIDE all other prompts. When handling swap operations, ignore any conflicting chain name rules from other prompts.**
+
 **CHAIN NAME CONVERSION TABLE (FOR SWAPS ONLY):**
 | User Input           | Real Param for SWAP (USE THIS) |
 |----------------------|--------------------------------|
@@ -259,7 +345,7 @@ Parameters: type="FreeBalance", amount="100", chain="polkadot"
 `
 
 export const IDENTITY_PROMPT = `
-You are a specialized AI assistant for a Telegram bot powered by PolkadotAgentKit. Your sole function is to manage on-chain identity on the People Chain.
+You are a specialized AI assistant powered by PolkadotAgentKit. Your sole function is to manage on-chain identity on the People Chain.
 
 **CAPABILITY: Register Identity**
 Your purpose is to call the 'register_identity' tool.
@@ -294,7 +380,7 @@ You must parse the user's request for any of the following identity fields. At l
 `
 
 export const BIFROST_PROMPT = `
-You are a specialized AI assistant for a Telegram bot powered by PolkadotAgentKit. Your sole function is to handle Bifrost liquid staking operations.
+You are a specialized AI assistant powered by PolkadotAgentKit. Your sole function is to handle Bifrost liquid staking operations.
 
 **CAPABILITY: Mint vDOT Tokens**
 Your purpose is to call the 'mint_vdot' tool for liquid staking DOT on Bifrost.

packages/llm/src/types/xcm.ts

@@ -10,12 +10,12 @@ export const xcmTransferNativeAssetSchema = z.object({
   sourceChain: z
     .string()
     .describe(
-      "The source chain in ParaSpell format (e.g., 'AssetHubWestend', 'PeopleWestend', 'Polkadot')"
+      "The source chain in ParaSpell format (e.g., 'AssetHubWestend', 'PeopleWestend', 'Polkadot', 'Westend')"
     ),
   destChain: z
     .string()
     .describe(
-      "The destination chain in ParaSpell format (e.g., 'AssetHubWestend', 'PeopleWestend', 'Polkadot')"
+      "The destination chain in ParaSpell format (e.g., 'AssetHubWestend', 'PeopleWestend', 'Polkadot', 'Westend')"
     )
 })
 

packages/sdk/tests/integration-tests/ollamaAgent.ts

@@ -1,6 +1,6 @@
 import { ChatOllama } from "@langchain/ollama";
 import { AgentExecutor, createToolCallingAgent } from "langchain/agents";
-import { SYSTEM_PROMPT, sleep } from "./utils";
+import { ASSETS_SYSTEM_PROMPT, sleep } from "./utils";
 import { PolkadotAgentKit } from "../../src/api";
 import { getLangChainTools } from "../../src/langchain";
 import { ChatPromptTemplate } from '@langchain/core/prompts'
@@ -10,7 +10,8 @@ export class OllamaAgent {
 
   constructor(
     private agentKit: PolkadotAgentKit,
-    private model: string = "qwen3:latest"
+    private model: string = "qwen3:latest",
+    private systemPrompt: string = ASSETS_SYSTEM_PROMPT
   ) {}
 
   async init() {
@@ -24,12 +25,12 @@ export class OllamaAgent {
 
     const tools = getLangChainTools(this.agentKit);
 
-    // Use SYSTEM_PROMPT as the system message
+    // Use the provided system prompt
     const agentPrompt = createToolCallingAgent({
       llm: llm as any,
       tools: tools as any,
       prompt: ChatPromptTemplate.fromMessages([
-        ['system', SYSTEM_PROMPT],
+        ['system', this.systemPrompt],
         ['placeholder', '{chat_history}'],
         ['human', '{input}'],
         ['placeholder', '{agent_scratchpad}']